Tool Calling: Единый цикл для Claude, GPT и Open Source

pip install openai

import json
import os
from openai import OpenAI

# Инициализация клиента OpenRouter
client = OpenAI(
    base_url="https://openrouter.ai/api/v1",
    api_key=os.environ["OPENROUTER_API_KEY"],
)

# Схема инструмента
tools = [{
    "type": "function",
    "function": {
        "name": "get_weather",
        "description": "Get the current weather for a location.",
        "parameters": {
            "type": "object",
            "properties": {
                "location": {"type": "string", "description": "City name, e.g. 'Paris'"},
                "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]}
            },
            "required": ["location"]
        }
    }
}]

def get_weather(location, unit="celsius"):
    # Здесь должна быть реальная логика (API запрос)
    return {"location": location, "temperature": 18, "unit": unit, "sky": "clear"}

def run_tool_loop(model, user_message):
    messages = [{"role": "user", "content": user_message}]
    
    # Шаг 1: Запрос к модели
    response = client.chat.completions.create(
        model=model,
        messages=messages,
        tools=tools
    )
    
    msg = response.choices[0].message
    
    # Если инструмент не нужен, возвращаем прямой ответ
    if not msg.tool_calls:
        return msg.content
    
    # Шаг 2: Добавляем ответ ассистента в контекст
    messages.append(msg)
    
    # Шаг 3: Исполняем инструменты
    for call in msg.tool_calls:
        args = json.loads(call.function.arguments)
        result = get_weather(**args)
        messages.append({
            "role": "tool",
            "tool_call_id": call.id,
            "content": json.dumps(result)
        })
    
    # Шаг 4: Запрос финального ответа
    final = client.chat.completions.create(
        model=model,
        messages=messages,
        tools=tools
    )
    
    return final.choices[0].message.content

import OpenAI from "openai";

const client = new OpenAI({
  baseURL: "https://openrouter.ai/api/v1",
  apiKey: process.env.OPENROUTER_API_KEY,
});

const tools = [{
  type: "function",
  function: {
    name: "get_weather",
    description: "Get the current weather for a location.",
    parameters: {
      type: "object",
      properties: {
        location: { type: "string", description: "City name, e.g. 'Paris'" },
        unit: { type: "string", enum: ["celsius", "fahrenheit"] }
      },
      required: ["location"]
    }
  }
}];

function getWeather(location, unit = "celsius") {
  return { location, temperature: 18, unit, sky: "clear" };
}

async function runToolLoop(model, userMessage) {
  const messages = [{ role: "user", content: userMessage }];
  
  const response = await client.chat.completions.create({ model, messages, tools });
  const msg = response.choices[0].message;
  
  if (!msg.tool_calls) return msg.content;
  
  messages.push(msg);
  
  for (const call of msg.tool_calls) {
    const args = JSON.parse(call.function.arguments);
    const result = getWeather(args.location, args.unit);
    messages.push({
      role: "tool",
      tool_call_id: call.id,
      content: JSON.stringify(result)
    });
  }
  
  const final = await client.chat.completions.create({ model, messages, tools });
  return final.choices[0].message.content;
}