Upload main.py

main.py

@@ -60,6 +60,7 @@ from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 
 import g4f
+from g4f.client import Client
 from g4f.client import Client as BingClient
 from g4f.cookies import set_cookies
 from g4f.Provider import BingCreateImages, OpenaiChat, Gemini
@@ -888,6 +889,22 @@ async def chatgpt4_turbo(
     except Exception as e:
         return SuccessResponse(status="False", randydev={"message": str(e)})
 
+@app.post("/ryuzaki/chatgpt-4o", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+async def chatgpt_4o(
+    item: ChatgptCustom,
+    api_key: None = Depends(validate_api_key)
+):
+    client = Client()
+    response = client.chat.completions.create(
+        model="gpt-4o",
+        messages=[{"role": "user", "content": item.query}],
+    )
+    answer = response.choices[0].message.content
+    return SuccessResponse(
+        status="True",
+        randydev={"message": answer}
+    )
+
 @app.post("/ryuzaki/chatgpt-custom", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 async def chatgpt_customs(
     item: ChatgptNewo,