From d8201867481cd118fa3f1f45f6f8eb4b1fce9700 Mon Sep 17 00:00:00 2001 From: Alisson Date: Mon, 5 Aug 2024 17:23:11 -0300 Subject: [PATCH] add print --- app/rerank.py | 1 + 1 file changed, 1 insertion(+) diff --git a/app/rerank.py b/app/rerank.py index 6e09c8f..4cc91a6 100644 --- a/app/rerank.py +++ b/app/rerank.py @@ -14,6 +14,7 @@ def rerank_chunks( threshold: float = RERANK_THRESHOLD, max_docs: int = 5, ): + print("[+ Rerank Chunks +]") co = cohere.Client(RERANK_API_KEY) responses = co.rerank( model=RERANK_MODEL,