From 9f60edb413db13f798a533a9a9576cf695b3d472 Mon Sep 17 00:00:00 2001 From: asemyanov Date: Mon, 18 Aug 2025 23:00:27 +0200 Subject: [PATCH] Revert model from Google Gemini to Moonshot AI Kimi K2 --- backend/agent/run.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/backend/agent/run.py b/backend/agent/run.py index 7127f56b..72b7718b 100644 --- a/backend/agent/run.py +++ b/backend/agent/run.py @@ -46,7 +46,7 @@ class AgentConfig: stream: bool native_max_auto_continues: int = 25 max_iterations: int = 100 - model_name: str = "openrouter/google/gemini-2.5-flash" + model_name: str = "openrouter/moonshotai/kimi-k2" enable_thinking: Optional[bool] = False reasoning_effort: Optional[str] = 'low' enable_context_manager: bool = True @@ -547,9 +547,9 @@ class AgentRunner: def get_max_tokens(self) -> Optional[int]: if "sonnet" in self.config.model_name.lower(): return 8192 - elif "gpt" in self.config.model_name.lower(): - return 8192 - elif "gemini-2.5" in self.config.model_name.lower(): + elif "gpt-4" in self.config.model_name.lower(): + return 4096 + elif "gemini-2.5-pro" in self.config.model_name.lower(): return 64000 elif "kimi-k2" in self.config.model_name.lower(): return 8192 @@ -741,7 +741,7 @@ async def run_agent( thread_manager: Optional[ThreadManager] = None, native_max_auto_continues: int = 25, max_iterations: int = 100, - model_name: str = "openrouter/google/gemini-2.5-flash", + model_name: str = "openrouter/moonshotai/kimi-k2", enable_thinking: Optional[bool] = False, reasoning_effort: Optional[str] = 'low', enable_context_manager: bool = True, @@ -751,10 +751,10 @@ async def run_agent( target_agent_id: Optional[str] = None ): effective_model = model_name - if model_name == "openrouter/google/gemini-2.5-flash" and agent_config and agent_config.get('model'): + if model_name == "openrouter/moonshotai/kimi-k2" and agent_config and agent_config.get('model'): effective_model = agent_config['model'] logger.debug(f"Using model from agent config: {effective_model} (no user selection)") - elif model_name != "openrouter/google/gemini-2.5-flash": + elif model_name != "openrouter/moonshotai/kimi-k2": logger.debug(f"Using user-selected model: {effective_model}") else: logger.debug(f"Using default model: {effective_model}")