Update app.py

app.py

@@ -1,8 +1,10 @@
 from fastapi import FastAPI, HTTPException
+from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 import requests
 import uuid
 from datetime import datetime
+import uvicorn
 
 app = FastAPI()
 
@@ -16,45 +18,31 @@ class ChatRequest(BaseModel):
     temperature: float = 0.7
 
 @app.get("/")
-def health_check():
-    return {"status": "OK", "time": datetime.now().isoformat()}
+async def health_check():
+    return {"status": "running", "timestamp": datetime.now().isoformat()}
 
 @app.post("/v1/chat/completions")
-def chat_completion(request: ChatRequest):
+async def chat_completion(request: ChatRequest):
     try:
-        # Prepare payload for MultiChatAI
-        payload = {
-            "chatSettings": {
-                "model": request.model,
-                "prompt": "You are a helpful assistant.",
-                "temperature": request.temperature,
-                "contextLength": 4096,
-                "includeProfileContext": False,
-                "includeWorkspaceInstructions": False,
-                "embeddingsProvider": "openai"
-            },
-            "messages": [
-                {"role": "system", "content": "You are a helpful assistant."},
-                *[{"role": msg.role, "content": msg.content} for msg in request.messages]
-            ]
-        }
-
-        # Call MultiChatAI API
         response = requests.post(
             "https://www.multichatai.com/api/chat/deepinfra",
-            json=payload,
-            headers={"Content-Type": "application/json"},
+            json={
+                "chatSettings": {
+                    "model": request.model,
+                    "prompt": "You are helpful.",
+                    "temperature": request.temperature,
+                    "contextLength": 4096
+                },
+                "messages": [
+                    {"role": "system", "content": "You are helpful."},
+                    *[{"role": m.role, "content": m.content} for m in request.messages]
+                ]
+            },
             timeout=10
         )
-
-        if response.status_code != 200:
-            raise HTTPException(
-                status_code=502,
-                detail=f"Upstream API error: {response.text}"
-            )
-
-        # Format response
-        return {
+        response.raise_for_status()
+        
+        return JSONResponse({
             "id": str(uuid.uuid4()),
             "object": "chat.completion",
             "created": int(datetime.now().timestamp()),
@@ -65,9 +53,9 @@ def chat_completion(request: ChatRequest):
                     "content": response.text.strip()
                 }
             }]
-        }
-
-    except requests.Timeout:
-        raise HTTPException(status_code=504, detail="Request timeout")
+        })
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+
+if __name__ == "__main__":
+    uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=False)