model=LiteLlm(model="openrouter/openai/gpt-4o-mini")

import datetime
from zoneinfo import ZoneInfo
from google.adk.agents import Agent
from google.adk.models.lite_llm import LiteLlm

def get_weather(city: str) -> dict:
    """Retrieves the current weather report for a specified city.

    Args:
        city (str): The name of the city for which to retrieve the weather report.

    Returns:
        dict: status and result or error msg.
    """
    if city.lower() == "new york":
        return {
            "status": "success",
            "report": (
                "The weather in New York is sunny with a temperature of 25 degrees"
                " Celsius (41 degrees Fahrenheit)."
            ),
        }
    else:
        return {
            "status": "error",
            "error_message": f"Weather information for '{city}' is not available.",
        }


def get_current_time(city: str) -> dict:
    """Returns the current time in a specified city.

    Args:
        city (str): The name of the city for which to retrieve the current time.

    Returns:
        dict: status and result or error msg.
    """

    if city.lower() == "new york":
        tz_identifier = "America/New_York"
    else:
        return {
            "status": "error",
            "error_message": (
                f"Sorry, I don't have timezone information for {city}."
            ),
        }

    tz = ZoneInfo(tz_identifier)
    now = datetime.datetime.now(tz)
    report = (
        f'The current time in {city} is {now.strftime("%Y-%m-%d %H:%M:%S %Z%z")}'
    )
    return {"status": "success", "report": report}


root_agent = Agent(
    name="weather_time_agent",
    model=LiteLlm(model="openrouter/openai/gpt-4o-mini"),
    description=(
        "Agent to answer questions about the time and weather in a city.  If a city"
        " is not provided, ask for clarification."
    ),
    instruction=(
        "You are a helpful agent. Your task is to answer user questions about"
        " time or weather using the provided tools. "
        "1. Identify the required tool (get_weather or get_current_time) and the city."
        "2. Call the identified tool **exactly once**."
        "3. After receiving the response from the tool, **immediately** formulate a final answer for the user based **only** on the tool's response."
        "4. **Do not call any tool more than once per user request.**"
        "5. If the user input is unclear or a tool returns an error, inform the user."
    ),
    tools=[get_weather, get_current_time],
)