Ensure default region

chroma-core · Jan 21, 2024 · fa9746f · fa9746f
1 parent 4426f64
commit fa9746f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/clients/js/src/embeddings/AmazonBedrockEmbeddingFunction.ts b/clients/js/src/embeddings/AmazonBedrockEmbeddingFunction.ts
@@ -4,12 +4,15 @@ let amazonBedrockApi: any;
 
 export class AmazonBedrockEmbeddingFunction implements IEmbeddingFunction {
   private model: string;
-  private readonly configuration: any; // Assume this is BedrockRuntimeClientConfig
+  private configuration: any; // Assume this is BedrockRuntimeClientConfig
   private amazonBedrockApi?: any;
   private invokeCommand?: any;
 
-  constructor({ config, model }: { config: any; model?: string }) {
+  constructor({ config, model }: { config: any; model?: string}) {
     this.configuration = config;
+    if (!this.configuration.region) {
+      this.configuration.region = "us-east-1";
+    }
     this.model = model || "amazon.titan-embed-text-v1";
   }