feat: add MCP wrapper tool system for progressive repository discovery #683
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Implement a 4-tool wrapper system (inspired by Klavis MCP) that reduces MCP tool explosion by wrapping repository operations through progressive discovery. **New Features:** - Add configurable MCP mode (`direct` or `wrapper`) via `RESTIFY_MCP_MODE` env variable - Create 4 wrapper tools for progressive discovery: - `discover-repositories`: List all MCP-enabled repositories - `get-repository-operations`: Get operations for a specific repository - `get-operation-details`: Get detailed schema for an operation - `execute-operation`: Execute an operation with parameters **Architecture:** - `ToolRegistry` service: Centralized registry for repository/operation metadata with caching - `WrapperToolHelpers` trait: Shared utilities for schema formatting and example generation - Multi-layer validation ensures only MCP-enabled repositories are accessible **Benefits:** - Reduces tool count from 50+ to 4 wrapper tools (in wrapper mode) - Better token efficiency for AI agents - Progressive discovery improves exploration - Backward compatible via config (defaults to `direct` mode) - Static tools remain unaffected regardless of mode **Validation:** - Only repositories with `HasMcpTools` trait are exposed - Each operation validates `mcpAllows*()` permissions - Clear error messages guide missing configuration 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
Add WrapperToolsIntegrationTest covering all 4 wrapper tools and their functionality. **Tests Added:** - Wrapper mode exposes correct tools (4 wrapper + static tools) - discover-repositories: Lists only MCP-enabled repositories - discover-repositories: Search functionality works correctly - get-repository-operations: Returns complete operations list - get-repository-operations: Rejects non-MCP repositories with clear error - get-operation-details: Returns schema with examples - execute-operation: Creates records via wrapper (store operation) - execute-operation: Lists records via wrapper (index operation) - execute-operation: Validates operation permissions - Complete workflow: Tests entire discover → operations → details → execute flow **Coverage:** - 10 test methods - 74 assertions - Tests all 4 wrapper tools - Tests error handling and validation - Tests end-to-end workflow 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
New Features:
directorwrapper) viaRESTIFY_MCP_MODEenv variablediscover-repositories: List all MCP-enabled repositoriesget-repository-operations: Get operations for a specific repositoryget-operation-details: Get detailed schema for an operationexecute-operation: Execute an operation with parametersArchitecture:
ToolRegistryservice: Centralized registry for repository/operation metadata with cachingWrapperToolHelperstrait: Shared utilities for schema formatting and example generationBenefits:
directmode)Validation:
HasMcpToolstrait are exposedmcpAllows*()permissions