nvidia-nat-crewai 1.3.dev0__py3-none-any.whl → 1.3.0.dev2__py3-none-any.whl

nat/plugins/crewai/llm.py

@@ -19,13 +19,59 @@ from nat.builder.builder import Builder
 from nat.builder.framework_enum import LLMFrameworkEnum
 from nat.cli.register_workflow import register_llm_client
 from nat.data_models.retry_mixin import RetryMixin
+from nat.llm.azure_openai_llm import AzureOpenAIModelConfig
 from nat.llm.nim_llm import NIMModelConfig
 from nat.llm.openai_llm import OpenAIModelConfig
 from nat.utils.exception_handlers.automatic_retries import patch_with_retry
 
 
+@register_llm_client(config_type=AzureOpenAIModelConfig, wrapper_type=LLMFrameworkEnum.CREWAI)
+async def azure_openai_crewai(llm_config: AzureOpenAIModelConfig, _builder: Builder):
+
+    from crewai import LLM
+
+    # 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 = config_obj.get("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")
+    os.environ["AZURE_API_KEY"] = api_key
+    api_base = (config_obj.get("azure_endpoint") or os.environ.get("AZURE_OPENAI_ENDPOINT")
+                or os.environ.get("AZURE_API_BASE"))
+    if api_base is None:
+        raise ValueError("Azure endpoint is not set")
+    os.environ["AZURE_API_BASE"] = api_base
+
+    os.environ["AZURE_API_VERSION"] = llm_config.api_version
+    model = config_obj.get("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)
+
+    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=NIMModelConfig, wrapper_type=LLMFrameworkEnum.CREWAI)
-async def nim_crewai(llm_config: NIMModelConfig, builder: Builder):
+async def nim_crewai(llm_config: NIMModelConfig, _builder: Builder):
 
     from crewai import LLM
 
@@ -60,7 +106,7 @@ async def nim_crewai(llm_config: NIMModelConfig, builder: Builder):
 
 
 @register_llm_client(config_type=OpenAIModelConfig, wrapper_type=LLMFrameworkEnum.CREWAI)
-async def openai_crewai(llm_config: OpenAIModelConfig, builder: Builder):
+async def openai_crewai(llm_config: OpenAIModelConfig, _builder: Builder):
 
     from crewai import LLM
 

{nvidia_nat_crewai-1.3.dev0.dist-info → nvidia_nat_crewai-1.3.0.dev2.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-crewai
-Version: 1.3.dev0
+Version: 1.3.0.dev2
 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-dev
+Requires-Dist: nvidia-nat==v1.3.0-dev2
 Requires-Dist: crewai~=0.95.0
 
 <!--

nvidia_nat_crewai-1.3.0.dev2.dist-info/RECORD

@@ -0,0 +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=LDOctDQC9qdba1SVGoVkceCOSYuDj_mnl3HCuq2nIuQ,8382
+nat/plugins/crewai/llm.py,sha256=ErsUXsHpQ-chWqFhYjp-OhARAnCl-HdV_tkrrTZzIOU,4665
+nat/plugins/crewai/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
+nat/plugins/crewai/tool_wrapper.py,sha256=BNKEPQQCLKtXNzGDAKBLCdmGJXe9lBOVI1hObha8hoI,1569
+nvidia_nat_crewai-1.3.0.dev2.dist-info/METADATA,sha256=WnIqCrZBOK3zpDqC789LaKaFJ1RO3L0XQDXWxTcMjT0,1445
+nvidia_nat_crewai-1.3.0.dev2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_crewai-1.3.0.dev2.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
+nvidia_nat_crewai-1.3.0.dev2.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_crewai-1.3.0.dev2.dist-info/RECORD,,

nvidia_nat_crewai-1.3.dev0.dist-info/RECORD

@@ -1,11 +0,0 @@
-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=LDOctDQC9qdba1SVGoVkceCOSYuDj_mnl3HCuq2nIuQ,8382
-nat/plugins/crewai/llm.py,sha256=_ykUOFJCc9czWPGQZIfQerfqH_cGZfjZ09U297I4xuo,3000
-nat/plugins/crewai/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
-nat/plugins/crewai/tool_wrapper.py,sha256=BNKEPQQCLKtXNzGDAKBLCdmGJXe9lBOVI1hObha8hoI,1569
-nvidia_nat_crewai-1.3.dev0.dist-info/METADATA,sha256=BtcRz8pDXVcK_giQGeY0N8xmWPs9k5mP79aFwgIEYE0,1440
-nvidia_nat_crewai-1.3.dev0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_crewai-1.3.dev0.dist-info/entry_points.txt,sha256=YF5PUdQGr_OUDXB4TykElHJTsKT8yKkuE0bMX5n_RXs,58
-nvidia_nat_crewai-1.3.dev0.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_crewai-1.3.dev0.dist-info/RECORD,,