crewplus 0.2.60__py3-none-any.whl → 0.2.62__py3-none-any.whl

crewplus/vectorstores/milvus/vdb_service.py

@@ -454,29 +454,17 @@ class VDBService(object):
             self.logger.error("get_vector_store called with no collection_name.")
             raise ValueError("collection_name must be provided.")
 
-        check_existence = True
-        # Check for a cached instance and validate its connection before returning.
+        # Check for a cached instance. If found, return it immediately.
         if collection_name in self._instances:
-            instance = self._instances[collection_name]
-            is_connected, collection_exists = self._is_good_connection(instance, collection_name)
-            
-            if is_connected and collection_exists:
-                self.logger.info(f"Returning existing vector store instance for collection: {collection_name}")
-                return instance
-            
-            self.logger.warning(f"Cached instance for '{collection_name}' is invalid. Removing it from cache.")
-            del self._instances[collection_name]
-            
-            if is_connected and not collection_exists:
-                # We know the collection doesn't exist, so no need to check again.
-                check_existence = False
+            self.logger.info(f"Returning existing vector store instance for collection: {collection_name}")
+            return self._instances[collection_name]
 
         self.logger.info(f"Creating new vector store instance for collection: {collection_name}")
         if embeddings is None:
             embeddings = self.get_embeddings()
 
         # Ensure the collection exists before proceeding.
-        self._ensure_collection_exists(collection_name, embeddings, check_existence=check_existence)
+        self._ensure_collection_exists(collection_name, embeddings, check_existence=True)
 
         # 1. Validate the embedding function before proceeding.
         try:
@@ -545,26 +533,16 @@ class VDBService(object):
             self.logger.error("aget_vector_store called with no collection_name.")
             raise ValueError("collection_name must be provided.")
 
-        check_existence = True
+        # Check for a cached instance. If found, return it immediately.
         if collection_name in self._async_instances:
-            instance = self._async_instances[collection_name]
-            is_connected, collection_exists = await self._ais_good_connection(instance, collection_name)
-            
-            if is_connected and collection_exists:
-                self.logger.info(f"Returning existing async vector store instance for collection: {collection_name}")
-                return instance
-            
-            self.logger.warning(f"Cached async instance for '{collection_name}' is invalid. Removing it from cache.")
-            del self._async_instances[collection_name]
-            
-            if is_connected and not collection_exists:
-                check_existence = False
+            self.logger.info(f"Returning existing async vector store instance for collection: {collection_name}")
+            return self._async_instances[collection_name]
 
         self.logger.info(f"Creating new async vector store instance for collection: {collection_name}")
         if embeddings is None:
             embeddings = self.get_embeddings()
 
-        # await self._aensure_collection_exists(collection_name, embeddings, check_existence=check_existence)
+        await self._aensure_collection_exists(collection_name, embeddings, check_existence=True)
 
         try:
             self.logger.info(f"Testing embedding function for collection '{collection_name}'...")
@@ -599,7 +577,8 @@ class VDBService(object):
         # After successful synchronous initialization, we hot-swap the alias on the
         # ASYNCHRONOUS client to ensure future async operations use the correct connection.
         self.logger.info(f"Swapping to async alias for instance of collection {collection_name}")
-        await self._get_or_create_async_client()
+        # DO NOT get the async client here, get it outside this function
+        #await self._get_or_create_async_client()
         vdb.aclient._using = self.async_alias
 
         self._async_instances[collection_name] = vdb

{crewplus-0.2.60.dist-info → crewplus-0.2.62.dist-info}/METADATA

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

{crewplus-0.2.60.dist-info → crewplus-0.2.62.dist-info}/RECORD

@@ -1,7 +1,7 @@
-crewplus-0.2.60.dist-info/METADATA,sha256=PBmO4kKCatXJ6RnPqFdLqyHtncR3HfDzCgnyly9kcsc,5424
-crewplus-0.2.60.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-crewplus-0.2.60.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-crewplus-0.2.60.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
+crewplus-0.2.62.dist-info/METADATA,sha256=VYAqDSFvN5iRr63wA0IH3I-sRON-PJbP7JoQ58Fzbd8,5424
+crewplus-0.2.62.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+crewplus-0.2.62.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+crewplus-0.2.62.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=qlmFWO_ZRXIeCGZGSEAlS2ciHhTPTH1Dg2ovovA1ZEY,38792
+crewplus/vectorstores/milvus/vdb_service.py,sha256=G8skLhBuAQ33eDVb6_qrFhDyC5zd-0L2hPZenwJCKgE,37749
 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.60.dist-info/RECORD,,
+crewplus-0.2.62.dist-info/RECORD,,