diff --git a/llm_processor.py b/llm_processor.py index 7558865..586b7d4 100644 --- a/llm_processor.py +++ b/llm_processor.py @@ -84,7 +84,7 @@ class LLMProcessor: env=None ) - self.exit_stack = asyncio.AsyncExitStack() + self.exit_stack = contextlib.AsyncExitStack() stdio_transport = await self.exit_stack.enter_async_context(stdio_client(server_params)) stdio, write = stdio_transport self.session = await self.exit_stack.enter_async_context(ClientSession(stdio, write))