crewplus 0.2.51__py3-none-any.whl → 0.2.53__py3-none-any.whl

crewplus/vectorstores/milvus/vdb_service.py

@@ -12,6 +12,7 @@ from langchain_openai import AzureOpenAIEmbeddings
 from pymilvus import MilvusClient, AsyncMilvusClient
 import time
 import asyncio
+import uuid
 
 from ...services.init_services import get_model_balancer
 from .schema_milvus import SchemaMilvus, DEFAULT_SCHEMA
@@ -24,6 +25,12 @@ class VDBService(object):
     This service centralizes the configuration and instantiation of the Milvus client
     and provides helper methods to get embedding functions and vector store instances.
 
+    This service generates a unique connection `alias` upon initialization. This `alias`
+    is propagated to all Milvus clients created by this service, including those
+    within `langchain_milvus` instances. This mechanism ensures that a single,
+    shared connection is used for all operations, preventing the creation of
+    multiple redundant connections and improving resource efficiency.
+
     Args:
         settings (dict, optional): A dictionary containing configuration for the vector store
                          and embedding models.
@@ -146,6 +153,14 @@ class VDBService(object):
             self.logger.error(msg)
             raise ValueError(msg)
 
+        # 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)
 
@@ -172,12 +187,15 @@ class VDBService(object):
                 "uri": uri,
                 "user": self.connection_args.get("user"),
                 "password": self.connection_args.get("password"),
-                "db_name": self.connection_args.get("db_name")
+                "db_name": self.connection_args.get("db_name"),
             }
             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.")
@@ -187,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
@@ -205,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.51.dist-info → crewplus-0.2.53.dist-info}/METADATA

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

{crewplus-0.2.51.dist-info → crewplus-0.2.53.dist-info}/RECORD

@@ -1,7 +1,7 @@
-crewplus-0.2.51.dist-info/METADATA,sha256=YWQyvTpALzQ1U499VEujqYeas-mC8dyfVSao_eWklf4,5459
-crewplus-0.2.51.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-crewplus-0.2.51.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-crewplus-0.2.51.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
+crewplus-0.2.53.dist-info/METADATA,sha256=Rr4VgoWnm7cdsr11euRhV6bth9s-2eP0y9K9NIrGSzk,5459
+crewplus-0.2.53.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+crewplus-0.2.53.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+crewplus-0.2.53.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
 crewplus/__init__.py,sha256=m46HkZL1Y4toD619NL47Sn2Qe084WFFSFD7e6VoYKZc,284
 crewplus/callbacks/__init__.py,sha256=YG7ieeb91qEjp1zF0-inEN7mjZ7yT_D2yzdWFT8Z1Ws,63
 crewplus/callbacks/async_langfuse_handler.py,sha256=A4uFeLpvOUdc58M7sZoE65_C1V98u0QCvx5jUquM0pM,7006
@@ -17,9 +17,9 @@ crewplus/utils/schema_document_updater.py,sha256=frvffxn2vbi71fHFPoGb9hq7gH2azmm
 crewplus/vectorstores/milvus/__init__.py,sha256=OeYv2rdyG7tcREIjBJPyt2TbE54NvyeRoWMe7LwopRE,245
 crewplus/vectorstores/milvus/milvus_schema_manager.py,sha256=-QRav-hzu-XWeJ_yKUMolal_EyMUspSg-nvh5sqlrlQ,11442
 crewplus/vectorstores/milvus/schema_milvus.py,sha256=wwNpfqsKS0xeozZES40IvB0iNwUtpCall_7Hkg0dL1g,27223
-crewplus/vectorstores/milvus/vdb_service.py,sha256=7nXpX8Gq6v0vI-zWPdbg1twwlvQ16gY4m7zBtzzfzEU,31347
+crewplus/vectorstores/milvus/vdb_service.py,sha256=xlMHhf47O5isQnsPMj8rHaDHMkXcTYh-y5BChla09Tw,32538
 docs/GeminiChatModel.md,sha256=zZYyl6RmjZTUsKxxMiC9O4yV70MC4TD-IGUmWhIDBKA,8677
 docs/ModelLoadBalancer.md,sha256=aGHES1dcXPz4c7Y8kB5-vsCNJjriH2SWmjBkSGoYKiI,4398
 docs/VDBService.md,sha256=Dw286Rrf_fsi13jyD3Bo4Sy7nZ_G7tYm7d8MZ2j9hxk,9375
 docs/index.md,sha256=3tlc15uR8lzFNM5WjdoZLw0Y9o1P1gwgbEnOdIBspqc,1643
-crewplus-0.2.51.dist-info/RECORD,,
+crewplus-0.2.53.dist-info/RECORD,,