Google ADK observability

import asyncio

from google.adk.agents import Agent
from google.adk.runners import InMemoryRunner
from google.genai import types
from lmnr import Laminar, observe

Laminar.initialize()


def get_weather(city: str) -> dict:
    """Returns the current weather for a city."""
    return {"city": city, "forecast": "sunny", "temperature_c": 22}


def get_local_time(city: str) -> dict:
    """Returns the current local time for a city."""
    return {"city": city, "time": "14:05"}


agent = Agent(
    name="travel_assistant",
    model="gemini-2.5-flash",
    instruction="You are a travel assistant. Use the tools to answer questions about a city.",
    tools=[get_weather, get_local_time],
)


@observe(name="travel-question")
async def main():
    runner = InMemoryRunner(agent=agent, app_name="travel_app")
    session = await runner.session_service.create_session(
        app_name="travel_app", user_id="user-123"
    )
    message = types.Content(
        role="user",
        parts=[types.Part(text="What's the weather and local time in Tokyo?")],
    )
    async for event in runner.run_async(
        user_id="user-123", session_id=session.id, new_message=message
    ):
        if event.is_final_response() and event.content and event.content.parts:
            print(event.content.parts[0].text)


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