Merge pull request #173 from microsoft/psl-thread-management

feat: Agent and thread management

Author: Roopan-Microsoft

src/backend/sql_agents/helpers/comms_manager.py

@@ -258,3 +258,29 @@ async def async_invoke(self) -> AsyncIterable[ChatMessageContent]:
 
                 await asyncio.sleep(current_delay)
                 current_delay *= self.backoff_factor
+
+    async def cleanup(self):
+        """Clean up all resources including internal threads."""
+        try:
+            if self.group_chat is not None:
+                self.logger.debug("Cleaning up AgentGroupChat resources...")
+                # Reset the group chat - this clears conversation state and deletes remote threads
+                await self.group_chat.reset()
+                self.logger.debug("AgentGroupChat cleanup completed successfully")
+
+        except Exception as e:
+            self.logger.error("Error during cleanup: %s", str(e))
+
+    def __del__(self):
+        """Destructor to ensure cleanup if not explicitly called."""
+        try:
+            # Only attempt cleanup if there's an active event loop
+            loop = asyncio.get_running_loop()
+            if loop and not loop.is_closed():
+                # Schedule cleanup as a task
+                loop.create_task(self.cleanup())
+        except RuntimeError:
+            # No event loop running, can't clean up asynchronously
+            self.logger.warning("No event loop available for cleanup in destructor")
+        except Exception as e:
+            self.logger.error("Error in destructor cleanup: %s", str(e))