From e5648ebeaf3b6f4329aaabde53cfbe5cb282c436 Mon Sep 17 00:00:00 2001 From: Sebastian Husch Lee Date: Mon, 8 Dec 2025 11:53:30 +0100 Subject: [PATCH 1/3] Use different model for integration test. More reliable than llama 3.3 --- integrations/togetherai/tests/test_togetherai_chat_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integrations/togetherai/tests/test_togetherai_chat_generator.py b/integrations/togetherai/tests/test_togetherai_chat_generator.py index d79f3d33bd..2564946215 100644 --- a/integrations/togetherai/tests/test_togetherai_chat_generator.py +++ b/integrations/togetherai/tests/test_togetherai_chat_generator.py @@ -394,7 +394,7 @@ def test_live_run_with_tools_and_response(self, tools): Integration test that the TogetherAIChatGenerator component can run with tools and get a response. """ initial_messages = [ChatMessage.from_user("What's the weather like in Paris and Berlin?")] - component = TogetherAIChatGenerator(tools=tools) + component = TogetherAIChatGenerator(model="openai/gpt-oss-20b", tools=tools) results = component.run(messages=initial_messages, generation_kwargs={"tool_choice": "auto"}) assert len(results["replies"]) == 1 From 79179a94a4ba815f3196c4b5703677a2a354ba54 Mon Sep 17 00:00:00 2001 From: Sebastian Husch Lee Date: Mon, 8 Dec 2025 12:07:31 +0100 Subject: [PATCH 2/3] Use different models --- integrations/togetherai/tests/test_togetherai_chat_generator.py | 2 +- .../togetherai/tests/test_togetherai_chat_generator_async.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/integrations/togetherai/tests/test_togetherai_chat_generator.py b/integrations/togetherai/tests/test_togetherai_chat_generator.py index 2564946215..907e4220aa 100644 --- a/integrations/togetherai/tests/test_togetherai_chat_generator.py +++ b/integrations/togetherai/tests/test_togetherai_chat_generator.py @@ -394,7 +394,7 @@ def test_live_run_with_tools_and_response(self, tools): Integration test that the TogetherAIChatGenerator component can run with tools and get a response. """ initial_messages = [ChatMessage.from_user("What's the weather like in Paris and Berlin?")] - component = TogetherAIChatGenerator(model="openai/gpt-oss-20b", tools=tools) + component = TogetherAIChatGenerator(model="openai/gpt-oss-120b", tools=tools) results = component.run(messages=initial_messages, generation_kwargs={"tool_choice": "auto"}) assert len(results["replies"]) == 1 diff --git a/integrations/togetherai/tests/test_togetherai_chat_generator_async.py b/integrations/togetherai/tests/test_togetherai_chat_generator_async.py index 92afcacb0c..7f0bf19cd7 100644 --- a/integrations/togetherai/tests/test_togetherai_chat_generator_async.py +++ b/integrations/togetherai/tests/test_togetherai_chat_generator_async.py @@ -176,7 +176,7 @@ async def callback(chunk: StreamingChunk): @pytest.mark.asyncio async def test_live_run_with_tools_and_response_async(self, tools): initial_messages = [ChatMessage.from_user("What's the weather like in Paris?")] - component = TogetherAIChatGenerator(tools=tools) + component = TogetherAIChatGenerator(model="openai/gpt-oss-20b", tools=tools) results = await component.run_async(messages=initial_messages, generation_kwargs={"tool_choice": "auto"}) assert len(results["replies"]) > 0, "No replies received" From d71d6f5d805bf1dc11c1222b6cbdf7bce943ac51 Mon Sep 17 00:00:00 2001 From: Sebastian Husch Lee Date: Mon, 8 Dec 2025 12:13:59 +0100 Subject: [PATCH 3/3] update to use openai's model --- integrations/togetherai/tests/test_togetherai_chat_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integrations/togetherai/tests/test_togetherai_chat_generator.py b/integrations/togetherai/tests/test_togetherai_chat_generator.py index 907e4220aa..c8d8f57a70 100644 --- a/integrations/togetherai/tests/test_togetherai_chat_generator.py +++ b/integrations/togetherai/tests/test_togetherai_chat_generator.py @@ -685,7 +685,7 @@ def test_live_run_with_mixed_tools(self, mixed_tools): initial_messages = [ ChatMessage.from_user("What's the weather like in Paris and what is the population of Berlin?") ] - component = TogetherAIChatGenerator(model="meta-llama/Llama-3.3-70B-Instruct-Turbo", tools=mixed_tools) + component = TogetherAIChatGenerator(model="openai/gpt-oss-120b", tools=mixed_tools) results = component.run(messages=initial_messages) assert len(results["replies"]) > 0, "No replies received"