Update generate_code.py

2023-12-12 10:04:28 +05:30 · 2023-12-12 10:04:28 +05:30 · 8ff1bf1976
commit 8ff1bf1976
parent 7eab735af4
1 changed files with 4 additions and 2 deletions
--- a/backend/routes/generate_code.py
+++ b/backend/routes/generate_code.py
@ -174,10 +174,12 @@ async def stream_code(websocket: WebSocket):
        try:
            if params.get("resultImage") and params["resultImage"]:
                prompt_messages = assemble_prompt(
-                    params["image"], valid_stack, params["resultImage"]
+                    params["images"], generated_code_config, params["resultImage"]
                )
            else:
-                prompt_messages = assemble_prompt(params["image"], valid_stack)
+                prompt_messages = assemble_prompt(
+                    params["images"], generated_code_config
+                )
        except:
            await websocket.send_json(
                {