Skip to content

Commit 20d4fa3

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 20d4fa3

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/mcp/server/session.py

Lines changed: 4 additions & 3 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,7 +104,9 @@ 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
111112
mcp_protocol_version: str | None = (

0 commit comments

Comments
 (0)