update

Author: eyw520

servers/fai-lambda/fai-scribe/src/utils/agent.py

@@ -150,14 +150,14 @@ async def run_editing_session(
                         await update_session_status(editing_id, "active")
                         LOGGER.info(f"Transitioned new session {editing_id} from STARTUP to ACTIVE")
 
-            is_interrupted, receipt_handle = check_if_interrupted_via_sqs(sqs_client)
-            if is_interrupted:
-                LOGGER.warning(f"Session interrupted via SQS: {editing_id}")
-                if receipt_handle:
-                    sqs_client.delete_message(receipt_handle)
-                if session_id is not None:
+            if session_id is not None:
+                is_interrupted, receipt_handle = check_if_interrupted_via_sqs(sqs_client)
+                if is_interrupted:
+                    LOGGER.warning(f"Session interrupted via SQS: {editing_id}")
+                    if receipt_handle:
+                        sqs_client.delete_message(receipt_handle)
                     await update_session_status(editing_id, "waiting")
-                raise SessionInterruptedError(f"Editing session {editing_id} was interrupted")
+                    raise SessionInterruptedError(f"Editing session {editing_id} was interrupted")
 
             if isinstance(message, AssistantMessage):
                 for block in message.content: