nvidia-nat-crewai 1.3.0a20251001__py3-none-any.whl → 1.3.0a20251004__py3-none-any.whl

nat/plugins/crewai/llm.py

@@ -23,6 +23,7 @@ from nat.data_models.llm import LLMBaseConfig
 from nat.data_models.retry_mixin import RetryMixin
 from nat.data_models.thinking_mixin import ThinkingMixin
 from nat.llm.azure_openai_llm import AzureOpenAIModelConfig
+from nat.llm.litellm_llm import LiteLlmModelConfig
 from nat.llm.nim_llm import NIMModelConfig
 from nat.llm.openai_llm import OpenAIModelConfig
 from nat.llm.utils.thinking import BaseThinkingInjector
@@ -126,3 +127,13 @@ async def openai_crewai(llm_config: OpenAIModelConfig, _builder: Builder):
     client = LLM(**llm_config.model_dump(exclude={"type", "thinking"}, by_alias=True, exclude_none=True))
 
     yield _patch_llm_based_on_config(client, llm_config)
+
+
+@register_llm_client(config_type=LiteLlmModelConfig, wrapper_type=LLMFrameworkEnum.CREWAI)
+async def litellm_crewai(llm_config: LiteLlmModelConfig, _builder: Builder):
+
+    from crewai import LLM
+
+    client = LLM(**llm_config.model_dump(exclude={"type", "thinking"}, by_alias=True, exclude_none=True))
+
+    yield _patch_llm_based_on_config(client, llm_config)

{nvidia_nat_crewai-1.3.0a20251001.dist-info → nvidia_nat_crewai-1.3.0a20251004.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-crewai
-Version: 1.3.0a20251001
+Version: 1.3.0a20251004
 Summary: Subpackage for CrewAI integration in NeMo Agent toolkit
 Keywords: ai,rag,agents
 Classifier: Programming Language :: Python
@@ -9,7 +9,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat[litellm]==v1.3.0a20251001
+Requires-Dist: nvidia-nat[litellm]==v1.3.0a20251004
 Requires-Dist: crewai~=0.193.2
 
 <!--

{nvidia_nat_crewai-1.3.0a20251001.dist-info → nvidia_nat_crewai-1.3.0a20251004.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=VKtRTCn4X5ReVD8F1mNCr4FYmcaE1vIcC3sKjvjmAXc,5079
+nat/plugins/crewai/llm.py,sha256=T57f7hZ5Aa4MUdc6PJLfU6TkF_dO87DWh2KvyAsCt7g,5493
 nat/plugins/crewai/register.py,sha256=_R3bhGmz___696_NwyIcpw3koMBiWqIFoWEFJ0VAgXs,831
 nat/plugins/crewai/tool_wrapper.py,sha256=BNKEPQQCLKtXNzGDAKBLCdmGJXe9lBOVI1hObha8hoI,1569
-nvidia_nat_crewai-1.3.0a20251001.dist-info/METADATA,sha256=4UpVNylqxQK_2xmoQUm4u0vsH3fiooFj-JA5RWxbzhc,1616
-nvidia_nat_crewai-1.3.0a20251001.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_crewai-1.3.0a20251001.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
-nvidia_nat_crewai-1.3.0a20251001.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_crewai-1.3.0a20251001.dist-info/RECORD,,
+nvidia_nat_crewai-1.3.0a20251004.dist-info/METADATA,sha256=2CnkBmdoBXudBrTyTlvAhF_e9Ng-vzd-otLKKVbIdq4,1616
+nvidia_nat_crewai-1.3.0a20251004.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_crewai-1.3.0a20251004.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
+nvidia_nat_crewai-1.3.0a20251004.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_crewai-1.3.0a20251004.dist-info/RECORD,,