use env var to set aws region

andrewyng · Aug 27, 2024 · ba56514 · ba56514
1 parent 75120e9
commit ba56514
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.env.sample b/.env.sample
@@ -1,4 +1,7 @@
 ANTHROPIC_API_KEY=""
+AWS_ACCESS_KEY_ID=""
+AWS_REGION_NAME=""
+AWS_SECRET_ACCESS_KEY=""
 FIREWORKS_API_KEY=""
 GROQ_API_KEY=""
 MISTRAL_API_KEY=""
@@ -7,5 +10,3 @@ OLLAMA_API_URL="http://localhost:11434"
 REPLICATE_API_KEY=""
 TOGETHER_API_KEY=""
 OCTO_API_KEY=""
-AWS_ACCESS_KEY_ID=""
-AWS_SECRET_ACCESS_KEY=""
diff --git a/aimodels/providers/aws_bedrock_interface.py b/aimodels/providers/aws_bedrock_interface.py
@@ -72,7 +72,7 @@ def __init__(self):
 
         self.aws_bedrock_client = boto3.client(
             service_name="bedrock-runtime",
-            region_name="us-west-2",
+            region_name=os.getenv("AWS_REGION_NAME"),
             aws_access_key_id=os.getenv("AWS_ACCESS_KEY_ID"),
             aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY"),
         )