Function test_function_call

test/test_api/test_openai_api.py:405–469

()

def test_function_call():
    client = LightLLMClient()

    # 定义函数
    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"],
                },
            },
        },
        {
            "type": "function",
            "function": {
                "name": "calculate",
                "description": "执行数学计算",
                "parameters": {
                    "type": "object",
                    "properties": {"expression": {"type": "string", "description": "数学表达式，例如：2+3*4"}},
                    "required": ["expression"],
                },
            },
        },
    ]

    try:
        # 测试天气查询
        print("用户: 北京今天天气怎么样？")
        result = client.function_call("北京今天天气怎么样？", tools)
        message = result["choices"][0]["message"]

        if message.get("tool_calls"):
            print("助手决定调用函数:")
            for tool_call in message["tool_calls"]:
                print(f"  函数名: {tool_call['function']['name']}")
                print(f"  参数: {tool_call['function']['arguments']}")
        else:
            print("助手:", message["content"])
        print()

        # 测试数学计算
        print("用户: 请计算 25 * 4 + 10 的结果")
        result = client.function_call("请计算 25 * 4 + 10 的结果", tools)
        message = result["choices"][0]["message"]

        if message.get("tool_calls"):
            print("助手决定调用函数:")
            for tool_call in message["tool_calls"]:
                print(f"  函数名: {tool_call['function']['name']}")