From 40e004cde8ee2eba9c0b1f77ace4bdfcad535c07 Mon Sep 17 00:00:00 2001 From: Saumya Date: Fri, 5 Sep 2025 14:36:14 +0530 Subject: [PATCH] fix: custom mcp enabled tools list fix --- backend/core/agent_tools.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/backend/core/agent_tools.py b/backend/core/agent_tools.py index 8a83a2fb..c4613a52 100644 --- a/backend/core/agent_tools.py +++ b/backend/core/agent_tools.py @@ -61,15 +61,21 @@ async def get_custom_mcp_tools_for_agent( except json.JSONDecodeError: logger.warning("Failed to parse X-MCP-Headers as JSON") - from mcp_module import mcp_service + from core.mcp_module import mcp_service discovery_result = await mcp_service.discover_custom_tools(mcp_type, mcp_config) existing_mcp = None for mcp in custom_mcps: - if (mcp.get('type') == mcp_type and - mcp.get('config', {}).get('url') == mcp_url): - existing_mcp = mcp - break + if mcp_type == 'composio': + if (mcp.get('type') == 'composio' and + mcp.get('config', {}).get('profile_id') == mcp_url): + existing_mcp = mcp + break + else: + if (mcp.get('customType') == mcp_type and + mcp.get('config', {}).get('url') == mcp_url): + existing_mcp = mcp + break tools = [] enabled_tools = existing_mcp.get('enabledTools', []) if existing_mcp else []