Fix bedrock example.

launchdarkly · Nov 4, 2024 · 817d4f0 · 817d4f0
1 parent 95fa2bc
commit 817d4f0
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 15 deletions.
diff --git a/packages/sdk/ai/examples/bedrock/src/index.ts b/packages/sdk/ai/examples/bedrock/src/index.ts
@@ -29,6 +29,11 @@ const context = {
 
 console.log('*** SDK successfully initialized');
 
+interface MyModelConfig {
+  modelId: string;
+  prompt: { role: ConversationRole; content: string }[];
+}
+
 function mapPromptToConversation(prompt: { role: ConversationRole; content: string }[]): Message[] {
   return prompt.map((item) => ({
     role: item.role,
@@ -47,23 +52,31 @@ async function main() {
     configValue = await aiClient.modelConfig(aiConfigKey, context, false, {
       myVariable: 'My User Defined Variable',
     });
-    tracker = configValue.tracker;
+    if (configValue === false) {
+      console.log('got default value for config');
+      process.exit(1);
+    } else {
+      tracker = configValue.tracker;
+    }
   } catch (error) {
     console.log(`*** SDK failed to initialize: ${error}`);
     process.exit(1);
   }
 
-  const completion = await tracker.trackBedrockConverse(
-    await awsClient.send(
-      new ConverseCommand({
-        modelId: configValue.config.config.modelId,
-        messages: mapPromptToConversation(configValue.config.prompt),
-      }),
-    ),
-  );
+  if (tracker) {
+    const modelConfig = configValue.config as MyModelConfig;
+    const completion = await tracker.trackBedrockConverse(
+      await awsClient.send(
+        new ConverseCommand({
+          modelId: modelConfig.modelId,
+          messages: mapPromptToConversation(modelConfig.prompt),
+        }),
+      ),
+    );
 
-  console.log('AI Response:', completion.output.message.content[0].text);
-  console.log('Success.');
+    console.log('AI Response:', completion.output.message.content[0].text);
+    console.log('Success.');
+  }
 }
 
 main();
diff --git a/packages/sdk/ai/examples/bedrock/tsconfig.json b/packages/sdk/ai/examples/bedrock/tsconfig.json
@@ -17,6 +17,6 @@
     "module": "CommonJS",
     "moduleResolution": "Node"
   },
-  "include": ["src", "test"],
+  "include": ["src"],
   "exclude": ["dist", "node_modules"]
 }
diff --git a/packages/sdk/ai/tsconfig.json b/packages/sdk/ai/tsconfig.json
@@ -1,14 +1,13 @@
 {
   "compilerOptions": {
-    // Uses "." so it can load package.json.
-    "rootDir": ".",
+    "rootDir": "src",
     "outDir": "dist",
     "target": "es2017",
     "lib": ["es6"],
     "module": "commonjs",
     "strict": true,
     "noImplicitOverride": true,
-    // Needed for CommonJS modules: markdown-it, fs-extra
+    // Needed for CommonJS modules.
     "allowSyntheticDefaultImports": true,
     "sourceMap": true,
     "declaration": true,