pip install google-adk

# 設(shè)置kimi官方開放平臺申請API密鑰
MOONSHOT_API_KEY="你的kimi-k3模型的API密鑰"
# 設(shè)置訪問的模型的基址
MOONSHOT_API_BASE="https://api.moonshot.cn/v1"

from . import agent

import os
import litellm
# LiteLLM 配置
# 禁用遠(yuǎn)程成本映射獲取（避免網(wǎng)絡(luò)問題）
litellm.local_model_cost_map = True
# 禁用遠(yuǎn)程成本映射獲?。ū苊饩W(wǎng)絡(luò)問題）
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
# 設(shè)置 LiteLLM 日志級別為 ERROR（避免過多日志輸出）
os.environ["LITELLM_LOG"] = "ERROR"
# 設(shè)置 LiteLLM 請求超時時間為 60 秒（避免網(wǎng)絡(luò)問題）
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加載當(dāng)前目錄的.env文件的環(huán)境變量 
from dotenv import load_dotenv
load_dotenv()
# 導(dǎo)入google adk的相關(guān)庫
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: 狀態(tài)和結(jié)果或錯誤的信息.
    """
    if city == "南昌":
        return {
            "status": "success",
            "report": (
                "今天南昌的天氣是晴朗，最高氣溫為35攝氏度。"
                " 天氣炎熱，請注意防暑降溫。"
            ),
        }
    else:
        return {
            "status": "error",
            "error_message": f"沒有找到 '{city}' 的天氣信息。",
        }

# 創(chuàng)建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"),
        }
    )
# 創(chuàng)建智能體
root_agent = Agent(  
    name="weather_assistant",  
    model=model,  
    instruction="你是一個有幫助的助手。需要時使用get_weather工具獲取天氣信息。",  
    description="一個獲取天氣信息的助手。",  
    tools=[get_weather]  
)

adk web