Upload 2 files

main.py

@@ -27,6 +27,7 @@ import shutil
 import random
 import tempfile
 import io
+import openai
 from io import BytesIO
 from datetime import datetime as dt
 from dotenv import load_dotenv
@@ -96,6 +97,7 @@ SOURCE_TIKTOK_TECH_URL = os.environ["SOURCE_TIKTOK_TECH_URL"]
 SOURCE_WHAT_GAY_URL = os.environ["SOURCE_WHAT_GAY_URL"]
 SOURCE_ASSISTANT_GOOGLE_AI = os.environ["SOURCE_ASSISTANT_GOOGLE_AI"]
 SOURCE_MONITOR_URL = os.environ["SOURCE_MONITOR_URL"]
+SOURCE_OPENAI_ACCESS_URL = os.environ["SOURCE_OPENAI_ACCESS_URL"]
 
 # api keys
 REVERSE_IMAGE_API = os.environ["REVERSE_IMAGE_API"]
@@ -572,7 +574,6 @@ def Anime_Styled(
         return SuccessResponse(status="True", randydev={"data": encoded_string})
     else:
         return SuccessResponse(status="False", randydev={"data": "Not found image data"})
-        
 
 @app.post("/ryuzaki/unsplash")
 def image_unsplash(item: GetImageUnsplash):
@@ -610,9 +611,7 @@ def chatgpt_model(item: ChatgptModel):
         answer = check_response.get("content")
         return SuccessResponse(
             status="True",
-            randydev={
-                "message": answer
-            }
+            randydev={"message": answer}
         )
     else:
         params = {"model_id": 5, "prompt": item.query}
@@ -623,27 +622,64 @@ def chatgpt_model(item: ChatgptModel):
         answer = check_response.get("content")
         return SuccessResponse(
             status="True",
-            randydev={
-                "message": answer
-            }
+            randydev={"message": answer}
         )
 
-
 @app.post("/ryuzaki/chatgpt3-turbo", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
-def chatgpt3_turbo(item: ChatgptCustom):
-    url = "https://lexica.qewertyy.me/models"
-    params = {"model_id": 5, "prompt": item.query}
-    response = requests.post(url, params=params)
-    if response.status_code != 200:
-        return f"Error status: {response.status_code}"
-    check_response = response.json()
-    answer = check_response.get("content")
-    return SuccessResponse(
-        status="True",
-        randydev={
-            "message": answer
-        }
+def chatgpt3_turbo(item: Chatgpt3Texts):
+    if item.is_openai_original:
+        openai.api_key = item.api_key
+        openai.api_base = SOURCE_OPENAI_ACCESS_URL
+        chat_completion = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": item.query}],
+            stream=True
+        )
+        if isinstance(chat_completion, dict):
+            print(chat_completion.choices[0].message.content)
+        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}
+                    )
+    else:
+        url = "https://lexica.qewertyy.me/models"
+        params = {"model_id": 5, "prompt": item.query}
+        response = requests.post(url, params=params)
+        if response.status_code != 200:
+            return f"Error status: {response.status_code}"
+        check_response = response.json()
+        answer = check_response.get("content")
+        return SuccessResponse(
+            status="True",
+            randydev={"message": answer}
+        )
+
+@app.post("/ryuzaki/chatgpt4-turbo", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def chatgpt4_turbo(
+    item: OpenaiTexts,
+    api_key: None = Depends(validate_api_key)
+):
+    openai.api_key = item.api_key
+    openai.api_base = SOURCE_OPENAI_ACCESS_URL
+    chat_completion = openai.ChatCompletion.create(
+        model="gpt-4-turbo",
+        messages=[{"role": "user", "content": item.query}],
+        stream=True
     )
+    if isinstance(chat_completion, dict):
+        print(chat_completion.choices[0].message.content)
+    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}
+                )
 
 @app.post("/ryuzaki/google-ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def v1beta3_google_ai(

models.py

@@ -132,5 +132,14 @@ class TranslateCustom(BaseModel):
 class AnimeStyled(BaseModel):
     query: str
 
+class OpenaiTexts(BaseModel):
+    query: str
+    api_key: str
+
+class Chatgpt3Texts(BaseModel):
+    query: str
+    api_key: Optional[str] = None
+    is_openai_original: Optional[bool] = False
+
 class TextCustom(BaseModel):
     query: str