From 76188a7465852e8d864a9101855edc59f7da94b2 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Mon, 16 Sep 2024 15:27:11 +0200 Subject: [PATCH] force image generation to be true on prod --- backend/routes/generate_code.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py index 0a1b766..489f6ad 100644 --- a/backend/routes/generate_code.py +++ b/backend/routes/generate_code.py @@ -193,7 +193,9 @@ async def extract_params( print("Using official OpenAI URL") # Get the image generation flag from the request. Fall back to True if not provided. - should_generate_images = bool(params.get("isImageGenerationEnabled", True)) + should_generate_images = ( + bool(params.get("isImageGenerationEnabled", True)) if not IS_PROD else True + ) return ExtractedParams( stack=validated_stack,