Skip to content

Commit 3261cbc

Browse files
committed
more tidy up
1 parent ad83eea commit 3261cbc

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

src/mcp/client/session.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def __init__(
142142
self._list_roots_callback = list_roots_callback or _default_list_roots_callback
143143
self._logging_callback = logging_callback or _default_logging_callback
144144
self._message_handler = message_handler or _default_message_handler
145-
self._tool_output_validator_provider = tool_output_validator_provider
145+
self._tool_output_validator_provider = tool_output_validator_provider or _default_tool_output_validator
146146

147147
async def initialize(self) -> types.InitializeResult:
148148
sampling = types.SamplingCapability()
@@ -183,10 +183,7 @@ async def initialize(self) -> types.InitializeResult:
183183
)
184184
)
185185

186-
tool_output_validator_provider = (
187-
self._tool_output_validator_provider or _default_tool_output_validator
188-
)
189-
self._tool_output_validator = await tool_output_validator_provider(self)
186+
self._tool_output_validator = await self._tool_output_validator_provider(self)
190187

191188
return result
192189

0 commit comments

Comments
 (0)