31 lines
973 B
Python
31 lines
973 B
Python
import importlib.util
|
|
import os
|
|
|
|
from fastapi import FastAPI, HTTPException
|
|
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("memgpt").origin), "server", "static_files")
|
|
if os.path.exists(static_files_path):
|
|
app.mount(
|
|
"/",
|
|
SPAStaticFiles(
|
|
directory=static_files_path,
|
|
html=True,
|
|
),
|
|
name="spa-static-files",
|
|
)
|