(self)
| 101 | return |
| 102 | |
| 103 | async def _init_client(self): |
| 104 | # If we have a session id (from checkpoint), attach to that server session. |
| 105 | if self._session_id: |
| 106 | transport = StreamableHttpTransport( |
| 107 | url=self.server_url, |
| 108 | headers={"mcp-session-id": self._session_id}, |
| 109 | ) |
| 110 | self._client = Client(transport) |
| 111 | else: |
| 112 | self._client = Client(self.server_url) |
| 113 | await self._client.__aenter__() # manually enter async context |
| 114 | self._update_session_id_from_transport() |
| 115 | |
| 116 | async def _close_client(self): |
| 117 | if self._client: |
no test coverage detected