Adapt story suite to the mcp-types package split after rebase

Author: maxisbey

examples/stories/_harness.py

@@ -19,13 +19,13 @@
 
 import anyio
 import httpx
+from mcp_types.version import LATEST_MODERN_VERSION
 
 from mcp import StdioServerParameters, stdio_client
 from mcp.client import Transport
 from mcp.client.streamable_http import streamable_http_client
 from mcp.server import Server
 from mcp.server.mcpserver import MCPServer
-from mcp.shared.version import LATEST_MODERN_VERSION
 
 if sys.version_info >= (3, 11):
     import tomllib

examples/stories/bearer_auth/server_lowlevel.py

@@ -2,10 +2,10 @@
 
 from typing import Any
 
+import mcp_types as types
 from pydantic import AnyHttpUrl
 from starlette.applications import Starlette
 
-from mcp import types
 from mcp.server.auth.middleware.auth_context import get_access_token
 from mcp.server.auth.settings import AuthSettings
 from mcp.server.context import ServerRequestContext

examples/stories/custom_methods/client.py

@@ -2,7 +2,8 @@
 
 from typing import Literal, cast
 
-from mcp import types
+import mcp_types as types
+
 from mcp.client import Client
 from stories._harness import Target, run_client
 

examples/stories/custom_methods/server.py

@@ -6,7 +6,8 @@
 
 from typing import Any
 
-from mcp import types
+import mcp_types as types
+
 from mcp.server.context import ServerRequestContext
 from mcp.server.lowlevel import Server
 from stories._hosting import run_server_from_args

examples/stories/dual_era/client.py

@@ -1,8 +1,9 @@
 """Connect to the same server factory twice — once per era, so `main` takes `targets` — and assert both are served."""
 
-from mcp import types
+import mcp_types as types
+from mcp_types.version import LATEST_HANDSHAKE_VERSION, LATEST_MODERN_VERSION
+
 from mcp.client import Client
-from mcp.shared.version import LATEST_HANDSHAKE_VERSION, LATEST_MODERN_VERSION
 from stories._harness import TargetFactory, run_client
 
 

examples/stories/dual_era/server.py

@@ -1,7 +1,8 @@
 """One MCPServer factory that serves both the 2025 handshake era and the 2026 stateless era."""
 
+from mcp_types.version import MODERN_PROTOCOL_VERSIONS
+
 from mcp.server.mcpserver import Context, MCPServer
-from mcp.shared.version import MODERN_PROTOCOL_VERSIONS
 from stories._hosting import run_server_from_args
 
 

examples/stories/dual_era/server_lowlevel.py

@@ -2,10 +2,11 @@
 
 from typing import Any
 
-from mcp import types
+import mcp_types as types
+from mcp_types.version import MODERN_PROTOCOL_VERSIONS
+
 from mcp.server.context import ServerRequestContext
 from mcp.server.lowlevel import Server
-from mcp.shared.version import MODERN_PROTOCOL_VERSIONS
 from stories._hosting import run_server_from_args
 
 GREET_INPUT_SCHEMA: dict[str, Any] = {

examples/stories/error_handling/client.py

@@ -1,8 +1,9 @@
 """Prove the two error channels: is_error results return; MCPError raises."""
 
+from mcp_types import INVALID_PARAMS, TextContent
+
 from mcp import MCPError
 from mcp.client import Client
-from mcp.types import INVALID_PARAMS, TextContent
 from stories._harness import Target, run_client
 
 

examples/stories/error_handling/server.py

@@ -1,9 +1,10 @@
 """Two error channels: ToolError -> is_error result; MCPError -> JSON-RPC protocol error."""
 
+from mcp_types import INVALID_PARAMS
+
 from mcp.server.mcpserver import MCPServer
 from mcp.server.mcpserver.exceptions import ToolError
 from mcp.shared.exceptions import MCPError
-from mcp.types import INVALID_PARAMS
 from stories._hosting import run_server_from_args
 
 

examples/stories/error_handling/server_lowlevel.py

@@ -2,7 +2,8 @@
 
 from typing import Any
 
-from mcp import types
+import mcp_types as types
+
 from mcp.server.context import ServerRequestContext
 from mcp.server.lowlevel import Server
 from mcp.shared.exceptions import MCPError