Update app.py

app.py

@@ -222,9 +222,14 @@ def run_agent(
 
     try:
         # Ensure retriever tool is included in tools
-        if "Retrieval" in st.session_state.tool_names[0] and st.session_state.retriever_tool:
-            if st.session_state.retriever_tool not in tools:
-                tools.append(st.session_state.retriever_tool)
+
+        if st.session_state.retriever_tool:
+            st.write("**Retriever tool has been added to the tools list.**")
+        if st.session_state.retriever_tool not in tools:
+            tools.append(st.session_state.retriever_tool)
+        else:
+            st.warning("Retriever tool is not available for querying.")
+
 
         llm = get_chat_model(model, temperature, [StreamHandler(st.empty())])
         if llm is None:
@@ -400,6 +405,11 @@ def get_retriever() -> None:
                     "this tool!"
                 ),
             )
+            if st.session_state.retriever_tool:
+                st.success("Retriever tool has been successfully initialized.")
+            else:
+                st.error("Failed to initialize retriever tool.")
+
             st.session_state.uploader_key += 1