nvidia-nat-crewai 1.3.0a20250901__py3-none-any.whl → 1.3.0a20250904__py3-none-any.whl

nat/plugins/crewai/llm.py

@@ -43,6 +43,12 @@ def _patch_llm_based_on_config(client: ModelType, llm_config: LLMBaseConfig) ->
             new_messages = [{"role": "system", "content": self.system_prompt}] + messages
             return FunctionArgumentWrapper(new_messages, *args, **kwargs)
 
+    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)
+
     if isinstance(llm_config, ThinkingMixin) and llm_config.thinking_system_prompt is not None:
         client = patch_with_thinking(
             client, CrewAIThinkingInjector(
@@ -50,12 +56,6 @@ def _patch_llm_based_on_config(client: ModelType, llm_config: LLMBaseConfig) ->
                 function_names=["call"],
             ))
 
-    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)
-
     return client
 
 

{nvidia_nat_crewai-1.3.0a20250901.dist-info → nvidia_nat_crewai-1.3.0a20250904.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-crewai
-Version: 1.3.0a20250901
+Version: 1.3.0a20250904
 Summary: Subpackage for CrewAI integration in NeMo Agent toolkit
 Keywords: ai,rag,agents
 Classifier: Programming Language :: Python
 Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat==v1.3.0a20250901
+Requires-Dist: nvidia-nat==v1.3.0a20250904
 Requires-Dist: crewai~=0.95.0
 
 <!--

{nvidia_nat_crewai-1.3.0a20250901.dist-info → nvidia_nat_crewai-1.3.0a20250904.dist-info}/RECORD

@@ -1,11 +1,11 @@
 nat/meta/pypi.md,sha256=T68FnThRzDGFf1LR8u-okM-r11-skSnKqSyI6HOktQY,1107
 nat/plugins/crewai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/plugins/crewai/crewai_callback_handler.py,sha256=nBpWWpYQ-yrAG1-6G3kvcYrb0jJQzckd_ftojvGVWj4,8333
-nat/plugins/crewai/llm.py,sha256=DlkLN_fCXS7LUuNku4vvOUrMAgIA_om5LaQsKJOAAPk,5024
+nat/plugins/crewai/llm.py,sha256=Snz67QawrXGd7SmeG7jC9CKeZ19fAiAmjuht9kIXfQc,5024
 nat/plugins/crewai/register.py,sha256=_R3bhGmz___696_NwyIcpw3koMBiWqIFoWEFJ0VAgXs,831
 nat/plugins/crewai/tool_wrapper.py,sha256=BNKEPQQCLKtXNzGDAKBLCdmGJXe9lBOVI1hObha8hoI,1569
-nvidia_nat_crewai-1.3.0a20250901.dist-info/METADATA,sha256=LpfSkWqkcYsUKbaVCcNJFEHFlHquPE-_iXAI5tE5QJE,1453
-nvidia_nat_crewai-1.3.0a20250901.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_crewai-1.3.0a20250901.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
-nvidia_nat_crewai-1.3.0a20250901.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_crewai-1.3.0a20250901.dist-info/RECORD,,
+nvidia_nat_crewai-1.3.0a20250904.dist-info/METADATA,sha256=DaNQo79xGwj9J_55N-S-pZJByIjaNLgx237pKGxu88A,1453
+nvidia_nat_crewai-1.3.0a20250904.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_crewai-1.3.0a20250904.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
+nvidia_nat_crewai-1.3.0a20250904.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_crewai-1.3.0a20250904.dist-info/RECORD,,