Upload 2 files

main.py

@@ -628,7 +628,7 @@ 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:
-        openai.api_key = ""
+        openai.api_key = item.api_key
         openai.api_base = SOURCE_OPENAI_ACCESS_URL
         chat_completion = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
@@ -636,14 +636,16 @@ def chatgpt3_turbo(item: Chatgpt3Texts):
             stream=True
         )
         if isinstance(chat_completion, dict):
-            print(chat_completion.choices[0].message.content)
+            answer = chat_completion.choices[0].message.content
+            answer = ""
         else:
             for token in chat_completion:
                 content = token["choices"][0]["delta"].get("content")
                 if content is not None:
+                    answer += content
                     return SuccessResponse(
                         status="True",
-                        randydev={"message": content}
+                        randydev={"message": answer}
                     )
     else:
         url = "https://lexica.qewertyy.me/models"
@@ -663,7 +665,7 @@ def chatgpt4_turbo(
     item: OpenaiTexts,
     api_key: None = Depends(validate_api_key)
 ):
-    openai.api_key = ""
+    openai.api_key = item.api_key
     openai.api_base = SOURCE_OPENAI_ACCESS_URL
     chat_completion = openai.ChatCompletion.create(
         model="gpt-4-turbo",
@@ -671,15 +673,17 @@ def chatgpt4_turbo(
         stream=True
     )
     if isinstance(chat_completion, dict):
-        print(chat_completion.choices[0].message.content)
+        answer = chat_completion.choices[0].message.content
+        answer = ""
     else:
         for token in chat_completion:
             content = token["choices"][0]["delta"].get("content")
-        if content is not None:
-            return SuccessResponse(
-                status="True",
-                randydev={"message": content}
-            )
+            if content is not None:
+                answer += content
+                return SuccessResponse(
+                    status="True",
+                    randydev={"message": answer}
+                )
 
 @app.post("/ryuzaki/google-ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def v1beta3_google_ai(

models.py

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