crewplus 0.2.61__tar.gz → 0.2.62__tar.gz

{crewplus-0.2.61 → crewplus-0.2.62}/PKG-INFO

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

{crewplus-0.2.61 → crewplus-0.2.62}/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}'...")

{crewplus-0.2.61 → crewplus-0.2.62}/pyproject.toml

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