adds caching

streamlit.py

@@ -285,6 +285,7 @@ QUERY_MESSAGES: list[tuple[Roles, str]] = [
 
 
 # pylint: enable=line-too-long
+@st.cache_data
 async def retrieve_docs(
         query: str, query_filter: dict[str, str]) -> list[docstore.Document]:
     # """Retrieves documents for a query
@@ -388,7 +389,7 @@ for page in PAGES:
             workable_customers.append(href)
 
 # workable_customers
-@st.cache
+@st.cache_data
 def get_paragraphs_workable(url):
     r = requests.get(url=url, headers=headers)
 
@@ -417,14 +418,19 @@ def loop(input):
         input = clean_text(input)
     return input
 
-workable_case_studies = []
-# for customer in customers:
-# TODO(fix)
-for customer in workable_customers:
-  url = customer
-  workable_case_studies.append((url,loop('<join>'.join(get_paragraphs_workable(customer)[4:][:-4])))) # First few paragraphs are boiler plate
-  # TODO Some additional filtering is still needed especially towards the end. We should probably discard things that are not in the main body.
-# workable_case_studies
+@st.cache_data
+def get_case_studies():
+    workable_case_studies = []
+    # for customer in customers:
+    # TODO(fix)
+    for customer in workable_customers:
+    url = customer
+    workable_case_studies.append((url,loop('<join>'.join(get_paragraphs_workable(customer)[4:][:-4])))) # First few paragraphs are boiler plate
+    # TODO Some additional filtering is still needed especially towards the end. We should probably discard things that are not in the main body.
+    # workable_case_studies
+    return workable_case_studies
+
+workable_case_studies = get_case_studies()
 
 
 # """## App logic"""