Fix CI issues related to missing dependency (#3096)

2025-07-03 18:52:50 -07:00
parent 2c496e6376
commit 7484eed4b2
32 changed files with 6150 additions and 828 deletions
--- a/tests/llms/test_vllm.py
+++ b/tests/llms/test_vllm.py
@@ -71,7 +71,13 @@ def test_generate_response_with_tools(mock_vllm_client):
    response = llm.generate_response(messages, tools=tools)

    mock_vllm_client.chat.completions.create.assert_called_once_with(
-        model="Qwen/Qwen2.5-32B-Instruct", messages=messages, temperature=0.7, max_tokens=100, top_p=1.0, tools=tools, tool_choice="auto"
+        model="Qwen/Qwen2.5-32B-Instruct",
+        messages=messages,
+        temperature=0.7,
+        max_tokens=100,
+        top_p=1.0,
+        tools=tools,
+        tool_choice="auto",
    )

    assert response["content"] == "I've added the memory for you."