From 1df35a93b20b92c65e9cb312c3e12d93cb9c2251 Mon Sep 17 00:00:00 2001
From: Devorun <130918800+devorun@users.noreply.github.com>
Date: Thu, 9 Apr 2026 16:57:49 +0300
Subject: [PATCH] Fix (mixture_of_agents): replace deprecated Gemini model and
 forward max_tokens to OpenRouter (#6621)

---
 tools/mixture_of_agents_tool.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tools/mixture_of_agents_tool.py b/tools/mixture_of_agents_tool.py
index 8bbc18792..a34e99aa8 100644
--- a/tools/mixture_of_agents_tool.py
+++ b/tools/mixture_of_agents_tool.py
@@ -62,7 +62,7 @@ logger = logging.getLogger(__name__)
 # Keep this list aligned with current top-tier OpenRouter frontier options.
 REFERENCE_MODELS = [
     "anthropic/claude-opus-4.6",
-    "google/gemini-3-pro-preview",
+    "google/gemini-2.5-pro",
     "openai/gpt-5.4-pro",
     "deepseek/deepseek-v3.2",
 ]
@@ -129,6 +129,7 @@ async def _run_reference_model_safe(
             api_params = {
                 "model": model,
                 "messages": [{"role": "user", "content": user_prompt}],
+                "max_tokens": max_tokens,
                 "extra_body": {
                     "reasoning": {
                         "enabled": True,
@@ -203,6 +204,7 @@ async def _run_aggregator_model(
             {"role": "system", "content": system_prompt},
             {"role": "user", "content": user_prompt}
         ],
+        "max_tokens": max_tokens,
         "extra_body": {
             "reasoning": {
                 "enabled": True,