aiqtoolkit-crewai 1.2.0.dev0__py3-none-any.whl → 1.2.0rc2__py3-none-any.whl

aiq/meta/pypi.md

@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![NVIDIA Agent Intelligence Toolkit](https://media.githubusercontent.com/media/NVIDIA/AIQToolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ toolkit banner image")
+![NVIDIA Agent Intelligence Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ toolkit banner image")
 
 # NVIDIA Agent Intelligence Toolkit Subpackage
 This is a subpackage for CrewAI integration in AIQ toolkit.

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.0.dev0.dist-info → aiqtoolkit_crewai-1.2.0rc2.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit-crewai
-Version: 1.2.0.dev0
+Version: 1.2.0rc2
 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.0-dev
+Requires-Dist: aiqtoolkit==v1.2.0-rc2
 Requires-Dist: crewai~=0.95.0
 
 <!--
@@ -26,7 +26,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![NVIDIA Agent Intelligence Toolkit](https://media.githubusercontent.com/media/NVIDIA/AIQToolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ toolkit banner image")
+![NVIDIA Agent Intelligence Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ toolkit banner image")
 
 # NVIDIA Agent Intelligence Toolkit Subpackage
 This is a subpackage for CrewAI integration in AIQ toolkit.

aiqtoolkit_crewai-1.2.0rc2.dist-info/RECORD

@@ -0,0 +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=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.0rc2.dist-info/METADATA,sha256=-D1_Oe22e1h_lNtE_gU-qXYcceqTQlI0qbWPxo7xAfs,1410
+aiqtoolkit_crewai-1.2.0rc2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqtoolkit_crewai-1.2.0rc2.dist-info/entry_points.txt,sha256=gY-k52LFDNWYEcSYiFs78LRKWeGijnGCkwcOroG4T3E,58
+aiqtoolkit_crewai-1.2.0rc2.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+aiqtoolkit_crewai-1.2.0rc2.dist-info/RECORD,,

{aiqtoolkit_crewai-1.2.0.dev0.dist-info → aiqtoolkit_crewai-1.2.0rc2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

aiqtoolkit_crewai-1.2.0.dev0.dist-info/RECORD

@@ -1,11 +0,0 @@
-aiq/meta/pypi.md,sha256=nbPN1EIxKvUGXgF4BsGOsw9taUNqexulfoqSF8PiMWA,1075
-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/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
-aiq/plugins/crewai/tool_wrapper.py,sha256=LuL5VHHxOXf-CBd5E6kA6OsUz2N_aGin_r5KF59srL0,1569
-aiqtoolkit_crewai-1.2.0.dev0.dist-info/METADATA,sha256=fkdISa9eA3HhOz_nnX33OUSoTbNMK82R-jHHS3UOTUY,1404
-aiqtoolkit_crewai-1.2.0.dev0.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-aiqtoolkit_crewai-1.2.0.dev0.dist-info/entry_points.txt,sha256=gY-k52LFDNWYEcSYiFs78LRKWeGijnGCkwcOroG4T3E,58
-aiqtoolkit_crewai-1.2.0.dev0.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit_crewai-1.2.0.dev0.dist-info/RECORD,,