本示例展示了如何使用图片生成模型通过 WhatsApp 应用生成图片。

代码

cookbook/apps/whatsapp/image_generation_model.py
from agno.agent import Agent
from agno.app.whatsapp.app import WhatsappAPI
from agno.models.google import Gemini

image_agentg = Agent(
    model=Gemini(
        id="gemini-2.0-flash-exp-image-generation",
        response_modalities=["Text", "Image"],
    ),
    debug_mode=True,
)

whatsapp_app = WhatsappAPI(
    agent=image_agent,
    name="图片生成模型",
    app_id="image_generation_model",
    description="一个使用 Gemini API 生成图片的模型。",
)

app = whatsapp_app.get_app()

if __name__ == "__main__":
    whatsapp_app.serve(app="image_generation_model:app", port=8000, reload=True)

用法

1

创建虚拟环境

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

python3 -m venv .venv
source .venv/bin/activate
2

设置您的 API 密钥

export GOOGLE_API_KEY=xxx
3

安装库

pip install -U agno google-generativeai "uvicorn[standard]"
4

运行 Agent

python cookbook/apps/whatsapp/image_generation_model.py