Function Calling (вызов функций)

from openai import OpenAI
import json

client = OpenAI(
    api_key="sk-aitunnel-xxx",
    base_url="https://api.aitunnel.ru/v1/"
)

# Определяем инструменты
tools = [{
    "type": "function",
    "function": {
        "name": "get_weather",
        "description": "Получить текущую погоду в городе",
        "parameters": {
            "type": "object",
            "properties": {
                "city": {
                    "type": "string",
                    "description": "Название города"
                }
            },
            "required": ["city"]
        }
    }
}]

# Функция для получения погоды
def get_weather(city):
    # Здесь был бы реальный API запрос
    return {"city": city, "temp": 20, "condition": "Солнечно"}

# Запрос к AI
response = client.chat.completions.create(
    model="gpt-5.2",
    messages=[{"role": "user", "content": "Какая погода в Москве?"}],
    tools=tools,
    tool_choice="auto"
)

# Обработка вызова функции
if response.choices[0].message.tool_calls:
    tool_call = response.choices[0].message.tool_calls[0]
    args = json.loads(tool_call.function.arguments)
    
    # Вызываем функцию
    result = get_weather(args["city"])
    
    # Отправляем результат обратно
    final_response = client.chat.completions.create(
        model="gpt-5.2",
        messages=[
            {"role": "user", "content": "Какая погода в Москве?"},
            response.choices[0].message,
            {
                "role": "tool",
                "tool_call_id": tool_call.id,
                "content": json.dumps(result)
            }
        ]
    )
    
    print(final_response.choices[0].message.content)