Switch to Chain Rerankers

agent.py

@@ -30,7 +30,17 @@ def create_assistant_tools(cfg):
         returns a response to a user question about electric vehicles.
         """,
         tool_args_schema = QueryElectricCars,
-        reranker = "multilingual_reranker_v1", rerank_k = 100, 
+        reranker = "chain", rerank_k = 100,
+        rerank_chain = [
+            {
+                "type": "slingshot",
+                "cutoff": 0.2
+            },
+            {
+                "type": "mmr",
+                "diversity_bias": 0.1
+            }
+        ],
         n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
         summary_num_results = 5,
         vectara_summarizer = summarizer,
@@ -60,7 +70,17 @@ def create_assistant_tools(cfg):
         You can ask this tool any question about laws passed by states or the federal government related to electric vehicles.
         """,
         tool_args_schema = QueryEVLaws,
-        reranker = "multilingual_reranker_v1", rerank_k = 100, 
+        reranker = "chain", rerank_k = 100, 
+        rerank_chain = [
+            {
+                "type": "slingshot",
+                "cutoff": 0.2
+            },
+            {
+                "type": "mmr",
+                "diversity_bias": 0.1
+            }
+        ],
         n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
         summary_num_results = 10,
         vectara_summarizer = summarizer,