diff --git a/main.py b/main.py index 4ccfcff..c2e7bcd 100644 --- a/main.py +++ b/main.py @@ -7,17 +7,23 @@ import time app = FastAPI() +os.environ['AWS_REGION'] = 'us-east-1' + # Define request body class BedrockRequest(BaseModel): question: str requestSessionId: str = 'user' - agentId: str = "ROJCGWHSC0" - agentAliasId: str ="TQ8VDTVQII" + agentId: str = "1" + agentAliasId: str ="1" # AWS Bedrock client setup def get_bedrock_client(): return boto3.client( - 'bedrock-agent-runtime' + 'bedrock-agent-runtime', + region_name=os.getenv('AWS_REGION', 'us-east-1'), + aws_access_key_id=os.getenv('AWS_ACCESS_KEY_ID'), + aws_secret_access_key=os.getenv('AWS_SECRET_ACCESS_KEY'), + aws_session_token=os.getenv('AWS_SESSION_TOKEN') ) @app.get("/health")