AmazonBedrockAgent

The AmazonBedrockAgent is a specialized agent class in the Multi-Agent Orchestrator that integrates directly with Amazon Bedrock agents.

Creating an AmazonBedrockAgent

Here are various examples showing different ways to create and configure an AmazonBedrockAgent:

Python Package

If you haven’t already installed the AWS-related dependencies, make sure to install them:

pip install "multi-agent-orchestrator[aws]"

Basic Examples

1. Minimal Configuration

TypeScript
Python

const agent = new AmazonBedrockAgent({
  name: 'My Bank Agent',
  description: 'A helpful and friendly agent that answers questions about loan-related inquiries',
  agentId: 'your-agent-id',
  agentAliasId: 'your-agent-alias-id'
});

agent = AmazonBedrockAgent(AmazonBedrockAgentOptions(
    name='My Bank Agent',
    description='A helpful and friendly agent that answers questions about loan-related inquiries',
    agent_id='your-agent-id',
    agent_alias_id='your-agent-alias-id'
))

2. Using Custom Client

TypeScript
Python

import { BedrockAgentRuntimeClient } from "@aws-sdk/client-bedrock-agent-runtime";
const customClient = new BedrockAgentRuntimeClient({ region: 'us-east-1' });
const agent = new AmazonBedrockAgent({
  name: 'My Bank Agent',
  description: 'A helpful and friendly agent for banking inquiries',
  agentId: 'your-agent-id',
  agentAliasId: 'your-agent-alias-id',
  client: customClient
});

import boto3
custom_client = boto3.client('bedrock-agent-runtime', region_name='us-east-1')
agent = AmazonBedrockAgent(AmazonBedrockAgentOptions(
name='My Bank Agent',
description='A helpful and friendly agent for banking inquiries',
agent_id='your-agent-id',
agent_alias_id='your-agent-alias-id',
client=custom_client
))

3. With Tracing Enabled

TypeScript
Python

const agent = new AmazonBedrockAgent({
  name: 'My Bank Agent',
  description: 'A banking agent with tracing enabled',
  agentId: 'your-agent-id',
  agentAliasId: 'your-agent-alias-id',
  enableTrace: true
});

agent = AmazonBedrockAgent(AmazonBedrockAgentOptions(
    name='My Bank Agent',
    description='A banking agent with tracing enabled',
    agent_id='your-agent-id',
    agent_alias_id='your-agent-alias-id',
    enable_trace=True
))

4. With Streaming Enabled

TypeScript
Python

const agent = new AmazonBedrockAgent({
  name: 'My Bank Agent',
  description: 'A streaming-enabled banking agent',
  agentId: 'your-agent-id',
  agentAliasId: 'your-agent-alias-id',
  streaming: true
});

agent = AmazonBedrockAgent(AmazonBedrockAgentOptions(
    name='My Bank Agent',
    description='A streaming-enabled banking agent',
    agent_id='your-agent-id',
    agent_alias_id='your-agent-alias-id',
    streaming=True
))

5. Complete Example with All Options

TypeScript
Python

import { AmazonBedrockAgent } from "multi-agent-orchestrator";
import { BedrockAgentRuntimeClient } from "@aws-sdk/client-bedrock-agent-runtime";
const agent = new AmazonBedrockAgent({
  // Required fields
  name: "Advanced Bank Agent",
  description: "A fully configured banking agent with all features enabled",
  agentId: "your-agent-id",
  agentAliasId: "your-agent-alias-id",
  // Optional fields
  region: "us-west-2",
  streaming: true,
  enableTrace: true,
  client: new BedrockAgentRuntimeClient({ region: "us-west-2" }),
});

import boto3
from multi_agent_orchestrator.agents import AmazonBedrockAgent, AmazonBedrockAgentOptions

custom_client = boto3.client('bedrock-agent-runtime', region_name='us-west-2')

agent = AmazonBedrockAgent(AmazonBedrockAgentOptions(
    # Required fields
    name='Advanced Bank Agent',
    description='A fully configured banking agent with all features enabled',
    agent_id='your-agent-id',
    agent_alias_id='your-agent-alias-id',

    # Optional fields
    region='us-west-2',
    streaming=True,
    enable_trace=True,
    client=custom_client
))

Option Explanations

name: (Required) Identifies the agent within your system.
description: (Required) Describes the agent’s purpose or capabilities.
agentId/agent_id: (Required) The ID of the Amazon Bedrock agent you want to use.
agentAliasId/agent_alias_id: (Required) The alias ID of the Amazon Bedrock agent.
region: (Optional) AWS region for the Bedrock service. If not provided, uses the default AWS region.
client: (Optional) Custom BedrockAgentRuntimeClient for specialized configurations.
enableTrace/enable_trace: (Optional) When set to true, enables tracing of the agent’s steps and reasoning process.
streaming: (Optional) Enables streaming for the final response. Defaults to false.

Adding the Agent to the Orchestrator

To integrate the AmazonBedrockAgent into your Multi-Agent Orchestrator, follow these steps:

First, ensure you have created an instance of the orchestrator:

TypeScript
Python

import { MultiAgentOrchestrator } from "multi-agent-orchestrator";

const orchestrator = new MultiAgentOrchestrator();

from multi_agent_orchestrator.orchestrator import MultiAgentOrchestrator

orchestrator = MultiAgentOrchestrator()

Then, add the agent to the orchestrator:

TypeScript
Python

orchestrator.addAgent(agent);

orchestrator.add_agent(agent)

Now you can use the orchestrator to route requests to the appropriate agent, including your Amazon Bedrock agent:

TypeScript
Python

const response = await orchestrator.routeRequest(
  "What is the base rate interest for 30 years?",
  "user123",
  "session456"
);

response = await orchestrator.route_request(
    "What is the base rate interest for 30 years?",
    "user123",
    "session456"
)

By leveraging the AmazonBedrockAgent, you can easily integrate pre-built Amazon Bedrock agents into your Multi-Agent Orchestrator.