aiqtoolkit-crewai 1.2.0a20250707__py3-none-any.whl → 1.2.0a20250730__py3-none-any.whl

aiq/plugins/crewai/llm.py

@@ -18,8 +18,10 @@ import os
 from aiq.builder.builder import Builder
 from aiq.builder.framework_enum import LLMFrameworkEnum
 from aiq.cli.register_workflow import register_llm_client
+from aiq.data_models.retry_mixin import RetryMixin
 from aiq.llm.nim_llm import NIMModelConfig
 from aiq.llm.openai_llm import OpenAIModelConfig
+from aiq.utils.exception_handlers.automatic_retries import patch_with_retry
 
 
 @register_llm_client(config_type=NIMModelConfig, wrapper_type=LLMFrameworkEnum.CREWAI)
@@ -45,7 +47,16 @@ async def nim_crewai(llm_config: NIMModelConfig, builder: Builder):
                 # Transfer the key to the correct environment variable for LiteLLM
                 os.environ["NVIDIA_NIM_API_KEY"] = nvidai_api_key
 
-    yield LLM(**config_obj)
+    client = LLM(**config_obj)
+
+    if isinstance(llm_config, RetryMixin):
+
+        client = patch_with_retry(client,
+                                  retries=llm_config.num_retries,
+                                  retry_codes=llm_config.retry_on_status_codes,
+                                  retry_on_messages=llm_config.retry_on_errors)
+
+    yield client
 
 
 @register_llm_client(config_type=OpenAIModelConfig, wrapper_type=LLMFrameworkEnum.CREWAI)
@@ -57,4 +68,13 @@ async def openai_crewai(llm_config: OpenAIModelConfig, builder: Builder):
         **llm_config.model_dump(exclude={"type"}, by_alias=True),
     }
 
-    yield LLM(**config_obj)
+    client = LLM(**config_obj)
+
+    if isinstance(llm_config, RetryMixin):
+
+        client = patch_with_retry(client,
+                                  retries=llm_config.num_retries,
+                                  retry_codes=llm_config.retry_on_status_codes,
+                                  retry_on_messages=llm_config.retry_on_errors)
+
+    yield client

{aiqtoolkit_crewai-1.2.0a20250707.dist-info → aiqtoolkit_crewai-1.2.0a20250730.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit-crewai
-Version: 1.2.0a20250707
+Version: 1.2.0a20250730
 Summary: Subpackage for CrewAI integration in AIQtoolkit
 Keywords: ai,rag,agents
 Classifier: Programming Language :: Python
 Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: aiqtoolkit==v1.2.0a20250707
+Requires-Dist: aiqtoolkit==v1.2.0a20250730
 Requires-Dist: crewai~=0.95.0
 
 <!--

{aiqtoolkit_crewai-1.2.0a20250707.dist-info → aiqtoolkit_crewai-1.2.0a20250730.dist-info}/RECORD

@@ -1,11 +1,11 @@
 aiq/meta/pypi.md,sha256=mAJKaBmH4UyD5wCe3_BfPGv3ihPTg8BNfS3zu2Z4-70,1083
 aiq/plugins/crewai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiq/plugins/crewai/crewai_callback_handler.py,sha256=mFNWmtDsOUVsa6LKw7BBjBl7lsytAEaYKZfPgl9yY1o,8396
-aiq/plugins/crewai/llm.py,sha256=a9M3FTvd0L1W2XGTf7hnUoCvaD-CXSp5ClNBao53QOg,2205
+aiq/plugins/crewai/llm.py,sha256=FsNt5TFYNpPU_FxB0dW0Cq_ZkviwJOmvivSYcVPtPvg,3000
 aiq/plugins/crewai/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
 aiq/plugins/crewai/tool_wrapper.py,sha256=LuL5VHHxOXf-CBd5E6kA6OsUz2N_aGin_r5KF59srL0,1569
-aiqtoolkit_crewai-1.2.0a20250707.dist-info/METADATA,sha256=ECNYAPLWYy7GwCR4PTNIwVYCuHoPWDfLH1jNmkeLrqI,1421
-aiqtoolkit_crewai-1.2.0a20250707.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiqtoolkit_crewai-1.2.0a20250707.dist-info/entry_points.txt,sha256=gY-k52LFDNWYEcSYiFs78LRKWeGijnGCkwcOroG4T3E,58
-aiqtoolkit_crewai-1.2.0a20250707.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit_crewai-1.2.0a20250707.dist-info/RECORD,,
+aiqtoolkit_crewai-1.2.0a20250730.dist-info/METADATA,sha256=7wSJikxGkTHto4UyLLg1RuMbWganWxY8fXUrgmuWPAA,1421
+aiqtoolkit_crewai-1.2.0a20250730.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqtoolkit_crewai-1.2.0a20250730.dist-info/entry_points.txt,sha256=gY-k52LFDNWYEcSYiFs78LRKWeGijnGCkwcOroG4T3E,58
+aiqtoolkit_crewai-1.2.0a20250730.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+aiqtoolkit_crewai-1.2.0a20250730.dist-info/RECORD,,