Update app/main.py

app/main.py

@@ -1,13 +1,12 @@
 import os
-import json
 from pathlib import Path
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import StreamingResponse, HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.middleware.cors import CORSMiddleware
-from agent_system import orchestrator
+from agent_system import orchestrator, BUILTIN_TOOLS
 
-app = FastAPI(title="PraisonChat", version="1.0.0")
+app = FastAPI(title="PraisonChat", version="2.0.0")
 
 app.add_middleware(
     CORSMiddleware,
@@ -17,6 +16,7 @@ app.add_middleware(
 )
 
 STATIC_DIR = Path(__file__).parent / "static"
+app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
 
 
 @app.get("/", response_class=HTMLResponse)
@@ -26,18 +26,28 @@ async def root():
 
 @app.get("/api/health")
 def health():
-    return {"status": "ok", "model": "LongCat-Flash-Lite", "version": "1.0.0"}
+    return {"status": "ok", "model": "LongCat-Flash-Lite", "version": "2.0.0"}
 
 
 @app.get("/api/models")
 def models():
-    return {
-        "models": [
-            {"id": "LongCat-Flash-Lite",         "name": "LongCat Flash Lite",         "context": "320K", "speed": "fastest"},
-            {"id": "LongCat-Flash-Chat",          "name": "LongCat Flash Chat",         "context": "256K", "speed": "fast"},
-            {"id": "LongCat-Flash-Thinking-2601", "name": "LongCat Flash Thinking",     "context": "256K", "speed": "medium"},
-        ]
-    }
+    return {"models": [
+        {"id": "LongCat-Flash-Lite",          "name": "LongCat Flash Lite",         "context": "320K", "speed": "⚡ Fastest", "quota": "50M/day"},
+        {"id": "LongCat-Flash-Chat",           "name": "LongCat Flash Chat",         "context": "256K", "speed": "🚀 Fast",    "quota": "500K/day"},
+        {"id": "LongCat-Flash-Thinking-2601",  "name": "LongCat Flash Thinking",     "context": "256K", "speed": "🧠 Deep",    "quota": "500K/day"},
+    ]}
+
+
+@app.get("/api/builtin-tools")
+def builtin_tools_list():
+    tools = [
+        {"name": "get_current_datetime",  "description": "Returns the current date and time", "icon": "🕐"},
+        {"name": "calculate_math",        "description": "Evaluates mathematical expressions", "icon": "🔢"},
+        {"name": "run_python_code",       "description": "Executes Python code in a sandbox",  "icon": "🐍"},
+        {"name": "create_voice_response", "description": "Converts text to spoken audio (gTTS)","icon": "🔊"},
+        {"name": "search_information",    "description": "Searches for information",            "icon": "🔍"},
+    ]
+    return {"tools": tools}
 
 
 @app.post("/api/chat")
@@ -47,12 +57,12 @@ async def chat(request: Request):
     except Exception:
         raise HTTPException(400, "Invalid JSON body")
 
-    messages  = body.get("messages", [])
-    api_key   = body.get("api_key") or os.getenv("LONGCAT_API_KEY", "")
+    messages = body.get("messages", [])
+    api_key  = body.get("api_key") or os.getenv("LONGCAT_API_KEY", "")
+    model    = body.get("model", "LongCat-Flash-Lite")
 
     if not api_key:
-        raise HTTPException(400, "LongCat API key is required. Set it in Settings or as LONGCAT_API_KEY env var.")
-
+        raise HTTPException(400, "LongCat API key required. Add it in Settings or set LONGCAT_API_KEY env var.")
     if not messages:
         raise HTTPException(400, "No messages provided")
 
@@ -60,8 +70,11 @@ async def chat(request: Request):
     history      = messages[:-1]
 
     async def event_stream():
-        async for chunk in orchestrator.stream_response(user_message, history, api_key):
+        async for chunk in orchestrator.stream_response(user_message, history, api_key, model):
             yield f"data: {chunk}\n\n"
 
-    return StreamingResponse(event_stream(), media_type="text/event-stream",
-                             headers={"X-Accel-Buffering": "no", "Cache-Control": "no-cache"})
+    return StreamingResponse(
+        event_stream(),
+        media_type="text/event-stream",
+        headers={"X-Accel-Buffering": "no", "Cache-Control": "no-cache", "Connection": "keep-alive"},
+    )