crewplus 0.2.46__py3-none-any.whl → 0.2.47__py3-none-any.whl

crewplus/services/gemini_chat_model.py

@@ -330,22 +330,22 @@ class GeminiChatModel(BaseChatModel):
 
     def invoke(self, input, config=None, **kwargs):
         """Override invoke to add tracing callbacks automatically."""
-        config = self._tracing_manager.add_callbacks_to_config(config)
+        config = self._tracing_manager.add_sync_callbacks_to_config(config)
         return super().invoke(input, config=config, **kwargs)
 
     async def ainvoke(self, input, config=None, **kwargs):
         """Override ainvoke to add tracing callbacks automatically."""
-        config = self._tracing_manager.add_callbacks_to_config(config)
+        config = self._tracing_manager.add_async_callbacks_to_config(config)
         return await super().ainvoke(input, config=config, **kwargs)
 
     def stream(self, input, config=None, **kwargs):
         """Override stream to add tracing callbacks automatically."""
-        config = self._tracing_manager.add_callbacks_to_config(config)
+        config = self._tracing_manager.add_sync_callbacks_to_config(config)
         return super().stream(input, config=config, **kwargs)
 
     async def astream(self, input, config=None, **kwargs):
         """Override astream to add tracing callbacks automatically."""
-        config = self._tracing_manager.add_callbacks_to_config(config)
+        config = self._tracing_manager.add_async_callbacks_to_config(config)
         # We must call an async generator,
         async for chunk in super().astream(input, config=config, **kwargs):
             yield chunk

crewplus/vectorstores/milvus/vdb_service.py

@@ -41,9 +41,9 @@ class VDBService(object):
         >>> # Initialize with a full settings dictionary
         >>> settings = {
         ...     "embedder": {
-        ...         "provider": "azure-openai",
+        ...         "provider": "azure-openai-embeddings",
         ...         "config": {
-        ...             "model": "text-embedding-3-small",
+        ...             "model": "text-embedding-ada-002",
         ...             "api_version": "2023-05-15",
         ...             "api_key": "YOUR_AZURE_OPENAI_KEY",
         ...             "openai_base_url": "YOUR_AZURE_OPENAI_ENDPOINT",

{crewplus-0.2.46.dist-info → crewplus-0.2.47.dist-info}/METADATA

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

{crewplus-0.2.46.dist-info → crewplus-0.2.47.dist-info}/RECORD

@@ -1,13 +1,13 @@
-crewplus-0.2.46.dist-info/METADATA,sha256=QSaSV8WkRmj_qHybxSCTNlyESu9UNuiYNxeLgolNBAA,5362
-crewplus-0.2.46.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-crewplus-0.2.46.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-crewplus-0.2.46.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
+crewplus-0.2.47.dist-info/METADATA,sha256=ZYdwfWzswjuz_ZN6PqvUrB_JDLRViZ1ou9iC3Ldjfqk,5362
+crewplus-0.2.47.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+crewplus-0.2.47.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+crewplus-0.2.47.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
 crewplus/services/__init__.py,sha256=V1CG8b2NOmRzNgQH7BPl4KVxWSYJH5vfEsW1wVErKNE,375
 crewplus/services/azure_chat_model.py,sha256=iWzJ2GQFSNmwJx-2O5_xKPSB6VVc-7T6bcfFI8_WezA,5521
-crewplus/services/gemini_chat_model.py,sha256=VsOB_st1qRmDkwLXzo-gCShhUsZHpk0V-G-ulQXGN3g,40081
+crewplus/services/gemini_chat_model.py,sha256=DYqz01H2TIHiCDQesSozVfOsMigno6QGwOtIweg7UHk,40103
 crewplus/services/init_services.py,sha256=7oZ1GmesK32EDB_DYnTzW17MEpXjXK41_U_1pmqu_m4,2183
 crewplus/services/model_load_balancer.py,sha256=Q9Gx3GrbKworU-Ytxeqp0ggHSgZ1Q6brtTk-nCl4sak,12095
 crewplus/services/tracing_manager.py,sha256=0KR-F0BKYEMdADANWofFZH9D9jcWDHzDICUE7nDhzJc,6579
@@ -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=2IZT61LVui21Pt5Z3y8YYS2dYcwzkgUKxMq2NA0-lQE,9222
 crewplus/vectorstores/milvus/schema_milvus.py,sha256=DtHP8jHRSpLqt9ixAnJE5R4CId9NLYXxOVqRxPCEyv4,26131
-crewplus/vectorstores/milvus/vdb_service.py,sha256=CaUMLIMeOCm2R4t5EKtAupIddFXQu0NSb8RpTkInGd4,22498
+crewplus/vectorstores/milvus/vdb_service.py,sha256=-cZmlLNAPBz4vBRWNC_rfNZh1VWKbmgYfCYsQFyeoSM,22509
 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.46.dist-info/RECORD,,
+crewplus-0.2.47.dist-info/RECORD,,