ofermend commited on
Commit
591e6ca
1 Parent(s): 464ef22

updated tool names

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- query_financial_reports = tools_factory.create_rag_tool(
56
- tool_name = "query_financial_reports",
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
- [query_financial_reports]
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):