Tool calling

import json

tools = [{
    "type": "function",
    "function": {
        "name": "get_weather",
        "description": "Current weather for a city",
        "parameters": {"type": "object", "properties": {"city": {"type": "string"}}, "required": ["city"]},
    },
}]
messages = [{"role": "user", "content": "What's the weather in Riyadh?"}]

first = client.chat.completions.create(model="a1", messages=messages, tools=tools)
call = first.choices[0].message.tool_calls[0]
result = {"city": json.loads(call.function.arguments)["city"], "temp_c": 31}

messages += [first.choices[0].message, {"role": "tool", "tool_call_id": call.id, "content": json.dumps(result)}]
second = client.chat.completions.create(model="a1", messages=messages, tools=tools)
print(second.choices[0].message.content)

tools = [{
    "name": "get_weather",
    "description": "Current weather for a city",
    "input_schema": {"type": "object", "properties": {"city": {"type": "string"}}, "required": ["city"]},
}]
messages = [{"role": "user", "content": "What's the weather in Riyadh?"}]

first = client.messages.create(model="a1", max_tokens=1024, tools=tools, messages=messages)
use = next(b for b in first.content if b.type == "tool_use")

messages += [
    {"role": "assistant", "content": first.content},
    {"role": "user", "content": [{"type": "tool_result", "tool_use_id": use.id, "content": '{"temp_c": 31}'}]},
]
second = client.messages.create(model="a1", max_tokens=1024, tools=tools, messages=messages)
print(next(b.text for b in second.content if b.type == "text"))