fix pylint

Author: Shreyas-Microsoft

src/backend/sql_agents/convert_script.py

@@ -4,7 +4,6 @@
 and updates the database with the results.
 """
 
-import asyncio
 import json
 import logging
 
@@ -45,8 +44,8 @@ async def convert_script(
     logger.info("Migrating query: %s\n", source_script)
 
     # Setup the group chat for the agents
-    chat = CommsManager(sql_agents.idx_agents).group_chat
-    #retry logic comms manager
+    # chat = CommsManager(sql_agents.idx_agents).group_chat
+    # retry logic comms manager
     comms_manager = CommsManager(
         sql_agents.idx_agents,
         max_retries=5,          # Retry up to 5 times for rate limits
@@ -216,7 +215,7 @@ async def convert_script(
                 #     ),
                 # )
                 # Safely parse response content to avoid crashing on malformed or incomplete JSON
-                #start
+                # start
                 try:
                     parsed_content = json.loads(response.content or "{}")
                 except json.JSONDecodeError:
@@ -240,7 +239,7 @@ async def convert_script(
                         FileResult.INFO,
                     ),
                 )
-                ##end
+                # end
                 await batch_service.create_file_log(
                     str(file.file_id),
                     description,
@@ -250,7 +249,7 @@ async def convert_script(
                     AuthorRole(response.role),
                 )
         except Exception as e:
-            #logger.error("Error during chat.invoke(): %s", str(e))
+            # logger.error("Error during chat.invoke(): %s", str(e))
             logger.error("Error during comms_manager.async_invoke(): %s", str(e))
             # Log the error to the batch service for tracking
             await batch_service.create_file_log(
@@ -356,4 +355,4 @@ async def validate_migration(
         author_role=AuthorRole.ASSISTANT,
     )
 
-    return True
+    return True

src/backend/sql_agents/helpers/comms_manager.py

@@ -115,7 +115,7 @@ async def should_agent_terminate(self, agent, history):
             return terminate
 
     def __init__(
-        self, 
+        self,
         agent_dict,
         exception_types: tuple = (Exception,),
         max_retries: int = 10,
@@ -124,7 +124,7 @@ def __init__(
         simple_truncation: int = None,
     ):
         """Initialize the CommsManager and agent_chat with the given agents.
-        
+
         Args:
             agent_dict: Dictionary of agents
             exception_types: Tuple of exception types that should trigger a retry
@@ -139,7 +139,7 @@ def __init__(
         self.backoff_factor = backoff_factor
         self.exception_types = exception_types
         self.simple_truncation = simple_truncation
-        
+
         # Initialize the group chat (maintaining original functionality)
         self.group_chat = AgentGroupChat(
             agents=agent_dict.values(),
@@ -168,12 +168,12 @@ async def async_invoke(self) -> AsyncIterable[ChatMessageContent]:
                 # Grab a snapshot of the history of the group chat
                 # Using copy to avoid getting a reference to the original list
                 history_snap = copy.deepcopy(self.group_chat.history)
-                
+
                 self.logger.debug(
                     "History before invoke: %s",
                     [msg.name for msg in self.group_chat.history],
                 )
-                
+
                 # Get a fresh iterator from the function
                 async_iter = self.group_chat.invoke()
 
@@ -257,4 +257,4 @@ async def async_invoke(self) -> AsyncIterable[ChatMessageContent]:
                 )
 
                 await asyncio.sleep(current_delay)
-                current_delay *= self.backoff_factor
+                current_delay *= self.backoff_factor

src/backend/sql_agents/process_batch.py

@@ -131,7 +131,7 @@ async def process_batch_async(
                 else:
                     await batch_service.update_file_counts(file["file_id"])
                 # TEMPORARY: awaiting bug fix for rate limits
-                #await asyncio.sleep(5)
+                # await asyncio.sleep(5)
             except UnicodeDecodeError as ucde:
                 logger.error("Error decoding file: %s", file)
                 logger.error("Error decoding file. %s", ucde)