jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

API: Fix return of current embeddings model

Browse source 
Return a ModelCard instead of a ModelList.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri 2024-08-01 13:43:31 -04:00
parent 0bcb4e4a7d
commit 54aeebaec1
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
endpoints/core/router.py
View file

@ -279,10 +279,11 @@ async def list_embedding_models(request: Request) -> ModelList:
"/v1/model/embedding",
dependencies=[Depends(check_api_key), Depends(check_embeddings_container)],
)
async def get_embedding_model() -> ModelList:
async def get_embedding_model() -> ModelCard:
"""Returns the currently loaded embedding model."""
models = await get_current_model_list(model_type="embedding")
return get_current_model_list(model_type="embedding")[0]
return models.data[0]
@router.post("/v1/model/embedding/load", dependencies=[Depends(check_admin_key)])