From fc70ecac6b19bf601af115f31007e7ede5a30f8e Mon Sep 17 00:00:00 2001
From: Abi Raja <abimanyuraja@gmail.com>
Date: Fri, 20 Sep 2024 13:34:55 +0200
Subject: [PATCH] when payment method is unknown, stop execution

---
 backend/routes/generate_code.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
index 9b94657..687b058 100644
--- a/backend/routes/generate_code.py
+++ b/backend/routes/generate_code.py
@@ -273,6 +273,10 @@ async def stream_code(websocket: WebSocket):
     should_generate_images = extracted_params.should_generate_images
     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
     code_generation_model = auto_upgrade_model(code_generation_model)