Skip to content

Commit 9f9b8c5

Browse files
committed
Push new observations
1 parent 81bf1b0 commit 9f9b8c5

1 file changed

Lines changed: 11 additions & 1 deletion

File tree

integrations/stackit/tests/test_stackit_chat_generator.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -284,11 +284,21 @@ def test_live_run_with_tools_and_response(self, tools):
284284
Integration test that the MistralChatGenerator component can run with tools and get a response.
285285
"""
286286
initial_messages = [ChatMessage.from_user("What's the weather like in Paris and Berlin?")]
287-
component = STACKITChatGenerator(model="neuralmagic/Meta-Llama-3.1-8B-Instruct-FP8", tools=tools)
287+
component = STACKITChatGenerator(
288+
# This one does indeed run, but for some reason the tool call is put into
289+
# chat_completion.choices[0].message.content instead chat_completion.choices[0].message.tool_calls
290+
model="cortecs/Llama-3.3-70B-Instruct-FP8-Dynamic",
291+
# Doesn't work get the "auto" tool choice requires --enable-auto-tool-choice and --tool-call-parser to be
292+
# set
293+
# model="google/gemma-3-27b-it",
294+
# model="neuralmagic/Meta-Llama-3.1-8B-Instruct-FP8",
295+
tools=tools
296+
)
288297
results = component.run(
289298
messages=initial_messages,
290299
generation_kwargs={"tool_choice": "auto"}
291300
)
301+
import pdb;pdb.set_trace()
292302

293303
assert len(results["replies"]) == 1
294304

0 commit comments

Comments
 (0)