open-webui/backend/apps/web/main.py

47 lines
1.2 KiB
Python
Raw Normal View History

from fastapi import FastAPI, Depends
from fastapi.routing import APIRoute
2023-11-19 01:47:12 +01:00
from fastapi.middleware.cors import CORSMiddleware
from starlette.middleware.authentication import AuthenticationMiddleware
2023-12-27 07:10:22 +01:00
from apps.web.routers import auths, users, chats, modelfiles, utils
from config import WEBUI_VERSION, WEBUI_AUTH
from utils.utils import verify_auth_token
2023-11-19 01:47:12 +01:00
app = FastAPI()
origins = ["*"]
app.add_middleware(
CORSMiddleware,
allow_origins=origins,
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
app.include_router(auths.router, prefix="/auths", tags=["auths"])
app.include_router(
users.router,
prefix="/users",
tags=["users"],
dependencies=[Depends(verify_auth_token)],
)
app.include_router(
chats.router,
prefix="/chats",
tags=["chats"],
dependencies=[Depends(verify_auth_token)],
)
app.include_router(
modelfiles.router,
prefix="/modelfiles",
tags=["modelfiles"],
dependencies=[Depends(verify_auth_token)],
)
2023-12-27 07:10:22 +01:00
app.include_router(utils.router, prefix="/utils", tags=["utils"])
2023-11-19 01:47:12 +01:00
@app.get("/")
async def get_status():
return {"status": True, "version": WEBUI_VERSION, "auth": WEBUI_AUTH}