Google ADK搭建基于Kimi K3的智能体详细过程

pip install google-adk

# 设置kimi官方开放平台申请API密钥
MOONSHOT_API_KEY="你的kimi-k3模型的API密钥"
# 设置访问的模型的基址
MOONSHOT_API_BASE="https://api.moonshot.cn/v1"

from . import agent

import os
import litellm
# LiteLLM 配置
# 禁用远程成本映射获取（避免网络问题）
litellm.local_model_cost_map = True
# 禁用远程成本映射获取（避免网络问题）
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
# 设置 LiteLLM 日志级别为 ERROR（避免过多日志输出）
os.environ["LITELLM_LOG"] = "ERROR"
# 设置 LiteLLM 请求超时时间为 60 秒（避免网络问题）
os.environ["LITELLM_REQUEST_TIMEOUT"] = "60"
# 为 LiteLLM 添加一个“非官方/自定义”kimi-k3模型的配置
# 注册 kimi-k3 模型成本
litellm.model_cost = litellm.model_cost or {}
litellm.model_cost["moonshot/kimi-k3"] = {
    "input_cost_per_token": 0.000001,
    "output_cost_per_token": 0.000002,
    "litellm_provider": "openai",
    "mode": "chat",
    "supports_system_messages": True,
    "supports_function_calls": True,
}
print("自定义LiteLLM 配置完成")

# 利用load_dotenv加载当前目录的.env文件的环境变量 
from dotenv import load_dotenv
load_dotenv()
# 导入google adk的相关库
from google.adk.models.lite_llm import LiteLlm
from google.adk.tools import FunctionTool
from google.adk.agents import Agent  

@FunctionTool
def get_weather(city: str) -> dict:
    """
    获取指定城市的天气信息.
    Args:
        city (str): 城市名称.
    Returns:
        dict: 状态和结果或错误的信息.
    """
    if city == "南昌":
        return {
            "status": "success",
            "report": (
                "今天南昌的天气是晴朗，最高气温为35摄氏度。"
                " 天气炎热，请注意防暑降温。"
            ),
        }
    else:
        return {
            "status": "error",
            "error_message": f"没有找到 '{city}' 的天气信息。",
        }

# 创建LiteLlm模型对象
model = LiteLlm(
        model="moonshot/kimi-k3",
        temperature=1,
        max_tokens=2048,
        extra_params={
            "api_base": os.getenv("MOONSHOT_API_BASE"),
            "api_key": os.getenv("MOONSHOT_API_KEY"),
        }
    )
# 创建智能体
root_agent = Agent(  
    name="weather_assistant",  
    model=model,  
    instruction="你是一个有帮助的助手。需要时使用get_weather工具获取天气信息。",  
    description="一个获取天气信息的助手。",  
    tools=[get_weather]  
)

adk web