OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-15 10:51:13 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

feat(sqlalchemy): reverted not needed api change

Browse Source
This commit is contained in:
Jonathan Rohde 2024-06-25 08:29:18 +02:00
parent 642c352c69
commit d4b6b7c4e8
4 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/apps/webui/routers/models.py
View File

@ -56,7 +56,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)

4
backend/test/apps/webui/routers/test_models.py
View File

@ -42,9 +42,9 @@ class TestModels(AbstractPostgresTest):
assert len(response.json()) == 1 assert len(response.json()) == 1
with mock_webui_user(id="2"): with mock_webui_user(id="2"):
response = self.fast_api_client.get(self.create_url("/my-model")) response = self.fast_api_client.get(self.create_url(query_params={"id": "my-model"}))
assert response.status_code == 200 assert response.status_code == 200
data = response.json() data = response.json()[0]
assert data["id"] == "my-model" assert data["id"] == "my-model"
assert data["name"] == "Hello World" assert data["name"] == "Hello World"

10
backend/test/util/abstract_integration_test.py
View File

@ -23,14 +23,20 @@ def get_fast_api_client():
class AbstractIntegrationTest: class AbstractIntegrationTest:
BASE_PATH = None BASE_PATH = None
def create_url(self, path): def create_url(self, path="", query_params=None):
if self.BASE_PATH is None: if self.BASE_PATH is None:
raise Exception("BASE_PATH is not set") raise Exception("BASE_PATH is not set")
parts = self.BASE_PATH.split("/") parts = self.BASE_PATH.split("/")
parts = [part.strip() for part in parts if part.strip() != ""] parts = [part.strip() for part in parts if part.strip() != ""]
path_parts = path.split("/") path_parts = path.split("/")
path_parts = [part.strip() for part in path_parts if part.strip() != ""] path_parts = [part.strip() for part in path_parts if part.strip() != ""]
return "/".join(parts + path_parts) query_parts = ""
if query_params:
query_parts = "&".join(
[f"{key}={value}" for key, value in query_params.items()]
)
query_parts = f"?{query_parts}"
return "/".join(parts + path_parts) + query_parts
@classmethod @classmethod
def setup_class(cls): def setup_class(cls):

5
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 searchParams = new URLSearchParams();
searchParams.append('id', id);
const res = await fetch(`${WEBUI_API_BASE_URL}/models?${searchParams.toString()}`, {
method: 'GET', method: 'GET',
headers: { headers: {
Accept: 'application/json', Accept: 'application/json',