refactor: if client exists close and reconnect

datastew/repository/weaviate.py

@@ -5,6 +5,7 @@
 from typing import List, Tuple, Union, Optional
 
 import weaviate
+from weaviate import WeaviateClient
 from weaviate.util import generate_uuid5
 from weaviate.classes.query import Filter, QueryReference, MetadataQuery
 
@@ -20,10 +21,14 @@ class WeaviateRepository(BaseRepository):
 
     def __init__(self, mode="memory", path=None, port=80, http_port=8079, grpc_port=50050):
         self.mode = mode
+        self.client: Union[None, WeaviateClient] = None
         try:
             if mode == "memory":
                 # Check if there is an existing instance of Weaviate client for the default ports
                 if self._is_port_in_use(http_port) and self._is_port_in_use(grpc_port):
+                    # if the client exists, first close then re-connect
+                    if self.client:
+                        self.client.close()
                     self.client = weaviate.connect_to_local(port=http_port, grpc_port=grpc_port)
                 else:
                     self.client = weaviate.connect_to_embedded(persistence_data_path="db")
@@ -32,6 +37,9 @@ def __init__(self, mode="memory", path=None, port=80, http_port=8079, grpc_port=
                     raise ValueError("Path must be provided for disk mode.")
                 # Check if there is an existing instance of Weaviate client for the default ports
                 if self._is_port_in_use(http_port) and self._is_port_in_use(grpc_port):
+                    # if the client exists, first close then re-connect
+                    if self.client:
+                        self.client.close()
                     self.client = weaviate.connect_to_local(port=http_port, grpc_port=grpc_port)
                 else:
                     self.client = weaviate.connect_to_embedded(persistence_data_path=path)