scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-02 02:11:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

fix(mcp): preserve the peer log schema for negotiation records

Browse source
This commit is contained in:
Joshua Valluru 2026-09-26 23:50:01 -07:00
parent d099c0f66f
commit 2c05e8bad8
2 changed files with 33 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/integration/_support/mcp.py
View file

@ -77,7 +77,7 @@ def math_service(
if ctx.method == "initialize":
requested: Final = InitializeRequestParams.model_validate(ctx.params).protocol_version
returned: Final = InitializeResult.model_validate(result).protocol_version
record({"body": {}, "negotiation": {"requested": requested, "returned": returned}})
record({"body": {}, "headers": {}, "negotiation": {"requested": requested, "returned": returned}})
return result
service.middleware.append(capture_negotiation)

32
tests/unit/integration_support/test_conformance.py
View file

@ -18,6 +18,38 @@ from tests.integration._support.conformance import (
from pydantic import ValidationError
@pytest.mark.asyncio
async def test_negotiation_records_preserve_the_peer_call_contract() -> None:
from integration._support.mcp import math_service
from integration.mcp.test_mcp_user_env_vars import UPSTREAM_CALLS
from mcp.server.context import ServerRequestContext
from mcp.types import InitializeResult
records: Final[list[dict[str, object]]] = []
service: Final = math_service(record=records.append)
context: Final = ServerRequestContext(
session=Mock(),
lifespan_context=None,
protocol_version="2025-03-26",
method="initialize",
params={"protocolVersion": "2025-03-26", "capabilities": {}, "clientInfo": {"name": "client", "version": "1"}},
)
async def initialize(request: ServerRequestContext[object, object]) -> InitializeResult:
return InitializeResult.model_validate(
{
"protocolVersion": request.protocol_version,
"capabilities": {},
"serverInfo": {"name": "peer", "version": "1"},
}
)
await service.middleware[-1](context, initialize)
calls: Final = UPSTREAM_CALLS.validate_python(tuple(records))
assert len(calls) == 1 and calls[0].headers == {} and calls[0].body == {}
assert records[0]["negotiation"] == {"requested": "2025-03-26", "returned": "2025-03-26"}
@pytest.mark.parametrize("explicit", (False, True))
def test_owned_proxy_isolates_automatic_coverage_unless_requested(
tmp_path: Path, monkeypatch: pytest.MonkeyPatch, explicit: bool