Merge branch 'main' of https://huggingface.co/spaces/vectara/legal-assistant

agent.py

@@ -172,13 +172,28 @@ def create_assistant_tools(cfg):
                                      vectara_customer_id=cfg.customer_id, 
                                      vectara_corpus_id=cfg.corpus_id)
     summarizer = 'vectara-experimental-summary-ext-2023-12-11-med-omni'
+
     ask_caselaw = vec_factory.create_rag_tool(
         tool_name = "ask_caselaw",
         tool_description = "A tool for asking questions about case law in Alaska.",
         tool_args_schema = QueryCaselawArgs,
-        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
+            },
+            {
+                "type": "udf",
+                "user_function": "max(1000 * get('$.score') - hours(seconds(to_unix_timestamp(now()) - to_unix_timestamp(datetime_parse(get('$.document_metadata.decision_date'), 'yyyy-MM-dd')))) / 24 / 365, 0)"
+            }
+        ],
         n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
-        summary_num_results = 10,
+        summary_num_results = 15,
         vectara_summarizer = summarizer,
         include_citations = False,
     )