DrishtiSharma
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -670,14 +670,12 @@ def set_tools() -> List[Tool]:
|
|
670 |
tool_options = ["ArXiv", "Wikipedia", "Python_REPL", "Retrieval"]
|
671 |
|
672 |
tool_dictionary = {
|
673 |
-
|
674 |
-
|
675 |
-
|
676 |
-
|
677 |
}
|
678 |
|
679 |
-
|
680 |
-
|
681 |
if st.session_state.bing_subscription_validity:
|
682 |
search = BingSearchAPIWrapper()
|
683 |
elif st.session_state.google_cse_id_validity:
|
|
|
670 |
tool_options = ["ArXiv", "Wikipedia", "Python_REPL", "Retrieval"]
|
671 |
|
672 |
tool_dictionary = {
|
673 |
+
"ArXiv": arxiv,
|
674 |
+
"Wikipedia": wikipedia,
|
675 |
+
"Python_REPL": python_repl,
|
676 |
+
"Retrieval": st.session_state.retriever_tool if st.session_state.retriever_tool else None
|
677 |
}
|
678 |
|
|
|
|
|
679 |
if st.session_state.bing_subscription_validity:
|
680 |
search = BingSearchAPIWrapper()
|
681 |
elif st.session_state.google_cse_id_validity:
|