crewplus 0.2.61__tar.gz → 0.2.63__tar.gz

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

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

{crewplus-0.2.61 → crewplus-0.2.63}/crewplus/vectorstores/milvus/vdb_service.py

@@ -454,41 +454,29 @@ 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)
-
-        # 1. Validate the embedding function before proceeding.
-        try:
-            self.logger.debug(f"Testing embedding function for collection '{collection_name}'...")
-            embeddings.embed_query("validation_test_string")
-            self.logger.debug("Embedding function is valid.")
-        except Exception as e:
-            self.logger.error(
-                f"The provided embedding function is invalid and failed with error: {e}. "
-                f"Cannot create a vector store for collection '{collection_name}'."
-            )
-            raise RuntimeError(f"Invalid embedding function provided.") from e
+        self._ensure_collection_exists(collection_name, embeddings, check_existence=True)
+
+        # # 1. Validate the embedding function before proceeding.
+        # try:
+        #     self.logger.info(f"Testing embedding function for collection '{collection_name}'...")
+        #     embeddings.embed_query("validation_test_string")
+        #     self.logger.info("Embedding function is valid.")
+        # except Exception as e:
+        #     self.logger.error(
+        #         f"The provided embedding function is invalid and failed with error: {e}. "
+        #         f"Cannot create a vector store for collection '{collection_name}'."
+        #     )
+        #     raise RuntimeError(f"Invalid embedding function provided.") from e
 
         # If embeddings are valid, proceed to create the Milvus instance.
         index_params = self.index_params or {
@@ -545,37 +533,27 @@ 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)
-
-        try:
-            self.logger.info(f"Testing embedding function for collection '{collection_name}'...")
-            await embeddings.aembed_query("validation_test_string")
-            self.logger.info("Embedding function is valid.")
-        except Exception as e:
-            self.logger.error(
-                f"The provided embedding function is invalid and failed with error: {e}. "
-                f"Cannot create a vector store for collection '{collection_name}'."
-            )
-            raise RuntimeError(f"Invalid embedding function provided.") from e
+        await self._aensure_collection_exists(collection_name, embeddings, check_existence=True)
+
+        # try:
+        #     self.logger.info(f"Testing embedding function for collection '{collection_name}'...")
+        #     await embeddings.aembed_query("validation_test_string")
+        #     self.logger.info("Embedding function is valid.")
+        # except Exception as e:
+        #     self.logger.error(
+        #         f"The provided embedding function is invalid and failed with error: {e}. "
+        #         f"Cannot create a vector store for collection '{collection_name}'."
+        #     )
+        #     raise RuntimeError(f"Invalid embedding function provided.") from e
 
         index_params = self.index_params or {
             "metric_type": metric_type,

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

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