Add UDF Rerank and Update Instructions

agent.py

@@ -41,38 +41,23 @@ def create_assistant_tools(cfg):
         returns a response to a user question about customer complaints for bank services.
         """,
         tool_args_schema = QueryCFPBComplaints,
-        reranker = "multilingual_reranker_v1", rerank_k = 100, 
+        reranker = "chain", rerank_k = 100, 
+        rerank_chain = [
+            {
+                "type": "slingshot",
+                "cutoff": 0.2
+            },
+            {
+                "type": "mmr",
+                "diversity_bias": 0.4,
+                "limit": 30
+            }
+        ],
         n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
-        summary_num_results = 30,
-        mmr_diversity_bias = 0.4,
         vectara_summarizer = summarizer,
         include_citations = True,
     )
 
-    # ask_complaints = vec_factory.create_rag_tool(
-    #     tool_name = "ask_complaints",
-    #     tool_description = """
-    #     Given a user query, 
-    #     returns a response to a user question about customer complaints for bank services.
-    #     """,
-    #     tool_args_schema = QueryCFPBComplaints,
-    #     reranker = "chain", rerank_k = 100, 
-    #     rerank_chain = [
-    #         {
-    #             "type": "slingshot",
-    #             "cutoff": 0.2
-    #         },
-    #         {
-    #             "type": "mmr",
-    #             "diversity_bias": 0.4,
-    #             "limit": 30
-    #         }
-    #     ],
-    #     n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
-    #     vectara_summarizer = summarizer,
-    #     include_citations = True,
-    # )
-
     tools_factory = ToolsFactory()
 
     db_tools = tools_factory.database_tools(
@@ -93,6 +78,8 @@ def initialize_agent(_cfg, agent_progress_callback=None):
       with expertise in finance and complaints from the CFPB (Consumer Financial Protection Bureau), 
       in conversation with a user.
     - For analytical/numeric questions, try to use the cfpb_load_data and other database tools.
+    - For questions about customers' complaints (the text of the complaint), use the ask_complaints tool.
+      Do not include the "References" section in your response.
     - Never discuss politics, and always respond politely.
     """