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

fix: tool calls of custom agents

This commit is contained in:
Saumya 2025-07-23 20:13:00 +05:30
parent 58ab4ace21
commit bdd7a005a8
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: if is_suna_default:
from agent.suna.config import SunaConfig from agent.suna.config import SunaConfig
system_prompt = SunaConfig.get_system_prompt() 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 agentpress_tools = SunaConfig.DEFAULT_TOOLS
else: else:
system_prompt = version_data.get('system_prompt', '') 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 from agent.suna.config import SunaConfig
config['system_prompt'] = SunaConfig.get_system_prompt() config['system_prompt'] = SunaConfig.get_system_prompt()
config['tools']['agentpress'] = SunaConfig.DEFAULT_TOOLS 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', {}) metadata = config.get('metadata', {})
config['avatar'] = metadata.get('avatar', agent_data.get('avatar')) config['avatar'] = metadata.get('avatar', agent_data.get('avatar'))
config['avatar_color'] = metadata.get('avatar_color', agent_data.get('avatar_color')) 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['configured_mcps'] = config.get('tools', {}).get('mcp', [])
config['custom_mcps'] = config.get('tools', {}).get('custom_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'] = metadata.get('avatar')
config['avatar_color'] = metadata.get('avatar_color') 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['configured_mcps'] = config.get('tools', {}).get('mcp', [])
config['custom_mcps'] = config.get('tools', {}).get('custom_mcp', []) config['custom_mcps'] = config.get('tools', {}).get('custom_mcp', [])