Update app.py

app.py

@@ -1,10 +1,10 @@
 from fastapi import FastAPI, Request, HTTPException
-from fastapi.responses import JSONResponse
+from fastapi.responses import PlainTextResponse
 import os
 import requests
 
-# Default model used internally; client doesn't need to specify it
-DEFAULT_MODEL = "deepseek/deepseek-chat-v3-0324:free"
+# Default model used internally; the client only sends plain text
+DEFAULT_MODEL = "kwaipilot/kat-coder-pro:free"
 
 app = FastAPI()
 
@@ -14,21 +14,15 @@ def health():
 
 @app.post("/chat")
 async def chat(request: Request):
-    # Accept plain text body; also accept JSON {"text": "..."} as fallback
-    text = ""
-    try:
-        raw = await request.body()
-        text = (raw or b"").decode().strip()
-    except Exception:
-        pass
-
+    # Accept plain text body; fallback to JSON {"text": "..."}
+    body = await request.body()
+    text = (body or b"").decode().strip()
     if not text:
         try:
             data = await request.json()
             text = str(data.get("text", "")).strip()
         except Exception:
             pass
-
     if not text:
         raise HTTPException(status_code=400, detail="Send plain text body or JSON {'text': '...'}")
 
@@ -36,10 +30,9 @@ async def chat(request: Request):
     if not api_key:
         raise HTTPException(status_code=500, detail="OPENROUTER_API_KEY not set")
 
-    # Forward to OpenRouter chat completions
     try:
         r = requests.post(
-            "https://openrouter.ai/api/v1/chat/completions",  # [web:21]
+            "https://openrouter.ai/api/v1/chat/completions",  # OpenRouter chat completions [web:21]
             headers={
                 "Authorization": f"Bearer {api_key}",
                 "Content-Type": "application/json",
@@ -53,21 +46,15 @@ async def chat(request: Request):
     except requests.RequestException as e:
         raise HTTPException(status_code=502, detail=str(e))
 
-    # Extract assistant text for a minimal response; fall back to raw JSON on parse issues
+    # Try to extract assistant text; if not possible, return the raw body from OpenRouter
+    content = None
     try:
         data = r.json()
-    except Exception:
-        return JSONResponse(content={"error": r.text}, status_code=r.status_code)
-
-    if r.status_code >= 400:
-        return JSONResponse(content=data, status_code=r.status_code)
-
-    content = ""
-    try:
+        if r.status_code >= 400:
+            return PlainTextResponse(str(data), status_code=r.status_code)
         content = data["choices"][0]["message"]["content"]
     except Exception:
-        pass
+        # Fall back to raw text (may be JSON or error text)
+        return PlainTextResponse(r.text, status_code=r.status_code)
 
-    if content:
-        return {"content": content}
-    return data
+    return PlainTextResponse(content)