Merge pull request #25 from andrewyng/update-aws-region-access

standsleeping · web-flow · commit 14138749300f · 2024-09-01T10:20:27.000-05:00
Update aws provider to use env variable for region
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"),
         )

Original file line number	Diff line number	Diff line change
`@@ -72,7 +72,7 @@ def __init__(self):`
`72`	`72`
`73`	`73`	`self.aws_bedrock_client = boto3.client(`
`74`	`74`	`service_name="bedrock-runtime",`
`75`		`- region_name="us-west-2",`
	`75`	`+ region_name=os.getenv("AWS_REGION_NAME"),`
`76`	`76`	`aws_access_key_id=os.getenv("AWS_ACCESS_KEY_ID"),`
`77`	`77`	`aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY"),`
`78`	`78`	`)`