mirror of
https://github.com/open-webui/open-webui
synced 2025-02-19 19:38:52 +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,
|
SRC_LOG_LEVELS,
|
||||||
WEBHOOK_URL,
|
WEBHOOK_URL,
|
||||||
ENABLE_ADMIN_EXPORT,
|
ENABLE_ADMIN_EXPORT,
|
||||||
MODEL_CONFIG,
|
|
||||||
)
|
)
|
||||||
from constants import ERROR_MESSAGES
|
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")
|
@app.get("/api/webhook")
|
||||||
async def get_webhook_url(user=Depends(get_admin_user)):
|
async def get_webhook_url(user=Depends(get_admin_user)):
|
||||||
return {
|
return {
|
||||||
|
Loading…
Reference in New Issue
Block a user