Spaces:
Running
Running
updated tool names
Browse files
app.py
CHANGED
@@ -52,11 +52,11 @@ def create_tools(cfg):
|
|
52 |
tools_factory = ToolsFactory(vectara_api_key=cfg.api_key,
|
53 |
vectara_customer_id=cfg.customer_id,
|
54 |
vectara_corpus_id=cfg.corpus_id)
|
55 |
-
|
56 |
-
tool_name = "
|
57 |
tool_description = """
|
58 |
Given a company name and year,
|
59 |
-
returns a response (str) to a user query about the company's financial reports for that year.
|
60 |
make sure to provide the a valid company ticker and year.
|
61 |
""",
|
62 |
tool_args_schema = QueryFinancialReportsArgs,
|
@@ -76,7 +76,7 @@ def create_tools(cfg):
|
|
76 |
tools_factory.standard_tools() +
|
77 |
tools_factory.financial_tools() +
|
78 |
tools_factory.guardrail_tools() +
|
79 |
-
[
|
80 |
)
|
81 |
|
82 |
def initialize_agent(agent_type: AgentType, _cfg):
|
|
|
52 |
tools_factory = ToolsFactory(vectara_api_key=cfg.api_key,
|
53 |
vectara_customer_id=cfg.customer_id,
|
54 |
vectara_corpus_id=cfg.corpus_id)
|
55 |
+
vectara_ask_analysts = tools_factory.create_rag_tool(
|
56 |
+
tool_name = "vectara_ask_analysts",
|
57 |
tool_description = """
|
58 |
Given a company name and year,
|
59 |
+
returns a response (str) to a user query based on analyst comments about the company's financial reports for that year.
|
60 |
make sure to provide the a valid company ticker and year.
|
61 |
""",
|
62 |
tool_args_schema = QueryFinancialReportsArgs,
|
|
|
76 |
tools_factory.standard_tools() +
|
77 |
tools_factory.financial_tools() +
|
78 |
tools_factory.guardrail_tools() +
|
79 |
+
[vectara_ask_analysts]
|
80 |
)
|
81 |
|
82 |
def initialize_agent(agent_type: AgentType, _cfg):
|