Update app.py

app.py

@@ -19,18 +19,13 @@ API_KEY = "sk-your-secret-key"
 executor = ThreadPoolExecutor(max_workers=5)
 
 # =====================================
-# FASTAPI
+# APP
 # =====================================
 
 app = FastAPI(
-    title="AI Gateway",
-    version="2.0.0"
+    title="AI Gateway"
 )
 
-# =====================================
-# CORS
-# =====================================
-
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -52,7 +47,7 @@ class ChatRequest(BaseModel):
     stream: bool = False
 
 # =====================================
-# API KEY
+# AUTH
 # =====================================
 
 def verify_api_key(auth: str):
@@ -66,7 +61,7 @@ def verify_api_key(auth: str):
     if not auth.startswith("Bearer "):
         raise HTTPException(
             status_code=401,
-            detail="Invalid Authorization Format"
+            detail="Invalid Authorization"
         )
 
     token = auth.replace("Bearer ", "")
@@ -78,68 +73,52 @@ def verify_api_key(auth: str):
         )
 
 # =====================================
-# NORMAL RESPONSE
+# GENERATE
 # =====================================
 
-def generate_response(model, messages):
-
-    try:
-
-        response = g4f.ChatCompletion.create(
-            model=model,
-            messages=messages
-        )
-
-        return response
+def generate(model, messages):
 
-    except Exception as e:
+    response = g4f.ChatCompletion.create(
+        model=model,
+        messages=messages
+    )
 
-        raise Exception(str(e))
+    return response
 
 # =====================================
-# STREAMING RESPONSE
+# STREAM
 # =====================================
 
-async def stream_response(model, messages):
+async def stream_generate(model, messages):
 
     loop = asyncio.get_event_loop()
 
-    try:
-
-        response = await loop.run_in_executor(
-            executor,
-            lambda: g4f.ChatCompletion.create(
-                model=model,
-                messages=messages,
-                stream=True
-            )
+    response = await loop.run_in_executor(
+        executor,
+        lambda: g4f.ChatCompletion.create(
+            model=model,
+            messages=messages,
+            stream=True
         )
+    )
 
-        for chunk in response:
+    for chunk in response:
 
-            if chunk:
+        if chunk:
 
-                payload = {
-                    "choices": [
-                        {
-                            "delta": {
-                                "content": chunk
-                            }
+            payload = {
+                "choices": [
+                    {
+                        "delta": {
+                            "content": chunk
                         }
-                    ]
-                }
-
-                yield f"data: {json.dumps(payload)}\\n\\n"
-
-        yield "data: [DONE]\\n\\n"
-
-    except Exception as e:
+                    }
+                ]
+            }
 
-        error_payload = {
-            "error": str(e)
-        }
+            yield f"data: {json.dumps(payload)}\\n\\n"
 
-        yield f"data: {json.dumps(error_payload)}\\n\\n"
+    yield "data: [DONE]\\n\\n"
 
 # =====================================
 # HOME
@@ -149,12 +128,11 @@ async def stream_response(model, messages):
 async def home():
 
     return {
-        "status": "online",
-        "service": "AI Gateway"
+        "status": "online"
     }
 
 # =====================================
-# CHAT COMPLETIONS
+# CHAT
 # =====================================
 
 @app.post("/v1/chat/completions")
@@ -163,7 +141,6 @@ async def chat(
     authorization: str = Header(None)
 ):
 
-    # تحقق API Key
     verify_api_key(authorization)
 
     messages = [
@@ -171,50 +148,41 @@ async def chat(
         for m in req.messages
     ]
 
-    # =================================
-    # STREAMING
-    # =================================
+    # =========================
+    # STREAM
+    # =========================
 
     if req.stream:
 
         return StreamingResponse(
-            stream_response(
+            stream_generate(
                 req.model,
                 messages
             ),
             media_type="text/event-stream"
         )
 
-    # =================================
-    # NORMAL RESPONSE
-    # =================================
+    # =========================
+    # NORMAL
+    # =========================
 
     loop = asyncio.get_event_loop()
 
-    try:
-
-        response = await loop.run_in_executor(
-            executor,
-            lambda: generate_response(
-                req.model,
-                messages
-            )
+    response = await loop.run_in_executor(
+        executor,
+        lambda: generate(
+            req.model,
+            messages
         )
+    )
 
-        return {
-            "choices": [
-                {
-                    "message": {
-                        "role": "assistant",
-                        "content": response
-                    }
+    return {
+        "choices": [
+            {
+                "message": {
+                    "role": "assistant",
+                    "content": response
                 }
-            ]
-        }
-
-    except Exception as e:
-
-        raise HTTPException(
-            status_code=500,
-            detail=str(e)
-        )
+            }
+        ]
+    }