nvidia-nat-semantic-kernel 1.2.0rc5__py3-none-any.whl → 1.2.0rc6__py3-none-any.whl

{aiq → nat}/meta/pypi.md

@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "NeMo Agent toolkit banner image")
+![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent toolkit banner image")
 
 # NVIDIA NeMo Agent Toolkit Subpackage
 This is a subpackage for Semantic-Kernel integration in NeMo Agent toolkit.

{aiq → nat}/plugins/semantic_kernel/llm.py

@@ -13,12 +13,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.cli.register_workflow import register_llm_client
-from aiq.data_models.retry_mixin import RetryMixin
-from aiq.llm.openai_llm import OpenAIModelConfig
-from aiq.utils.exception_handlers.automatic_retries import patch_with_retry
+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.openai_llm import OpenAIModelConfig
+from nat.utils.exception_handlers.automatic_retries import patch_with_retry
 
 
 @register_llm_client(config_type=OpenAIModelConfig, wrapper_type=LLMFrameworkEnum.SEMANTIC_KERNEL)

{aiq → nat}/plugins/semantic_kernel/tool_wrapper.py

@@ -24,10 +24,10 @@ from typing import get_origin
 
 from pydantic import BaseModel
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.builder.function import Function
-from aiq.cli.register_workflow import register_tool_wrapper
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.builder.function import Function
+from nat.cli.register_workflow import register_tool_wrapper
 
 logger = logging.getLogger(__name__)
 
@@ -75,37 +75,37 @@ def semantic_kernel_tool_wrapper(name: str, fn: Function, builder: Builder):
         async for item in fn.acall_stream(*args, **kwargs):
             yield item
 
-    def aiq_kernel_function(
+    def nat_kernel_function(
         func: Callable[..., object] | None = None,
-        aiq_function: Function | None = None,
+        nat_function: Function | None = None,
         name: str | None = None,
         description: str | None = None,
     ) -> Callable[..., Any]:
         """
         Modified version of Semantic Kernel's kernel_function decorator.
 
-        Uses `aiq` Function properties instead of doing type inference on the function's inner
+        Uses `nat` Function properties instead of doing type inference on the function's inner
         """
 
         def decorator(func: Callable[..., object]) -> Callable[..., object]:
             """The actual decorator function."""
             setattr(func, "__kernel_function__", True)
-            setattr(func, "__kernel_function_description__", description or aiq_function.description)
-            setattr(func, "__kernel_function_name__", name or aiq_function.config.type)
+            setattr(func, "__kernel_function_description__", description or nat_function.description)
+            setattr(func, "__kernel_function_name__", name or nat_function.config.type)
 
             # Always defer to single output schema, if present, for now
             # No need to check streaming output is present given one of the two is always present
-            has_single = aiq_function.has_single_output
-            has_streaming = aiq_function.has_streaming_output
-            output_schema = aiq_function.single_output_schema if has_single else aiq_function.streaming_output_schema
-            setattr(func, "__kernel_function_streaming__", not aiq_function.has_single_output if has_single else True)
+            has_single = nat_function.has_single_output
+            has_streaming = nat_function.has_streaming_output
+            output_schema = nat_function.single_output_schema if has_single else nat_function.streaming_output_schema
+            setattr(func, "__kernel_function_streaming__", not nat_function.has_single_output if has_single else True)
 
             if has_single and has_streaming:
                 logger.warning("Function has both single and streaming output schemas. "
                                "Defaulting to single output schema.")
 
             input_annotations = []
-            for arg_name, annotation in aiq_function.input_schema.model_fields.items():
+            for arg_name, annotation in nat_function.input_schema.model_fields.items():
                 type_obj = resolve_type(annotation.annotation)
                 include_in_choices = True
                 if isinstance(type_obj, type) and (issubclass(type_obj, BaseModel) or is_dataclass(type_obj)):
@@ -156,8 +156,8 @@ def semantic_kernel_tool_wrapper(name: str, fn: Function, builder: Builder):
         return decorator
 
     if fn.has_streaming_output and not fn.has_single_output:
-        kernel_func = aiq_kernel_function(func=callable_astream, aiq_function=fn, name=name, description=fn.description)
+        kernel_func = nat_kernel_function(func=callable_astream, nat_function=fn, name=name, description=fn.description)
     else:
-        kernel_func = aiq_kernel_function(func=callable_ainvoke, aiq_function=fn, name=name, description=fn.description)
+        kernel_func = nat_kernel_function(func=callable_ainvoke, nat_function=fn, name=name, description=fn.description)
 
     return {name: kernel_func}

{nvidia_nat_semantic_kernel-1.2.0rc5.dist-info → nvidia_nat_semantic_kernel-1.2.0rc6.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-semantic-kernel
-Version: 1.2.0rc5
-Summary: Subpackage for Semantic-Kernel integration in AIQtoolkit
+Version: 1.2.0rc6
+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~=1.2
+Requires-Dist: nvidia-nat==v1.2.0-rc6
 Requires-Dist: semantic-kernel~=1.24.0
 
 <!--
@@ -26,7 +26,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "NeMo Agent toolkit banner image")
+![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent toolkit banner image")
 
 # NVIDIA NeMo Agent Toolkit Subpackage
 This is a subpackage for Semantic-Kernel integration in NeMo Agent toolkit.

nvidia_nat_semantic_kernel-1.2.0rc6.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=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.2.0rc6.dist-info/METADATA,sha256=8YGoAQnvVPfkSM-4n_EeiVZqGha_7N4eKdA8weiUv7c,1478
+nvidia_nat_semantic_kernel-1.2.0rc6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_semantic_kernel-1.2.0rc6.dist-info/entry_points.txt,sha256=0jCtQBAn5Ohs9XoVCF34WvNCV33OwAsH8bjFzgw_ByM,76
+nvidia_nat_semantic_kernel-1.2.0rc6.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_semantic_kernel-1.2.0rc6.dist-info/RECORD,,

nvidia_nat_semantic_kernel-1.2.0rc6.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[nat.components]
+nat_semantic_kernel = nat.plugins.semantic_kernel.register

nvidia_nat_semantic_kernel-1.2.0rc6.dist-info/top_level.txt

@@ -0,0 +1 @@
+nat

nvidia_nat_semantic_kernel-1.2.0rc5.dist-info/RECORD

@@ -1,10 +0,0 @@
-aiq/meta/pypi.md,sha256=Ki8bvrAo3d2iW7q6vTszp2AY2fapDP--el1wKuOtWcQ,1127
-aiq/plugins/semantic_kernel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiq/plugins/semantic_kernel/llm.py,sha256=l13R5Wv50iQ8OQ2juAFCoPoll6ZtIbojTnWjyGjepGI,1744
-aiq/plugins/semantic_kernel/register.py,sha256=RKXyuaXy4ftA5IL2RrCofIBjpie_-2lP9YZoHAiyPU0,863
-aiq/plugins/semantic_kernel/tool_wrapper.py,sha256=miogE5cp3o5H-KxaKp1dMqqJZnV-UMsYb1PQ4ox-xKY,7175
-nvidia_nat_semantic_kernel-1.2.0rc5.dist-info/METADATA,sha256=nc7IguzX3ahlmWrPjnadRw_UUP11w8QoJyexsM7Q8sM,1474
-nvidia_nat_semantic_kernel-1.2.0rc5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_semantic_kernel-1.2.0rc5.dist-info/entry_points.txt,sha256=qGCOsf9tkWvc5EUonXrUI96cDrTJ3qoueyy3DgClK3M,76
-nvidia_nat_semantic_kernel-1.2.0rc5.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-nvidia_nat_semantic_kernel-1.2.0rc5.dist-info/RECORD,,

nvidia_nat_semantic_kernel-1.2.0rc5.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[aiq.components]
-aiq_semantic_kernel = aiq.plugins.semantic_kernel.register

nvidia_nat_semantic_kernel-1.2.0rc5.dist-info/top_level.txt

@@ -1 +0,0 @@
-aiq