함수 호출 (Function Calling)

from openai import OpenAI
import json
 
client = OpenAI(
    base_url="https://api.ofox.ai/v1",
    api_key="<OFOXAI_API_KEY>"
)
 
# 1. 도구 정의
tools = [{
    "type": "function",
    "function": {
        "name": "get_weather",
        "description": "지정한 도시의 실시간 날씨 정보를 가져옵니다",
        "parameters": {
            "type": "object",
            "properties": {
                "city": {
                    "type": "string",
                    "description": "도시 이름, 예: 서울, 부산"
                },
                "unit": {
                    "type": "string",
                    "enum": ["celsius", "fahrenheit"],
                    "description": "온도 단위"
                }
            },
            "required": ["city"]
        }
    }
}]
 
# 2. 요청 전송
messages = [{"role": "user", "content": "서울 오늘 날씨 어때요?"}]
 
response = client.chat.completions.create(
    model="openai/gpt-4o",
    messages=messages,
    tools=tools,
    tool_choice="auto"
)
 
message = response.choices[0].message
 
# 3. 도구 호출 처리
if message.tool_calls:
    for tool_call in message.tool_calls:
        args = json.loads(tool_call.function.arguments)
 
        # 4. 함수 실행
        result = get_weather(args["city"])  # 직접 구현한 함수
 
        # 5. 결과를 모델에 다시 전송
        messages.append(message)
        messages.append({
            "role": "tool",
            "tool_call_id": tool_call.id,
            "content": json.dumps(result)
        })
 
    # 최종 응답 받기
    final = client.chat.completions.create(
        model="openai/gpt-4o",
        messages=messages,
        tools=tools
    )
    print(final.choices[0].message.content)

import OpenAI from 'openai'
 
const client = new OpenAI({
  baseURL: 'https://api.ofox.ai/v1',
  apiKey: '<OFOXAI_API_KEY>'
})
 
// 1. 도구 정의
const tools: OpenAI.ChatCompletionTool[] = [{
  type: 'function',
  function: {
    name: 'get_weather',
    description: '지정한 도시의 실시간 날씨 정보를 가져옵니다',
    parameters: {
      type: 'object',
      properties: {
        city: { type: 'string', description: '도시 이름' },
        unit: { type: 'string', enum: ['celsius', 'fahrenheit'] }
      },
      required: ['city']
    }
  }
}]
 
// 2. 요청 전송
const messages: OpenAI.ChatCompletionMessageParam[] = [
  { role: 'user', content: '서울 오늘 날씨 어때요?' }
]
 
const response = await client.chat.completions.create({
  model: 'openai/gpt-4o',
  messages,
  tools,
  tool_choice: 'auto'
})
 
const message = response.choices[0].message
 
// 3. 도구 호출 처리
if (message.tool_calls) {
  messages.push(message)
 
  for (const toolCall of message.tool_calls) {
    const args = JSON.parse(toolCall.function.arguments)
    const result = await getWeather(args.city)  // 직접 구현한 함수
 
    messages.push({
      role: 'tool',
      tool_call_id: toolCall.id,
      content: JSON.stringify(result)
    })
  }
 
  const final = await client.chat.completions.create({
    model: 'openai/gpt-4o',
    messages,
    tools
  })
 
  console.log(final.choices[0].message.content)
}