changing the name of the query funtcion

app.py

@@ -7,7 +7,6 @@ import requests
 
 import streamlit as st
 
-from utils.haystack import query, start_haystack
 from utils.ui import reset_results, set_initial_state, sidebar
 from utils.config import TWITTER_BEARER_TOKEN, SERPER_KEY, OPENAI_API_KEY
 
@@ -38,16 +37,15 @@ run_query = (
 if run_query and question:
     reset_results()
     st.session_state.question = question
-    with st.spinner("🔎"):
-        try:
-            st.session_state.result = query(agent, question)
-        except JSONDecodeError as je:
-            st.error(
-                "👓    An error occurred reading the results. Is the document store working?"
-            )    
-        except Exception as e:
-            logging.exception(e)
-            st.error("🐞    An error occurred during the request.")            
+    try:
+        st.session_state.result = run_agent(agent, question)
+    except JSONDecodeError as je:
+        st.error(
+            "👓    An error occurred reading the results. Is the document store working?"
+        )    
+    except Exception as e:
+        logging.exception(e)
+        st.error("🐞    An error occurred during the request.")            
             
 if st.session_state.result:
     result = st.session_state.result

utils/haystack.py

@@ -36,10 +36,9 @@ def start_haystack(openai_key, twitter_bearer, serper_key, last_k_tweets):
     
 
 @st.cache_data(show_spinner=True)
-def query(_agent, question):
+def run_agent(_agent, question):
     try:
         result = _agent.run(question)
     except Exception as e:
-        print(e)
         result = ["Life isn't ideal sometimes and this Agent fails at doing a good job.. Maybe try another query..."]
     return result