Update main.py

main.py

@@ -894,16 +894,19 @@ async def chatgpt_4o(
     item: ChatgptCustom,
     api_key: None = Depends(validate_api_key)
 ):
-    client = Client()
-    response = client.chat.completions.create(
-        model="gpt-4o",
-        messages=[{"role": "user", "content": item.query}],
-    )
-    answer = response.choices[0].message.content
-    return SuccessResponse(
-        status="True",
-        randydev={"message": answer}
-    )
+    try:
+        client = Client()
+        response = client.chat.completions.create(
+            model="gpt-4o",
+            messages=[{"role": "user", "content": item.query}]
+        )
+        answer = response.choices[0].message.content
+        return SuccessResponse(
+            status="True",
+            randydev={"message": answer}
+        )
+    except Exception as e:
+        return SuccessResponse(status="False", randydev={"message": str(e)})
 
 @app.post("/ryuzaki/chatgpt-custom", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 async def chatgpt_customs(