Route plugin imports through NAT public API

Author: bbednarski9

src/nat/plugins/redis/memory.py

@@ -13,13 +13,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from nat.builder.builder import Builder
-from nat.cli.register_workflow import register_memory
-from nat.data_models.common import OptionalSecretStr, get_secret_value
-from nat.data_models.component_ref import EmbedderRef
-from nat.data_models.memory import MemoryBaseConfig
 from pydantic import Field
 
+from nvidia_nat_redis._nat_api import (
+    Builder,
+    EmbedderRef,
+    LLMFrameworkEnum,
+    MemoryBaseConfig,
+    OptionalSecretStr,
+    get_secret_value,
+    register_memory,
+)
+
 
 class RedisMemoryClientConfig(MemoryBaseConfig, name="redis_memory"):
     host: str = Field(default="localhost", description="Redis server host")
@@ -35,8 +40,6 @@ class RedisMemoryClientConfig(MemoryBaseConfig, name="redis_memory"):
 @register_memory(config_type=RedisMemoryClientConfig)
 async def redis_memory_client(config: RedisMemoryClientConfig, builder: Builder):
 
-    from nat.builder.framework_enum import LLMFrameworkEnum
-
     import redis.asyncio as redis
 
     from .redis_editor import RedisEditor

src/nat/plugins/redis/object_store.py

@@ -13,12 +13,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from nat.builder.builder import Builder
-from nat.cli.register_workflow import register_object_store
-from nat.data_models.common import OptionalSecretStr
-from nat.data_models.object_store import ObjectStoreBaseConfig
 from pydantic import Field, field_validator
 
+from nvidia_nat_redis._nat_api import Builder, ObjectStoreBaseConfig, OptionalSecretStr, register_object_store
+
 
 class RedisObjectStoreClientConfig(ObjectStoreBaseConfig, name="redis"):
     """

src/nat/plugins/redis/redis_editor.py

@@ -16,12 +16,12 @@
 import logging
 import secrets
 
-from nat.memory.interfaces import MemoryEditor
-from nat.memory.models import MemoryItem
 from redisvl.index import AsyncSearchIndex
 from redisvl.query import VectorQuery
 from redisvl.query.filter import Tag
 
+from nvidia_nat_redis._nat_api import MemoryEditor, MemoryItem
+
 logger = logging.getLogger(__name__)
 
 

src/nat/plugins/redis/redis_object_store.py

@@ -15,12 +15,8 @@
 
 import logging
 
-from nat.data_models.object_store import KeyAlreadyExistsError, NoSuchKeyError
-from nat.object_store.interfaces import ObjectStore
-from nat.object_store.models import ObjectStoreItem
-from nat.utils.type_utils import override
-
 import redis.asyncio as redis
+from nvidia_nat_redis._nat_api import KeyAlreadyExistsError, NoSuchKeyError, ObjectStore, ObjectStoreItem
 
 logger = logging.getLogger(__name__)
 
@@ -88,7 +84,6 @@ async def __aexit__(self, exc_type, exc_value, traceback) -> None:
     def _make_key(self, key: str) -> str:
         return f"nat/object_store/{self._bucket_name}/{key}"
 
-    @override
     async def put_object(self, key: str, item: ObjectStoreItem):
 
         if not self._client:
@@ -103,7 +98,6 @@ async def put_object(self, key: str, item: ObjectStoreItem):
                 key=key, additional_message=f"Redis bucket {self._bucket_name} already has key {key}"
             )
 
-    @override
     async def upsert_object(self, key: str, item: ObjectStoreItem):
 
         if not self._client:
@@ -113,7 +107,6 @@ async def upsert_object(self, key: str, item: ObjectStoreItem):
         item_json = item.model_dump_json()
         await self._client.set(full_key, item_json, ex=self._ttl)
 
-    @override
     async def get_object(self, key: str) -> ObjectStoreItem:
 
         if not self._client:
@@ -127,7 +120,6 @@ async def get_object(self, key: str) -> ObjectStoreItem:
             )
         return ObjectStoreItem.model_validate_json(data)
 
-    @override
     async def delete_object(self, key: str):
 
         if not self._client:

src/nvidia_nat_redis/_nat_api.py

@@ -0,0 +1,82 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026, Redis
+# SPDX-License-Identifier: Apache-2.0
+
+"""NAT public plugin API imports with compatibility for older supported NAT releases.
+
+NAT's third-party plugin guidance asks runtime plugin code to import plugin-authoring
+symbols from ``nat.plugin_api``. The current supported range also includes older NAT
+releases that do not ship that facade, so this module is the only compatibility
+fallback to implementation imports.
+"""
+
+from __future__ import annotations
+
+try:
+    from nat.plugin_api import (
+        Builder,
+        EmbedderRef,
+        Function,
+        FunctionBaseConfig,
+        FunctionInfo,
+        FunctionRef,
+        KeyAlreadyExistsError,
+        LLMFrameworkEnum,
+        MemoryBaseConfig,
+        MemoryEditor,
+        MemoryItem,
+        MemoryRef,
+        NoSuchKeyError,
+        ObjectStore,
+        ObjectStoreBaseConfig,
+        ObjectStoreItem,
+        OptionalSecretStr,
+        get_secret_value,
+        register_function,
+        register_memory,
+        register_object_store,
+    )
+
+    USING_PLUGIN_API = True
+except ModuleNotFoundError:  # pragma: no cover - exercised only on older NAT releases
+    from nat.builder.builder import Builder
+    from nat.builder.framework_enum import LLMFrameworkEnum
+    from nat.builder.function import Function
+    from nat.builder.function_info import FunctionInfo
+    from nat.cli.register_workflow import register_function, register_memory, register_object_store
+    from nat.data_models.common import OptionalSecretStr, get_secret_value
+    from nat.data_models.component_ref import EmbedderRef, FunctionRef, MemoryRef
+    from nat.data_models.function import FunctionBaseConfig
+    from nat.data_models.memory import MemoryBaseConfig
+    from nat.data_models.object_store import KeyAlreadyExistsError, NoSuchKeyError, ObjectStoreBaseConfig
+    from nat.memory.interfaces import MemoryEditor
+    from nat.memory.models import MemoryItem
+    from nat.object_store.interfaces import ObjectStore
+    from nat.object_store.models import ObjectStoreItem
+
+    USING_PLUGIN_API = False
+
+
+__all__ = [
+    "Builder",
+    "EmbedderRef",
+    "Function",
+    "FunctionBaseConfig",
+    "FunctionInfo",
+    "FunctionRef",
+    "KeyAlreadyExistsError",
+    "LLMFrameworkEnum",
+    "MemoryBaseConfig",
+    "MemoryEditor",
+    "MemoryItem",
+    "MemoryRef",
+    "NoSuchKeyError",
+    "ObjectStore",
+    "ObjectStoreBaseConfig",
+    "ObjectStoreItem",
+    "OptionalSecretStr",
+    "USING_PLUGIN_API",
+    "get_secret_value",
+    "register_function",
+    "register_memory",
+    "register_object_store",
+]

src/nvidia_nat_redis/redis_agent_memory/auto_memory/config.py

@@ -7,10 +7,10 @@
 
 from typing import TYPE_CHECKING, Any, Literal
 
-from nat.data_models.component_ref import FunctionRef, MemoryRef
-from nat.data_models.function import FunctionBaseConfig
 from pydantic import BaseModel, Field, PositiveInt, field_validator
 
+from nvidia_nat_redis._nat_api import FunctionBaseConfig, FunctionRef, MemoryRef
+
 from .._text import normalize_optional_string
 
 if TYPE_CHECKING:

src/nvidia_nat_redis/redis_agent_memory/auto_memory/register.py

@@ -8,12 +8,14 @@
 from collections.abc import AsyncGenerator
 
 from agent_memory_client import create_memory_client
-from nat.builder.builder import Builder
-from nat.builder.function_info import FunctionInfo
-from nat.cli.register_workflow import register_function
+
+# Chat request models and retry patching are runtime helpers not exported by
+# nat.plugin_api in the currently supported NAT versions.
 from nat.data_models.api_server import ChatRequest, ChatRequestOrMessage, ChatResponse
 from nat.utils.exception_handlers.automatic_retries import patch_with_retry
 
+from nvidia_nat_redis._nat_api import Builder, FunctionInfo, register_function
+
 from ..memory import RedisAgentMemoryBackendConfig
 from .config import RedisAgentMemoryAutoMemoryConfig
 from .service import RedisAgentMemoryAutoMemoryService

src/nvidia_nat_redis/redis_agent_memory/auto_memory/service.py

@@ -9,8 +9,9 @@
 from datetime import UTC, datetime
 from typing import Any
 
+# Context, chat request models, and type conversion are runtime helpers not
+# exported by nat.plugin_api in the currently supported NAT versions.
 from nat.builder.context import Context
-from nat.builder.function import Function
 from nat.data_models.api_server import (
     ChatRequest,
     ChatRequestOrMessage,
@@ -20,6 +21,8 @@
 )
 from nat.utils.type_converter import GlobalTypeConverter
 
+from nvidia_nat_redis._nat_api import Function
+
 from .._text import message_content_to_text, normalize_optional_string
 from .config import RedisAgentMemoryAutoMemoryConfig
 

src/nvidia_nat_redis/redis_agent_memory/editor.py

@@ -11,9 +11,11 @@
 from typing import TYPE_CHECKING, Any
 
 from agent_memory_client import MemoryAPIClient
+
+# Runtime request context is not exported by nat.plugin_api.
 from nat.builder.context import Context
-from nat.memory.interfaces import MemoryEditor
-from nat.memory.models import MemoryItem
+
+from nvidia_nat_redis._nat_api import MemoryEditor, MemoryItem
 
 from ._text import message_content_to_text
 

src/nvidia_nat_redis/redis_agent_memory/memory.py

@@ -8,14 +8,15 @@
 from collections.abc import AsyncGenerator
 
 from agent_memory_client import create_memory_client
-from nat.builder.builder import Builder
-from nat.cli.register_workflow import register_memory
-from nat.data_models.memory import MemoryBaseConfig
+
+# RetryMixin and patch_with_retry are retry implementation helpers not exported
+# by nat.plugin_api in the currently supported NAT versions.
 from nat.data_models.retry_mixin import RetryMixin
-from nat.memory.interfaces import MemoryEditor
 from nat.utils.exception_handlers.automatic_retries import patch_with_retry
 from pydantic import Field
 
+from nvidia_nat_redis._nat_api import Builder, MemoryBaseConfig, MemoryEditor, register_memory
+
 from .editor import RedisAgentMemoryEditor