函数调用(tool use)

import json
from openai import OpenAI

client = OpenAI(
    api_key="sk-你的密钥",
    base_url="https://www.ruapi.ai/v1",
)

# 这是你自己的真实代码。模型并不会运行它,只会请求你去调用。
def get_weather(city: str) -> str:
    # 这里本该是一次真实的天气 API 调用
    return f"{city}当前 +18°C,局部多云"

# 给模型的函数描述:它能做什么、接收哪些参数
tools = [
    {
        "type": "function",
        "function": {
            "name": "get_weather",
            "description": "返回指定城市的当前天气",
            "parameters": {
                "type": "object",
                "properties": {
                    "city": {
                        "type": "string",
                        "description": "城市名,例如「北京」",
                    }
                },
                "required": ["city"],
            },
        },
    }
]

messages = [{"role": "user", "content": "北京现在天气怎么样?"}]

# 第 1 步。第一次请求——把问题和工具列表交给模型
response = client.chat.completions.create(
    model="claude-opus-4-8",  # 准确名称见主站定价页
    messages=messages,
    tools=tools,
)

message = response.choices[0].message

# 第 2 步。检查模型是否想调用函数
if message.tool_calls:
    # 重要:先把模型自己的回复(带 tool_calls 的那条)追加进历史
    messages.append(message)

    # 第 3 步。逐个处理调用(可能不止一个)
    for tool_call in message.tool_calls:
        # 参数是以 JSON 字符串形式传来的——必须先解析
        args = json.loads(tool_call.function.arguments)

        if tool_call.function.name == "get_weather":
            result = get_weather(args["city"])
        else:
            result = "未知函数"

        # 把调用结果作为单独的 role="tool" 消息发回去
        messages.append(
            {
                "role": "tool",
                "tool_call_id": tool_call.id,  # 与这次具体调用绑定
                "content": result,
            }
        )

    # 第 4 步。第二次请求——此时模型看到结果,给出最终回答
    final = client.chat.completions.create(
        model="claude-opus-4-8",
        messages=messages,
        tools=tools,
    )
    print(final.choices[0].message.content)
else:
    # 模型选择直接回答,没有调用函数
    print(message.content)