main.py 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. from fastapi import FastAPI, Depends
  2. from fastapi.routing import APIRoute
  3. from fastapi.middleware.cors import CORSMiddleware
  4. from apps.web.routers import auths, users, chats, modelfiles, utils
  5. from config import WEBUI_VERSION, WEBUI_AUTH
  6. from utils.utils import verify_auth_token
  7. app = FastAPI()
  8. origins = ["*"]
  9. app.add_middleware(
  10. CORSMiddleware,
  11. allow_origins=origins,
  12. allow_credentials=True,
  13. allow_methods=["*"],
  14. allow_headers=["*"],
  15. )
  16. app.include_router(auths.router, prefix="/auths", tags=["auths"])
  17. app.include_router(
  18. users.router,
  19. prefix="/users",
  20. tags=["users"],
  21. dependencies=[Depends(verify_auth_token)],
  22. )
  23. app.include_router(
  24. chats.router,
  25. prefix="/chats",
  26. tags=["chats"],
  27. dependencies=[Depends(verify_auth_token)],
  28. )
  29. app.include_router(
  30. modelfiles.router,
  31. prefix="/modelfiles",
  32. tags=["modelfiles"],
  33. dependencies=[Depends(verify_auth_token)],
  34. )
  35. app.include_router(utils.router, prefix="/utils", tags=["utils"])
  36. @app.get("/")
  37. async def get_status():
  38. return {"status": True, "version": WEBUI_VERSION, "auth": WEBUI_AUTH}