Tool calling

weather_tool = {
    "type": "function",
    "function": {
        "name": "get_weather",
        "description": "Get the weather for a city.",
        "parameters": {
            "type": "object",
            "properties": {"city": {"type": "string"}},
            "required": ["city"],
        },
    },
}

import json

import routehub


def get_weather(city):
    return f"Sunny in {city}"


chat = [{"role": "user", "content": "What's the weather in Paris and Tokyo?"}]

while True:
    response = routehub.completion(
        model="claude-sonnet-4-6", messages=chat, tools=[weather_tool]
    )
    message = response.choices[0].message
    chat.append(message.model_dump(exclude_none=True))
    if not message.tool_calls:
        break
    for call in message.tool_calls:
        result = get_weather(**json.loads(call.function.arguments))
        chat.append(
            {"role": "tool", "tool_call_id": call.id, "content": result}
        )

print(message.content)

response = routehub.completion(
    model="gpt-5.4-mini",
    messages=chat,
    tools=[weather_tool],
    tool_choice={"type": "function", "function": {"name": "get_weather"}},
    parallel_tool_calls=False,
)