add more cohere nodes

towardsai · Jul 24, 2024 · e34ffae · e34ffae
1 parent 23da9d9
commit e34ffae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/custom_retriever.py b/scripts/custom_retriever.py
@@ -72,7 +72,7 @@ def filter_nodes_by_unique_doc_id(nodes):
             else:
                 nodes_context.append(node)
 
-        reranker = CohereRerank(top_n=5, model="rerank-english-v3.0")
+        reranker = CohereRerank(top_n=8, model="rerank-english-v3.0")
         nodes_context = reranker.postprocess_nodes(nodes_context, query_bundle)
         logfire.info(f"Cohere raranking to {len(nodes_context)} nodes")