main.py 1.2 KB

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