Skip to content

Commit 81bf1b0

Browse files
committed
Update test to show more informative error about only being able to use one tool
1 parent a4bab78 commit 81bf1b0

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

integrations/stackit/tests/test_stackit_chat_generator.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,10 @@ def test_live_run_with_tools_and_response(self, tools):
285285
"""
286286
initial_messages = [ChatMessage.from_user("What's the weather like in Paris and Berlin?")]
287287
component = STACKITChatGenerator(model="neuralmagic/Meta-Llama-3.1-8B-Instruct-FP8", tools=tools)
288-
results = component.run(messages=initial_messages, generation_kwargs={"tool_choice": "any"})
288+
results = component.run(
289+
messages=initial_messages,
290+
generation_kwargs={"tool_choice": "auto"}
291+
)
289292

290293
assert len(results["replies"]) == 1
291294

0 commit comments

Comments
 (0)