github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-30 03:00:41 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat: partial loading of routes and ui files

Browse Source
This commit is contained in:
Nicholas Tindle
2024-07-12 00:01:36 -05:00
parent b376e84556
commit bded7916c2
2 changed files with 17 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
rnd/autogpt_server/autogpt_server/server/server.py
View File

@@ -5,7 +5,7 @@ from typing import Annotated, Any, Dict, MutableMapping
from starlette.responses import Response
from starlette.exceptions import HTTPException as StarletteHTTPException
import uvicorn
from fastapi.responses import JSONResponse
from fastapi.responses import HTMLResponse, JSONResponse, RedirectResponse
from fastapi.staticfiles import StaticFiles
from contextlib import asynccontextmanager
@@ -137,29 +137,37 @@ class AgentServer(AppService):
methods=["POST"],
)
app.add_exception_handler(500, self.handle_internal_error) # type: ignore
app.add_exception_handler(500, self.handle_internal_error)
class SPAStaticFiles(StaticFiles):
async def get_response(self, path: str, scope: MutableMapping[str, Any]) -> Response:
async def get_response(
self, path: str, scope: MutableMapping[str, Any]
) -> Response:
try:
return await super().get_response(path, scope)
except (HTTPException, StarletteHTTPException) as ex:
if ex.status_code == 404:
return await super().get_response("index.html", scope)
return await super().get_response("build.html", scope)
else:
raise ex
app.mount(
path="/",
app=SPAStaticFiles(directory=get_frontend_path(), html=True),
name="example_files",
)
app.include_router(router)
app.mount(
path="/_next",
app=SPAStaticFiles(directory=get_frontend_path() / "_next", html=True),
name="example_files",
)
@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket): # type: ignore
await self.websocket_router(websocket)
@app.get("/", response_class=HTMLResponse)
async def catch_all():
with open(get_frontend_path() / "build.html") as file:
return str(file.read())
uvicorn.run(app, host="0.0.0.0", port=8000)
@property

19
rnd/example_files/index.html
View File

@@ -1,19 +0,0 @@
<!DOCTYPE html>
<html lang="en">
<head>
<title>Example Files</title>
<meta charset="utf-8">
<meta name="viewport" content="width=device-width, initial-scale=1">
</head>
<body>
<h1>Example Files</h1>
<ul>
<li><a href="build.html">Example 1</a></li>
<li><a href="example2.txt">Example 2</a></li>
<li><a href="example3.txt">Example 3</a></li>
</ul>
</body>
</html>