追踪 Google ADK 应用

from langsmith.integrations.google_adk import configure_google_adk

configure_google_adk(
    project_name="my-adk-project",  # 可选：默认为 LANGSMITH_PROJECT 环境变量
)

import asyncio

from dotenv import load_dotenv  # 可选
from google.adk.agents import Agent
from google.adk.runners import Runner
from google.adk.sessions import InMemorySessionService
from google.genai import types
from langsmith.integrations.google_adk import configure_google_adk

load_dotenv()  # 可选


async def main():
    # 配置 LangSmith 追踪
    configure_google_adk()

    # 定义一个工具
    def get_weather(city: str) -> dict:
        """获取城市的天气信息。"""
        return {"city": city, "temperature": "72°F", "conditions": "Sunny"}

    # 创建代理
    agent = Agent(
        name="weather_agent",
        model="gemini-2.0-flash",
        description="提供天气信息。",
        instruction="使用 get_weather 工具回答天气问题。",
        tools=[get_weather],
    )

    # 设置会话和运行器
    session_service = InMemorySessionService()
    session = await session_service.create_session(
        app_name="weather_app",
        user_id="user_123",
        session_id="session_456",
    )

    runner = Runner(
        agent=agent,
        app_name="weather_app",
        session_service=session_service,
    )

    # 运行代理
    async for event in runner.run_async(
        user_id="user_123",
        session_id=session.id,
        new_message=types.Content(
            role="user",
            parts=[types.Part(text="旧金山的天气怎么样？")],
        ),
    ):
        if event.is_final_response():
            print(event.content.parts[0].text)


if __name__ == "__main__":
    asyncio.run(main())

from langsmith.integrations.google_adk import configure_google_adk

configure_google_adk(
    project_name="production-agents",
    metadata={
        "environment": "production",
        "team": "ml-platform",
    },
    tags=["adk", "weather", "v2"],
)

import asyncio

from dotenv import load_dotenv  # 可选
from google.adk.agents import Agent, SequentialAgent
from google.adk.runners import Runner
from google.adk.sessions import InMemorySessionService
from google.genai import types
from langsmith.integrations.google_adk import configure_google_adk

load_dotenv()  # 可选


async def main():
    # 配置 LangSmith 追踪
    # 默认情况下，追踪数据会发送到 LANGSMITH_PROJECT 环境变量指定的项目。
    # 可通过传递 project_name="my-project" 来覆盖。
    configure_google_adk()

    # 创建子代理
    translator = Agent(
        name="translator",
        model="gemini-2.0-flash",
        description="将文本翻译成英语。",
    )

    summarizer = Agent(
        name="summarizer",
        model="gemini-2.0-flash",
        description="简洁地总结文本。",
    )

    # 创建一个按顺序运行子代理的顺序代理
    pipeline = SequentialAgent(
        name="translate_and_summarize",
        sub_agents=[translator, summarizer],
        description="先翻译文本，然后进行总结。",
    )

    # 设置并运行
    session_service = InMemorySessionService()
    session = await session_service.create_session(
        app_name="pipeline_app",
        user_id="user_123",
        session_id="session_456",
    )

    runner = Runner(
        agent=pipeline,
        app_name="pipeline_app",
        session_service=session_service,
    )

    events = runner.run_async(
        user_id="user_123",
        session_id=session.id,
        new_message=types.Content(
            role="user",
            parts=[types.Part(text="Quelle est la plus haute tour de Paris?")],
        ),
    )

    async for event in events:
        if event.is_final_response():
            print(event.content.parts[0].text)


if __name__ == "__main__":
    asyncio.run(main())