Skip to content

Commit f85ce19

Browse files
authored
test: replace tool calling model in tests with Qwen2.5-72B-Instruct (#9500)
1 parent 8e21c50 commit f85ce19

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

test/components/generators/chat/test_hugging_face_api.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -821,13 +821,13 @@ def test_live_run_with_tools(self, tools):
821821
"""
822822
We test the round trip: generate tool call, pass tool message, generate response.
823823
824-
The model used here (Hermes-3-Llama-3.1-8B) is not gated and kept in a warm state.
824+
The model used here (Qwen/Qwen2.5-72B-Instruct) is not gated and kept in a warm state.
825825
"""
826826

827827
chat_messages = [ChatMessage.from_user("What's the weather like in Paris?")]
828828
generator = HuggingFaceAPIChatGenerator(
829829
api_type=HFGenerationAPIType.SERVERLESS_INFERENCE_API,
830-
api_params={"model": "NousResearch/Hermes-3-Llama-3.1-8B"},
830+
api_params={"model": "Qwen/Qwen2.5-72B-Instruct", "provider": "hf-inference"},
831831
generation_kwargs={"temperature": 0.5},
832832
)
833833

@@ -852,7 +852,7 @@ def test_live_run_with_tools(self, tools):
852852
final_message = results["replies"][0]
853853
assert not final_message.tool_calls
854854
assert len(final_message.text) > 0
855-
assert "paris" in final_message.text.lower()
855+
assert "paris" in final_message.text.lower() and "22" in final_message.text
856856

857857
@pytest.mark.asyncio
858858
async def test_run_async(self, mock_check_valid_model, mock_chat_completion_async, chat_messages):

0 commit comments

Comments
 (0)