Skip to content

Conversation

pranramesh
Copy link

@pranramesh pranramesh commented Oct 4, 2025

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:

@pranramesh pranramesh marked this pull request as ready for review October 4, 2025 03:21
@carenthomas carenthomas force-pushed the main branch 2 times, most recently from 42a8fad to 30781c4 Compare October 8, 2025 01:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants