From 196cf5865b80ac264eb170ae113d490140f1ff53 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Tue, 19 Mar 2024 18:21:17 -0400 Subject: [PATCH] set a default for inputMode param --- backend/routes/generate_code.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py index 9a1b2c0..1d75ca1 100644 --- a/backend/routes/generate_code.py +++ b/backend/routes/generate_code.py @@ -76,7 +76,7 @@ async def stream_code(websocket: WebSocket): valid_stack = cast(Stack, generated_code_config) # Validate the input mode - input_mode = params.get("inputMode") + input_mode = params.get("inputMode", "image") if not input_mode in get_args(InputMode): await throw_error(f"Invalid input mode: {input_mode}") raise Exception(f"Invalid input mode: {input_mode}")