Joshua Sundance Bailey
commited on
Commit
·
d8714b0
1
Parent(s):
617f868
requests
Browse files
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
|