-
Notifications
You must be signed in to change notification settings - Fork 1.9k
feat: Added support for per-tool and per-agent MCP schema override #3016
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Open
pranramesh
wants to merge
25
commits into
letta-ai:main
Choose a base branch
from
pranramesh:feature/mcp_schema_override_task
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Open
feat: Added support for per-tool and per-agent MCP schema override #3016
pranramesh
wants to merge
25
commits into
letta-ai:main
from
pranramesh:feature/mcp_schema_override_task
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
…e concepts, and navigation restructure (letta-ai#3010)
42a8fad
to
30781c4
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
Please describe the purpose of this pull request.
New feature: Added support for per-tool and per-agent MCP schema override
How to test
I've added a pytest file for validation: tests/mcp_tests/test_mcp_schema_override.py (auto-generated acting as placeholder; not robust yet)
Additionally, I've included an example usage of the client-side updates and walkthrough of testing as a python script to simulate how the changes persist in local host settings. Please refer to examples/override_mcp_schema.py
With this file, you can additionally view schema overrides in the ADE (self-hosted projects mode)
Have you tested this PR?
I've validated my tests by running the pytest file and testing locally with local host. For reference, locally tested in-depth to ensure the following:
per tool overrides properly are persisted in DB and accessible via other client-side calls.
per agent overrides are also done at the tool level, and persist changes across agents in a multi agent scenario
Related issues or PRs
N/A
Is your PR over 500 lines of code?
Yes. Involved substantial changes to server side code to allow for MCP schema overriding. Additionally, added test files which had substantial line count.
Additional context
I've additionally made modifications to the client-side, editing the client and raw client class to expose the new functionality to users. Specifically I've done so in the following files:
Per Agent:
Per Tool: