Change to new model that is less flaky (#2632)

sjrl · web-flow · commit 328d58262199 · 2025-12-30T10:17:26.000+01:00
diff --git 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="openai/gpt-oss-120b", tools=mixed_tools)
+        component = TogetherAIChatGenerator(model="moonshotai/kimi-k2-thinking", tools=mixed_tools)
         results = component.run(messages=initial_messages)
 
         assert len(results["replies"]) > 0, "No replies received"

Original file line number	Diff line number	Diff line change
`@@ -685,7 +685,7 @@ def test_live_run_with_mixed_tools(self, mixed_tools):`
`685`	`685`	`initial_messages = [`
`686`	`686`	`ChatMessage.from_user("What's the weather like in Paris and what is the population of Berlin?")`
`687`	`687`	`]`
`688`		`- component = TogetherAIChatGenerator(model="openai/gpt-oss-120b", tools=mixed_tools)`
	`688`	`+ component = TogetherAIChatGenerator(model="moonshotai/kimi-k2-thinking", tools=mixed_tools)`
`689`	`689`	`results = component.run(messages=initial_messages)`
`690`	`690`
`691`	`691`	`assert len(results["replies"]) > 0, "No replies received"`