Update app.py

app.py

@@ -1,13 +1,24 @@
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, HTTPException
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 import requests
 import uuid
 from datetime import datetime
+from fastapi.middleware.cors import CORSMiddleware
 
-app = FastAPI()
+# Initialize FastAPI app
+app = FastAPI(title="MultiChatAI to OpenAI API Wrapper")
 
-# OpenAI-compatible request model
+# Configure CORS - adjust these based on your needs
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Request models
 class ChatMessage(BaseModel):
     role: str
     content: str
@@ -15,13 +26,19 @@ class ChatMessage(BaseModel):
 class ChatCompletionRequest(BaseModel):
     model: str = "deepseek-ai/DeepSeek-V3"
     messages: list[ChatMessage]
-    temperature: float = 0.5
+    temperature: float = 0.7
     max_tokens: int = None
 
+# Health check endpoint
+@app.get("/")
+async def health_check():
+    return {"status": "OK", "service": "MultiChatAI Proxy", "timestamp": datetime.now().isoformat()}
+
+# Main API endpoint
 @app.post("/v1/chat/completions")
 async def chat_completion(request: ChatCompletionRequest):
     try:
-        # Prepare the MultiChatAI request
+        # Prepare request for MultiChatAI
         multi_chat_body = {
             "chatSettings": {
                 "model": request.model,
@@ -33,31 +50,32 @@ async def chat_completion(request: ChatCompletionRequest):
                 "embeddingsProvider": "openai"
             },
             "messages": [
-                {"role": "system", "content": f"Today is {datetime.now().strftime('%m/%d/%Y')}.\n\nYou are a helpful AI assistant."},
+                {"role": "system", "content": f"Today is {datetime.now().strftime('%m/%d/%Y')}.\nYou are a helpful AI assistant."},
                 *[{"role": msg.role, "content": msg.content} for msg in request.messages]
             ],
             "customModelId": ""
         }
         
         headers = {
-            "accept": "*/*",
+            "accept": "application/json",
             "content-type": "application/json",
         }
         
-        # Make request to MultiChatAI
+        # Call MultiChatAI API
         response = requests.post(
             "https://www.multichatai.com/api/chat/deepinfra",
             headers=headers,
-            json=multi_chat_body
+            json=multi_chat_body,
+            timeout=30  # Add timeout to prevent hanging
         )
         
-        if response.status_code != 200:
+        if not response.ok:
             raise HTTPException(
                 status_code=response.status_code,
                 detail=f"MultiChatAI API error: {response.text}"
             )
         
-        # Transform response to OpenAI format
+        # Format response in OpenAI style
         return JSONResponse({
             "id": f"chatcmpl-{uuid.uuid4()}",
             "object": "chat.completion",
@@ -72,24 +90,18 @@ async def chat_completion(request: ChatCompletionRequest):
                 "finish_reason": "stop"
             }],
             "usage": {
-                "prompt_tokens": 0,
+                "prompt_tokens": 0,  # You can implement token counting later
                 "completion_tokens": 0,
                 "total_tokens": 0
             }
         })
         
+    except requests.Timeout:
+        raise HTTPException(status_code=504, detail="Upstream service timeout")
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@app.get("/")
-async def health_check():
-    return {"status": "healthy"}
-
-# Add CORS middleware if needed
-@app.middleware("http")
-async def add_cors_header(request: Request, call_next):
-    response = await call_next(request)
-    response.headers["Access-Control-Allow-Origin"] = "*"
-    response.headers["Access-Control-Allow-Methods"] = "GET, POST, OPTIONS"
-    response.headers["Access-Control-Allow-Headers"] = "Content-Type"
-    return response
+# Add this if you need to support OPTIONS requests
+@app.options("/v1/chat/completions")
+async def options_handler():
+    return JSONResponse(content={}, status_code=200)