【干货】一步步带你认识大模型的 Function Call

import openaiimport requests
# 设置OpenAI的API密钥openai.api_key = "xxx"
# 定义一个函数，用于获取指定地点的当前天气信息def get_current_weather(location: str, unit: str = "metric"):    """    获取指定地点的当前天气信息。    参数:        location (str): 地点名称，例如 "Beijing" 或 "Shanghai"。        unit (str): 温度单位，"metric" 表示摄氏度，"imperial" 表示华氏度，默认为 "metric"。    返回:        dict: 包含温度和天气描述的字典。    """    api_key = "xxx"  # 替换为你的OpenWeatherMap API密钥     # 构造请求的URL    url = f"http://api.openweathermap.org/data/2.5/weather?q={location}&units={unit}&appid={api_key}"     try:        # 发送HTTP GET请求        response = requests.get(url)        # 检查响应状态码是否为200（成功）        if response.status_code == 200:            data = response.json()            # 从JSON数据中提取温度和天气描述            return {                "temperature": data["main"]["temp"],                "description": data["weather"][0]["description"]            }        else:            # 如果响应状态码不是200，打印错误信息            print(f"请求天气API失败，状态码：{response.status_code}")            return None
    except Exception as e:        # 捕获异常并打印错误信息        print(f"解析天气API时发生错误：{e}")        return None
# 定义函数调用的元数据functions = [    {        "name": "get_current_weather",        "description": "获取指定城市的当前天气",        "parameters": {            "type": "object",            "properties": {                "location": {                    "type": "string",                    "description": "城市名称"                },                "unit": {                    "type": "string",                    "description": "温度单位（metric 或 imperial）",                    "default": "metric"                }            },            "required": ["location"]        }    }]
# 用户输入user_input = "沈阳今天的天气如何？"
# 调用OpenAI的ChatCompletion接口response = openai.ChatCompletion.create(    model="gpt-4",    messages=[        {"role": "user", "content": user_input}    ],    functions=functions,    function_call="auto"  # 自动决定是否调用函数)
# 检查是否触发了函数调用if response.choices[0].finish_reason == "function_call":    function_args = response.choices[0].message.function_call.arguments    function_name = response.choices[0].message.function_call.name    # 如果触发的函数是get_current_weather    if function_name == "get_current_weather":        # 调用get_current_weather函数获取天气信息        weather_result = get_current_weather(            location=function_args["location"],            unit=function_args.get("unit", "metric")        )
        # 如果获取到天气信息        if weather_result:            # 再次调用ChatCompletion接口，将天气信息作为上下文传递            second_response = openai.ChatCompletion.create(                model="gpt-4",                messages=[                    {"role": "user", "content": user_input},                    {"role": "assistant", "content": response.choices[0].message.content},                    {"role": "function", "name": function_name, "arguments": function_args},                    {"role": "function", "name": function_name, "content": weather_result}                ]            )
            # 打印最终的回答            print(second_response.choices[0].message.content)        else:            # 如果天气信息获取失败，提示用户            print("无法获取天气信息，请检查网络连接或稍后再试。")
else:    # 如果没有触发函数调用，直接打印模型的回答    print(response.choices[0].message.content)