thread management

Author: Shreyas-Microsoft

src/backend/sql_agents/convert_script.py

@@ -275,6 +275,8 @@ async def convert_script(
     logger.info(
         "Analysis of source and migrated queries:\n%s", "semantic verifier response"
     )
+    # Delete the thread using CommsManager
+    # await comms_manager.close()
 
     return migrated_query
 

src/backend/sql_agents/helpers/comms_manager.py

@@ -258,3 +258,11 @@ async def async_invoke(self) -> AsyncIterable[ChatMessageContent]:
 
                 await asyncio.sleep(current_delay)
                 current_delay *= self.backoff_factor
+
+    async def close(self):
+        """Clean up the thread after agent execution is complete."""
+        try:
+            await self.group_chat.delete()
+            self.logger.info("Deleted group chat thread successfully.")
+        except Exception as ex:
+            self.logger.warning("Failed to delete group chat thread: %s", str(ex))