Fix (mixture_of_agents): replace deprecated Gemini model and forward max_tokens to OpenRouter (#6621)

This commit is contained in:
Devorun 2026-04-09 16:57:49 +03:00 committed by Teknium
parent 9599271180
commit 1df35a93b2

View file

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