图片生成模型

本示例展示了如何使用图片生成模型通过 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 虚拟环境。

   Mac

   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

   Mac

   python cookbook/apps/whatsapp/image_generation_model.py