Merge pull request #1378 from asemyanov/fix-search-mcp-only

Fix search_mcp_servers function to handle paginated response structure
This commit is contained in:
Marko Kraemer 2025-08-17 18:27:33 -07:00 committed by GitHub
commit e9b1103674
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions

View File

@ -53,9 +53,11 @@ class MCPSearchTool(AgentBuilderBaseTool):
integration_service = get_integration_service()
if query:
toolkits = await integration_service.search_toolkits(query, category=category)
toolkits_response = await integration_service.search_toolkits(query, category=category)
toolkits = toolkits_response.get("items", [])
else:
toolkits = await toolkit_service.list_toolkits(limit=limit, category=category)
toolkits_response = await toolkit_service.list_toolkits(limit=limit, category=category)
toolkits = toolkits_response.get("items", [])
if len(toolkits) > limit:
toolkits = toolkits[:limit]