UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-21 23:57:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: models api

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-11-19 18:45:26 -08:00
parent 7c5b845d16
commit 7966367107
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/open_webui/apps/webui/routers/models.py
View File

@ -83,7 +83,8 @@ async def create_new_model(
########################### ###########################
@router.get("/id/{id}", response_model=Optional[ModelResponse]) # Note: We're not using the typical url path param here, but instead using a query parameter to allow '/' in the id
@router.get("/model", response_model=Optional[ModelResponse])
async def get_model_by_id(id: str, user=Depends(get_verified_user)): async def get_model_by_id(id: str, user=Depends(get_verified_user)):
model = Models.get_model_by_id(id) model = Models.get_model_by_id(id)
if model: if model:
@ -105,7 +106,7 @@ async def get_model_by_id(id: str, user=Depends(get_verified_user)):
############################ ############################
@router.post("/id/{id}/toggle", response_model=Optional[ModelResponse]) @router.post("/model/toggle", response_model=Optional[ModelResponse])
async def toggle_model_by_id(id: str, user=Depends(get_verified_user)): async def toggle_model_by_id(id: str, user=Depends(get_verified_user)):
model = Models.get_model_by_id(id) model = Models.get_model_by_id(id)
if model: if model:
@ -140,7 +141,7 @@ async def toggle_model_by_id(id: str, user=Depends(get_verified_user)):
############################ ############################
@router.post("/id/{id}/update", response_model=Optional[ModelModel]) @router.post("/model/update", response_model=Optional[ModelModel])
async def update_model_by_id( async def update_model_by_id(
id: str, id: str,
form_data: ModelForm, form_data: ModelForm,
@ -163,7 +164,7 @@ async def update_model_by_id(
############################ ############################
@router.delete("/id/{id}/delete", response_model=bool) @router.delete("/model/delete", response_model=bool)
async def delete_model_by_id(id: str, user=Depends(get_verified_user)): async def delete_model_by_id(id: str, user=Depends(get_verified_user)):
model = Models.get_model_by_id(id) model = Models.get_model_by_id(id)
if not model: if not model:

8
src/lib/apis/models/index.ts
View File

@ -97,7 +97,7 @@ export const getModelById = async (token: string, id: string) => {
const searchParams = new URLSearchParams(); const searchParams = new URLSearchParams();
searchParams.append('id', id); searchParams.append('id', id);
const res = await fetch(`${WEBUI_API_BASE_URL}/models/id/${id}`, { const res = await fetch(`${WEBUI_API_BASE_URL}/models/model?${searchParams.toString()}`, {
method: 'GET', method: 'GET',
headers: { headers: {
Accept: 'application/json', Accept: 'application/json',
@ -132,7 +132,7 @@ export const toggleModelById = async (token: string, id: string) => {
const searchParams = new URLSearchParams(); const searchParams = new URLSearchParams();
searchParams.append('id', id); searchParams.append('id', id);
const res = await fetch(`${WEBUI_API_BASE_URL}/models/id/${id}/toggle`, { const res = await fetch(`${WEBUI_API_BASE_URL}/models/model/toggle?${searchParams.toString()}`, {
method: 'POST', method: 'POST',
headers: { headers: {
Accept: 'application/json', Accept: 'application/json',
@ -167,7 +167,7 @@ export const updateModelById = async (token: string, id: string, model: object)
const searchParams = new URLSearchParams(); const searchParams = new URLSearchParams();
searchParams.append('id', id); searchParams.append('id', id);
const res = await fetch(`${WEBUI_API_BASE_URL}/models/id/${id}/update`, { const res = await fetch(`${WEBUI_API_BASE_URL}/models/model/update?${searchParams.toString()}`, {
method: 'POST', method: 'POST',
headers: { headers: {
Accept: 'application/json', Accept: 'application/json',
@ -203,7 +203,7 @@ export const deleteModelById = async (token: string, id: string) => {
const searchParams = new URLSearchParams(); const searchParams = new URLSearchParams();
searchParams.append('id', id); searchParams.append('id', id);
const res = await fetch(`${WEBUI_API_BASE_URL}/models/id/${id}/delete`, { const res = await fetch(`${WEBUI_API_BASE_URL}/models/model/delete?${searchParams.toString()}`, {
method: 'DELETE', method: 'DELETE',
headers: { headers: {
Accept: 'application/json', Accept: 'application/json',