nvidia-nat-semantic-kernel 1.3a20250819__py3-none-any.whl → 1.3.0.dev2__py3-none-any.whl

nat/plugins/semantic_kernel/llm.py

@@ -17,20 +17,38 @@ 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.openai_llm import OpenAIModelConfig
 from nat.utils.exception_handlers.automatic_retries import patch_with_retry
 
 
+@register_llm_client(config_type=AzureOpenAIModelConfig, wrapper_type=LLMFrameworkEnum.SEMANTIC_KERNEL)
+async def azure_openai_semantic_kernel(llm_config: AzureOpenAIModelConfig, _builder: Builder):
+
+    from semantic_kernel.connectors.ai.open_ai import AzureChatCompletion
+
+    llm = AzureChatCompletion(
+        api_key=llm_config.api_key,
+        api_version=llm_config.api_version,
+        endpoint=llm_config.azure_endpoint,
+        deployment_name=llm_config.azure_deployment,
+    )
+
+    if isinstance(llm_config, RetryMixin):
+        llm = patch_with_retry(llm,
+                               retries=llm_config.num_retries,
+                               retry_codes=llm_config.retry_on_status_codes,
+                               retry_on_messages=llm_config.retry_on_errors)
+
+    yield llm
+
+
 @register_llm_client(config_type=OpenAIModelConfig, wrapper_type=LLMFrameworkEnum.SEMANTIC_KERNEL)
-async def openai_semantic_kernel(llm_config: OpenAIModelConfig, builder: Builder):
+async def openai_semantic_kernel(llm_config: OpenAIModelConfig, _builder: Builder):
 
     from semantic_kernel.connectors.ai.open_ai import OpenAIChatCompletion
 
-    config_obj = {
-        **llm_config.model_dump(exclude={"type"}, by_alias=True),
-    }
-
-    llm = OpenAIChatCompletion(ai_model_id=config_obj.get("model"))
+    llm = OpenAIChatCompletion(ai_model_id=llm_config.model_name)
 
     if isinstance(llm_config, RetryMixin):
         llm = patch_with_retry(llm,

{nvidia_nat_semantic_kernel-1.3a20250819.dist-info → nvidia_nat_semantic_kernel-1.3.0.dev2.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-semantic-kernel
-Version: 1.3a20250819
+Version: 1.3.0.dev2
 Summary: Subpackage for Semantic-Kernel 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.3a20250819
+Requires-Dist: nvidia-nat==v1.3.0-dev2
 Requires-Dist: semantic-kernel~=1.24.0
 
 <!--

nvidia_nat_semantic_kernel-1.3.0.dev2.dist-info/RECORD

@@ -0,0 +1,10 @@
+nat/meta/pypi.md,sha256=rFmwVds3akmoz0TE1SOjCjCUbB6SWfaRex_Vi5OfUAk,1116
+nat/plugins/semantic_kernel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nat/plugins/semantic_kernel/llm.py,sha256=5HNry_FGjgGMWL1yjF85zLyNsU2I0Fd1Y2FIAdeHJEQ,2514
+nat/plugins/semantic_kernel/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
+nat/plugins/semantic_kernel/tool_wrapper.py,sha256=PEtzcYhZ73ftsg80Jz9yWTWQtks1y7PDZdS89seFb3I,7175
+nvidia_nat_semantic_kernel-1.3.0.dev2.dist-info/METADATA,sha256=IYpz6ttyVDq8Uf9tJ3ESiRnI9-k9Mu06-OnxdzhDmz8,1481
+nvidia_nat_semantic_kernel-1.3.0.dev2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_semantic_kernel-1.3.0.dev2.dist-info/entry_points.txt,sha256=0jCtQBAn5Ohs9XoVCF34WvNCV33OwAsH8bjFzgw_ByM,76
+nvidia_nat_semantic_kernel-1.3.0.dev2.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_semantic_kernel-1.3.0.dev2.dist-info/RECORD,,

nvidia_nat_semantic_kernel-1.3a20250819.dist-info/RECORD

@@ -1,10 +0,0 @@
-nat/meta/pypi.md,sha256=rFmwVds3akmoz0TE1SOjCjCUbB6SWfaRex_Vi5OfUAk,1116
-nat/plugins/semantic_kernel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nat/plugins/semantic_kernel/llm.py,sha256=C2ISvuCkz_YN0yOgyf1QJPtbJmX_1kgLpQumwpKfcOI,1744
-nat/plugins/semantic_kernel/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
-nat/plugins/semantic_kernel/tool_wrapper.py,sha256=PEtzcYhZ73ftsg80Jz9yWTWQtks1y7PDZdS89seFb3I,7175
-nvidia_nat_semantic_kernel-1.3a20250819.dist-info/METADATA,sha256=xpBRgLqdAeYExPSyS_5wOuKbMxkfinlNN_Ncs1q9cbM,1485
-nvidia_nat_semantic_kernel-1.3a20250819.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_semantic_kernel-1.3a20250819.dist-info/entry_points.txt,sha256=0jCtQBAn5Ohs9XoVCF34WvNCV33OwAsH8bjFzgw_ByM,76
-nvidia_nat_semantic_kernel-1.3a20250819.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_semantic_kernel-1.3a20250819.dist-info/RECORD,,