Disconnect when deleted. (run-llama#13611)

Mateusz-Switala · Jun 13, 2024 · 41577c2 · 41577c2
1 parent e0b9c12
commit 41577c2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/...ctor_stores/llama-index-vector-stores-supabase/llama_index/vector_stores/supabase/base.py b/...ctor_stores/llama-index-vector-stores-supabase/llama_index/vector_stores/supabase/base.py
@@ -81,6 +81,11 @@ def __init__(
                 name=collection_name, dimension=dimension
             )
 
+    def __del__(self) -> None:
+        """Close the client when the object is deleted."""
+        if self._client is not None:
+            self._client.disconnect()
+
     @property
     def client(self) -> None:
         """Get client."""

diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-supabase/pyproject.toml b/llama-index-integrations/vector_stores/llama-index-vector-stores-supabase/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-vector-stores-supabase"
 readme = "README.md"
-version = "0.1.3"
+version = "0.1.4"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"