Generative AI Use Cases JP (略称:GenU) でサポートされたAmazon Bedrock AgentCore連携を試してみた

strands-agents
strands-agents-tools
bedrock-agentcore
bedrock-agentcore-starter-toolkit

pip install -r requirements.txt

import os
from strands import Agent, tool
from strands.models import BedrockModel
from bedrock_agentcore.runtime import BedrockAgentCoreApp

model_id = os.getenv("BEDROCK_MODEL_ID", "anthropic.claude-3-5-haiku-20241022-v1:0")
model = BedrockModel(model_id=model_id, params={"max_tokens": 4096, "temperature": 0.7}, region="us-west-2")

app = BedrockAgentCoreApp()

@tool
def get_weather(city: str) -> str:
    """Get the weather for a given city"""
    return f"The weather in {city} is sunny"

@app.entrypoint
async def entrypoint(payload):
    agent = Agent(model=model, tools=[get_weather])
    message = payload.get("prompt", "")
    stream_messages = agent.stream_async(message)
    async for message in stream_messages:
        if "event" in message:
            yield message

if __name__ == "__main__":
    app.run()

async for message in stream_messages:
        if "event" in message:
            yield message

agentcore configure --entrypoint agent.py

agentcore launch

Agent ARN:arn:aws:bedrock-agentcore:us-west-2:xxx:runtime/agent-yyy

git clone https://github.com/aws-samples/generative-ai-use-cases.git

npm ci

const envs: Record<string, Partial<StackInput>> = {

  // If you want to define an anonymous environment, uncomment the following and the content of cdk.json will be ignored.
  // If you want to define an anonymous environment in parameter.ts, uncomment the following and the content of cdk.json will be ignored.
  '': {
     // Parameters for anonymous environment
     // If you want to override the default settings, add the following
     modelRegion: 'us-west-2',
     imageGenerationModelIds: [],
     videoGenerationModelIds: [],
     speechToSpeechModelIds: [],
     createGenericAgentCoreRuntime: true,
     agentCoreRegion: 'us-west-2',
     agentCoreExternalRuntimes: [
      {
        name: 'SimpleAgentCore',
        arn: 'arn:aws:bedrock-agentcore:us-west-2:xxx:runtime/agent-yyy',
      },
    ]
   },
  dev: {
    // Parameters for development environment
  },
  staging: {
    // Parameters for staging environment
  },
  prod: {
    // Parameters for production environment
  },
  // If you need other environments, customize them as needed
};

npx -w packages/cdk cdk bootstrap

docker run --privileged --rm tonistiigi/binfmt --install arm64

npm run cdk:deploy

GenerativeAiUseCasesStack.WebUrl = https://xxx.cloudfront.net

{
  "_comment": "Generic AgentCore Runtime Configuration",
  "_agentcore_requirements": {
    "platform": "linux/arm64",
    "port": 8080,
    "endpoints": {
      "/ping": "GET - Health check endpoint",
      "/invocations": "POST - Main inference endpoint"
    },
    "aws_credentials": "Required for Bedrock model access and S3 operations"
  },
  "mcpServers": {
    "time": {
      "command": "uvx",
      "args": ["mcp-server-time"]
    },
    "awslabs.aws-documentation-mcp-server": {
      "command": "uvx",
      "args": ["awslabs.aws-documentation-mcp-server@latest"]
    },
    "awslabs.cdk-mcp-server": {
      "command": "uvx",
      "args": ["awslabs.cdk-mcp-server@latest"]
    },
    "awslabs.aws-diagram-mcp-server": {
      "command": "uvx",
      "args": ["awslabs.aws-diagram-mcp-server@latest"]
    },
    "awslabs.nova-canvas-mcp-server": {
      "command": "uvx",
      "args": ["awslabs.nova-canvas-mcp-server@latest"],
      "env": {
        "AWS_REGION": "us-east-1"
      }
    }
  }
}

import os
from strands import Agent, tool
from strands.models import BedrockModel
from bedrock_agentcore.runtime import BedrockAgentCoreApp
app = BedrockAgentCoreApp()

@tool
def get_weather(city: str) -> str:
    """Get the weather for a given city"""
    return f"The weather in {city} is sunny"

@app.entrypoint
async def entrypoint(payload):
    message = payload.get("prompt", "")
    model = payload.get("model", {})
    model_id = model.get("modelId","anthropic.claude-3-5-haiku-20241022-v1:0")
    model = BedrockModel(model_id=model_id, params={"max_tokens": 4096, "temperature": 0.7}, region="us-west-2")
    agent = Agent(model=model, tools=[get_weather])
    stream_messages = agent.stream_async(message)
    async for message in stream_messages:
        if "event" in message:
            yield message

if __name__ == "__main__":
    app.run()

agentcore launch