mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
85faf5f474
Co-authored-by: Charles Packer <packercharles@gmail.com> Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Shubham Naik <shub@memgpt.ai>
75 lines
3.1 KiB
Python
75 lines
3.1 KiB
Python
import importlib.util
|
|
import os
|
|
|
|
from fastapi import FastAPI, HTTPException
|
|
from fastapi.responses import FileResponse
|
|
from starlette.exceptions import HTTPException as StarletteHTTPException
|
|
from starlette.staticfiles import StaticFiles
|
|
|
|
|
|
class SPAStaticFiles(StaticFiles):
|
|
async def get_response(self, path: str, scope):
|
|
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)
|
|
else:
|
|
raise ex
|
|
|
|
|
|
def mount_static_files(app: FastAPI):
|
|
static_files_path = os.path.join(os.path.dirname(importlib.util.find_spec("letta").origin), "server", "static_files")
|
|
if os.path.exists(static_files_path):
|
|
app.mount("/assets", StaticFiles(directory=os.path.join(static_files_path, "assets")), name="assets")
|
|
|
|
@app.get("/letta_logo_transparent.png", include_in_schema=False)
|
|
async def serve_spa():
|
|
return FileResponse(os.path.join(static_files_path, "letta_logo_transparent.png"))
|
|
|
|
@app.get("/", include_in_schema=False)
|
|
async def serve_spa():
|
|
return FileResponse(os.path.join(static_files_path, "index.html"))
|
|
|
|
@app.get("/agents", include_in_schema=False)
|
|
async def serve_spa():
|
|
return FileResponse(os.path.join(static_files_path, "index.html"))
|
|
|
|
@app.get("/data-sources", include_in_schema=False)
|
|
async def serve_spa():
|
|
return FileResponse(os.path.join(static_files_path, "index.html"))
|
|
|
|
@app.get("/tools", include_in_schema=False)
|
|
async def serve_spa():
|
|
return FileResponse(os.path.join(static_files_path, "index.html"))
|
|
|
|
@app.get("/agent-templates", include_in_schema=False)
|
|
async def serve_spa():
|
|
return FileResponse(os.path.join(static_files_path, "index.html"))
|
|
|
|
@app.get("/human-templates", include_in_schema=False)
|
|
async def serve_spa():
|
|
return FileResponse(os.path.join(static_files_path, "index.html"))
|
|
|
|
@app.get("/settings/profile", include_in_schema=False)
|
|
async def serve_spa():
|
|
return FileResponse(os.path.join(static_files_path, "index.html"))
|
|
|
|
@app.get("/agents/{agent-id}/chat", include_in_schema=False)
|
|
async def serve_spa():
|
|
return FileResponse(os.path.join(static_files_path, "index.html"))
|
|
|
|
|
|
# def mount_static_files(app: FastAPI):
|
|
# static_files_path = os.path.join(os.path.dirname(importlib.util.find_spec("letta").origin), "server", "static_files")
|
|
# if os.path.exists(static_files_path):
|
|
|
|
# @app.get("/{full_path:path}")
|
|
# async def serve_spa(full_path: str):
|
|
# if full_path.startswith("v1"):
|
|
# raise HTTPException(status_code=404, detail="Not found")
|
|
# file_path = os.path.join(static_files_path, full_path)
|
|
# if os.path.isfile(file_path):
|
|
# return FileResponse(file_path)
|
|
# return FileResponse(os.path.join(static_files_path, "index.html"))
|