fix: ensure that connection to weaviate is ready

jabra · Sep 15, 2023 · 7eb56c0 · 7eb56c0
1 parent 579b0fd
commit 7eb56c0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/streamlit_examples/utils/weaviate.py b/streamlit_examples/utils/weaviate.py
@@ -3,7 +3,10 @@
 from streamlit_examples.utils.cohere import cohere_api_key
 
 
-@st.cache_resource(show_spinner="Connecting to Weaviate...")
+@st.cache_resource(
+ show_spinner="Connecting to Weaviate...",
+ validate=lambda client: client.is_ready(),
+)
 def connect_weaviate():
  # Connect to the Weaviate demo database containing 10M wikipedia vectors
  # This uses a public READ-ONLY Weaviate API key
@@ -18,7 +21,6 @@ def connect_weaviate():
  },
  )
 
- client.is_ready()
  return client