使用Python调用OpenAI的function calling源码

一、第三方库

pip install openai

二、示例代码

# -*- coding: utf-8 -*-  
# @Author ：Kan 
# @Date ：2025/8/1 9:12  
# @File ：1_function_calling.py  
import json  
import random  
from openai import OpenAI  
from datetime import datetime  
  
  
# ================================================================  
# 1. Function Calling（函数调用）  
# ================================================================  
  
class ToolsUsage:  
    tools = [  
        # 工具2 获取指定城市的天气  
        {  
            "type": "function",  
            "function": {  
                "name": "get_current_weather",  
                "description": "获取指定地点的天气信息",  
                "parameters": {  
                    "type": "object",  
                    "properties": {  
                        "location": {"type": "string", "description": "地点名称"}  
                    },  
                    "required": ["location"]  
                }  
            }  
        },  
        # 工具1 获取当前时刻的时间  
        {  
            "type": "function",  
            "function": {  
                "name": "get_current_time",  
                "description": "当你想知道现在的时间时非常有用。",  
                # 因为获取当前时间无需输入参数，因此parameters为空字典  
                "parameters": {},  
            },  
        },  
    ]  
  
    @staticmethod  
    # 天气查询工具。返回结果示例：“北京今天是雨天。”  
    def get_current_weather(location):  
        weather_conditions = ["晴天", "多云", "雨天"]  
        # 随机选择一个天气条件  
        random_weather = random.choice(weather_conditions)  
        # 返回格式化信息  
        return f"{location}今天是{random_weather}。"  
  
    @staticmethod  
    # 查询当前时间的工具。返回结果示例：“当前时间：2024-04-15 17:15:18。“  
    def get_current_time():  
        current_datetime = datetime.now()  
        # 格式化日期  
        formatted_time = current_datetime.strftime("%Y-%m-%d %H:%M:%S")  
        return f"当前时间：{formatted_time}。"  
  
    @staticmethod  
    def execute_tools(func_name, func_args):  
        func_dic = {  
            "get_current_weather": ToolsUsage.get_current_weather,  
            "get_current_time": ToolsUsage.get_current_time,  
        }  
        return func_dic[func_name](**func_args)  
  
  
class ChatAgent:  
    def __init__(self, api_key: str, url: str, model_name: str):  
        self.client = OpenAI(api_key=api_key, base_url=url)  
        self.model_name = model_name  
  
    def request_chat(self, messages: list):  
        response = self.client.chat.completions.create(  
            model=self.model_name,  
            messages=messages,  
            tools=ToolsUsage.tools,  
            extra_body={"enable_thinking": False},  
            tool_choice="auto",  
        )  
        return response  
  
    def execute_chat(self):  
        print("\n")  
        messages = [  
            {  
                # 提问示例："现在几点了？" "一个小时后几点" "北京天气如何？"  
                "content": input(  
                    "请输入问题:"  
                ),  
                "role": "user",  
            }  
        ]  
        print("-*" * 60)  
        # 模型调用次数  
        i = 1  
        first_response = self.request_chat(messages)  
        assistant_output = first_response.choices[0].message  
        print(f"\n第{i}轮大模型输出信息：{assistant_output}\n")  
        # 不需要调用工具，则直接返回答案  
        if not assistant_output.tool_calls:  
            print(f"无需调用工具，直接回复：{assistant_output.content}")  
            return  
        tool_calls_result = assistant_output.tool_calls  
        # 如果需要调用工具，则进行模型的多轮调用，直到模型判断无需调用工具  
        while tool_calls_result:  
            # 执行工具调用  
            for tool_call in assistant_output.tool_calls:  
                tool_info = {  
                    "content": "",  
                    "role": "tool",  
                    "tool_call_id": assistant_output.tool_calls[0].id,  
                }  
                func_name = tool_call.function.name  
                func_args = json.loads(tool_call.function.arguments)  
                tools_result = ToolsUsage.execute_tools(func_name, func_args)  
                print(f"当前调用工具:{func_name},参数:{func_args},输出信息：{tools_result}\n")  
                tool_info["content"] = tools_result  
                messages.append(tool_info)  
                print("-*" * 60)  
            second_response = self.request_chat(messages)  
            assistant_output = second_response.choices[0].message  
            i += 1  
            print(f"第{i}轮大模型输出信息：{assistant_output}\n")  
            # 指导调用工具为空时终止  
            if not assistant_output.tool_calls:  
                tool_calls_result = None  
        print(f"最终答案:\n {assistant_output.content}")  
  
  
if __name__ == "__main__":  
    api_key = 'xxxxxx'  
    base_url = "http://xxxxxxxx/v1"  
    model = "qwen3"  
    chat_service = ChatAgent(api_key, base_url, model)  
    chat_service.execute_chat()