UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-06 00:32:05 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

feat: add API endpoint for updating configs

Browse Source
This commit is contained in:
Jun Siang Cheah 2024-05-09 22:41:07 +08:00
parent 8effff6524
commit 0dbddebcb0
1 changed files with 32 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
backend/main.py
View File

@ -58,7 +58,6 @@ from config import (
SRC_LOG_LEVELS,
WEBHOOK_URL,
ENABLE_ADMIN_EXPORT,
MODEL_CONFIG,
)
from constants import ERROR_MESSAGES
@ -287,6 +286,38 @@ async def update_model_filter_config(
}
class ModelConfig(BaseModel):
id: str
name: str
description: str
vision_capable: bool
class SetModelConfigForm(BaseModel):
ollama: List[ModelConfig]
litellm: List[ModelConfig]
openai: List[ModelConfig]
@app.post("/api/config/models")
async def update_model_config(
form_data: SetModelConfigForm, user=Depends(get_admin_user)
):
data = form_data.model_dump()
ollama_app.state.MODEL_CONFIG = data.get("ollama", [])
openai_app.state.MODEL_CONFIG = data.get("openai", [])
litellm_app.state.MODEL_CONFIG = data.get("litellm", [])
return {
"ollama": ollama_app.state.MODEL_CONFIG,
"openai": openai_app.state.MODEL_CONFIG,
"litellm": litellm_app.state.MODEL_CONFIG,
}
@app.get("/api/webhook")
async def get_webhook_url(user=Depends(get_admin_user)):
return {