crewplus 0.2.52__tar.gz → 0.2.53__tar.gz

{crewplus-0.2.52 → crewplus-0.2.53}/PKG-INFO

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

{crewplus-0.2.52 → crewplus-0.2.53}/crewplus/vectorstores/milvus/vdb_service.py

@@ -153,8 +153,13 @@ class VDBService(object):
             self.logger.error(msg)
             raise ValueError(msg)
 
-        self.alias = f"crewplus-vdb-{uuid.uuid4()}"
-        self.connection_args['alias'] = self.alias
+        # Create separate aliases for sync and async clients to avoid connection handler race conditions.
+        self.sync_alias = f"crewplus-vdb-sync-{uuid.uuid4()}"
+        self.async_alias = f"crewplus-vdb-async-{uuid.uuid4()}"
+
+        # The default alias in connection_args should be the sync one, as langchain_milvus
+        # primarily uses a synchronous client and will pick up this alias.
+        self.connection_args['alias'] = self.sync_alias
 
         self._client = self._initialize_milvus_client(provider)
         self._async_client = self._initialize_async_milvus_client(provider)
@@ -183,12 +188,14 @@ class VDBService(object):
                 "user": self.connection_args.get("user"),
                 "password": self.connection_args.get("password"),
                 "db_name": self.connection_args.get("db_name"),
-                "alias": self.connection_args.get("alias")
             }
             return {k: v for k, v in client_args.items() if v is not None}
 
         elif provider == "zilliz":
-            return self.connection_args
+            # Return a copy without the default alias, as it will be added specifically for sync/async clients.
+            zilliz_args = self.connection_args.copy()
+            zilliz_args.pop('alias', None)
+            return zilliz_args
         else:
             self.logger.error(f"Unsupported vector store provider: {provider}")
             raise NotImplementedError(f"Vector store provider '{provider}' is not supported.")
@@ -198,6 +205,7 @@ class VDBService(object):
         Initializes and returns a MilvusClient with a retry mechanism.
         """
         client_args = self._get_milvus_client_args(provider)
+        client_args["alias"] = self.sync_alias
 
         try:
             # First attempt to connect
@@ -216,6 +224,8 @@ class VDBService(object):
         Initializes and returns an AsyncMilvusClient with a retry mechanism.
         """
         client_args = self._get_milvus_client_args(provider)
+        client_args["alias"] = self.async_alias
+        
         try:
             return AsyncMilvusClient(**client_args)
         except Exception as e:

{crewplus-0.2.52 → crewplus-0.2.53}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "crewplus"
-version = "0.2.52"
+version = "0.2.53"
 description = "Base services for CrewPlus AI applications"
 authors = [
     { name = "Tim Liu", email = "tim@opsmateai.com" },