refactor: refactor mcp toolset to hide non-public field

PiperOrigin-RevId: 758494601
This commit is contained in:
Xiang (Sean) Zhou
2025-05-13 21:05:22 -07:00
committed by Copybara-Service
parent 14cf910ce6
commit fc40226ec0
3 changed files with 31 additions and 29 deletions
@@ -138,15 +138,15 @@ class MCPSessionManager:
errlog: (Optional) TextIO stream for error logging. Use only for
initializing a local stdio MCP session.
"""
self.connection_params = connection_params
self.exit_stack = exit_stack
self.errlog = errlog
self._connection_params = connection_params
self._exit_stack = exit_stack
self._errlog = errlog
async def create_session(self) -> ClientSession:
return await MCPSessionManager.initialize_session(
connection_params=self.connection_params,
exit_stack=self.exit_stack,
errlog=self.errlog,
connection_params=self._connection_params,
exit_stack=self._exit_stack,
errlog=self._errlog,
)
@classmethod