Spaces:
Running
Running
fahmiaziz98
commited on
Commit
·
3b88f19
1
Parent(s):
9e5acab
[UPDATE]: Change version + add prefix rank endpoint
Browse files- src/api/routers/rerank.py +2 -2
- src/config/settings.py +2 -2
src/api/routers/rerank.py
CHANGED
|
@@ -21,11 +21,11 @@ from src.core.exceptions import (
|
|
| 21 |
from src.api.dependencies import get_model_manager
|
| 22 |
from src.utils.validators import extract_embedding_kwargs
|
| 23 |
|
| 24 |
-
router = APIRouter(tags=["rerank"])
|
| 25 |
|
| 26 |
|
| 27 |
@router.post(
|
| 28 |
-
"
|
| 29 |
response_model=RerankResponse,
|
| 30 |
summary="Rerank documents",
|
| 31 |
description="Reranks the provided documents based on the given query.",
|
|
|
|
| 21 |
from src.api.dependencies import get_model_manager
|
| 22 |
from src.utils.validators import extract_embedding_kwargs
|
| 23 |
|
| 24 |
+
router = APIRouter(prefix="/rerank",tags=["rerank"])
|
| 25 |
|
| 26 |
|
| 27 |
@router.post(
|
| 28 |
+
"",
|
| 29 |
response_model=RerankResponse,
|
| 30 |
summary="Rerank documents",
|
| 31 |
description="Reranks the provided documents based on the given query.",
|
src/config/settings.py
CHANGED
|
@@ -19,8 +19,8 @@ class Settings(BaseSettings):
|
|
| 19 |
"""
|
| 20 |
|
| 21 |
# Application Info
|
| 22 |
-
APP_NAME: str = "Unified Embedding API - Dense &
|
| 23 |
-
VERSION: str = "3.5.
|
| 24 |
DEBUG: bool = False
|
| 25 |
ENVIRONMENT: str = "development" # development, staging, production
|
| 26 |
|
|
|
|
| 19 |
"""
|
| 20 |
|
| 21 |
# Application Info
|
| 22 |
+
APP_NAME: str = "Unified Embedding API - Dense, Sparse & Reranking"
|
| 23 |
+
VERSION: str = "3.5.1"
|
| 24 |
DEBUG: bool = False
|
| 25 |
ENVIRONMENT: str = "development" # development, staging, production
|
| 26 |
|