DrishtiSharma commited on
Commit
da89558
·
verified ·
1 Parent(s): c5009ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -668,13 +668,17 @@ def set_tools() -> List[Tool]:
668
  # python_repl = PythonREPLTool()
669
 
670
  tool_options = ["ArXiv", "Wikipedia", "Python_REPL", "Retrieval"]
 
671
  tool_dictionary = {
672
- "ArXiv": arxiv,
673
- "Wikipedia": wikipedia,
674
- "Python_REPL": python_repl,
675
- "Retrieval": st.session_state.retriever_tool
676
  }
677
 
 
 
 
 
678
  if st.session_state.bing_subscription_validity:
679
  search = BingSearchAPIWrapper()
680
  elif st.session_state.google_cse_id_validity:
 
668
  # python_repl = PythonREPLTool()
669
 
670
  tool_options = ["ArXiv", "Wikipedia", "Python_REPL", "Retrieval"]
671
+
672
  tool_dictionary = {
673
+ "ArXiv": arxiv,
674
+ "Wikipedia": wikipedia,
675
+ "Python_REPL": python_repl
 
676
  }
677
 
678
+ if st.session_state.retriever_tool:
679
+ tool_dictionary["Retrieval"] = st.session_state.retriever_tool
680
+
681
+
682
  if st.session_state.bing_subscription_validity:
683
  search = BingSearchAPIWrapper()
684
  elif st.session_state.google_cse_id_validity: