mypy-boto3-bedrock-agentcore

import boto3 from typing import TYPE_CHECKING, List import os # These imports are only for type checking, not for runtime execution if TYPE_CHECKING: from mypy_boto3_bedrock_agentcore import BedrockAgentCoreClient from mypy_boto3_bedrock_agentcore.type_defs import AgentSummaryTypeDef, ListAgentsResponseTypeDef def get_bedrock_agent_core_client() -> "BedrockAgentCoreClient": """ Returns a type-hinted BedrockAgentCore client. """ # boto3.client returns a botocore.client.BedrockAgentCore instance, # which mypy-boto3 stubs enhance with static type information. # Ensure AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, and AWS_REGION are set as environment variables client: BedrockAgentCoreClient = boto3.client( "bedrock-agent-core", region_name=os.environ.get('AWS_REGION', 'us-east-1') # Default region if not set ) return client def list_bedrock_agents(): """ Lists Bedrock Agents with type hints. """ client = get_bedrock_agent_core_client() print("Listing Bedrock Agents...") try: # The response is type-checked against ListAgentsResponseTypeDef response: ListAgentsResponseTypeDef = client.list_agents() agents: List[AgentSummaryTypeDef] = response.get("agentSummaries", []) if agents: for agent in agents: print(f" Agent Name: {agent.get('agentName')}, Status: {agent.get('agentStatus')}") else: print(" No Bedrock Agents found in the configured region.") except Exception as e: print(f"Error listing agents: {e}") if __name__ == "__main__": # Example of setting environment variables (replace with your actual credentials/region) # os.environ['AWS_ACCESS_KEY_ID'] = 'YOUR_ACCESS_KEY' # os.environ['AWS_SECRET_ACCESS_KEY'] = 'YOUR_SECRET_KEY' # os.environ['AWS_REGION'] = 'us-east-1' # This will attempt to use credentials from environment variables, ~/.aws/credentials, or IAM role. list_bedrock_agents()