mypy-boto3-bedrock-agent

import os import boto3 from mypy_boto3_bedrock_agent import BedrockAgentClient from botocore.exceptions import ClientError # Ensure boto3 is installed and configured (e.g., via AWS environment variables # AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION, or IAM roles). # For local testing, ensure these are set or provide dummy values. # Type hint the client to enable static analysis by Mypy # region_name is typically required for Bedrock services. client: BedrockAgentClient = boto3.client( "bedrock-agent", region_name=os.environ.get("AWS_REGION", "us-east-1") ) try: # Example operation: List existing agents print("Attempting to list Bedrock Agents...") response = client.list_agents(maxResults=5) agents = response.get("agentSummaries", []) if agents: print("Successfully listed agents:") for agent_summary in agents: print(f" - Agent ID: {agent_summary['agentId']}, Name: {agent_summary['agentName']}") else: print("No Bedrock agents found in this region.") except ClientError as e: error_code = e.response.get("Error", {}).get("Code") error_message = e.response.get("Error", {}).get("Message") if error_code == "AccessDeniedException": print(f"Access Denied: You might not have permissions to list agents. Error: {error_message}") else: print(f"An AWS Client Error occurred: {error_code} - {error_message}") except Exception as e: print(f"An unexpected error occurred: {e}") # Mypy will now correctly type-check method calls on `client` # and the structure of `response` based on the AWS API definition.