Added milvus vdb prototype impl

morpheus/controllers/milvus_vector_db_controller.py

@@ -26,7 +26,7 @@
 from pymilvus import utility
 
 from morpheus.controllers.vector_db_controller import VectorDBController
-from morpheus.controllers.vector_db_controller import with_mutex
+from morpheus.utils.vector_db_utils import with_mutex
 
 logger = logging.getLogger(__name__)
 

morpheus/controllers/qdrant_vector_db_controller.py

@@ -15,6 +15,7 @@
 from qdrant_openapi_client.qdrant_client import QdrantClient
 
 from morpheus.controllers.vector_db_controller import VectorDatabaseController
+from morpheus.utils.vector_db_utils import with_mutex
 
 
 class QdrantVectorDBController(VectorDatabaseController):

morpheus/controllers/vector_db_controller.py

@@ -17,26 +17,13 @@
 from abc import abstractmethod
 
 
-def with_mutex(lock_name):
-
-    def decorator(func):
-
-        def wrapper(*args, **kwargs):
-            with getattr(args[0], lock_name):
-                return func(*args, **kwargs)
-
-        return wrapper
-
-    return decorator
-
-
 class VectorDBController(ABC):
-
-    _mutex = threading.Lock()
     """
     Abstract class for vector document store implementation.
     """
 
+    _mutex = threading.Lock()
+
     @abstractmethod
     def insert(self, name, data, **kwargs):
         """
@@ -58,46 +45,18 @@ def drop(self):
     # @abstractmethod
     # def update(self, vector: list[float], vector_id: str):
     #     """
-    #     Update an existing vector in the vector document store.
-
-    #     Parameters:
-    #     - vector (List[float]): The updated vector data.
-    #     - vector_id (str): The unique identifier of the vector to update.
-
-    #     Returns:
-    #     - None
-
-    #     Raises:
-    #     - RuntimeError: If an error occurs while updating the vector.
     #     """
     #     pass
 
     # @abstractmethod
     # def get_by_name(self, name: str) -> list[float]:
     #     """
-    #     Retrieve a vector from the vector document store by its ID.
-
-    #     Parameters:
-    #     - vector_id (str): The unique identifier of the vector to retrieve.
-
-    #     Returns:
-    #     - List[float]: The vector data.
-
-    #     Raises:
-    #     - RuntimeError: If an error occurs while retrieving the vector.
     #     """
     #     pass
 
     # @abstractmethod
     # def count(self) -> int:
     #     """
-    #     Get the total count of vectors in the vector document store.
-
-    #     Returns:
-    #     - int: The total count of vectors.
-
-    #     Raises:
-    #     - RuntimeError: If an error occurs while counting the vectors.
     #     """
     #     pass
 

morpheus/utils/vector_db_utils.py

@@ -18,6 +18,21 @@
 from morpheus.controllers.vector_db_controller import VectorDBController
 
 
+def with_mutex(lock_name):
+    """
+    """
+
+    def decorator(func):
+
+        def wrapper(*args, **kwargs):
+            with getattr(args[0], lock_name):
+                return func(*args, **kwargs)
+
+        return wrapper
+
+    return decorator
+
+
 def load_handler_by_path(handler_class_path, kwargs):
     """
     Dynamically loads and instantiates a handler class specified by its import path.