Function test_stream_function_call

test/test_api/test_openai_api.py:472–508

()

def test_stream_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"]},
                    },
                    "required": ["city"],
                },
            },
        }
    ]

    try:
        print("用户: 上海今天天气怎么样？")
        print("助手: ", end="", flush=True)

        for chunk in client.stream_function_call("上海今天天气怎么样？", tools):
            if chunk["type"] == "content":
                print(chunk["data"], end="", flush=True)
            elif chunk["type"] == "tool_call":
                print(f"\n[函数调用: {chunk['data']['function']['name']}]")
                if chunk["data"]["function"].get("arguments"):
                    print(f"参数: {chunk['data']['function']['arguments']}")
        print("\n")

    except Exception as e:
        print(f"错误: {e}")


def test_token_completions():