带有图像输入的代理

AWS Bedrock 支持使用 amazon.nova-pro-v1:0 等模型进行图像输入。您可以使用此功能分析图像并获取相关信息。

代码

cookbook/models/aws/bedrock/image_agent.py

from pathlib import Path
from agno.agent import Agent
from agno.media import Image
from agno.models.aws import AwsBedrock
from agno.tools.duckduckgo import DuckDuckGoTools

agent = Agent(
    model=AwsBedrock(id="amazon.nova-pro-v1:0"),
    tools=[DuckDuckGoTools()],
    markdown=True,
)

image_path = Path(__file__).parent.joinpath("sample.jpg")

# Read the image file content as bytes
with open(image_path, "rb") as img_file:
    image_bytes = img_file.read()

agent.print_response(
    "Tell me about this image and give me the latest news about it.",
    images=[
        Image(content=image_bytes, format="jpeg"),
    ],
)

用法

1. 创建虚拟环境

   打开 Terminal 并创建一个 python 虚拟环境。

   Mac

   python3 -m venv .venv
   source .venv/bin/activate
   
2. 设置您的 AWS 凭证

   export AWS_ACCESS_KEY_ID=***
   export AWS_SECRET_ACCESS_KEY=***
   export AWS_REGION=***
   
3. 安装库

   pip install -U boto3 duckduckgo-search agno
   
4. 添加图像

   将名为 sample.jpg 的图像文件放置在脚本的同一目录下。

5. 运行代理

   Mac

   python cookbook/models/aws/bedrock/image_agent.py