ofermend commited on
Commit
6fabb15
·
1 Parent(s): b8315ca

added tavily tools

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -128,7 +128,7 @@ def create_tools(cfg):
128
  texts = [get_story_text(story_id) for story_id in stories[:n_stories]]
129
  all_stories = '---------\n\n'.join(texts)
130
  return summarize_text(all_stories)
131
-
132
  return (
133
  tools_factory.get_tools(
134
  [
@@ -137,9 +137,10 @@ def create_tools(cfg):
137
  get_ask_stories,
138
  get_story_details,
139
  get_story_text,
140
- whats_new
141
  ]
142
  ) +
 
143
  tools_factory.standard_tools() +
144
  tools_factory.guardrail_tools() +
145
  [ask_hackernews]
@@ -201,7 +202,8 @@ def launch_bot():
201
  'customer_id': str(os.environ['VECTARA_CUSTOMER_ID']),
202
  'corpus_id': str(os.environ['VECTARA_CORPUS_ID']),
203
  'api_key': str(os.environ['VECTARA_API_KEY']),
204
- 'examples': os.environ.get('QUERY_EXAMPLES', None)
 
205
  })
206
  st.session_state.cfg = cfg
207
  st.session_state.ex_prompt = None
 
128
  texts = [get_story_text(story_id) for story_id in stories[:n_stories]]
129
  all_stories = '---------\n\n'.join(texts)
130
  return summarize_text(all_stories)
131
+
132
  return (
133
  tools_factory.get_tools(
134
  [
 
137
  get_ask_stories,
138
  get_story_details,
139
  get_story_text,
140
+ whats_new,
141
  ]
142
  ) +
143
+ tools_factory.get_llama_index_tools("tavily_research", "TavilyToolSpec", api_key=cfg.tavily_api_key) +
144
  tools_factory.standard_tools() +
145
  tools_factory.guardrail_tools() +
146
  [ask_hackernews]
 
202
  'customer_id': str(os.environ['VECTARA_CUSTOMER_ID']),
203
  'corpus_id': str(os.environ['VECTARA_CORPUS_ID']),
204
  'api_key': str(os.environ['VECTARA_API_KEY']),
205
+ 'examples': os.environ.get('QUERY_EXAMPLES', None),
206
+ 'tavily_api_key': str(os.environ['TAVILY_API_KEY']),
207
  })
208
  st.session_state.cfg = cfg
209
  st.session_state.ex_prompt = None