Update app.py

app.py

@@ -1,37 +1,31 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
+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(title="MultiChatAI to OpenAI Compatible API")
-
-# Allow CORS if needed
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+app = FastAPI()
 
 # OpenAI-compatible request model
+class ChatMessage(BaseModel):
+    role: str
+    content: str
+
 class ChatCompletionRequest(BaseModel):
     model: str = "deepseek-ai/DeepSeek-V3"
-    messages: list[dict]
+    messages: list[ChatMessage]
     temperature: float = 0.5
     max_tokens: int = None
-    stream: bool = False
 
 @app.post("/v1/chat/completions")
 async def chat_completion(request: ChatCompletionRequest):
     try:
-        # Transform OpenAI format to MultiChatAI format
+        # Prepare the MultiChatAI request
         multi_chat_body = {
             "chatSettings": {
                 "model": request.model,
-                "prompt": "You are a friendly, helpful AI assistant.",
+                "prompt": "You are a helpful AI assistant.",
                 "temperature": request.temperature,
                 "contextLength": 32000,
                 "includeProfileContext": True,
@@ -39,17 +33,15 @@ async def chat_completion(request: ChatCompletionRequest):
                 "embeddingsProvider": "openai"
             },
             "messages": [
-                {"role": "system", "content": f"Today is {datetime.now().strftime('%m/%d/%Y')}.\n\nUser Instructions:\nYou are a friendly, helpful AI assistant."},
-                *request.messages
+                {"role": "system", "content": f"Today is {datetime.now().strftime('%m/%d/%Y')}.\n\nYou are a helpful AI assistant."},
+                *[{"role": msg.role, "content": msg.content} for msg in request.messages]
             ],
             "customModelId": ""
         }
         
         headers = {
             "accept": "*/*",
-            "accept-language": "en-US,en;q=0.9",
-            "content-type": "text/plain;charset=UTF-8",
-            "priority": "u=1, i"
+            "content-type": "application/json",
         }
         
         # Make request to MultiChatAI
@@ -60,13 +52,14 @@ async def chat_completion(request: ChatCompletionRequest):
         )
         
         if response.status_code != 200:
-            raise HTTPException(status_code=response.status_code, detail="MultiChatAI API error")
+            raise HTTPException(
+                status_code=response.status_code,
+                detail=f"MultiChatAI API error: {response.text}"
+            )
         
-        # Transform MultiChatAI response to OpenAI format
-        multi_chat_response = response.text
-        
-        openai_response = {
-            "id": f"chatcmpl-{str(uuid.uuid4())}",
+        # Transform response to OpenAI format
+        return JSONResponse({
+            "id": f"chatcmpl-{uuid.uuid4()}",
             "object": "chat.completion",
             "created": int(datetime.now().timestamp()),
             "model": request.model,
@@ -74,18 +67,29 @@ async def chat_completion(request: ChatCompletionRequest):
                 "index": 0,
                 "message": {
                     "role": "assistant",
-                    "content": multi_chat_response
+                    "content": response.text.strip()
                 },
                 "finish_reason": "stop"
             }],
             "usage": {
-                "prompt_tokens": 0,  # You might need to calculate these
+                "prompt_tokens": 0,
                 "completion_tokens": 0,
                 "total_tokens": 0
             }
-        }
-        
-        return openai_response
+        })
         
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(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