Upload 2 files

main.py

@@ -628,16 +628,24 @@ def chatgpt_model(item: ChatgptModel):
 @app.post("/ryuzaki/chatgpt3-turbo", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def chatgpt3_turbo(item: Chatgpt3Texts):
     if item.is_openai_original:
-        GPTbase = SOURCE_OPENAI_ACCESS_URL
-        response = OpenAiToken(api_key=item.api_key, api_base=GPTbase).chat_message_turbo(
-            query=item.query,
-            model="gpt-3.5-turbo"
-        )
-        answer = response["choices"][0]["message"]["content"]
-        return SuccessResponse(
-            status="True",
-            randydev={"message": answer}
-        )
+        try:
+            GPTbase = SOURCE_OPENAI_ACCESS_URL
+            response = OpenAiToken(api_key=item.api_key, api_base=GPTbase).chat_message_turbo(
+                query=item.query,
+                model="gpt-3.5-turbo",
+                is_stream=item.is_stream
+            )
+            answer = response[0]
+            continue_chat = response[1]
+            return SuccessResponse(
+                status="True",
+                randydev={
+                    "message": answer,
+                    "chat_history": continue_chat
+                }
+            )
+        except Exception as e:
+            return SuccessResponse(status="False", randydev={"message": f"Error responding: {e}"})
     else:
         url = "https://lexica.qewertyy.me/models"
         params = {"model_id": 5, "prompt": item.query}
@@ -656,21 +664,24 @@ def chatgpt4_turbo(
     item: OpenaiTexts,
     api_key: None = Depends(validate_api_key)
 ):
-    GPTbase = SOURCE_OPENAI_ACCESS_URL
-    response = OpenAiToken(api_key=item.api_key, api_base=GPTbase).chat_message_turbo(
-        query=item.query,
-        model="gpt-4",
-        is_stream=item.is_stream
-    )
-    answer = response[0]
-    continue_chat = response[1]
-    return SuccessResponse(
-        status="True",
-        randydev={
-            "message": answer,
-            "chat_history": continue_chat
-        }
-    )
+    try:
+        GPTbase = SOURCE_OPENAI_ACCESS_URL
+        response = OpenAiToken(api_key=item.api_key, api_base=GPTbase).chat_message_turbo(
+            query=item.query,
+            model="gpt-4",
+            is_stream=item.is_stream
+        )
+        answer = response[0]
+        continue_chat = response[1]
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "message": answer,
+                "chat_history": continue_chat
+            }
+        )
+    except Exception as e:
+        return SuccessResponse(status="False", randydev={"message": f"Error responding: {e}"})
 
 @app.post("/ryuzaki/google-ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def v1beta3_google_ai(

models.py

@@ -134,12 +134,13 @@ class AnimeStyled(BaseModel):
 
 class OpenaiTexts(BaseModel):
     query: str
-    api_key: str
+    api_key: Optional[str] = ""
     is_stream: Optional[bool] = False
 
 class Chatgpt3Texts(BaseModel):
     query: str
-    api_key: str
+    api_key: Optional[str] = ""
+    is_stream: Optional[bool] = False
     is_openai_original: Optional[bool] = False
 
 class TextCustom(BaseModel):