Ling0925
/
suna
mirror of https://github.com/kortix-ai/suna.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Revert model from Google Gemini to Moonshot AI Kimi K2

This commit is contained in:
asemyanov 2025-08-18 23:00:27 +02:00
parent 6acbd2b2cd
commit 9f60edb413
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
backend/agent/run.py
View File

@ -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}")