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

agent builder works with config

This commit is contained in:
Saumya 2025-07-24 15:15:38 +05:30
parent 7d3479d171
commit 1e6488ca45
2 changed files with 49 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
backend/agent/tools/agent_builder_tools/agent_config_tool.py
View File

@ -235,31 +235,51 @@ class AgentConfigTool(AgentBuilderBaseTool):
) )
async def get_current_agent_config(self) -> ToolResult: async def get_current_agent_config(self) -> ToolResult:
try: try:
agent = await self._get_agent_data() agent_data = await self._get_agent_data()
if not agent: if not agent_data:
return self.fail_response("Agent not found") return self.fail_response("Agent not found")
version_data = None
if agent_data.get('current_version_id'):
try:
from agent.versioning.facade import version_manager
account_id = await self._get_current_account_id()
version_dict = await version_manager.get_version(
agent_id=self.agent_id,
version_id=agent_data['current_version_id'],
user_id=account_id
)
version_data = version_dict
except Exception as e:
logger.warning(f"Failed to get version data for agent config tool: {e}")
from agent.config_helper import extract_agent_config
agent_config = extract_agent_config(agent_data, version_data)
config_summary = { config_summary = {
"agent_id": agent["agent_id"], "agent_id": agent_config["agent_id"],
"name": agent.get("name", "Untitled Agent"), "name": agent_config.get("name", "Untitled Agent"),
"description": agent.get("description", "No description set"), "description": agent_config.get("description", "No description set"),
"system_prompt": agent.get("system_prompt", "No system prompt set"), "system_prompt": agent_config.get("system_prompt", "No system prompt set"),
"avatar": agent.get("avatar", "🤖"), "avatar": agent_config.get("avatar", "🤖"),
"avatar_color": agent.get("avatar_color", "#6B7280"), "avatar_color": agent_config.get("avatar_color", "#6B7280"),
"agentpress_tools": agent.get("agentpress_tools", {}), "agentpress_tools": agent_config.get("agentpress_tools", {}),
"configured_mcps": agent.get("configured_mcps", []), "configured_mcps": agent_config.get("configured_mcps", []),
"custom_mcps": agent.get("custom_mcps", []), "custom_mcps": agent_config.get("custom_mcps", []),
"created_at": agent.get("created_at"), "created_at": agent_data.get("created_at"),
"updated_at": agent.get("updated_at") "updated_at": agent_data.get("updated_at"),
"current_version": agent_config.get("version_name", "v1") if version_data else "No version data"
} }
tools_count = len([t for t, cfg in config_summary["agentpress_tools"].items() if cfg.get("enabled")]) tools_count = len([t for t, cfg in config_summary["agentpress_tools"].items() if cfg.get("enabled")])
mcps_count = len(config_summary["configured_mcps"]) mcps_count = len(config_summary["configured_mcps"])
custom_mcps_count = len(config_summary["custom_mcps"]) custom_mcps_count = len(config_summary["custom_mcps"])
summary_text = f"Agent '{config_summary['name']}' (version: {config_summary['current_version']}) has {tools_count} tools enabled, {mcps_count} MCP servers configured, and {custom_mcps_count} custom MCP integrations."
return self.success_response({ return self.success_response({
"summary": f"Agent '{config_summary['name']}' has {tools_count} tools enabled, {mcps_count} MCP servers configured, and {custom_mcps_count} custom MCP integrations.", "summary": summary_text,
"configuration": config_summary "configuration": config_summary
}) })

18
backend/agent/tools/agent_builder_tools/workflow_tool.py
View File

@ -14,13 +14,25 @@ class WorkflowTool(AgentBuilderBaseTool):
async def _get_available_tools_for_agent(self) -> List[str]: async def _get_available_tools_for_agent(self) -> List[str]:
try: try:
client = await self.db.client client = await self.db.client
agent_result = await client.table('agents').select('*, agent_versions!current_version_id(*)').eq('agent_id', self.agent_id).execute() agent_result = await client.table('agents').select('*').eq('agent_id', self.agent_id).execute()
if not agent_result.data: if not agent_result.data:
return [] return []
agent_data = agent_result.data[0] agent_data = agent_result.data[0]
version_data = agent_data.get('agent_versions') version_data = None
if agent_data.get('current_version_id'):
try:
from agent.versioning.facade import version_manager
account_id = await self._get_current_account_id()
version_dict = await version_manager.get_version(
agent_id=self.agent_id,
version_id=agent_data['current_version_id'],
user_id=account_id
)
version_data = version_dict
except Exception as e:
logger.warning(f"Failed to get version data for workflow tool: {e}")
agent_config = extract_agent_config(agent_data, version_data) agent_config = extract_agent_config(agent_data, version_data)