DrishtiSharma commited on
Commit
c3bf3f1
·
verified ·
1 Parent(s): 5b36b4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
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
- "ArXiv": arxiv,
674
- "Wikipedia": wikipedia,
675
- "Python_REPL": python_repl,
676
- "Retrieval": st.session_state.retriever_tool
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: