crewplus 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

crewplus/vectorstores/milvus/vdb_service.py

@@ -22,17 +22,22 @@ class VDBService(object):
     and provides helper methods to get embedding functions and vector store instances.
 
     Args:
-        settings (dict): A dictionary containing configuration for the vector store
+        settings (dict, optional): A dictionary containing configuration for the vector store
                          and embedding models.
+        endpoint (str, optional): The URI for the Zilliz cluster. Can be used for simple
+                                  initialization instead of `settings`.
+        token (str, optional): The token for authenticating with Zilliz. Must be provided
+                               with `endpoint`.
         schema (str, optional): The schema definition for a collection. Defaults to None.
         logger (logging.Logger, optional): An optional logger instance. Defaults to None.
 
     Raises:
-        ValueError: If required configurations are missing from the settings dictionary.
+        ValueError: If required configurations are missing.
         NotImplementedError: If an unsupported provider is specified.
         RuntimeError: If the MilvusClient fails to initialize after a retry.
 
     Example:
+        >>> # Initialize with a full settings dictionary
         >>> settings = {
         ...     "embedder": {
         ...         "provider": "azure-openai",
@@ -61,6 +66,10 @@ class VDBService(object):
         ...     }
         ... }
         >>> vdb_service = VDBService(settings=settings)
+        >>>
+        >>> # Alternatively, initialize with an endpoint and token for Zilliz
+        >>> # vdb_service_zilliz = VDBService(endpoint="YOUR_ZILLIZ_ENDPOINT", token="YOUR_ZILLIZ_TOKEN")
+        >>>
         >>> # Get the raw Milvus client
         >>> client = vdb_service.get_vector_client()
         >>> print(client.list_collections())
@@ -82,17 +91,41 @@ class VDBService(object):
     connection_args: dict
     settings: dict
     
-    def __init__(self, settings: dict, schema: str = None, logger: logging.Logger = None):
+    def __init__(self, settings: dict = None, endpoint: str = None, token: str = None, schema: str = None, logger: logging.Logger = None):
         """
         Initializes the VDBService.
+
+        Can be initialized in two ways:
+        1. By providing a full `settings` dictionary for complex configurations.
+        2. By providing `endpoint` and `token` for a direct Zilliz connection.
+           Note: When using this method, an `embedder` configuration is not created.
+           You must either use the `ModelLoadBalancer` or pass an `Embeddings` object
+           directly to methods like `get_vector_store`.
         
         Args:
-            settings (dict): Configuration dictionary for the service.
+            settings (dict, optional): Configuration dictionary for the service. Defaults to None.
+            endpoint (str, optional): The URI for the Zilliz cluster. Used if `settings` is not provided.
+            token (str, optional): The token for authenticating with the Zilliz cluster.
             schema (str, optional): Default schema for new collections. Defaults to None.
             logger (logging.Logger, optional): Logger instance. Defaults to None.
         """
         self.logger = logger or logging.getLogger(__name__)
-        self.settings = settings
+
+        if settings:
+            self.settings = settings
+        elif endpoint and token:
+            self.logger.info("Initializing VDBService with endpoint and token for a Zilliz connection.")
+            self.settings = {
+                "vector_store": {
+                    "provider": "zilliz",
+                    "config": {
+                        "uri": endpoint,
+                        "token": token
+                    }
+                }
+            }
+        else:
+            raise ValueError("VDBService must be initialized with either a 'settings' dictionary or both 'endpoint' and 'token'.")
 
         vector_store_settings = self.settings.get("vector_store")
         if not vector_store_settings:
@@ -262,6 +295,7 @@ class VDBService(object):
 
         Args:
             url (str): source url
+            vdb (Zilliz): Zilliz instance
         """
         self.logger.info(f"Delete old indexes of the same source_url:{url}")
 
@@ -269,7 +303,7 @@ class VDBService(object):
             return None
 
         # Delete indexes of the same source_url
-        expr = "source in [\"" + url + "\"]"
+        expr = f'source_url == "{url}" or source == "{url}"'
         pks = vdb.get_pks(expr)
 
         # Delete entities by pks
@@ -278,19 +312,19 @@ class VDBService(object):
             self.logger.info("Deleted old indexes result: " + str(res))
             return res
 
-    def delete_old_indexes_by_id(self, id: str = None, vdb: Zilliz = None) -> (bool | None):
+    def delete_old_indexes_by_id(self, source_id: str = None, vdb: Zilliz = None) -> (bool | None):
         """ Delete old indexes of the same source_id
 
         Args:
-            id (str): source id
+            source_id (str): source id
         """
-        self.logger.info(f"Delete old indexes of the same source_id:{id}")
+        self.logger.info(f"Delete old indexes of the same source_id:{source_id}")
 
-        if id is None or vdb is None:
+        if source_id is None or vdb is None:
             return None
 
         # Delete indexes of the same source_id
-        expr = "source_id in [\"" + id + "\"]"
+        expr = f'source_id == "{source_id}"'
         pks = vdb.get_pks(expr)
 
         # Delete entities by pks

{crewplus-0.2.2.dist-info → crewplus-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crewplus
-Version: 0.2.2
+Version: 0.2.3
 Summary: Base services for CrewPlus AI applications
 Author-Email: Tim Liu <tim@opsmateai.com>
 License: MIT

{crewplus-0.2.2.dist-info → crewplus-0.2.3.dist-info}/RECORD

@@ -1,7 +1,7 @@
-crewplus-0.2.2.dist-info/METADATA,sha256=HDMXrtxk0z3R13fno8IVVGUErJjRq2gINuilhmcGOQE,4909
-crewplus-0.2.2.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-crewplus-0.2.2.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-crewplus-0.2.2.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
+crewplus-0.2.3.dist-info/METADATA,sha256=3kbzrxbpcbp1zW-lV4j4rYmxnOV5iFiF1oDi5bQ1K-I,4909
+crewplus-0.2.3.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+crewplus-0.2.3.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+crewplus-0.2.3.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
 crewplus/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 crewplus/services/__init__.py,sha256=MmH2v3N0ZMsuqFNAupkXENjUqvgf5ehQ99H6EzPqLZU,48
 crewplus/services/gemini_chat_model.py,sha256=i9p5KvSJYaHSUBLPKM_bpyGVLWCDQoNeah_WjQVJRXs,26227
@@ -12,5 +12,5 @@ crewplus/utils/schema_document_updater.py,sha256=frvffxn2vbi71fHFPoGb9hq7gH2azmm
 crewplus/vectorstores/milvus/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 crewplus/vectorstores/milvus/milvus_schema_manager.py,sha256=qHMVIM0NS3rLfACb8d3-tQS9hJo6_7_YP8AxVx4t1Cc,9019
 crewplus/vectorstores/milvus/schema_milvus.py,sha256=GhHTtCH5HsIJc3RHa25RXl3aZdkS3Rba5KeuUk_Hi0k,11425
-crewplus/vectorstores/milvus/vdb_service.py,sha256=M5HdLAvjDeVAqIoI48miXaa_OAjW8ixru04DuVX93w8,14714
-crewplus-0.2.2.dist-info/RECORD,,
+crewplus/vectorstores/milvus/vdb_service.py,sha256=J7B6TOZmJl9_K2euJFKJFvSYqvruKbXuYkFiugWnXXs,16657
+crewplus-0.2.3.dist-info/RECORD,,