Spaces:
Sleeping
Sleeping
Commit
·
e3e32d4
1
Parent(s):
1d03974
Update query params
Browse files
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(
|
21 |
-
|
22 |
-
|
|
|
|
|
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 =
|
34 |
-
summary_num_results =
|
|
|
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 |
)
|