mirror of
https://github.com/open-webui/open-webui
synced 2024-11-24 21:13:59 +00:00
feat: add API endpoint for updating configs
This commit is contained in:
parent
8effff6524
commit
0dbddebcb0
@ -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 {
|
||||
|
Loading…
Reference in New Issue
Block a user