Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
open-webui
Commits
0dbddebc
Commit
0dbddebc
authored
May 09, 2024
by
Jun Siang Cheah
Browse files
feat: add API endpoint for updating configs
parent
8effff65
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
32 additions
and
1 deletion
+32
-1
backend/main.py
backend/main.py
+32
-1
No files found.
backend/main.py
View file @
0dbddebc
...
@@ -58,7 +58,6 @@ from config import (
...
@@ -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(
...
@@ -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
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment