From 4255f788c90318e236e4fc2aab5055704701da37 Mon Sep 17 00:00:00 2001 From: Mauro Poggianella Date: Mon, 30 Jun 2025 13:52:34 +0200 Subject: [PATCH] AWS provider use credentials from config --- aisuite/providers/aws_provider.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/aisuite/providers/aws_provider.py b/aisuite/providers/aws_provider.py index c0f00daa..55221652 100644 --- a/aisuite/providers/aws_provider.py +++ b/aisuite/providers/aws_provider.py @@ -16,9 +16,26 @@ def __init__(self, **config): self.region_name = config.get( "region_name", os.getenv("AWS_REGION", "us-west-2") ) + self.access_key_id = config.get( + "access_key_id", os.getenv("AWS_ACCESS_KEY_ID") + ) + if not self.access_key_id: + raise ValueError("For AWS, access_key_id is required.") + self.secret_access_key = config.get( + "secret_access_key", os.getenv("AWS_SECRET_ACCESS_KEY") + ) + if not self.secret_access_key: + raise ValueError("For AWS, secret_access_key is required.") def create_client(self): - return boto3.client("bedrock-runtime", region_name=self.region_name) + session = boto3.Session( + aws_access_key_id=self.access_key_id, + aws_secret_access_key=self.secret_access_key, + ) + return session.client( + "bedrock-runtime", + region_name=self.region_name, + ) # AWS Bedrock API Example -