Spaces:
Running
Running
added tavily tools
Browse files
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
|