函数调用

用户提问 → 模型判断需要调用函数 → 返回函数名和参数
    → 你的程序执行函数 → 将结果发回模型 → 模型生成最终回复

from openai import OpenAI
import json
 
client = OpenAI(
    api_key="<ACCELE_AI_API_KEY>",
    base_url="https://api.acceleai.cn/v1"
)
 
# 第一步：定义工具
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"]
            }
        }
    }
]
 
# 第二步：发送请求，让模型决定是否调用工具
response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "user", "content": "北京今天天气怎么样？"}
    ],
    tools=tools,
    tool_choice="auto"
)
 
message = response.choices[0].message
 
# 第三步：检查模型是否要求调用函数
if message.tool_calls:
    tool_call = message.tool_calls[0]
    function_name = tool_call.function.name
    arguments = json.loads(tool_call.function.arguments)
    
    print(f"模型请求调用: {function_name}({arguments})")
    # 输出: 模型请求调用: get_weather({"city": "北京"})
    
    # 第四步：执行函数（这里用模拟数据）
    weather_result = json.dumps({
        "city": "北京",
        "temperature": 22,
        "condition": "晴",
        "humidity": 45
    }, ensure_ascii=False)
    
    # 第五步：将函数结果发回模型
    final_response = client.chat.completions.create(
        model="gpt-4o",
        messages=[
            {"role": "user", "content": "北京今天天气怎么样？"},
            message,  # 包含 tool_calls 的助手消息
            {
                "role": "tool",
                "tool_call_id": tool_call.id,
                "content": weather_result
            }
        ],
        tools=tools
    )
    
    print(final_response.choices[0].message.content)
    # 输出: 北京今天天气晴朗，气温22°C，湿度45%，适合外出活动。

curl https://api.acceleai.cn/v1/chat/completions \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer <ACCELE_AI_API_KEY>" \
  -d '{
    "model": "gpt-4o",
    "messages": [
      {"role": "user", "content": "北京今天天气怎么样？"}
    ],
    "tools": [
      {
        "type": "function",
        "function": {
          "name": "get_weather",
          "description": "获取指定城市的当前天气信息",
          "parameters": {
            "type": "object",
            "properties": {
              "city": {"type": "string", "description": "城市名称"},
              "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]}
            },
            "required": ["city"]
          }
        }
      }
    ],
    "tool_choice": "auto"
  }'

tools = [
    {
        "type": "function",
        "function": {
            "name": "get_weather",
            "description": "获取指定城市的当前天气",
            "parameters": {
                "type": "object",
                "properties": {
                    "city": {"type": "string"}
                },
                "required": ["city"]
            }
        }
    },
    {
        "type": "function",
        "function": {
            "name": "search_restaurants",
            "description": "搜索指定城市的餐厅",
            "parameters": {
                "type": "object",
                "properties": {
                    "city": {"type": "string"},
                    "cuisine": {"type": "string", "description": "菜系，如：中餐、日料、西餐"}
                },
                "required": ["city"]
            }
        }
    }
]
 
response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "user", "content": "北京天气怎么样？顺便推荐几家好吃的日料店。"}
    ],
    tools=tools,
    tool_choice="auto"
)
 
# 模型可能同时调用 get_weather 和 search_restaurants
for tool_call in response.choices[0].message.tool_calls:
    print(f"调用: {tool_call.function.name}({tool_call.function.arguments})")

stream = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": "北京天气如何？"}],
    tools=tools,
    tool_choice="auto",
    stream=True
)
 
tool_calls_data = {}
for chunk in stream:
    delta = chunk.choices[0].delta
    if delta.tool_calls:
        for tc in delta.tool_calls:
            idx = tc.index
            if idx not in tool_calls_data:
                tool_calls_data[idx] = {"id": tc.id, "name": tc.function.name, "arguments": ""}
            if tc.function.arguments:
                tool_calls_data[idx]["arguments"] += tc.function.arguments
 
for idx, tc in tool_calls_data.items():
    print(f"函数: {tc['name']}, 参数: {tc['arguments']}")