Add Metadata Filters

agent.py

@@ -16,6 +16,16 @@ def create_assistant_tools(cfg):
 
     class QueryCFPBComplaints(BaseModel):
         query: str = Field(description="The user query.")
+        Company: Optional[str] = Field(
+            default=None,
+            description="The company that the complaint is about.",
+            examples=['CAPITAL ONE FINANCIAL CORPORATION', 'BANK OF AMERICA, NATIONAL ASSOCIATION', 'CITIBANK, N.A.', 'WELLS FARGO & COMPANY', 'JPMORGAN CHASE & CO.']
+        )
+        State: Optional[str] = Field(
+            default=None,
+            descripition="The two-character state code where the consumer lives.",
+            examples=['CA', 'FL', 'NY', 'TX', 'GA']
+        )
 
     vec_factory = VectaraToolFactory(
         vectara_api_key=cfg.api_keys,
@@ -39,6 +49,30 @@ def create_assistant_tools(cfg):
         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(