Split protocol types into a standalone mcp-types package (#2973)

tests/server/lowlevel/test_server_discover.py

@@ -8,11 +8,11 @@ the handler's contract in isolation from the dispatch pipeline.
 import importlib.metadata
 from typing import Any, cast
 
+import mcp_types as types
 import pytest
+from mcp_types.version import MODERN_PROTOCOL_VERSIONS
 
-from mcp import types
 from mcp.server import Server, ServerRequestContext
-from mcp.shared.version import MODERN_PROTOCOL_VERSIONS
 
 # `Server._handle_discover` ignores its `ctx` argument entirely (it derives the
 # result from server state), so a sentinel keeps the call site type-correct

tests/server/lowlevel/test_server_listing.py

@@ -1,10 +1,7 @@
 """Basic tests for list_prompts, list_resources, and list_tools handlers without pagination."""
 
 import pytest
-
-from mcp import Client
-from mcp.server import Server, ServerRequestContext
-from mcp.types import (
+from mcp_types import (
     ListPromptsResult,
     ListResourcesResult,
     ListToolsResult,
@@ -14,6 +11,9 @@ from mcp.types import (
     Tool,
 )
 
+from mcp import Client
+from mcp.server import Server, ServerRequestContext
+
 
 @pytest.mark.anyio
 async def test_list_prompts_basic() -> None:

tests/server/lowlevel/test_server_pagination.py

@@ -1,14 +1,14 @@
 import pytest
-
-from mcp import Client
-from mcp.server import Server, ServerRequestContext
-from mcp.types import (
+from mcp_types import (
     ListPromptsResult,
     ListResourcesResult,
     ListToolsResult,
     PaginatedRequestParams,
 )
 
+from mcp import Client
+from mcp.server import Server, ServerRequestContext
+
 
 @pytest.mark.anyio
 async def test_list_prompts_pagination() -> None:

tests/server/mcpserver/prompts/test_base.py

@@ -2,10 +2,10 @@ import threading
 from typing import Any
 
 import pytest
+from mcp_types import EmbeddedResource, TextContent, TextResourceContents
 
 from mcp.server.mcpserver import Context
 from mcp.server.mcpserver.prompts.base import AssistantMessage, Message, Prompt, UserMessage
-from mcp.types import EmbeddedResource, TextContent, TextResourceContents
 
 
 class TestRenderPrompt:

tests/server/mcpserver/prompts/test_manager.py

@@ -1,9 +1,9 @@
 import pytest
+from mcp_types import TextContent
 
 from mcp.server.mcpserver import Context
 from mcp.server.mcpserver.prompts.base import Prompt, UserMessage
 from mcp.server.mcpserver.prompts.manager import PromptManager
-from mcp.types import TextContent
 
 
 class TestPromptManager:

tests/server/mcpserver/resources/test_resource_template.py

@@ -3,12 +3,12 @@ import threading
 from typing import Any
 
 import pytest
+from mcp_types import Annotations
 from pydantic import BaseModel
 
 from mcp.server.mcpserver import Context, MCPServer
 from mcp.server.mcpserver.exceptions import ResourceError
 from mcp.server.mcpserver.resources import FunctionResource, ResourceTemplate
-from mcp.types import Annotations
 
 
 class TestResourceTemplate:

tests/server/mcpserver/resources/test_resources.py

@@ -1,8 +1,8 @@
 import pytest
+from mcp_types import Annotations
 
 from mcp.server.mcpserver import MCPServer
 from mcp.server.mcpserver.resources import FunctionResource, Resource
-from mcp.types import Annotations
 
 
 class TestResourceValidation:

tests/server/mcpserver/test_elicitation.py

@@ -2,14 +2,15 @@
 
 from typing import Any, Literal
 
+import mcp_types as types
 import pytest
+from mcp_types import ElicitRequestParams, ElicitResult, TextContent
 from pydantic import BaseModel, Field
 
-from mcp import Client, types
+from mcp import Client
 from mcp.client import ClientRequestContext
 from mcp.client.session import ElicitationFnT
 from mcp.server.mcpserver import Context, MCPServer
-from mcp.types import ElicitRequestParams, ElicitResult, TextContent
 
 
 # Shared schema for basic tests

tests/server/mcpserver/test_func_metadata.py

@@ -10,11 +10,11 @@ from typing import Annotated, Any, Final, NamedTuple, TypedDict
 import annotated_types
 import pytest
 from dirty_equals import IsPartialDict
+from mcp_types import CallToolResult
 from pydantic import BaseModel, Field
 
 from mcp.server.mcpserver.exceptions import InvalidSignature
 from mcp.server.mcpserver.utilities.func_metadata import func_metadata
-from mcp.types import CallToolResult
 
 
 class SomeInputModelA(BaseModel):

tests/server/mcpserver/test_integration.py

@@ -13,22 +13,7 @@ import json
 
 import pytest
 from inline_snapshot import snapshot
-
-from examples.snippets.servers import (
-    basic_prompt,
-    basic_resource,
-    basic_tool,
-    completion,
-    elicitation,
-    mcpserver_quickstart,
-    notifications,
-    sampling,
-    structured_output,
-    tool_progress,
-)
-from mcp.client import Client, ClientRequestContext
-from mcp.shared.session import RequestResponder
-from mcp.types import (
+from mcp_types import (
     ClientResult,
     CreateMessageRequestParams,
     CreateMessageResult,
@@ -51,6 +36,21 @@ from mcp.types import (
     ToolListChangedNotification,
 )
 
+from examples.snippets.servers import (
+    basic_prompt,
+    basic_resource,
+    basic_tool,
+    completion,
+    elicitation,
+    mcpserver_quickstart,
+    notifications,
+    sampling,
+    structured_output,
+    tool_progress,
+)
+from mcp.client import Client, ClientRequestContext
+from mcp.shared.session import RequestResponder
+
 pytestmark = pytest.mark.anyio
 
 

tests/server/mcpserver/test_server.py

@@ -5,20 +5,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 from inline_snapshot import snapshot
-from pydantic import BaseModel
-from starlette.applications import Starlette
-from starlette.routing import Mount, Route
-
-from mcp.client import Client
-from mcp.server.context import ServerRequestContext
-from mcp.server.mcpserver import Context, MCPServer
-from mcp.server.mcpserver.exceptions import ResourceNotFoundError, ToolError
-from mcp.server.mcpserver.prompts.base import Message, UserMessage
-from mcp.server.mcpserver.resources import FileResource, FunctionResource
-from mcp.server.mcpserver.utilities.types import Audio, Image
-from mcp.server.transport_security import TransportSecuritySettings
-from mcp.shared.exceptions import MCPError
-from mcp.types import (
+from mcp_types import (
     INTERNAL_ERROR,
     INVALID_PARAMS,
     AudioContent,
@@ -43,6 +30,19 @@ from mcp.types import (
     TextContent,
     TextResourceContents,
 )
+from pydantic import BaseModel
+from starlette.applications import Starlette
+from starlette.routing import Mount, Route
+
+from mcp.client import Client
+from mcp.server.context import ServerRequestContext
+from mcp.server.mcpserver import Context, MCPServer
+from mcp.server.mcpserver.exceptions import ResourceNotFoundError, ToolError
+from mcp.server.mcpserver.prompts.base import Message, UserMessage
+from mcp.server.mcpserver.resources import FileResource, FunctionResource
+from mcp.server.mcpserver.utilities.types import Audio, Image
+from mcp.server.transport_security import TransportSecuritySettings
+from mcp.shared.exceptions import MCPError
 
 pytestmark = pytest.mark.anyio
 

tests/server/mcpserver/test_title.py

@@ -1,12 +1,12 @@
 """Integration tests for title field functionality."""
 
 import pytest
+from mcp_types import Prompt, Resource, ResourceTemplate, Tool, ToolAnnotations
 
 from mcp import Client
 from mcp.server.mcpserver import MCPServer
 from mcp.server.mcpserver.resources import FunctionResource
 from mcp.shared.metadata_utils import get_display_name
-from mcp.types import Prompt, Resource, ResourceTemplate, Tool, ToolAnnotations
 
 
 @pytest.mark.anyio

tests/server/mcpserver/test_tool_manager.py

@@ -4,6 +4,7 @@ from dataclasses import dataclass
 from typing import Any, TypedDict
 
 import pytest
+from mcp_types import CallToolResult, TextContent, ToolAnnotations
 from pydantic import BaseModel
 
 from mcp.server.context import LifespanContextT, RequestT
@@ -11,7 +12,6 @@ from mcp.server.mcpserver import Context, MCPServer
 from mcp.server.mcpserver.exceptions import ToolError
 from mcp.server.mcpserver.tools import Tool, ToolManager
 from mcp.server.mcpserver.utilities.func_metadata import ArgModelBase, FuncMetadata
-from mcp.types import CallToolResult, TextContent, ToolAnnotations
 
 
 class TestAddTools:

tests/server/mcpserver/test_url_elicitation.py

@@ -1,14 +1,15 @@
 """Test URL mode elicitation feature (SEP 1036)."""
 
 import anyio
+import mcp_types as types
 import pytest
+from mcp_types import ElicitRequestParams, ElicitResult, TextContent
 from pydantic import BaseModel, Field
 
-from mcp import Client, types
+from mcp import Client
 from mcp.client import ClientRequestContext
 from mcp.server.elicitation import CancelledElicitation, DeclinedElicitation, elicit_url
 from mcp.server.mcpserver import Context, MCPServer
-from mcp.types import ElicitRequestParams, ElicitResult, TextContent
 
 
 @pytest.mark.anyio

tests/server/mcpserver/test_url_elicitation_error_throw.py

@@ -1,9 +1,10 @@
 """Test that UrlElicitationRequiredError is properly propagated as MCP error."""
 
+import mcp_types as types
 import pytest
 from inline_snapshot import snapshot
 
-from mcp import Client, ErrorData, types
+from mcp import Client, ErrorData
 from mcp.server.mcpserver import Context, MCPServer
 from mcp.shared.exceptions import MCPError, UrlElicitationRequiredError
 

tests/server/mcpserver/tools/test_base.py

@@ -1,6 +1,7 @@
+import mcp_types as types
 import pytest
 
-from mcp import Client, types
+from mcp import Client
 from mcp.server.mcpserver import Context, MCPServer
 from mcp.server.mcpserver.tools.base import Tool
 from mcp.shared.exceptions import MCPError

tests/server/test_cancel_handling.py

@@ -2,13 +2,7 @@
 
 import anyio
 import pytest
-
-from mcp import Client
-from mcp.server import Server, ServerRequestContext
-from mcp.shared.exceptions import MCPError
-from mcp.shared.message import SessionMessage
-from mcp.shared.version import LATEST_HANDSHAKE_VERSION
-from mcp.types import (
+from mcp_types import (
     CallToolRequest,
     CallToolRequestParams,
     CallToolResult,
@@ -24,6 +18,12 @@ from mcp.types import (
     TextContent,
     Tool,
 )
+from mcp_types.version import LATEST_HANDSHAKE_VERSION
+
+from mcp import Client
+from mcp.server import Server, ServerRequestContext
+from mcp.shared.exceptions import MCPError
+from mcp.shared.message import SessionMessage
 
 
 @pytest.mark.anyio

tests/server/test_completion_with_context.py

@@ -1,10 +1,7 @@
 """Tests for completion handler with context functionality."""
 
 import pytest
-
-from mcp import Client
-from mcp.server import Server, ServerRequestContext
-from mcp.types import (
+from mcp_types import (
     CompleteRequestParams,
     CompleteResult,
     Completion,
@@ -12,6 +9,9 @@ from mcp.types import (
     ResourceTemplateReference,
 )
 
+from mcp import Client
+from mcp.server import Server, ServerRequestContext
+
 
 @pytest.mark.anyio
 async def test_completion_handler_receives_context():

tests/server/test_connection.py

@@ -14,13 +14,7 @@ from typing import Any, Literal
 
 import anyio
 import pytest
-from pydantic import BaseModel, ValidationError
-
-from mcp.server.connection import Connection
-from mcp.shared.dispatcher import CallOptions
-from mcp.shared.exceptions import NoBackChannelError
-from mcp.shared.version import LATEST_HANDSHAKE_VERSION, LATEST_MODERN_VERSION
-from mcp.types import (
+from mcp_types import (
     LATEST_PROTOCOL_VERSION,
     ClientCapabilities,
     CreateMessageRequest,
@@ -38,6 +32,12 @@ from mcp.types import (
     SamplingContextCapability,
     SamplingToolsCapability,
 )
+from mcp_types.version import LATEST_HANDSHAKE_VERSION, LATEST_MODERN_VERSION
+from pydantic import BaseModel, ValidationError
+
+from mcp.server.connection import Connection
+from mcp.shared.dispatcher import CallOptions
+from mcp.shared.exceptions import NoBackChannelError
 
 _CLIENT_INFO = Implementation(name="t", version="0")
 

tests/server/test_lifespan.py

@@ -5,14 +5,7 @@ from contextlib import asynccontextmanager
 
 import anyio
 import pytest
-from pydantic import TypeAdapter
-
-from mcp.server import ServerRequestContext
-from mcp.server.lowlevel.server import NotificationOptions, Server
-from mcp.server.mcpserver import Context, MCPServer
-from mcp.server.models import InitializationOptions
-from mcp.shared.message import SessionMessage
-from mcp.types import (
+from mcp_types import (
     CallToolRequestParams,
     CallToolResult,
     ClientCapabilities,
@@ -24,6 +17,13 @@ from mcp.types import (
     JSONRPCResponse,
     TextContent,
 )
+from pydantic import TypeAdapter
+
+from mcp.server import ServerRequestContext
+from mcp.server.lowlevel.server import NotificationOptions, Server
+from mcp.server.mcpserver import Context, MCPServer
+from mcp.server.models import InitializationOptions
+from mcp.shared.message import SessionMessage
 
 
 @pytest.mark.anyio