adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(mcp): MCP tool binds the information of the tool marketplace

Browse Source
This commit is contained in:
Timebomb2018
2026-03-06 16:32:33 +08:00
parent bccbeaabe4
commit 391cd602a2
4 changed files with 47 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
api/app/services/tool_service.py
View File

@@ -85,7 +85,7 @@ class ToolService:
"""检查工具名称是否重复"""
query = self.db.query(ToolConfig).filter(
ToolConfig.name == name,
ToolConfig.tool_type == tool_type.value,
ToolConfig.tool_type == tool_type,
ToolConfig.tenant_id == tenant_id
)
if exclude_id:
@@ -965,7 +965,11 @@ class ToolService:
id=tool_config.id,
server_url=config.get("server_url"),
connection_config=config.get("connection_config", {}),
available_tools=config.get("available_tools", [])
available_tools=config.get("available_tools", []),
source_channel=config.get("source_channel", "self_hosted"),
market_id=config.get("market_id"),
market_config_id=config.get("market_config_id"),
mcp_service_id=config.get("mcp_service_id"),
)
self.db.add(mcp_config)
@@ -1018,6 +1022,14 @@ class ToolService:
mcp_config.server_url = config.get("server_url")
mcp_config.connection_config = config.get("connection_config", {})
mcp_config.available_tools = config.get("available_tools", [])
if config.get("source_channel") is not None:
mcp_config.source_channel = config.get("source_channel")
if config.get("market_id") is not None:
mcp_config.market_id = config.get("market_id")
if config.get("market_config_id") is not None:
mcp_config.market_config_id = config.get("market_config_id")
if config.get("mcp_service_id") is not None:
mcp_config.mcp_service_id = config.get("mcp_service_id")
@staticmethod
def _determine_initial_status(tool_info: Dict[str, Any]) -> str: