main.py 871 B

12345678910111213141516171819202122232425262728293031
  1. from fastapi import FastAPI, Request, Depends, HTTPException
  2. from fastapi.middleware.cors import CORSMiddleware
  3. from apps.web.routers import auths, users, chats, modelfiles, utils
  4. from config import WEBUI_VERSION, WEBUI_AUTH
  5. app = FastAPI()
  6. origins = ["*"]
  7. app.add_middleware(
  8. CORSMiddleware,
  9. allow_origins=origins,
  10. allow_credentials=True,
  11. allow_methods=["*"],
  12. allow_headers=["*"],
  13. )
  14. app.include_router(auths.router, prefix="/auths", tags=["auths"])
  15. app.include_router(users.router, prefix="/users", tags=["users"])
  16. app.include_router(chats.router, prefix="/chats", tags=["chats"])
  17. app.include_router(modelfiles.router, prefix="/modelfiles", tags=["modelfiles"])
  18. app.include_router(utils.router, prefix="/utils", tags=["utils"])
  19. @app.get("/")
  20. async def get_status():
  21. return {"status": True, "version": WEBUI_VERSION, "auth": WEBUI_AUTH}