Chat Completions

response1 = client.chat.completions.create(
    model="HCX-GOV-THINK",
    messages=[
        {"role": "user", "content": "안녕하세요!"}
    ],
    max_tokens=4096,
    stop=["<|im_end|><|endofturn|>", "<|im_end|><|stop|>"],
    extra_body={"skip_special_tokens": False},
)

messages = [
    {"role": "user", "content": "안녕하세요!"},
    {"role": "assistant", "content": response1.choices[0].message.content},
    {"role": "user", "content": "오늘 날씨가 어때요?"}
]

response2 = client.chat.completions.create(
    model="HCX-GOV-THINK",
    messages=messages,
    max_tokens=4096,
    stop=["<|im_end|><|endofturn|>", "<|im_end|><|stop|>"],
    extra_body={"skip_special_tokens": False},
)

import json
from openai import OpenAI

client = OpenAI(
    base_url="https://api.clovastudio.go.kr/api/v1",
    api_key="YOUR_API_KEY",
)

tools = [
    {
        "type": "function",
        "function": {
            "name": "get_document_info",
            "description": "문서 관리 시스템에서 문서 정보를 조회합니다.",
            "parameters": {
                "type": "object",
                "required": ["document_id"],
                "properties": {
                    "document_id": {
                        "type": "string",
                        "description": "조회할 문서의 고유 ID"
                    }
                }
            }
        }
    }
]

response = client.chat.completions.create(
    model="HCX-GOV-THINK",
    messages=[
        {"role": "system", "content": "도구가 필요할 때는 tool_calls만 반환하세요."},
        {"role": "user", "content": "문서 ID 'DOC-2024-001'의 정보를 알려주세요."}
    ],
    tools=tools,
    max_tokens=512,
)

tool_call = response.choices[0].message.tool_calls[0]
function_name = tool_call.function.name
arguments = json.loads(tool_call.function.arguments)

tool_result = {"title": "2024년도 예산안", "status": "승인", "date": "2024-01-15"}

messages = [
    {"role": "system", "content": "도구가 필요할 때는 tool_calls만 반환하세요."},
    {"role": "user", "content": "문서 ID 'DOC-2024-001'의 정보를 알려주세요."},
    {
        "role": "assistant",
        "content": "",
        "tool_calls": [
            {
                "id": tool_call.id,
                "type": "function",
                "function": {
                    "name": tool_call.function.name,
                    "arguments": tool_call.function.arguments
                }
            }
        ]
    },
    {
        "role": "tool",
        "tool_call_id": tool_call.id,
        "content": json.dumps(tool_result, ensure_ascii=False)
    }
]

final_response = client.chat.completions.create(
    model="HCX-GOV-THINK",
    messages=messages,
    tools=tools,
    max_tokens=1024,
)
print(final_response.choices[0].message.content)