fix bug with history not being updated if image generation fails

sud0su · Dec 14, 2023 · f1a9859 · f1a9859
1 parent 9494f4c
commit f1a9859
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
@@ -254,6 +254,9 @@ async def process_chunk(content: str):
     except Exception as e:
         traceback.print_exc()
         print("Image generation failed", e)
+        # Send set code even if image generation fails since that triggers
+        # the frontend to update history
+        await websocket.send_json({"type": "setCode", "value": completion})
         await websocket.send_json(
             {"type": "status", "value": "Image generation failed but code is complete."}
         )