chore: wip

Author: chrisbbreuer

config/ai.ts

@@ -12,7 +12,7 @@ export default {
 
   models: [
     // 'amazon.titan-embed-text-v1',
-    // 'amazon.titan-text-express-v1',
+    'amazon.titan-text-express-v1',
     // 'amazon.titan-embed-image-v1',
     // 'amazon.titan-image-generator-v1',
     // 'anthropic.claude-v1',

storage/framework/.stacks/core/cloud/src/cloud/ai.ts

@@ -23,13 +23,15 @@ export class AiStack {
 
     // Defining the Node.js Lambda function
     const aiLambda = new lambda.Function(scope, 'LambdaFunction', {
+      functionName: `${props.slug}-${props.appEnv}-ai`,
       runtime: lambda.Runtime.NODEJS_20_X,
       handler: 'index.handler',
       code: lambda.Code.fromAsset('src/cloud/lambda'), // path relative to the cloud root package dir
       layers: [awsSdkLayer],
     })
 
     const api = new apigateway.LambdaRestApi(scope, 'ApiGateway', {
+      restApiName: `${props.slug}-${props.appEnv}-ai`,
       handler: aiLambda,
     })
 

storage/framework/.stacks/core/cloud/src/cloud/lambda/index.js

@@ -8,7 +8,7 @@ async function handler(event) {
   console.log(`Question received: ${question}`)
   const bedrockRuntime = new AWS.BedrockRuntime({ apiVersion: '2023-09-30' })
   const res = await bedrockRuntime.invokeModel({
-    modelId: 'ai21.j2-ultra-v1',
+    modelId: 'amazon.titan-text-express-v1',
     contentType: 'application/json',
     accept: '*/*',
     body: `{"prompt": "${question}","maxTokens":200,"temperature":0.7,"topP":1,"stopSequences":[],"countPenalty":{"scale":0},"presencePenalty":{"scale":0},"frequencyPenalty":{"scale":0}}`,