Skip to content

Commit bdfa1ee

Browse files
committed
address review: remove type cast and use direct dict access for method
1 parent 3796200 commit bdfa1ee

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

src/mcp/shared/session.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ async def send_request(
274274
# Store the callback for this request
275275
self._progress_callbacks[request_id] = progress_callback
276276

277-
method: str = request_data["method"]
277+
method = request_data["method"]
278278
span = start_client_span(method, request_data.get("params"))
279279

280280
try:
@@ -365,10 +365,7 @@ async def _receive_loop(self) -> None:
365365
by_name=False,
366366
)
367367
request_data = message.message.model_dump(by_alias=True, mode="json", exclude_none=True)
368-
server_span = start_server_span(
369-
request_data.get("method", ""),
370-
request_data.get("params"),
371-
)
368+
server_span = start_server_span(request_data["method"], request_data.get("params"))
372369
responder = RequestResponder(
373370
request_id=message.message.id,
374371
request_meta=validated_request.params.meta if validated_request.params else None,

0 commit comments

Comments
 (0)