init

ADKLearning/google-adk-tutorial/code/deploy_demo.py

@@ -0,0 +1,201 @@
+"""
+Google ADK 部署配置完整示例
+展示各种部署方式的配置
+
+对应教程：第09章 - 部署指南
+"""
+
+# 导入 ADK 核心模块
+from google.adk.agents import Agent          # Agent 类
+from google.adk.tools import google_search  # Google 搜索工具
+
+
+# ========================================
+# 定义 Agent（部署入口）
+# ========================================
+
+def get_weather(city: str) -> dict:
+    """
+    获取天气信息
+
+    Args:
+        city (str): 城市名称
+
+    Returns:
+        dict: 天气信息
+    """
+    weather_data = {                        # 天气数据
+        "北京": {"temp": "25°C", "condition": "晴天"},
+        "上海": {"temp": "28°C", "condition": "多云"},
+    }
+    data = weather_data.get(city)           # 查找数据
+    if not data:                            # 如果找不到
+        return {"status": "error", "error_message": f"未找到'{city}'的天气信息"}
+    return {"status": "success", "city": city, **data}  # 返回天气
+
+
+# root_agent 是 ADK 的入口点
+# 部署时 ADK 会自动查找这个变量
+root_agent = Agent(
+    name="deploy_agent",                    # Agent 名称
+    model="gemini-2.0-flash",               # 模型
+    description="部署演示 Agent。",          # 描述
+    instruction=(                           # 系统指令
+        "你是一个多功能助手。\n"
+        "可以查询天气和搜索信息。\n"
+        "使用中文回答。"
+    ),
+    tools=[get_weather, google_search],     # 注册工具
+)
+
+
+# ========================================
+# 以下为部署配置参考
+# ========================================
+
+# ----------------------------------------
+# Dockerfile 内容（保存为 Dockerfile）
+# ----------------------------------------
+"""
+# 使用 Python 3.11 作为基础镜像
+FROM python:3.11-slim
+
+# 设置工作目录
+WORKDIR /app
+
+# 设置环境变量
+ENV PYTHONUNBUFFERED=1
+
+# 复制依赖文件
+COPY requirements.txt .
+
+# 安装依赖
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 复制 Agent 代码
+COPY deploy_demo.py .
+
+# 暴露端口
+EXPOSE 8080
+
+# 启动 API Server
+CMD ["adk", "api_server", "--port", "8080", "--host", "0.0.0.0"]
+"""
+
+# ----------------------------------------
+# requirements.txt 内容
+# ----------------------------------------
+"""
+google-adk>=1.0.0
+"""
+
+# ----------------------------------------
+# start.sh 启动脚本
+# ----------------------------------------
+"""
+#!/bin/bash
+# ADK Agent 启动脚本
+
+# 从环境变量读取 API Key
+export GOOGLE_API_KEY="${GOOGLE_API_KEY}"
+
+# 启动 API Server
+adk api_server \\
+    --port 8080 \\
+    --host 0.0.0.0
+"""
+
+# ----------------------------------------
+# docker-compose.yml 内容
+# ----------------------------------------
+"""
+version: '3.8'
+
+services:
+  adk-agent:
+    build: .
+    ports:
+      - "8080:8080"
+    environment:
+      - GOOGLE_API_KEY=${GOOGLE_API_KEY}
+    restart: unless-stopped
+"""
+
+# ----------------------------------------
+# 部署命令参考
+# ----------------------------------------
+"""
+# 构建镜像
+docker build -t my-adk-agent .
+
+# 运行容器
+docker run -d \\
+    --name my-agent \\
+    -p 8080:8080 \\
+    -e GOOGLE_API_KEY="your_api_key" \\
+    my-adk-agent
+
+# 使用 adk deploy 部署到 Cloud Run
+adk deploy . --platform cloud-run
+
+# 使用 adk deploy 部署到 Vertex AI
+adk deploy . --platform vertex-ai
+"""
+
+
+# ========================================
+# API 调用示例
+# ========================================
+
+def api_call_example():
+    """
+    调用 ADK API Server 的示例代码
+    需要先启动 API Server: adk api_server --port 8080
+    """
+
+    import requests                          # HTTP 请求库
+    import json                              # JSON 处理
+
+    API_BASE = "http://localhost:8080"       # API 地址
+    APP_NAME = "deploy_agent"               # 应用名称
+    USER_ID = "user_001"                    # 用户 ID
+    SESSION_ID = "session_001"              # 会话 ID
+
+    # 创建会话
+    session_url = f"{API_BASE}/apps/{APP_NAME}/users/{USER_ID}/sessions"
+    response = requests.post(                # 发送请求
+        session_url,                         # URL
+        json={"session_id": SESSION_ID},     # 请求体
+    )
+    print(f"会话创建: {response.json()}")     # 打印结果
+
+    # 发送消息
+    run_url = f"{session_url}/{SESSION_ID}:run"
+    payload = {                             # 请求体
+        "user_id": USER_ID,                 # 用户 ID
+        "session_id": SESSION_ID,           # 会话 ID
+        "new_message": {                    # 新消息
+            "role": "user",                 # 角色
+            "parts": [{"text": "北京天气怎么样？"}],  # 内容
+        },
+    }
+
+    response = requests.post(                # 发送请求
+        run_url,                             # URL
+        json=payload,                        # 请求体
+        stream=True,                         # 流式响应
+    )
+
+    for line in response.iter_lines():       # 逐行读取
+        if line:                             # 如果有内容
+            data = json.loads(line)          # 解析 JSON
+            print(f"事件: {data}")           # 打印事件
+
+
+if __name__ == "__main__":
+    print("✅ 部署配置示例")                 # 打印信息
+    print("请参考文件中的注释配置部署。")      # 提示
+    print("\n快速启动命令:")                  # 打印命令
+    print("  adk api_server --port 8080     # 启动 API Server")
+    print("  adk web --port 8000            # 启动 Web UI")
+    print("  adk deploy . --platform cloud-run  # 部署到 Cloud Run")