add print

weni-ai · Aug 5, 2024 · d820186 · d820186
1 parent 7e3fa1e
commit d820186
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,