Upload 3 files

aworld/cmd/web/api_server.py

@@ -0,0 +1,22 @@
+import logging
+from fastapi import FastAPI
+import uvicorn
+
+
+logger = logging.getLogger(__name__)
+
+app = FastAPI()
+
+from .routers import chats,workspaces,sessions
+
+app.include_router(chats.router, prefix=chats.prefix)
+app.include_router(workspaces.router, prefix=workspaces.prefix)
+app.include_router(sessions.router, prefix=sessions.prefix)
+
+def run_server(port, args=None, **kwargs):
+    logger.info(f"Running API server on port {port}")
+    uvicorn.run(
+        app,
+        host="0.0.0.0",
+        port=port,
+    )

aworld/cmd/web/web_server.py

@@ -0,0 +1,79 @@
+import asyncio
+import logging
+from fastapi import FastAPI, Request, Response
+from fastapi.responses import RedirectResponse
+import uvicorn
+import os
+import subprocess
+from fastapi.staticfiles import StaticFiles
+from starlette.middleware.base import BaseHTTPMiddleware
+
+logger = logging.getLogger(__name__)
+
+app = FastAPI()
+
+
+@app.get("/")
+async def root():
+    return RedirectResponse("/index.html")
+
+
+def get_user_id_from_jwt(request: Request) -> str:
+    return "test_user_1"
+
+
+from .routers import chats, workspaces, sessions, traces  # noqa
+
+app.include_router(chats.router, prefix=chats.prefix)
+app.include_router(workspaces.router, prefix=workspaces.prefix)
+app.include_router(sessions.router, prefix=sessions.prefix)
+app.include_router(traces.router, prefix=traces.prefix)
+
+
+def build_webui(force_rebuild: bool = False) -> str:
+    webui_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "webui")
+    static_path = os.path.join(webui_path, "dist")
+
+    if (not os.path.exists(static_path)) or force_rebuild:
+        logger.warning(f"Build WebUI at {webui_path}")
+
+        p = subprocess.Popen(
+            ["sh", "-c", "npm install && npm run build"],
+            cwd=webui_path,
+        )
+        p.wait()
+        if p.returncode != 0:
+            raise Exception(f"Failed to build WebUI, error code: {p.returncode}")
+        else:
+            logger.info("WebUI build successfully")
+
+    return static_path
+
+
+static_path = build_webui(force_rebuild=True)
+logger.info(f"Mounting static files from {static_path}")
+app.mount("/", StaticFiles(directory=static_path, html=True), name="static")
+
+
+class TimeoutMiddleware(BaseHTTPMiddleware):
+    def __init__(self, app, timeout: int = 300):
+        super().__init__(app)
+        self.timeout = timeout
+
+    async def dispatch(self, request: Request, call_next):
+        try:
+            return await asyncio.wait_for(call_next(request), timeout=self.timeout)
+        except asyncio.TimeoutError:
+            return Response("Request timeout", status_code=408)
+
+
+app.add_middleware(TimeoutMiddleware, timeout=300)
+
+
+def run_server(port, args=None, **kwargs):
+    logger.info(f"Running Web server on port {port}")
+    uvicorn.run(
+        app,
+        host="0.0.0.0",
+        port=port,
+    )