fix: aws region error

traceloop · Feb 13, 2024 · 5c36343 · 5c36343
1 parent a04722d
commit 5c36343
Show file tree

Hide file tree

Showing 5 changed files with 40 additions and 5 deletions.
diff --git a/packages/instrumentation-bedrock/tests/ai21.test.ts b/packages/instrumentation-bedrock/tests/ai21.test.ts
@@ -55,7 +55,14 @@ describe("Test Ai21 with AWS Bedrock Instrumentation", () => {
     instrumentation.setTracerProvider(provider);
     bedrock = await import("@aws-sdk/client-bedrock-runtime");
 
-    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient();
+    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient(
+      process.env.RECORD_MODE !== "NEW"
+        ? {
+            region: "us-west-2",
+            credentials: { accessKeyId: "test", secretAccessKey: "test" },
+          }
+        : {},
+    );
   });
 
   beforeEach(function () {

diff --git a/packages/instrumentation-bedrock/tests/amazon.test.ts b/packages/instrumentation-bedrock/tests/amazon.test.ts
@@ -55,7 +55,14 @@ describe("Test Amazon Titan with AWS Bedrock Instrumentation", () => {
     instrumentation.setTracerProvider(provider);
     bedrock = await import("@aws-sdk/client-bedrock-runtime");
 
-    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient();
+    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient(
+      process.env.RECORD_MODE !== "NEW"
+        ? {
+            region: "us-west-2",
+            credentials: { accessKeyId: "test", secretAccessKey: "test" },
+          }
+        : {},
+    );
   });
 
   beforeEach(function () {

diff --git a/packages/instrumentation-bedrock/tests/anthropic.test.ts b/packages/instrumentation-bedrock/tests/anthropic.test.ts
@@ -55,7 +55,14 @@ describe("Test Anthropic with AWS Bedrock Instrumentation", () => {
     instrumentation.setTracerProvider(provider);
     bedrock = await import("@aws-sdk/client-bedrock-runtime");
 
-    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient();
+    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient(
+      process.env.RECORD_MODE !== "NEW"
+        ? {
+            region: "us-west-2",
+            credentials: { accessKeyId: "test", secretAccessKey: "test" },
+          }
+        : {},
+    );
   });
 
   beforeEach(function () {

diff --git a/packages/instrumentation-bedrock/tests/cohere.test.ts b/packages/instrumentation-bedrock/tests/cohere.test.ts
@@ -55,7 +55,14 @@ describe("Test Cohere with AWS Bedrock Instrumentation", () => {
     instrumentation.setTracerProvider(provider);
     bedrock = await import("@aws-sdk/client-bedrock-runtime");
 
-    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient();
+    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient(
+      process.env.RECORD_MODE !== "NEW"
+        ? {
+            region: "us-west-2",
+            credentials: { accessKeyId: "test", secretAccessKey: "test" },
+          }
+        : {},
+    );
   });
 
   beforeEach(function () {

diff --git a/packages/instrumentation-bedrock/tests/meta.test.ts b/packages/instrumentation-bedrock/tests/meta.test.ts
@@ -55,7 +55,14 @@ describe("Test Meta with AWS Bedrock Instrumentation", () => {
     instrumentation.setTracerProvider(provider);
     bedrock = await import("@aws-sdk/client-bedrock-runtime");
 
-    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient();
+    bedrockRuntimeClient = new bedrock.BedrockRuntimeClient(
+      process.env.RECORD_MODE !== "NEW"
+        ? {
+            region: "us-west-2",
+            credentials: { accessKeyId: "test", secretAccessKey: "test" },
+          }
+        : {},
+    );
   });
 
   beforeEach(function () {