Joshua Sundance Bailey commited on
Commit
d8714b0
1 Parent(s): 617f868
Files changed (1) hide show
  1. langchain-streamlit-demo/app.py +3 -2
langchain-streamlit-demo/app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from datetime import datetime
2
  from typing import Tuple, List, Dict, Any, Union, Optional
3
 
@@ -444,11 +445,11 @@ if st.session_state.llm:
444
  # stream_handler = StreamHandler(message_placeholder)
445
  # callbacks.append(stream_handler)
446
  message_placeholder = st.empty()
447
-
448
  default_tools = [
449
  DuckDuckGoSearchRun(),
450
  WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper()),
451
  ]
 
452
  if st.session_state.provider in ("Azure OpenAI", "OpenAI"):
453
  research_assistant_chain = get_research_assistant_chain(
454
  search_llm=get_llm(**get_llm_args_temp_zero), # type: ignore
@@ -462,7 +463,7 @@ if st.session_state.llm:
462
  # config=get_config(callbacks),
463
  ),
464
  name="web-research-assistant",
465
- description="this assistant returns a comprehensive report based on web research. for quick facts, use duckduckgo instead.",
466
  )
467
 
468
  TOOLS = [research_assistant_tool] + default_tools
 
1
+ from langchain.agents import load_tools
2
  from datetime import datetime
3
  from typing import Tuple, List, Dict, Any, Union, Optional
4
 
 
445
  # stream_handler = StreamHandler(message_placeholder)
446
  # callbacks.append(stream_handler)
447
  message_placeholder = st.empty()
 
448
  default_tools = [
449
  DuckDuckGoSearchRun(),
450
  WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper()),
451
  ]
452
+ default_tools += load_tools(["requests_get"])
453
  if st.session_state.provider in ("Azure OpenAI", "OpenAI"):
454
  research_assistant_chain = get_research_assistant_chain(
455
  search_llm=get_llm(**get_llm_args_temp_zero), # type: ignore
 
463
  # config=get_config(callbacks),
464
  ),
465
  name="web-research-assistant",
466
+ description="this assistant returns a comprehensive report based on web research. it's relatively expensive, so use it sparingly. for quick facts, use duckduckgo instead.",
467
  )
468
 
469
  TOOLS = [research_assistant_tool] + default_tools