File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 )])
3030async 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 } " ,
You can’t perform that action at this time.
0 commit comments