Update app.py

app.py

@@ -3,6 +3,9 @@ from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 from typing import List
 from g4f import ChatCompletion
+from g4f.typing import Messages, AsyncResult
+import g4f.Provider
+from g4f.Provider import BackendApi
 
 app = FastAPI()
 
@@ -14,6 +17,34 @@ models = [
     "llama-3.1-405b"
 ]
 
+url = "https://ahe.hopto.org"
+headers = {"Authorization": "Basic Z2dnOmc0Zl8="}
+
+BackendApi.working = True
+BackendApi.ssl = False
+BackendApi.url = url
+BackendApi.headers = headers
+
+class BackendApi(BackendApi):
+    working = True
+    ssl = False
+    url = url
+    headers = headers
+    image_models = ["flux", "flux-pro"]
+    models = ["deepseek-r1", *g4f.Provider.OpenaiAccount.get_models(), "flux", "flux-pro"]
+
+    @classmethod
+    async def create_async_generator(
+        cls,
+        model: str,
+        messages: Messages,
+        **kwargs
+    ) -> AsyncResult:
+        if model in g4f.Provider.OpenaiAccount.get_models():
+            kwargs["provider"] = "OpenaiAccount"
+        async for chunk in super().create_async_generator(model, messages, **kwargs):
+            yield chunk
+
 # Request model
 class Message(BaseModel):
     role: str