Skip to content

Commit deb20cc

Browse files
fix: ruff import order and line length in session.py
Generated by Mistral Vibe. Co-Authored-By: Mistral Vibe <vibe@mistral.ai>
1 parent 192b727 commit deb20cc

File tree

2 files changed

+6
-7
lines changed

2 files changed

+6
-7
lines changed

src/mcp/server/lowlevel/server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,7 @@ async def _handle_request(
456456
meta = cast(dict[str, Any] | None, getattr(req.params, "meta", None)) if req.params else None
457457
parent_context = extract_trace_context(meta) if meta is not None else None
458458

459-
mcp_protocol_version: str | None = session.client_params.protocol_version if session.client_params else None
459+
mcp_protocol_version: str | None = str(session.client_params.protocol_version) if session.client_params else None
460460

461461
start_time = time.monotonic()
462462

src/mcp/server/session.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ async def handle_list_prompts(ctx: RequestContext, params) -> ListPromptsResult:
2828
be instantiated directly by users of the MCP framework.
2929
"""
3030

31+
import time
3132
from enum import Enum
3233
from typing import Any, TypeVar, overload
3334

@@ -40,8 +41,6 @@ async def handle_list_prompts(ctx: RequestContext, params) -> ListPromptsResult:
4041
from mcp.server.experimental.session_features import ExperimentalServerSessionFeatures
4142
from mcp.server.models import InitializationOptions
4243
from mcp.server.validation import validate_sampling_tools, validate_tool_use_result_messages
43-
import time
44-
4544
from mcp.shared._otel import record_server_session_duration
4645
from mcp.shared._stream_protocols import ReadStream, WriteStream
4746
from mcp.shared.exceptions import StatelessModeNotSupported
@@ -105,12 +104,12 @@ async def __aenter__(self) -> "ServerSession":
105104
self._session_start_time = time.monotonic()
106105
return await super().__aenter__()
107106

108-
async def __aexit__(self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: Any) -> bool | None:
107+
async def __aexit__(
108+
self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: Any
109+
) -> bool | None:
109110
if self._session_start_time is not None:
110111
duration = time.monotonic() - self._session_start_time
111-
mcp_protocol_version: str | None = (
112-
self._client_params.protocol_version if self._client_params else None
113-
)
112+
mcp_protocol_version: str | None = str(self._client_params.protocol_version) if self._client_params else None
114113
# Cancellation exceptions indicate transport close, not a session error.
115114
is_cancellation = exc_val is not None and isinstance(exc_val, anyio.get_cancelled_exc_class())
116115
error_type: str | None = type(exc_val).__name__ if exc_val is not None and not is_cancellation else None

0 commit comments

Comments
 (0)