도구 호출

import OpenAI from "openai";

const client = new OpenAI({
  baseURL: "https://api.arouter.ai/v1",
  apiKey: "lr_live_xxxx",
});

// 도구 정의
const tools: OpenAI.Chat.ChatCompletionTool[] = [
  {
    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. 'San Francisco'",
          },
          unit: { type: "string", enum: ["celsius", "fahrenheit"] },
        },
        required: ["location"],
      },
    },
  },
];

const messages: OpenAI.Chat.ChatCompletionMessageParam[] = [
  { role: "user", content: "What is the weather in San Francisco?" },
];

// 1단계: 첫 번째 추론
const response = await client.chat.completions.create({
  model: "openai/gpt-5.4",
  messages,
  tools,
  tool_choice: "auto",
});

const assistantMessage = response.choices[0].message;
messages.push(assistantMessage);

// 2단계: 도구 실행
if (assistantMessage.tool_calls) {
  for (const toolCall of assistantMessage.tool_calls) {
    if (toolCall.function.name === "get_weather") {
      const args = JSON.parse(toolCall.function.arguments);
      const result = {
        temperature: 72,
        condition: "sunny",
        unit: args.unit ?? "fahrenheit",
      };

      messages.push({
        role: "tool",
        tool_call_id: toolCall.id,
        content: JSON.stringify(result),
      });
    }
  }
}

// 3단계: 최종 추론
const finalResponse = await client.chat.completions.create({
  model: "openai/gpt-5.4",
  messages,
  tools,
});

console.log(finalResponse.choices[0].message.content);

package main

import (
    "context"
    "encoding/json"
    "fmt"
    "log"

    "github.com/arouter-ai/arouter-go"
)

func main() {
    client := arouter.NewClient("lr_live_xxxx",
        arouter.WithBaseURL("https://api.arouter.ai/v1"),
    )

    tools := []arouter.Tool{
        {
            Type: "function",
            Function: &arouter.FunctionDefinition{
                Name:        "get_weather",
                Description: "Get the current weather for a location",
                Parameters: map[string]any{
                    "type": "object",
                    "properties": map[string]any{
                        "location": map[string]any{
                            "type":        "string",
                            "description": "City name",
                        },
                        "unit": map[string]any{
                            "type": "string",
                            "enum": []string{"celsius", "fahrenheit"},
                        },
                    },
                    "required": []string{"location"},
                },
            },
        },
    }

    messages := []arouter.Message{
        {Role: "user", Content: "What is the weather in San Francisco?"},
    }

    // 1단계
    resp, err := client.CreateChatCompletion(context.Background(), arouter.ChatCompletionRequest{
        Model:    "openai/gpt-5.4",
        Messages: messages,
        Tools:    tools,
    })
    if err != nil {
        log.Fatal(err)
    }

    assistantMsg := resp.Choices[0].Message
    messages = append(messages, assistantMsg)

    // 2단계 & 3단계
    for _, tc := range assistantMsg.ToolCalls {
        var args map[string]string
        json.Unmarshal([]byte(tc.Function.Arguments), &args)
        result := fmt.Sprintf(`{"temperature":72,"condition":"sunny","unit":"%s"}`, args["unit"])
        messages = append(messages, arouter.Message{
            Role:       "tool",
            ToolCallID: tc.ID,
            Content:    result,
        })
    }

    finalResp, err := client.CreateChatCompletion(context.Background(), arouter.ChatCompletionRequest{
        Model:    "openai/gpt-5.4",
        Messages: messages,
        Tools:    tools,
    })
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println(finalResp.Choices[0].Message.Content)
}

# 1단계: 도구를 포함한 추론
curl https://api.arouter.ai/v1/chat/completions \
  -H "Authorization: Bearer lr_live_xxxx" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "openai/gpt-5.4",
    "messages": [
      {"role": "user", "content": "What is the weather in San Francisco?"}
    ],
    "tools": [
      {
        "type": "function",
        "function": {
          "name": "get_weather",
          "description": "Get the current weather for a location",
          "parameters": {
            "type": "object",
            "properties": {
              "location": {"type": "string"},
              "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]}
            },
            "required": ["location"]
          }
        }
      }
    ],
    "tool_choice": "auto"
  }'

# 3단계: 도구 결과 전송
curl https://api.arouter.ai/v1/chat/completions \
  -H "Authorization: Bearer lr_live_xxxx" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "openai/gpt-5.4",
    "messages": [
      {"role": "user", "content": "What is the weather in San Francisco?"},
      {
        "role": "assistant",
        "content": null,
        "tool_calls": [{"id": "call_abc123", "type": "function", "function": {"name": "get_weather", "arguments": "{\"location\": \"San Francisco\"}"}}]
      },
      {
        "role": "tool",
        "tool_call_id": "call_abc123",
        "content": "{\"temperature\": 72, \"condition\": \"sunny\"}"
      }
    ]
  }'