Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
150a741e92 |
@@ -345,9 +345,6 @@ class MCPClient:
|
|||||||
|
|
||||||
async def cleanup(self):
|
async def cleanup(self):
|
||||||
"""Clean up resources including old exit stacks from reconnections"""
|
"""Clean up resources including old exit stacks from reconnections"""
|
||||||
# Set running_event first to unblock any waiting tasks
|
|
||||||
self.running_event.set()
|
|
||||||
|
|
||||||
# Close current exit stack
|
# Close current exit stack
|
||||||
try:
|
try:
|
||||||
await self.exit_stack.aclose()
|
await self.exit_stack.aclose()
|
||||||
@@ -359,6 +356,9 @@ class MCPClient:
|
|||||||
# Just clear the list to release references
|
# Just clear the list to release references
|
||||||
self._old_exit_stacks.clear()
|
self._old_exit_stacks.clear()
|
||||||
|
|
||||||
|
# Set running_event first to unblock any waiting tasks
|
||||||
|
self.running_event.set()
|
||||||
|
|
||||||
|
|
||||||
class MCPTool(FunctionTool, Generic[TContext]):
|
class MCPTool(FunctionTool, Generic[TContext]):
|
||||||
"""A function tool that calls an MCP service."""
|
"""A function tool that calls an MCP service."""
|
||||||
|
|||||||
Reference in New Issue
Block a user