nvidia-nat 1.3.0a20250910__py3-none-any.whl → 1.3.0a20250922__py3-none-any.whl

nat/utils/exception_handlers/automatic_retries.py

@@ -118,6 +118,7 @@ def _retry_decorator(
     retry_codes: Sequence[CodePattern] | None = None,
     retry_on_messages: Sequence[str] | None = None,
     deepcopy: bool = False,
+    instance_context_aware: bool = False,
 ) -> Callable[[Callable[..., T]], Callable[..., T]]:
     """
     Build a decorator that retries with exponential back-off *iff*:
@@ -130,69 +131,122 @@ def _retry_decorator(
     deepcopy:
         If True, each retry receives deep‑copied *args and **kwargs* to avoid
         mutating shared state between attempts.
+
+    instance_context_aware:
+        If True, the decorator will check for a retry context flag on the first
+        argument (assumed to be 'self'). If the flag is set, retries are skipped
+        to prevent retry storms in nested method calls.
     """
 
     def decorate(fn: Callable[..., T]) -> Callable[..., T]:
         use_deepcopy = deepcopy
+        use_context_aware = instance_context_aware
 
-        async def _call_with_retry_async(*args, **kw) -> T:
-            delay = base_delay
-            for attempt in range(retries):
-                call_args = copy.deepcopy(args) if use_deepcopy else args
-                call_kwargs = copy.deepcopy(kw) if use_deepcopy else kw
+        class _RetryContext:
+            """Context manager for instance-level retry gating."""
+
+            __slots__ = ("_obj", "_enabled", "_active")
+
+            def __init__(self, args: tuple[Any, ...]):
+                self._obj = args[0] if (use_context_aware and args) else None
+                self._enabled = bool(self._obj)
+                self._active = False
+
+            def __enter__(self):
+                if not self._enabled:
+                    return False
                 try:
-                    return await fn(*call_args, **call_kwargs)
-                except retry_on as exc:
-                    if (not _want_retry(exc, code_patterns=retry_codes, msg_substrings=retry_on_messages)
-                            or attempt == retries - 1):
-                        raise
-                    await asyncio.sleep(delay)
-                    delay *= backoff
+                    # If already in retry context, signal caller to skip retries
+                    if getattr(self._obj, "_in_retry_context", False):
+                        return True
+                    object.__setattr__(self._obj, "_in_retry_context", True)
+                    self._active = True
+                    return False
+                except Exception:
+                    # If we cannot set the attribute, behave as if context isn't enabled
+                    self._enabled = False
+                    return False
+
+            def __exit__(self, _exc_type, _exc, _tb):
+                if self._enabled and self._active:
+                    try:
+                        object.__setattr__(self._obj, "_in_retry_context", False)
+                    except Exception:
+                        pass
+
+        async def _call_with_retry_async(*args, **kw) -> T:
+            with _RetryContext(args) as already_in_context:
+                if already_in_context:
+                    return await fn(*args, **kw)
+                delay = base_delay
+                for attempt in range(retries):
+                    call_args = copy.deepcopy(args) if use_deepcopy else args
+                    call_kwargs = copy.deepcopy(kw) if use_deepcopy else kw
+                    try:
+                        return await fn(*call_args, **call_kwargs)
+                    except retry_on as exc:
+                        if (not _want_retry(exc, code_patterns=retry_codes, msg_substrings=retry_on_messages)
+                                or attempt == retries - 1):
+                            raise
+                        await asyncio.sleep(delay)
+                        delay *= backoff
 
         async def _agen_with_retry(*args, **kw):
-            delay = base_delay
-            for attempt in range(retries):
-                call_args = copy.deepcopy(args) if use_deepcopy else args
-                call_kwargs = copy.deepcopy(kw) if use_deepcopy else kw
-                try:
-                    async for item in fn(*call_args, **call_kwargs):
+            with _RetryContext(args) as already_in_context:
+                if already_in_context:
+                    async for item in fn(*args, **kw):
                         yield item
                     return
-                except retry_on as exc:
-                    if (not _want_retry(exc, code_patterns=retry_codes, msg_substrings=retry_on_messages)
-                            or attempt == retries - 1):
-                        raise
-                    await asyncio.sleep(delay)
-                    delay *= backoff
+                delay = base_delay
+                for attempt in range(retries):
+                    call_args = copy.deepcopy(args) if use_deepcopy else args
+                    call_kwargs = copy.deepcopy(kw) if use_deepcopy else kw
+                    try:
+                        async for item in fn(*call_args, **call_kwargs):
+                            yield item
+                        return
+                    except retry_on as exc:
+                        if (not _want_retry(exc, code_patterns=retry_codes, msg_substrings=retry_on_messages)
+                                or attempt == retries - 1):
+                            raise
+                        await asyncio.sleep(delay)
+                        delay *= backoff
 
         def _gen_with_retry(*args, **kw) -> Iterable[Any]:
-            delay = base_delay
-            for attempt in range(retries):
-                call_args = copy.deepcopy(args) if use_deepcopy else args
-                call_kwargs = copy.deepcopy(kw) if use_deepcopy else kw
-                try:
-                    yield from fn(*call_args, **call_kwargs)
+            with _RetryContext(args) as already_in_context:
+                if already_in_context:
+                    yield from fn(*args, **kw)
                     return
-                except retry_on as exc:
-                    if (not _want_retry(exc, code_patterns=retry_codes, msg_substrings=retry_on_messages)
-                            or attempt == retries - 1):
-                        raise
-                    time.sleep(delay)
-                    delay *= backoff
+                delay = base_delay
+                for attempt in range(retries):
+                    call_args = copy.deepcopy(args) if use_deepcopy else args
+                    call_kwargs = copy.deepcopy(kw) if use_deepcopy else kw
+                    try:
+                        yield from fn(*call_args, **call_kwargs)
+                        return
+                    except retry_on as exc:
+                        if (not _want_retry(exc, code_patterns=retry_codes, msg_substrings=retry_on_messages)
+                                or attempt == retries - 1):
+                            raise
+                        time.sleep(delay)
+                        delay *= backoff
 
         def _sync_with_retry(*args, **kw) -> T:
-            delay = base_delay
-            for attempt in range(retries):
-                call_args = copy.deepcopy(args) if use_deepcopy else args
-                call_kwargs = copy.deepcopy(kw) if use_deepcopy else kw
-                try:
-                    return fn(*call_args, **call_kwargs)
-                except retry_on as exc:
-                    if (not _want_retry(exc, code_patterns=retry_codes, msg_substrings=retry_on_messages)
-                            or attempt == retries - 1):
-                        raise
-                    time.sleep(delay)
-                    delay *= backoff
+            with _RetryContext(args) as already_in_context:
+                if already_in_context:
+                    return fn(*args, **kw)
+                delay = base_delay
+                for attempt in range(retries):
+                    call_args = copy.deepcopy(args) if use_deepcopy else args
+                    call_kwargs = copy.deepcopy(kw) if use_deepcopy else kw
+                    try:
+                        return fn(*call_args, **call_kwargs)
+                    except retry_on as exc:
+                        if (not _want_retry(exc, code_patterns=retry_codes, msg_substrings=retry_on_messages)
+                                or attempt == retries - 1):
+                            raise
+                        time.sleep(delay)
+                        delay *= backoff
 
         # Decide which wrapper to return
         if inspect.iscoroutinefunction(fn):
@@ -245,6 +299,7 @@ def patch_with_retry(
         retry_codes=retry_codes,
         retry_on_messages=retry_on_messages,
         deepcopy=deepcopy,
+        instance_context_aware=True,  # Prevent retry storms
     )
 
     # Choose attribute source: the *class* to avoid triggering __getattr__

nat/utils/log_levels.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import logging
+
+# Define log level choices
+LOG_LEVELS = {
+    'DEBUG': logging.DEBUG,
+    'INFO': logging.INFO,
+    'WARNING': logging.WARNING,
+    'ERROR': logging.ERROR,
+    'CRITICAL': logging.CRITICAL
+}

nat/utils/type_utils.py

@@ -469,6 +469,10 @@ class DecomposedType:
             # Handle generic types that can't use issubclass
             pass
 
+        # Check direct equality (works for both regular and generic types)
+        if source_type == target_type:
+            return True
+
         # Check if source outputs list[T] and target expects T
         source_decomposed = DecomposedType(source_type)
         if source_decomposed.origin is list and source_decomposed.args:

{nvidia_nat-1.3.0a20250910.dist-info → nvidia_nat-1.3.0a20250922.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat
-Version: 1.3.0a20250910
+Version: 1.3.0a20250922
 Summary: NVIDIA NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -225,11 +225,13 @@ Requires-Dist: httpx~=0.27
 Requires-Dist: jinja2~=3.1
 Requires-Dist: jsonpath-ng~=1.7
 Requires-Dist: mcp~=1.13
+Requires-Dist: nest-asyncio~=1.6
 Requires-Dist: networkx~=3.4
 Requires-Dist: numpy~=1.26; python_version < "3.12"
 Requires-Dist: numpy~=2.3; python_version >= "3.12"
 Requires-Dist: openinference-semantic-conventions~=0.1.14
 Requires-Dist: openpyxl~=3.1
+Requires-Dist: optuna~=4.4.0
 Requires-Dist: pkce==1.0.3
 Requires-Dist: pkginfo~=1.12
 Requires-Dist: platformdirs~=4.3
@@ -296,7 +298,9 @@ Requires-Dist: nat_por_to_jiratickets; extra == "examples"
 Requires-Dist: nat_profiler_agent; extra == "examples"
 Requires-Dist: nat_redact_pii; extra == "examples"
 Requires-Dist: nat_retail_sales_agent; extra == "examples"
+Requires-Dist: nat_router_agent; extra == "examples"
 Requires-Dist: nat_semantic_kernel_demo; extra == "examples"
+Requires-Dist: nat_sequential_executor; extra == "examples"
 Requires-Dist: nat_simple_auth; extra == "examples"
 Requires-Dist: nat_simple_web_query; extra == "examples"
 Requires-Dist: nat_simple_web_query_eval; extra == "examples"
@@ -311,6 +315,11 @@ Requires-Dist: nat_swe_bench; extra == "examples"
 Requires-Dist: nat_user_report; extra == "examples"
 Provides-Extra: gunicorn
 Requires-Dist: gunicorn~=23.0; extra == "gunicorn"
+Provides-Extra: async-endpoints
+Requires-Dist: aiosqlite~=0.21; extra == "async-endpoints"
+Requires-Dist: dask==2023.6; extra == "async-endpoints"
+Requires-Dist: distributed==2023.6; extra == "async-endpoints"
+Requires-Dist: sqlalchemy[asyncio]~=2.0; extra == "async-endpoints"
 Dynamic: license-file
 
 <!--