Skip to content

Commit 2bfc4e0

Browse files
committed
lazy init mcp session
Signed-off-by: wuhang <wuhang6@huawei.com>
1 parent 5be7ee5 commit 2bfc4e0

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

vllm/entrypoints/context.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,9 @@ async def __aenter__(self):
315315
async with self._reference_count_lock:
316316
self._reference_count += 1
317317
if self._async_exit_stack is None:
318+
assert (self._reference_count == 1
319+
), "Reference count of exit stack should be "
320+
"1 when initializing exit stack."
318321
self._async_exit_stack = AsyncExitStack()
319322
await self._async_exit_stack.__aenter__()
320323
return self

0 commit comments

Comments
 (0)