diff --git a/.env.sample b/.env.sample index db4e9561..3201a9eb 100644 --- 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="" \ No newline at end of file diff --git a/aimodels/providers/aws_bedrock_interface.py b/aimodels/providers/aws_bedrock_interface.py index 1b6b0aca..7afbe330 100644 --- 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"), )