Method _run

src/mcphub/__init__.py:94–129

(self)

        return fut.result()

    async def _run(self):
        stack = AsyncExitStack()
        sessions: Dict[str, ClientSession] = {}
        stop_fut = None
        try:
            while True:
                action, payload, fut = await self._queue.get()
                if action == "stop":
                    stop_fut = fut
                    break
                try:
                    if action == "open":
                        name, cmd, env = payload
                        params = StdioServerParameters(
                            command=cmd[0], args=cmd[1:], env=env
                        )
                        read, write = await stack.enter_async_context(
                            stdio_client(params)
                        )
                        session = await stack.enter_async_context(
                            ClientSession(read, write)
                        )
                        await session.initialize()
                        sessions[name] = session
                        _set(fut, None)
                    elif action == "list":
                        _set(fut, await sessions[payload].list_tools())
                    elif action == "call":
                        name, tool, args = payload
                        _set(fut, await sessions[name].call_tool(tool, args))
                except Exception as exc:
                    _set(fut, exc, error=True)
        finally:
            await stack.aclose()
            if stop_fut is not None:
                _set(stop_fut, None)

    def open(self, name, cmd, env):
        self._submit("open", (name, cmd, env))