david-oplatka commited on
Commit
e3e32d4
·
1 Parent(s): 1d03974

Update query params

Browse files
Files changed (1) hide show
  1. agent.py +8 -5
agent.py CHANGED
@@ -17,9 +17,11 @@ def create_assistant_tools(cfg):
17
  class QueryCFPBComplaints(BaseModel):
18
  query: str = Field(description="The user query.")
19
 
20
- vec_factory = VectaraToolFactory(vectara_api_key=cfg.api_keys,
21
- vectara_customer_id=cfg.customer_id,
22
- vectara_corpus_id=cfg.corpus_ids)
 
 
23
 
24
  summarizer = 'vectara-experimental-summary-ext-2023-12-11-med-omni'
25
  ask_complaints = vec_factory.create_rag_tool(
@@ -30,8 +32,9 @@ def create_assistant_tools(cfg):
30
  """,
31
  tool_args_schema = QueryCFPBComplaints,
32
  reranker = "multilingual_reranker_v1", rerank_k = 100,
33
- n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
34
- summary_num_results = 15,
 
35
  vectara_summarizer = summarizer,
36
  include_citations = True,
37
  )
 
17
  class QueryCFPBComplaints(BaseModel):
18
  query: str = Field(description="The user query.")
19
 
20
+ vec_factory = VectaraToolFactory(
21
+ vectara_api_key=cfg.api_keys,
22
+ vectara_customer_id=cfg.customer_id,
23
+ vectara_corpus_id=cfg.corpus_ids
24
+ )
25
 
26
  summarizer = 'vectara-experimental-summary-ext-2023-12-11-med-omni'
27
  ask_complaints = vec_factory.create_rag_tool(
 
32
  """,
33
  tool_args_schema = QueryCFPBComplaints,
34
  reranker = "multilingual_reranker_v1", rerank_k = 100,
35
+ n_sentences_before = 1, n_sentences_after = 1, lambda_val = 0.005,
36
+ summary_num_results = 30,
37
+ mmr_diversity_bias = 0.4,
38
  vectara_summarizer = summarizer,
39
  include_citations = True,
40
  )