fix: aws region error

traceloop · Feb 13, 2024 · 15f6f42 · 15f6f42
1 parent a04722d
commit 15f6f42
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 0 deletions.
diff --git a/packages/instrumentation-bedrock/tests/ai21.test.ts b/packages/instrumentation-bedrock/tests/ai21.test.ts
@@ -50,6 +50,10 @@ describe("Test Ai21 with AWS Bedrock Instrumentation", () => {
   });
 
   before(async () => {
+    if (process.env.RECORD_MODE !== "NEW") {
+      process.env.AWS_REGION = "us-west-2";
+    }
+
     provider.addSpanProcessor(new SimpleSpanProcessor(memoryExporter));
     instrumentation = new BedrockInstrumentation();
     instrumentation.setTracerProvider(provider);

diff --git a/packages/instrumentation-bedrock/tests/amazon.test.ts b/packages/instrumentation-bedrock/tests/amazon.test.ts
@@ -50,6 +50,10 @@ describe("Test Amazon Titan with AWS Bedrock Instrumentation", () => {
   });
 
   before(async () => {
+    if (process.env.RECORD_MODE !== "NEW") {
+      process.env.AWS_REGION = "us-west-2";
+    }
+
     provider.addSpanProcessor(new SimpleSpanProcessor(memoryExporter));
     instrumentation = new BedrockInstrumentation();
     instrumentation.setTracerProvider(provider);

diff --git a/packages/instrumentation-bedrock/tests/anthropic.test.ts b/packages/instrumentation-bedrock/tests/anthropic.test.ts
@@ -50,6 +50,10 @@ describe("Test Anthropic with AWS Bedrock Instrumentation", () => {
   });
 
   before(async () => {
+    if (process.env.RECORD_MODE !== "NEW") {
+      process.env.AWS_REGION = "us-west-2";
+    }
+
     provider.addSpanProcessor(new SimpleSpanProcessor(memoryExporter));
     instrumentation = new BedrockInstrumentation();
     instrumentation.setTracerProvider(provider);

diff --git a/packages/instrumentation-bedrock/tests/cohere.test.ts b/packages/instrumentation-bedrock/tests/cohere.test.ts
@@ -50,6 +50,10 @@ describe("Test Cohere with AWS Bedrock Instrumentation", () => {
   });
 
   before(async () => {
+    if (process.env.RECORD_MODE !== "NEW") {
+      process.env.AWS_REGION = "us-west-2";
+    }
+
     provider.addSpanProcessor(new SimpleSpanProcessor(memoryExporter));
     instrumentation = new BedrockInstrumentation();
     instrumentation.setTracerProvider(provider);

diff --git a/packages/instrumentation-bedrock/tests/meta.test.ts b/packages/instrumentation-bedrock/tests/meta.test.ts
@@ -50,6 +50,10 @@ describe("Test Meta with AWS Bedrock Instrumentation", () => {
   });
 
   before(async () => {
+    if (process.env.RECORD_MODE !== "NEW") {
+      process.env.AWS_REGION = "us-west-2";
+    }
+
     provider.addSpanProcessor(new SimpleSpanProcessor(memoryExporter));
     instrumentation = new BedrockInstrumentation();
     instrumentation.setTracerProvider(provider);