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

Merge pull request #1049 from escapade-mckv/custom-suna-agent 

fix: tool calls of custom agents
This commit is contained in:
Bobbie 2025-07-23 20:21:22 +05:30 committed by GitHub
parent dbdfd8f2fa bdd7a005a8
commit dec25956ac
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
backend/agent/config_helper.py
View File

@ -18,6 +18,7 @@ def extract_agent_config(agent_data: Dict[str, Any], version_data: Optional[Dict
if is_suna_default:
from agent.suna.config import SunaConfig
system_prompt = SunaConfig.get_system_prompt()
# For Suna agents, always use DEFAULT_TOOLS which is in the correct format for run.py
agentpress_tools = SunaConfig.DEFAULT_TOOLS
else:
system_prompt = version_data.get('system_prompt', '')
@ -66,13 +67,15 @@ def extract_agent_config(agent_data: Dict[str, Any], version_data: Optional[Dict
from agent.suna.config import SunaConfig
config['system_prompt'] = SunaConfig.get_system_prompt()
config['tools']['agentpress'] = SunaConfig.DEFAULT_TOOLS
# For Suna agents, use DEFAULT_TOOLS directly as it's already in the correct format
config['agentpress_tools'] = SunaConfig.DEFAULT_TOOLS
else:
config['agentpress_tools'] = extract_tools_for_agent_run(config)
metadata = config.get('metadata', {})
config['avatar'] = metadata.get('avatar', agent_data.get('avatar'))
config['avatar_color'] = metadata.get('avatar_color', agent_data.get('avatar_color'))
config['agentpress_tools'] = extract_tools_for_agent_run(config)
config['configured_mcps'] = config.get('tools', {}).get('mcp', [])
config['custom_mcps'] = config.get('tools', {}).get('custom_mcp', [])
@ -104,7 +107,11 @@ def extract_agent_config(agent_data: Dict[str, Any], version_data: Optional[Dict
config['avatar'] = metadata.get('avatar')
config['avatar_color'] = metadata.get('avatar_color')
config['agentpress_tools'] = extract_tools_for_agent_run(config)
if is_suna_default:
from agent.suna.config import SunaConfig
config['agentpress_tools'] = SunaConfig.DEFAULT_TOOLS
else:
config['agentpress_tools'] = extract_tools_for_agent_run(config)
config['configured_mcps'] = config.get('tools', {}).get('mcp', [])
config['custom_mcps'] = config.get('tools', {}).get('custom_mcp', [])