feat(tool): 推断 MCP session_affinity 以支持 MCP_REMOTE proxy 模式#111
Merged
OhYee merged 2 commits intoJun 2, 2026
Merged
Conversation
Signed-off-by: 黑曜 <haotian.qht@alibaba-inc.com>
Contributor
There was a problem hiding this comment.
Pull request overview
Adds inference of MCP session_affinity from protocol_spec for the MCP_REMOTE + proxy_enabled scenario when mcp_config.session_affinity is empty, so the proxy mode can correctly route to streamable-http upstream MCP servers.
Changes:
- Add
_infer_protocol_spec_mcp_session_affinityhelper that parsesprotocol_spec.mcpServers[*].transportTypeto deriveMCP_STREAMABLE/MCP_SSE. - Update
_get_mcp_endpointto call the inference when session_affinity is empty under MCP_REMOTE proxy mode, falling back toMCP_SSE. - Add unit tests covering inference for missing and empty
session_affinity.
Reviewed changes
Copilot reviewed 3 out of 3 changed files in this pull request and generated no comments.
| File | Description |
|---|---|
| agentrun/tool/tool.py | Adds inference helper and integrates it into _get_mcp_endpoint. |
| agentrun/tool/__tool_async_template.py | Mirrors the same change in the async template (codegen source). |
| tests/unittests/tool/test_tool.py | Adds two test cases covering proxy-mode streamable inference. |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
292cb08 to
b26e01b
Compare
Signed-off-by: 117503445 <t117503445@gmail.com>
b26e01b to
2656ee2
Compare
OhYee
approved these changes
Jun 2, 2026
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
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.
Signed-off-by: 黑曜 haotian.qht@alibaba-inc.com
Fix bugs
Bug detail
Pull request tasks
Update docs
Reason for update
Pull request tasks
Add contributor
Contributed content
Content detail
Others
Reason for update