forked from open-webui/open-webui
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
|
from fastapi import Response, Request
|
||
|
|
from fastapi import Depends, FastAPI, HTTPException, status
|
||
|
|
from datetime import datetime, timedelta
|
||
|
|
from typing import List, Union
|
||
|
|
|
||
|
|
from fastapi import APIRouter
|
||
|
|
from pydantic import BaseModel
|
||
|
|
import time
|
||
|
|
import uuid
|
||
|
|
|
||
|
|
from apps.web.models.users import Users
|
||
|
|
|
||
|
|
|
||
|
|
from utils.utils import get_password_hash, get_current_user, create_token
|
||
|
|
from utils.misc import get_gravatar_url, validate_email_format
|
||
|
|
from constants import ERROR_MESSAGES
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
|
||
|
|
class SetDefaultModelsForm(BaseModel):
|
||
|
|
models: str
|
||
|
|
|
||
|
|
|
||
|
|
############################
|
||
|
|
# SetDefaultModels
|
||
|
|
############################
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/default/models", response_model=str)
|
||
|
|
async def set_global_default_models(
|
||
|
|
request: Request, form_data: SetDefaultModelsForm, user=Depends(get_current_user)
|
||
|
|
):
|
||
|
|
if user.role == "admin":
|
||
|
|
request.app.state.DEFAULT_MODELS = form_data.models
|
||
|
|
return request.app.state.DEFAULT_MODELS
|
||
|
|
else:
|
||
|
|
raise HTTPException(
|
||
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
||
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
||
|
|
)
|