nvidia-nat-crewai 1.3.0a20250906__py3-none-any.whl → 1.3.0a20250909__py3-none-any.whl

nat/plugins/crewai/llm.py

@@ -66,15 +66,6 @@ async def azure_openai_crewai(llm_config: AzureOpenAIModelConfig, _builder: Buil
 
     # https://docs.crewai.com/en/concepts/llms#azure
 
-    config_obj = {
-        **llm_config.model_dump(exclude={
-            "type",
-            "api_key",
-            "azure_endpoint",
-            "azure_deployment",
-        }, by_alias=True),
-    }
-
     api_key = llm_config.api_key or os.environ.get("AZURE_OPENAI_API_KEY") or os.environ.get("AZURE_API_KEY")
     if api_key is None:
         raise ValueError("Azure API key is not set")
@@ -89,9 +80,20 @@ async def azure_openai_crewai(llm_config: AzureOpenAIModelConfig, _builder: Buil
     model = llm_config.azure_deployment or os.environ.get("AZURE_MODEL_DEPLOYMENT")
     if model is None:
         raise ValueError("Azure model deployment is not set")
-    config_obj["model"] = model
 
-    client = LLM(**config_obj)
+    client = LLM(
+        **llm_config.model_dump(
+            exclude={
+                "type",
+                "api_key",
+                "azure_endpoint",
+                "azure_deployment",
+                "thinking",
+            },
+            by_alias=True,
+        ),
+        model=model,
+    )
 
     yield _patch_llm_based_on_config(client, llm_config)
 
@@ -101,18 +103,16 @@ async def nim_crewai(llm_config: NIMModelConfig, _builder: Builder):
 
     from crewai import LLM
 
-    config_obj = {
-        **llm_config.model_dump(exclude={"type"}, by_alias=True),
-        "model": f"nvidia_nim/{llm_config.model_name}",
-    }
-
     # Because CrewAI uses a different environment variable for the API key, we need to set it here manually
-    if config_obj.get("api_key") is None and "NVIDIA_NIM_API_KEY" not in os.environ:
+    if llm_config.api_key is None and "NVIDIA_NIM_API_KEY" not in os.environ:
         nvidia_api_key = os.getenv("NVIDIA_API_KEY")
         if nvidia_api_key is not None:
             os.environ["NVIDIA_NIM_API_KEY"] = nvidia_api_key
 
-    client = LLM(**config_obj)
+    client = LLM(
+        **llm_config.model_dump(exclude={"type", "model_name", "thinking"}, by_alias=True),
+        model=f"nvidia_nim/{llm_config.model_name}",
+    )
 
     yield _patch_llm_based_on_config(client, llm_config)
 
@@ -122,10 +122,6 @@ async def openai_crewai(llm_config: OpenAIModelConfig, _builder: Builder):
 
     from crewai import LLM
 
-    config_obj = {
-        **llm_config.model_dump(exclude={"type"}, by_alias=True),
-    }
-
-    client = LLM(**config_obj)
+    client = LLM(**llm_config.model_dump(exclude={"type", "thinking"}, by_alias=True))
 
     yield _patch_llm_based_on_config(client, llm_config)

{nvidia_nat_crewai-1.3.0a20250906.dist-info → nvidia_nat_crewai-1.3.0a20250909.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-crewai
-Version: 1.3.0a20250906
+Version: 1.3.0a20250909
 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==v1.3.0a20250906
+Requires-Dist: nvidia-nat==v1.3.0a20250909
 Requires-Dist: crewai~=0.95.0
 
 <!--

{nvidia_nat_crewai-1.3.0a20250906.dist-info → nvidia_nat_crewai-1.3.0a20250909.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=Snz67QawrXGd7SmeG7jC9CKeZ19fAiAmjuht9kIXfQc,5024
+nat/plugins/crewai/llm.py,sha256=5-73k7G0kcztcFPEgyQs_cVzV_Qz8ve9uRrjG5AM28c,5010
 nat/plugins/crewai/register.py,sha256=_R3bhGmz___696_NwyIcpw3koMBiWqIFoWEFJ0VAgXs,831
 nat/plugins/crewai/tool_wrapper.py,sha256=BNKEPQQCLKtXNzGDAKBLCdmGJXe9lBOVI1hObha8hoI,1569
-nvidia_nat_crewai-1.3.0a20250906.dist-info/METADATA,sha256=cHgcFH7qeTSyRO6AkPo0qOqn4kypWZARhEGmj1zS-Dk,1606
-nvidia_nat_crewai-1.3.0a20250906.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_crewai-1.3.0a20250906.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
-nvidia_nat_crewai-1.3.0a20250906.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_crewai-1.3.0a20250906.dist-info/RECORD,,
+nvidia_nat_crewai-1.3.0a20250909.dist-info/METADATA,sha256=2t73T_buu4Yjqb-q8I5JKD4T5Fzjdo0qjnbeh5VqDZY,1606
+nvidia_nat_crewai-1.3.0a20250909.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_crewai-1.3.0a20250909.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
+nvidia_nat_crewai-1.3.0a20250909.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_crewai-1.3.0a20250909.dist-info/RECORD,,