Upload 2 files

main.py

@@ -68,7 +68,7 @@ from pymongo import MongoClient
 
 from RyuzakiLib.hackertools.chatgpt import RendyDevChat
 from RyuzakiLib.hackertools.gemini import GeminiLatest
-from RyuzakiLib.hackertools.openai_api import OpenAiToken
+from RyuzakiLib.hackertools.openai_api import OpenAI
 from RyuzakiLib.hackertools.huggingface import FaceAI
 
 from RyuzakiLib.mental import BadWordsList
@@ -628,22 +628,6 @@ def chatgpt_olds(item: ChatgptCustom):
     except Exception as e:
         return SuccessResponse(status="False", randydev={"message": contact_support})
 
-@app.get("/ryuzaki/freechatgpt-beta", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
-def free_chatgpt4_beta(item: ChatgptCustom):
-    try:
-        response = RendyDevChat(item.query).get_response_beta(joke=True)
-        return SuccessResponse(status="True", randydev={"message": response})
-    except:
-        return SuccessResponse(status="False", randydev={"message": contact_support})
-
-@app.get("/ryuzaki/freechatgpt-bing", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
-def free_chatgpt4_bing(item: ChatgptCustom):
-    try:
-        response = RendyDevChat(item.query).get_response_bing(bing=True)
-        return SuccessResponse(status="True", randydev={"message": response})
-    except:
-        return SuccessResponse(status="False", randydev={"message": contact_support})
-
 @app.post("/ryuzaki/ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def ryuzaki_ai(
     item: RyuzakiAi,
@@ -861,20 +845,16 @@ def chatgpt3_turbo(
     if item.is_openai_original:
         try:
             GPTbase = SOURCE_OPENAI_ACCESS_URL
-            response = OpenAiToken(api_key=item.api_key, api_base=GPTbase).chat_message_turbo(
+            response = await OpenAI(api_key=item.api_key, api_base=GPTbase).chat_message_turbo(
                 query=item.query,
                 model=item.model,
                 user_id=user_id,
                 is_stream=item.is_stream
             )
-            answer = response[0]
-            continue_chat = response[1]
+            answer = response
             return SuccessResponse(
                 status="True",
-                randydev={
-                    "message": answer,
-                    "chat_history": continue_chat
-                }
+                randydev={"message": answer}
             )
         except Exception:
             return SuccessResponse(status="False", randydev={"message": contact_support})
@@ -895,19 +875,31 @@ def chatgpt4_turbo(
 ):
     try:
         GPTbase = SOURCE_OPENAI_ACCESS_URL
-        response = OpenAiToken(api_key=item.api_key, api_base=GPTbase).chat_message_turbo(
+        response = await OpenAI(api_key=item.api_key, api_base=GPTbase).chat_message_turbo(
             query=item.query,
             model=item.model,
             is_stream=item.is_stream
         )
-        answer = response[0]
-        continue_chat = response[1]
+        answer = response
         return SuccessResponse(
             status="True",
-            randydev={
-                "message": answer,
-                "chat_history": continue_chat
-            }
+            randydev={"message": answer}
+        )
+    except Exception:
+        return SuccessResponse(status="False", randydev={"message": contact_support})
+
+@app.post("/ryuzaki/chatgpt-custom", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def chatgpt_customs(
+    item: ChatgptNewo,
+    api_key: None = Depends(validate_api_key)
+):
+    try:
+        GPTbase = SOURCE_OPENAI_ACCESS_URL
+        x = OpenAI(api_key=api_key, api_base=GPTbase)
+        response = await x.api_chat(query=item.query, model=item.model)
+        return SuccessResponse(
+            status="True",
+            randydev={"message": answer}
         )
     except Exception:
         return SuccessResponse(status="False", randydev={"message": contact_support})

models.py

@@ -164,5 +164,9 @@ class Chatgpt3Texts(BaseModel):
 class TextCustom(BaseModel):
     query: str
 
+class ChatgptNewo(BaseModel):
+    query: str
+    model: str
+
 class Picsart(BaseModel):
     image_path: str