UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 13:40:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: models

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-25 13:48:45 -07:00
parent 3d0b3eb577
commit afcab78cab
5 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/apps/web/routers/models.py
View File

@ -53,7 +53,7 @@ async def add_new_model(
############################ ############################
@router.get("/{id}", response_model=Optional[ModelModel]) @router.get("/", response_model=Optional[ModelModel])
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)
@ -71,7 +71,7 @@ async def get_model_by_id(id: str, user=Depends(get_verified_user)):
############################ ############################
@router.post("/{id}/update", response_model=Optional[ModelModel]) @router.post("/update", response_model=Optional[ModelModel])
async def update_model_by_id( async def update_model_by_id(
request: Request, id: str, form_data: ModelForm, user=Depends(get_admin_user) request: Request, id: str, form_data: ModelForm, user=Depends(get_admin_user)
): ):
@ -102,7 +102,7 @@ async def update_model_by_id(
############################ ############################
@router.delete("/{id}/delete", response_model=bool) @router.delete("/delete", response_model=bool)
async def delete_model_by_id(id: str, user=Depends(get_admin_user)): async def delete_model_by_id(id: str, user=Depends(get_admin_user)):
result = Models.delete_model_by_id(id) result = Models.delete_model_by_id(id)
return result return result

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

@ -63,7 +63,10 @@ export const getModelInfos = async (token: string = '') => {
export const getModelById = async (token: string, id: string) => { export const getModelById = async (token: string, id: string) => {
let error = null; let error = null;
const res = await fetch(`${WEBUI_API_BASE_URL}/models/${id}`, { const url = new URL(`${WEBUI_API_BASE_URL}/models`);
url.searchParams.append('id', id);
const res = await fetch(url.toString(), {
method: 'GET', method: 'GET',
headers: { headers: {
Accept: 'application/json', Accept: 'application/json',
@ -95,7 +98,10 @@ export const getModelById = async (token: string, id: string) => {
export const updateModelById = async (token: string, id: string, model: object) => { export const updateModelById = async (token: string, id: string, model: object) => {
let error = null; let error = null;
const res = await fetch(`${WEBUI_API_BASE_URL}/models/${id}/update`, { const url = new URL(`${WEBUI_API_BASE_URL}/models/update`);
url.searchParams.append('id', id);
const res = await fetch(url.toString(), {
method: 'POST', method: 'POST',
headers: { headers: {
Accept: 'application/json', Accept: 'application/json',
@ -128,7 +134,10 @@ export const updateModelById = async (token: string, id: string, model: object)
export const deleteModelById = async (token: string, id: string) => { export const deleteModelById = async (token: string, id: string) => {
let error = null; let error = null;
const res = await fetch(`${WEBUI_API_BASE_URL}/models/${id}/delete`, { const url = new URL(`${WEBUI_API_BASE_URL}/models/delete`);
url.searchParams.append('id', id);
const res = await fetch(url.toString(), {
method: 'DELETE', method: 'DELETE',
headers: { headers: {
Accept: 'application/json', Accept: 'application/json',

2
src/lib/components/chat/Settings/General.svelte
View File

@ -302,7 +302,7 @@
system: system !== '' ? system : undefined, system: system !== '' ? system : undefined,
params: { params: {
seed: (params.seed !== 0 ? params.seed : undefined) ?? undefined, seed: (params.seed !== 0 ? params.seed : undefined) ?? undefined,
stop: params.stop !== null ? params.stop.split(',').filter((e) => e) : undefined, stop: params.stop ? params.stop.split(',').filter((e) => e) : undefined,
temperature: params.temperature !== '' ? params.temperature : undefined, temperature: params.temperature !== '' ? params.temperature : undefined,
frequency_penalty: frequency_penalty:
params.frequency_penalty !== '' ? params.frequency_penalty : undefined, params.frequency_penalty !== '' ? params.frequency_penalty : undefined,

2
src/routes/(app)/workspace/models/create/+page.svelte
View File

@ -62,7 +62,7 @@
info.id = id; info.id = id;
info.name = name; info.name = name;
info.meta.capabilities = capabilities; info.meta.capabilities = capabilities;
info.params.stop = params.stop !== null ? params.stop.split(',').filter((s) => s.trim()) : null; info.params.stop = params.stop ? params.stop.split(',').filter((s) => s.trim()) : null;
if ($models.find((m) => m.id === info.id)) { if ($models.find((m) => m.id === info.id)) {
toast.error( toast.error(

2
src/routes/(app)/workspace/models/edit/+page.svelte
View File

@ -63,7 +63,7 @@
info.id = id; info.id = id;
info.name = name; info.name = name;
info.meta.capabilities = capabilities; info.meta.capabilities = capabilities;
info.params.stop = params.stop !== null ? params.stop.split(',').filter((s) => s.trim()) : null; info.params.stop = params.stop ? params.stop.split(',').filter((s) => s.trim()) : null;
const res = await updateModelById(localStorage.token, info.id, info); const res = await updateModelById(localStorage.token, info.id, info);