2023-12-30 10:53:33 +00:00
|
|
|
from fastapi import FastAPI, Depends
|
|
|
|
from fastapi.routing import APIRoute
|
2023-11-19 00:47:12 +00:00
|
|
|
from fastapi.middleware.cors import CORSMiddleware
|
2024-01-03 05:00:50 +00:00
|
|
|
from apps.web.routers import auths, users, chats, modelfiles, prompts, configs, utils
|
2023-11-19 08:41:29 +00:00
|
|
|
from config import WEBUI_VERSION, WEBUI_AUTH
|
2023-11-19 00:47:12 +00:00
|
|
|
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
|
|
origins = ["*"]
|
|
|
|
|
2024-01-01 20:32:28 +00:00
|
|
|
app.state.ENABLE_SIGNUP = True
|
2024-01-03 00:48:49 +00:00
|
|
|
app.state.DEFAULT_MODELS = None
|
2024-01-01 20:32:28 +00:00
|
|
|
|
2023-11-19 00:47:12 +00:00
|
|
|
app.add_middleware(
|
|
|
|
CORSMiddleware,
|
|
|
|
allow_origins=origins,
|
|
|
|
allow_credentials=True,
|
|
|
|
allow_methods=["*"],
|
|
|
|
allow_headers=["*"],
|
|
|
|
)
|
|
|
|
|
|
|
|
app.include_router(auths.router, prefix="/auths", tags=["auths"])
|
2024-01-01 08:55:50 +00:00
|
|
|
app.include_router(users.router, prefix="/users", tags=["users"])
|
|
|
|
app.include_router(chats.router, prefix="/chats", tags=["chats"])
|
|
|
|
app.include_router(modelfiles.router, prefix="/modelfiles", tags=["modelfiles"])
|
2024-01-03 05:00:50 +00:00
|
|
|
app.include_router(prompts.router, prefix="/prompts", tags=["prompts"])
|
|
|
|
|
|
|
|
|
2024-01-03 00:48:10 +00:00
|
|
|
app.include_router(configs.router, prefix="/configs", tags=["configs"])
|
2023-12-27 06:10:22 +00:00
|
|
|
app.include_router(utils.router, prefix="/utils", tags=["utils"])
|
2023-11-19 00:47:12 +00:00
|
|
|
|
|
|
|
|
|
|
|
@app.get("/")
|
|
|
|
async def get_status():
|
2024-01-03 00:48:10 +00:00
|
|
|
return {
|
|
|
|
"status": True,
|
|
|
|
"version": WEBUI_VERSION,
|
|
|
|
"auth": WEBUI_AUTH,
|
|
|
|
"default_models": app.state.DEFAULT_MODELS,
|
|
|
|
}
|