Skip to content

Commit 8a1314a

Browse files
committed
🦄 refactor: tiny refactor
1 parent 03893c4 commit 8a1314a

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/api/router/v1/reranker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,14 +28,14 @@ async def list_rerankers() -> ListRerankerResponse:
2828

2929
@reranker_router.post("/rerank", response_model=RerankResponse, dependencies=[Depends(auth_secret_key)])
3030
async def cohere_rerank(request: RerankRequest) -> RerankResponse:
31-
logger.info(f"[Cohere] use model: {request.model}")
31+
logger.info(f"[Reranker] use model: {request.model}")
3232
if not request.query or not request.documents:
3333
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Invalid request, query and documents are required")
3434

3535
try:
3636
engine = RERANKER_ENGINE_MAPPING[request.model]
3737
except KeyError:
38-
logger.error(f"[Cohere Rerank] Invalid model name: {request.model}")
38+
logger.error(f"[Reranker] Invalid model name: {request.model}")
3939
raise HTTPException(
4040
status_code=status.HTTP_404_NOT_FOUND,
4141
detail=f"Invalid model name: {request.model}",

0 commit comments

Comments
 (0)