main.py 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  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 (
  5. auths,
  6. users,
  7. chats,
  8. documents,
  9. modelfiles,
  10. prompts,
  11. configs,
  12. utils,
  13. )
  14. from config import WEBUI_VERSION, WEBUI_AUTH
  15. app = FastAPI()
  16. origins = ["*"]
  17. app.state.ENABLE_SIGNUP = True
  18. app.state.DEFAULT_MODELS = None
  19. app.add_middleware(
  20. CORSMiddleware,
  21. allow_origins=origins,
  22. allow_credentials=True,
  23. allow_methods=["*"],
  24. allow_headers=["*"],
  25. )
  26. app.include_router(auths.router, prefix="/auths", tags=["auths"])
  27. app.include_router(users.router, prefix="/users", tags=["users"])
  28. app.include_router(chats.router, prefix="/chats", tags=["chats"])
  29. app.include_router(documents.router, prefix="/documents", tags=["documents"])
  30. app.include_router(modelfiles.router, prefix="/modelfiles", tags=["modelfiles"])
  31. app.include_router(prompts.router, prefix="/prompts", tags=["prompts"])
  32. app.include_router(configs.router, prefix="/configs", tags=["configs"])
  33. app.include_router(utils.router, prefix="/utils", tags=["utils"])
  34. @app.get("/")
  35. async def get_status():
  36. return {
  37. "status": True,
  38. "version": WEBUI_VERSION,
  39. "auth": WEBUI_AUTH,
  40. "default_models": app.state.DEFAULT_MODELS,
  41. }