Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

when payment method is unknown, stop execution

Browse Source
This commit is contained in:
Abi Raja 2024-09-20 13:34:55 +02:00
parent 06ce9f8c32
commit fc70ecac6b
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/routes/generate_code.py
View File

@ -273,6 +273,10 @@ async def stream_code(websocket: WebSocket):
should_generate_images = extracted_params.should_generate_images should_generate_images = extracted_params.should_generate_images
payment_method = extracted_params.payment_method payment_method = extracted_params.payment_method
# If the payment method is unknown, we shouldn't proceed
if payment_method is PaymentMethod.UNKNOWN:
return
# Auto-upgrade usage of older models # Auto-upgrade usage of older models
code_generation_model = auto_upgrade_model(code_generation_model) code_generation_model = auto_upgrade_model(code_generation_model)