From 0f731598ddf69517c9d50b67016b692350decbb4 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Wed, 31 Jul 2024 10:10:13 -0400 Subject: [PATCH] refactor to get .get() --- backend/routes/generate_code.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py index 6d4f0e4..055853f 100644 --- a/backend/routes/generate_code.py +++ b/backend/routes/generate_code.py @@ -107,13 +107,11 @@ async def stream_code(websocket: WebSocket): print("Received params") - # Read the code config settings (stack) from the request. Fall back to default if not provided. - generated_code_config = "" - if "generatedCodeConfig" in params and params["generatedCodeConfig"]: - generated_code_config = params["generatedCodeConfig"] + # Read the code config settings (stack) from the request. + generated_code_config = params.get("generatedCodeConfig", "") if not generated_code_config in get_args(Stack): await throw_error(f"Invalid generated code config: {generated_code_config}") - return + raise Exception(f"Invalid generated code config: {generated_code_config}") # Cast the variable to the Stack type valid_stack = cast(Stack, generated_code_config)