crewplus 0.2.5__py3-none-any.whl → 0.2.6__py3-none-any.whl

crewplus/vectorstores/milvus/vdb_service.py

@@ -248,8 +248,8 @@ class VDBService(object):
     def get_vector_store(self, collection_name: str, embeddings: Embeddings = None, metric_type: str = "L2") -> Zilliz:
         """
         Gets a vector store instance, creating it if it doesn't exist for the collection.
-
-        This method caches instances by collection name to avoid re-instantiation.
+        This method validates both the embedding function and the vector store connection
+        before caching the instance to prevent faulty instances from being reused.
 
         Args:
             collection_name (str): The name of the collection in the vector database.
@@ -272,6 +272,19 @@ class VDBService(object):
         if embeddings is None:
             embeddings = self.get_embeddings()
 
+        # 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
+
+        # If embeddings are valid, proceed to create the Zilliz instance.
         index_params = self.index_params or {
             "metric_type": metric_type,
             "index_type": "AUTOINDEX",

{crewplus-0.2.5.dist-info → crewplus-0.2.6.dist-info}/METADATA

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

{crewplus-0.2.5.dist-info → crewplus-0.2.6.dist-info}/RECORD

@@ -1,7 +1,7 @@
-crewplus-0.2.5.dist-info/METADATA,sha256=D9FcMOIX_tv8VbPdxDfA_0WwIjvT5Q3Wkyy6x8OAtpY,4909
-crewplus-0.2.5.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-crewplus-0.2.5.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-crewplus-0.2.5.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
+crewplus-0.2.6.dist-info/METADATA,sha256=EWHY3KjrDXJjYeO-WqNITUVJ_3gzhjMwE6f6trstGD0,4909
+crewplus-0.2.6.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+crewplus-0.2.6.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+crewplus-0.2.6.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
 crewplus/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 crewplus/services/__init__.py,sha256=MmH2v3N0ZMsuqFNAupkXENjUqvgf5ehQ99H6EzPqLZU,48
 crewplus/services/gemini_chat_model.py,sha256=i9p5KvSJYaHSUBLPKM_bpyGVLWCDQoNeah_WjQVJRXs,26227
@@ -12,9 +12,9 @@ crewplus/utils/schema_document_updater.py,sha256=frvffxn2vbi71fHFPoGb9hq7gH2azmm
 crewplus/vectorstores/milvus/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 crewplus/vectorstores/milvus/milvus_schema_manager.py,sha256=qHMVIM0NS3rLfACb8d3-tQS9hJo6_7_YP8AxVx4t1Cc,9019
 crewplus/vectorstores/milvus/schema_milvus.py,sha256=GhHTtCH5HsIJc3RHa25RXl3aZdkS3Rba5KeuUk_Hi0k,11425
-crewplus/vectorstores/milvus/vdb_service.py,sha256=QfoQaB1JBdhX8XY-2iz65i7OqPn3u8Vzyq_w0YIpZpE,16718
+crewplus/vectorstores/milvus/vdb_service.py,sha256=iCiSXcmtpD9tW1WYyJ1P0SRJEFEsUoH3lY65mnwf6R8,17524
 docs/GeminiChatModel.md,sha256=_IQyup3ofAa2HxfSurO1GYUEezTHYYt5Q1khYNVThGM,8040
 docs/ModelLoadBalancer.md,sha256=aGHES1dcXPz4c7Y8kB5-vsCNJjriH2SWmjBkSGoYKiI,4398
-docs/VDBService.md,sha256=vrdYqI5BCLgwOIsEP0NSCQgKDIHXFQ6nj1-9nENZOjQ,9364
+docs/VDBService.md,sha256=Dw286Rrf_fsi13jyD3Bo4Sy7nZ_G7tYm7d8MZ2j9hxk,9375
 docs/index.md,sha256=3tlc15uR8lzFNM5WjdoZLw0Y9o1P1gwgbEnOdIBspqc,1643
-crewplus-0.2.5.dist-info/RECORD,,
+crewplus-0.2.6.dist-info/RECORD,,

docs/VDBService.md

@@ -101,7 +101,7 @@ This requires the `ModelLoadBalancer` to have been initialized, as shown in the
 # The service will call get_model_balancer() internally to get the initialized instance.
 embedding_model = vdb_service.get_embeddings(
     from_model_balancer=True,
-    provider="azure-openai", 
+    provider="azure-openai-embeddings", 
     model_type="embedding-large" # Specify the model type configured in the balancer
 )