crewplus 0.2.50__tar.gz → 0.2.52__tar.gz

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

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

{crewplus-0.2.50 → crewplus-0.2.52}/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.
@@ -115,7 +122,6 @@ class VDBService(object):
         """
         self.logger = logger or logging.getLogger(__name__)
         self.collection_schema = None
-        self.schema_manager = MilvusSchemaManager(client=self.client, async_client=self.aclient)
 
         if settings:
             self.settings = settings
@@ -147,12 +153,17 @@ class VDBService(object):
             self.logger.error(msg)
             raise ValueError(msg)
 
+        self.alias = f"crewplus-vdb-{uuid.uuid4()}"
+        self.connection_args['alias'] = self.alias
+
         self._client = self._initialize_milvus_client(provider)
         self._async_client = self._initialize_async_milvus_client(provider)
 
         self.schema = schema
         self.index_params = self.settings.get("index_params")
         
+        self.schema_manager = MilvusSchemaManager(client=self._client, async_client=self._async_client)
+        
         self.logger.info("VDBService initialized successfully")
 
     def _get_milvus_client_args(self, provider: str) -> dict:
@@ -171,7 +182,8 @@ 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"),
+                "alias": self.connection_args.get("alias")
             }
             return {k: v for k, v in client_args.items() if v is not None}
 

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

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