From 8ff1bf1976ae2f497461af6bf3d6b49a91522cfa Mon Sep 17 00:00:00 2001 From: Piyush Mishra Date: Tue, 12 Dec 2023 10:04:28 +0530 Subject: [PATCH] Update generate_code.py --- backend/routes/generate_code.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py index e7186fc..4de8412 100644 --- 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( {