mirror of
https://github.com/cocoindex-io/cocoindex-code.git
synced 2026-09-14 16:39:38 +08:00
JunZ
0c18fe33c3
`MCPServer`'s `version` parameter defaults to `""`, so the daemon-backed server advertised an empty `serverInfo.version`. Under the v1 SDK the field was auto-filled with the SDK's own version, which was already the wrong number to report — clients saw the MCP SDK release, not ours. Pass `_version.__version__`, the same value `ccc version`, the daemon handshake, and the version-mismatch check already use.
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import pytest
|
|
from mcp import Client
|
|
|
|
from cocoindex_code import client as daemon_client
|
|
from cocoindex_code._version import __version__
|
|
from cocoindex_code.protocol import SearchResponse
|
|
from cocoindex_code.server import create_mcp_server
|
|
|
|
|
|
async def test_mcp_server_uses_v2_protocol(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(
|
|
daemon_client,
|
|
"search",
|
|
lambda **kwargs: SearchResponse(success=True, offset=kwargs["offset"]),
|
|
)
|
|
|
|
server = create_mcp_server(".")
|
|
async with Client(server, raise_exceptions=True) as client:
|
|
tools = await client.list_tools()
|
|
result = await client.call_tool(
|
|
"search",
|
|
{"query": "authentication", "refresh_index": False},
|
|
)
|
|
|
|
assert [tool.name for tool in tools.tools] == ["search"]
|
|
assert result.structured_content == {
|
|
"success": True,
|
|
"results": [],
|
|
"total_returned": 0,
|
|
"offset": 0,
|
|
"message": None,
|
|
}
|
|
|
|
|
|
async def test_mcp_server_reports_own_version() -> None:
|
|
"""The handshake advertises our version, not the SDK's or an empty string."""
|
|
server = create_mcp_server(".")
|
|
async with Client(server, raise_exceptions=True) as client:
|
|
assert client.server_info is not None
|
|
assert client.server_info.name == "cocoindex-code"
|
|
assert client.server_info.version == __version__
|