diff --git a/integrations/togetherai/tests/test_togetherai_chat_generator.py b/integrations/togetherai/tests/test_togetherai_chat_generator.py index d79f3d33bd..c8d8f57a70 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-120b", tools=tools) results = component.run(messages=initial_messages, generation_kwargs={"tool_choice": "auto"}) assert len(results["replies"]) == 1 @@ -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" 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"