tools.py

"""
Tool definitions for Qwen3 tool calling examples.

Single source of truth - imported by both main.py and server.py.
Same tools as functionGemma for direct comparison.
"""

from langchain_core.tools import tool


# --- Simulated Data ---

WEATHER_DATA = {
    "london": "15C, cloudy",
    "tokyo": "22C, sunny",
    "new york": "18C, partly cloudy",
    "mumbai": "32C, humid",
}

CONTACTS = {
    "alice": "alice@example.com | +1-555-0101",
    "bob": "bob@example.com | +1-555-0102",
    "charlie": "charlie@example.com | +1-555-0103",
}


# --- Tool Definitions ---

@tool
def get_weather(city: str) -> str:
    """Get the current weather for a given city.

    Args:
        city: Name of the city to get weather for.
    """
    return WEATHER_DATA.get(city.lower(), f"No weather data available for {city}")


@tool
def add_numbers(a: int, b: int) -> int:
    """Add two numbers together and return the result.

    Args:
        a: First number.
        b: Second number.
    """
    return a + b


@tool
def search_contacts(name: str) -> str:
    """Search for a contact by name in the address book.

    Args:
        name: Name of the contact to search for.
    """
    return CONTACTS.get(name.lower(), f"No contact found for '{name}'")


ALL_TOOLS = [get_weather, add_numbers, search_contacts]
TOOL_MAP = {t.name: t for t in ALL_TOOLS}