crewplus 0.2.56__py3-none-any.whl → 0.2.58__py3-none-any.whl

crewplus/vectorstores/milvus/vdb_service.py

@@ -16,7 +16,7 @@ import uuid
 
 from ...services.init_services import get_model_balancer
 from .schema_milvus import SchemaMilvus, DEFAULT_SCHEMA
-from .milvus_schema_manager import MilvusSchemaManager
+#from .milvus_schema_manager import MilvusSchemaManager
 
 class VDBService(object):
     """
@@ -93,7 +93,7 @@ class VDBService(object):
         >>> assert vector_store is same_vector_store
     """
     _client: MilvusClient
-    _async_client: AsyncMilvusClient
+    _async_client: Optional[AsyncMilvusClient] = None
     _instances: Dict[str, Milvus] = {}
     _async_instances: Dict[str, Milvus] = {}
 
@@ -154,6 +154,8 @@ class VDBService(object):
             self.logger.error(msg)
             raise ValueError(msg)
 
+        self._provider = provider # Store provider for lazy initialization
+
         # 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()}"
@@ -163,12 +165,13 @@ class VDBService(object):
         self.connection_args['alias'] = self.sync_alias
 
         self._client = self._initialize_milvus_client(provider)
-        self._async_client = self._initialize_async_milvus_client(provider)
+        # lazy-initialize async milvus
+        # 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.schema_manager = MilvusSchemaManager(client=self._client, async_client=self._async_client)
         
         self.logger.info("VDBService initialized successfully")
 
@@ -240,20 +243,28 @@ class VDBService(object):
 
     def get_vector_client(self) -> MilvusClient:
         """
-        Returns the active MilvusClient instance.
+        Returns the active MilvusClient instance, initializing it if necessary.
 
         Returns:
             MilvusClient: The initialized client for interacting with the vector database.
         """
+        if self._client is None:
+            self.logger.debug("Initializing synchronous MilvusClient...")
+            self._client = self._initialize_milvus_client(self._provider)
+
         return self._client
 
     def get_async_vector_client(self) -> AsyncMilvusClient:
         """
-        Returns the active AsyncMilvusClient instance.
+        Returns the active AsyncMilvusClient instance, initializing it if necessary.
 
         Returns:
             AsyncMilvusClient: The initialized async client for interacting with the vector database.
         """
+        if self._async_client is None:
+            self.logger.debug("Initializing asynchronous AsyncMilvusClient...")
+            self._async_client = self._initialize_async_milvus_client(self._provider)
+        
         return self._async_client
 
     def get_vector_field(self, collection_name: str) -> str:
@@ -374,6 +385,9 @@ class VDBService(object):
                     index_params=self.index_params
                 )
 
+                #ensure using async connection alias
+                schema_milvus.aclient._using = self.async_alias
+
                 schema_to_use = self.schema or DEFAULT_SCHEMA
                 if not self.schema:
                     self.logger.warning(f"No schema provided for VDBService. Using DEFAULT_SCHEMA for collection '{collection_name}'.")

{crewplus-0.2.56.dist-info → crewplus-0.2.58.dist-info}/METADATA

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

{crewplus-0.2.56.dist-info → crewplus-0.2.58.dist-info}/RECORD

@@ -1,7 +1,7 @@
-crewplus-0.2.56.dist-info/METADATA,sha256=ccwA_6ip1D6ZEviEIiTtshF4wqO54TS5THUBV7wDSJY,5459
-crewplus-0.2.56.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-crewplus-0.2.56.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-crewplus-0.2.56.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
+crewplus-0.2.58.dist-info/METADATA,sha256=bcVORIdlBj4keo_14dBUc9N6UDu9ch2p8JmBXSvFqIk,5459
+crewplus-0.2.58.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+crewplus-0.2.58.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+crewplus-0.2.58.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=AomZTlwKaw1r_9xgWPVJGBJlJjMlwb-Jn1CWBSsS2Eg,34387
+crewplus/vectorstores/milvus/vdb_service.py,sha256=sVDThSfsSq_aISJToU4yWY0nRfm9cOyIpIj_0YeJCXg,35111
 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.56.dist-info/RECORD,,
+crewplus-0.2.58.dist-info/RECORD,,