Upload 2 files

main.py

@@ -632,7 +632,7 @@ def chatgpt3_turbo(item: Chatgpt3Texts):
             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",
+                model=item.model,
                 is_stream=item.is_stream
             )
             answer = response[0]
@@ -668,7 +668,7 @@ def chatgpt4_turbo(
         GPTbase = SOURCE_OPENAI_ACCESS_URL
         response = OpenAiToken(api_key=item.api_key, api_base=GPTbase).chat_message_turbo(
             query=item.query,
-            model="gpt-4",
+            model=item.model,
             is_stream=item.is_stream
         )
         answer = response[0]

models.py

@@ -135,11 +135,13 @@ class AnimeStyled(BaseModel):
 class OpenaiTexts(BaseModel):
     query: str
     api_key: Optional[str] = ""
+    model: Optional[str] = "gpt-4"
     is_stream: Optional[bool] = False
 
 class Chatgpt3Texts(BaseModel):
     query: str
     api_key: Optional[str] = ""
+    model: Optional[str] = "gpt-3.5-turbo"
     is_stream: Optional[bool] = False
     is_openai_original: Optional[bool] = False