Skip to content

Commit 1a8c14a

Browse files
authored
refactor: replace mcp.types as types to from mcp import types (#1986)
1 parent 4fc49c6 commit 1a8c14a

File tree

46 files changed

+66
-73
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

46 files changed

+66
-73
lines changed

README.md

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -682,7 +682,7 @@ The Context object provides the following capabilities:
682682
- `ctx.session` - Access to the underlying session for advanced communication (see [Session Properties and Methods](#session-properties-and-methods))
683683
- `ctx.request_context` - Access to request-specific data and lifespan resources (see [Request Context Properties](#request-context-properties))
684684
- `await ctx.debug(message)` - Send debug log message
685-
- `await ctx.info(message)` - Send info log message
685+
- `await ctx.info(message)` - Send info log message
686686
- `await ctx.warning(message)` - Send warning log message
687687
- `await ctx.error(message)` - Send error log message
688688
- `await ctx.log(level, message, logger_name=None)` - Send log with custom level
@@ -1110,13 +1110,13 @@ The session object accessible via `ctx.session` provides advanced control over c
11101110
async def notify_data_update(resource_uri: str, ctx: Context) -> str:
11111111
"""Update data and notify clients of the change."""
11121112
# Perform data update logic here
1113-
1113+
11141114
# Notify clients that this specific resource changed
11151115
await ctx.session.send_resource_updated(AnyUrl(resource_uri))
1116-
1116+
11171117
# If this affects the overall resource list, notify about that too
11181118
await ctx.session.send_resource_list_changed()
1119-
1119+
11201120
return f"Updated {resource_uri} and notified clients"
11211121
```
11221122

@@ -1145,11 +1145,11 @@ def query_with_config(query: str, ctx: Context) -> str:
11451145
"""Execute a query using shared database and configuration."""
11461146
# Access typed lifespan context
11471147
app_ctx: AppContext = ctx.request_context.lifespan_context
1148-
1148+
11491149
# Use shared resources
11501150
connection = app_ctx.db
11511151
settings = app_ctx.config
1152-
1152+
11531153
# Execute query with configuration
11541154
result = connection.execute(query, timeout=settings.query_timeout)
11551155
return str(result)
@@ -1644,7 +1644,7 @@ from contextlib import asynccontextmanager
16441644
from typing import Any
16451645

16461646
import mcp.server.stdio
1647-
import mcp.types as types
1647+
from mcp import types
16481648
from mcp.server.lowlevel import NotificationOptions, Server
16491649
from mcp.server.models import InitializationOptions
16501650

@@ -1758,7 +1758,7 @@ uv run examples/snippets/servers/lowlevel/basic.py
17581758
import asyncio
17591759

17601760
import mcp.server.stdio
1761-
import mcp.types as types
1761+
from mcp import types
17621762
from mcp.server.lowlevel import NotificationOptions, Server
17631763
from mcp.server.models import InitializationOptions
17641764

@@ -1837,7 +1837,7 @@ import asyncio
18371837
from typing import Any
18381838

18391839
import mcp.server.stdio
1840-
import mcp.types as types
1840+
from mcp import types
18411841
from mcp.server.lowlevel import NotificationOptions, Server
18421842
from mcp.server.models import InitializationOptions
18431843

@@ -1939,7 +1939,7 @@ import asyncio
19391939
from typing import Any
19401940

19411941
import mcp.server.stdio
1942-
import mcp.types as types
1942+
from mcp import types
19431943
from mcp.server.lowlevel import NotificationOptions, Server
19441944
from mcp.server.models import InitializationOptions
19451945

@@ -2012,7 +2012,7 @@ For servers that need to handle large datasets, the low-level server provides pa
20122012
```python
20132013
"""Example of implementing pagination with MCP server decorators."""
20142014

2015-
import mcp.types as types
2015+
from mcp import types
20162016
from mcp.server.lowlevel import Server
20172017

20182018
# Initialize the server

README.v2.md

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -683,7 +683,7 @@ The Context object provides the following capabilities:
683683
- `ctx.session` - Access to the underlying session for advanced communication (see [Session Properties and Methods](#session-properties-and-methods))
684684
- `ctx.request_context` - Access to request-specific data and lifespan resources (see [Request Context Properties](#request-context-properties))
685685
- `await ctx.debug(message)` - Send debug log message
686-
- `await ctx.info(message)` - Send info log message
686+
- `await ctx.info(message)` - Send info log message
687687
- `await ctx.warning(message)` - Send warning log message
688688
- `await ctx.error(message)` - Send error log message
689689
- `await ctx.log(level, message, logger_name=None)` - Send log with custom level
@@ -1111,13 +1111,13 @@ The session object accessible via `ctx.session` provides advanced control over c
11111111
async def notify_data_update(resource_uri: str, ctx: Context) -> str:
11121112
"""Update data and notify clients of the change."""
11131113
# Perform data update logic here
1114-
1114+
11151115
# Notify clients that this specific resource changed
11161116
await ctx.session.send_resource_updated(AnyUrl(resource_uri))
1117-
1117+
11181118
# If this affects the overall resource list, notify about that too
11191119
await ctx.session.send_resource_list_changed()
1120-
1120+
11211121
return f"Updated {resource_uri} and notified clients"
11221122
```
11231123

@@ -1146,11 +1146,11 @@ def query_with_config(query: str, ctx: Context) -> str:
11461146
"""Execute a query using shared database and configuration."""
11471147
# Access typed lifespan context
11481148
app_ctx: AppContext = ctx.request_context.lifespan_context
1149-
1149+
11501150
# Use shared resources
11511151
connection = app_ctx.db
11521152
settings = app_ctx.config
1153-
1153+
11541154
# Execute query with configuration
11551155
result = connection.execute(query, timeout=settings.query_timeout)
11561156
return str(result)
@@ -1645,7 +1645,7 @@ from contextlib import asynccontextmanager
16451645
from typing import Any
16461646

16471647
import mcp.server.stdio
1648-
import mcp.types as types
1648+
from mcp import types
16491649
from mcp.server.lowlevel import NotificationOptions, Server
16501650
from mcp.server.models import InitializationOptions
16511651

@@ -1759,7 +1759,7 @@ uv run examples/snippets/servers/lowlevel/basic.py
17591759
import asyncio
17601760

17611761
import mcp.server.stdio
1762-
import mcp.types as types
1762+
from mcp import types
17631763
from mcp.server.lowlevel import NotificationOptions, Server
17641764
from mcp.server.models import InitializationOptions
17651765

@@ -1838,7 +1838,7 @@ import asyncio
18381838
from typing import Any
18391839

18401840
import mcp.server.stdio
1841-
import mcp.types as types
1841+
from mcp import types
18421842
from mcp.server.lowlevel import NotificationOptions, Server
18431843
from mcp.server.models import InitializationOptions
18441844

@@ -1940,7 +1940,7 @@ import asyncio
19401940
from typing import Any
19411941

19421942
import mcp.server.stdio
1943-
import mcp.types as types
1943+
from mcp import types
19441944
from mcp.server.lowlevel import NotificationOptions, Server
19451945
from mcp.server.models import InitializationOptions
19461946

@@ -2013,7 +2013,7 @@ For servers that need to handle large datasets, the low-level server provides pa
20132013
```python
20142014
"""Example of implementing pagination with MCP server decorators."""
20152015

2016-
import mcp.types as types
2016+
from mcp import types
20172017
from mcp.server.lowlevel import Server
20182018

20192019
# Initialize the server

examples/servers/simple-pagination/mcp_simple_pagination/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
import anyio
1010
import click
11-
import mcp.types as types
11+
from mcp import types
1212
from mcp.server.lowlevel import Server
1313
from starlette.requests import Request
1414

examples/servers/simple-prompt/mcp_simple_prompt/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import anyio
22
import click
3-
import mcp.types as types
3+
from mcp import types
44
from mcp.server.lowlevel import Server
55
from starlette.requests import Request
66

examples/servers/simple-resource/mcp_simple_resource/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import anyio
22
import click
3-
import mcp.types as types
3+
from mcp import types
44
from mcp.server.lowlevel import Server
55
from mcp.server.lowlevel.helper_types import ReadResourceContents
66
from starlette.requests import Request

examples/servers/simple-streamablehttp-stateless/mcp_simple_streamablehttp_stateless/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55

66
import anyio
77
import click
8-
import mcp.types as types
98
import uvicorn
9+
from mcp import types
1010
from mcp.server.lowlevel import Server
1111
from mcp.server.streamable_http_manager import StreamableHTTPSessionManager
1212
from starlette.applications import Starlette

examples/servers/simple-streamablehttp/mcp_simple_streamablehttp/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
import anyio
77
import click
8-
import mcp.types as types
8+
from mcp import types
99
from mcp.server.lowlevel import Server
1010
from mcp.server.streamable_http_manager import StreamableHTTPSessionManager
1111
from starlette.applications import Starlette

examples/servers/simple-task-interactive/mcp_simple_task_interactive/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@
1111
from typing import Any
1212

1313
import click
14-
import mcp.types as types
1514
import uvicorn
15+
from mcp import types
1616
from mcp.server.experimental.task_context import ServerTaskContext
1717
from mcp.server.lowlevel import Server
1818
from mcp.server.streamable_http_manager import StreamableHTTPSessionManager

examples/servers/simple-task/mcp_simple_task/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66

77
import anyio
88
import click
9-
import mcp.types as types
109
import uvicorn
10+
from mcp import types
1111
from mcp.server.experimental.task_context import ServerTaskContext
1212
from mcp.server.lowlevel import Server
1313
from mcp.server.streamable_http_manager import StreamableHTTPSessionManager

examples/servers/simple-tool/mcp_simple_tool/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import anyio
44
import click
5-
import mcp.types as types
5+
from mcp import types
66
from mcp.server.lowlevel import Server
77
from mcp.shared._httpx_utils import create_mcp_http_client
88
from starlette.requests import Request

0 commit comments

Comments
 (0)