Manage Agents

# Create an agent using an existing API key
response = client.agents.create(
    agent_name="my-openai-agent",
    hub="openai",
    api_key_name="my-openai-key",  # Use existing API key
    model_name="gpt-4o-mini",
    agent_system_prompt="You are a helpful AI assistant."
)
print(f"Created agent: {response}")

# Create an agent and automatically create a new API key
response = client.agents.create(
    agent_name="my-claude-agent",
    hub="anthropic",
    api_key_value="sk-ant-api03-...",  # Your actual API key
    model_name="claude-3-sonnet-20240229",
    agent_system_prompt="You are an expert code reviewer.",
    rate_limit_per_interval=50,
    rate_limit_interval=60
)
print(f"Created agent with new API key: {response}")

# Create a Bedrock agent with specific configuration
response = client.agents.create(
    agent_name="my-bedrock-agent",
    hub="bedrockAgents",
    agent_id="ABCD1234",
    agent_alias_id="ALIAS123",
    hub_config={
        "region": "us-east-1",
        "access_key": "your-access-key",
        "secret_access_key": "your-secret-key",
        "agent_id": "ABCD1234",
        "agent_alias_id": "ALIAS123"
    }
)
print(f"Created Bedrock agent: {response}")

# List all active agents
agents = client.agents.list()
print("Active agents:")
for agent in agents:
    print(f"- {agent.get('agent_name')} (Hub: {agent.get('hub')}, Model: {agent.get('model_name')})")

# Update an existing agent's name and model
response = client.agents.update(
    agent_name="my-agent",  # Current agent name
    new_agent_name="updated-agent",  # New name
    model_name="updated-model",  # Updated model
    hub="updated-hub",  # Updated hub
    api_key_name="updated-api-key",  # Updated API key
    agent_url="updated-url",  # Updated URL
    agent_system_prompt="updated-system-prompt"
)
print(f"Updated agent: {response}")

# Delete an agent by name (archives the agent)
try:
    response = client.agents.delete(agent_name="my-openai-agent")
    print(f"Successfully deleted agent: {response}")
except ValueError as e:
    print(f"Error deleting agent: {e}")