digitalkin 1.0.5.dev6__tar.gz → 1.0.5.dev7__tar.gz

{digitalkin-1.0.5.dev6 → digitalkin-1.0.5.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 1.0.5.dev6
+Version: 1.0.5.dev7
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -452,9 +452,8 @@ Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: ag-ui-protocol==0.1.21.dev1787220377
+Requires-Dist: ag-ui-protocol==1.0.0
 Requires-Dist: agentic-mesh-protocol==1.2.0
-Requires-Dist: agno<4,>=3.1.1
 Requires-Dist: anyio>=4.13.0
 Requires-Dist: cel-python>=0.5
 Requires-Dist: grpcio-health-checking==1.82.1
@@ -474,190 +473,136 @@ Requires-Dist: viztracer>=1.1.1; extra == "profiling"
 Requires-Dist: yappi>=1.7.6; extra == "profiling"
 Dynamic: license-file
 
+<!-- --8<-- [start:intro] -->
 # DigitalKin Python SDK
 
-[![CI](https://github.com/DigitalKin-ai/digitalkin/actions/workflows/ci.yml/badge.svg)](https://github.com/DigitalKin-ai/digitalkin/actions/workflows/ci.yml)
-[![PyPI](https://img.shields.io/pypi/v/digitalkin.svg)](https://pypi.org/project/digitalkin/)
-[![Python Version](https://img.shields.io/pypi/pyversions/digitalkin.svg)](https://pypi.org/project/digitalkin/)
-[![License](https://img.shields.io/github/license/DigitalKin-ai/digitalkin)](https://github.com/DigitalKin-ai/digitalkin/blob/main/LICENSE)
+`digitalkin` is the Python SDK for writing modules (tools and archetypes) of the DigitalKin agentic mesh.
+You declare typed Pydantic contracts and trigger handlers; the SDK serves them over gRPC.
+Each module process embeds a gateway that persists outputs to Redis Streams and pushes them to the consumer.
 
-The DigitalKin Python SDK for building and managing modules within the
-DigitalKin agentic mesh. Create custom **Tools** and **Archetypes** that
-communicate over gRPC, register with a service mesh, and scale independently.
+**What it is not**
 
-## Features
+- Not a standalone gateway process: the gateway runs inside each module server.
+- Not a registry server: modules register with an external registry in remote mode.
+- Not a client library: the consumer calls `StartStream` and hosts the dial-back `Stream` endpoint itself.
+- Not the wire contract: protos come from the `agentic-mesh-protocol` package.
 
-- **Async-native gRPC module system** — every module is a gRPC server built on
-  `grpcio` with full async support
-- **Typed module contracts** — Pydantic models for Input, Output, Setup, and
-  Secret schemas with protocol-based trigger dispatch
-- **Module-to-module communication** — tools and archetypes discover each other
-  via the registry and exchange requests over gRPC
-- **Tool resolution** — archetypes dynamically resolve and invoke tool modules
-  at runtime
-- **Admission queue & backpressure** — built-in request admission with
-  configurable concurrency limits
-- **Healthcheck protocols** — automatic ping, services, and status healthcheck
-  triggers registered on every module
-- **Profiling** — optional `[profiling]` extra with asyncio-inspector,
-  pyinstrument, viztracer, and yappi
-- **Batched history writes** — efficient storage writes for conversation history
-- **Redis Streams** — durable message passing, crash recovery, and reconnection
+## Install
 
-## Installation
+`uv add digitalkin` or `pip install digitalkin` (Python 3.10+); add `[agno]` for the Agno integration.
+Extras: `agno` (Agno integration), `performance` (uvloop), `profiling`.
 
-```bash
-# With uv (recommended)
-uv add digitalkin
-
-# With pip
-pip install digitalkin
-```
-
-**Optional extras:**
-
-```bash
-# Async profiling tools
-uv add "digitalkin[profiling]"
-
-# uvloop for faster event loop
-uv add "digitalkin[performance]"
-```
+## Quick start
 
-## Quick Start
-
-### 1. Define your models
+Files: `echo_module.py`, `server.py`, `triggers/__init__.py`, `triggers/echo_trigger.py` (discovery imports `*_trigger.py`).
 
 ```python
+# echo_module.py
+from typing import Any, ClassVar, Literal
 from pydantic import BaseModel
-from digitalkin.models.module.base_types import DataModel, DataTrigger
-
+from digitalkin import ModuleContext, ToolModule
+from digitalkin.models.module import DataModel, DataTrigger, SetupModel
+from digitalkin.utils.package_discover import ModuleDiscoverer
 
-class MessageInput(DataTrigger):
-    protocol: str = "message"
-    content: str
+class EchoIn(DataTrigger):
+    protocol: Literal["message"] = "message"
+    text: str
 
-
-class InputModel(DataModel[MessageInput]):
-    root: MessageInput
-
-
-class MessageOutput(DataTrigger):
-    protocol: str = "message"
+class EchoOut(DataTrigger):
+    protocol: Literal["message"] = "message"
     reply: str
 
+class EchoInput(DataModel[EchoIn]):
+    root: EchoIn
 
-class OutputModel(DataModel[MessageOutput]):
-    root: MessageOutput
-```
-
-### 2. Create a module and trigger
+class EchoOutput(DataModel[EchoOut]):
+    root: EchoOut
 
-```python
-from digitalkin import ArchetypeModule, ModuleContext, TriggerHandler
-from digitalkin.models.module.setup_types import SetupModel
+class EchoSetup(SetupModel):
+    prefix: str = "echo: "
 
+class EchoModule(ToolModule[EchoInput, EchoOutput, EchoSetup, BaseModel]):
+    name = "EchoModule"
+    description = "Echoes the input text."
+    input_format = EchoInput
+    output_format = EchoOutput
+    setup_format = EchoSetup
+    secret_format = BaseModel  # no secrets
+    metadata: ClassVar[dict[str, Any]] = {"module_id": "modules:echo", "version": "0.1.0"}
+    services_config_params: ClassVar[dict[str, Any]] = {"storage": {"config": {}}, "cost": {"config": {}}}
+    triggers_discoverer = ModuleDiscoverer(packages=["triggers"])
 
-class MyArchetype(ArchetypeModule[InputModel, OutputModel, SetupModel, BaseModel]):
-    async def initialize(self, context: ModuleContext, setup_data: SetupModel) -> None:
-        pass
+    async def initialize(self, context: ModuleContext, setup_data: EchoSetup) -> None:
+        pass  # open clients, build agents
 
-    async def cleanup(self, context: ModuleContext) -> None:
+    async def cleanup(self) -> None:
         pass
+```
 
+```python
+# triggers/echo_trigger.py
+from digitalkin import ModuleContext, TriggerHandler
+from echo_module import EchoIn, EchoModule, EchoOut, EchoOutput, EchoSetup
 
-@MyArchetype.register
-class MessageTrigger(TriggerHandler[InputModel, SetupModel, OutputModel]):
+@EchoModule.register
+class EchoTrigger(TriggerHandler):
     protocol = "message"
-    input_format = InputModel
-    output_format = OutputModel
-
-    def __init__(self, context: ModuleContext) -> None:
-        super().__init__(context)
+    input_format = EchoIn
+    output_format = EchoOut
 
-    async def handle(
-        self,
-        input_data: InputModel,
-        setup_data: SetupModel,
-        context: ModuleContext,
-    ) -> None:
-        output = OutputModel(root=MessageOutput(reply=f"Echo: {input_data.root.content}"))
-        await self.send_message(context, output)
+    async def handle(self, input_data: EchoIn, setup_data: EchoSetup, context: ModuleContext) -> None:
+        reply = EchoOut(reply=setup_data.prefix + input_data.text)
+        await context.callbacks.send_message(EchoOutput(root=reply))
 ```
 
-### 3. Run the server
-
 ```python
+# server.py
 import asyncio
 from digitalkin.grpc_servers.module_server import ModuleServer
+from echo_module import EchoModule
 
 async def main() -> None:
-    server = ModuleServer(MyArchetype)
+    server = ModuleServer(EchoModule)
     await server.start_async()
-    await server.await_termination()
+    try:
+        await server.await_termination()
+    finally:
+        await server.stop_async()
 
 asyncio.run(main())
 ```
 
-## Redis Gateway
-
-The embedded gateway enables real-time bidirectional communication between
-modules via Redis Streams, with crash recovery and horizontal scaling.
-
-- **Durable Streaming**: Output persisted to Redis Streams — reconnection via `from_seq`.
-- **Zero-Copy Proto**: Binary proto serialization to Redis — no JSON intermediary.
-- **Horizontal Scaling**: Each module instance embeds its own gateway. Scale by adding replicas behind a load balancer.
-
-
-## Development
-
-### Prerequisites
-
-- Python 3.10+
-- [uv](https://astral.sh/uv) — modern Python package management
-- [Task](https://taskfile.dev/) — task runner
-
-### Setup
-
 ```bash
-git clone --recurse-submodules https://github.com/DigitalKin-ai/digitalkin.git
-cd digitalkin
-
-task setup-dev
-source .venv/bin/activate
+docker compose -f examples/redis_demo/docker-compose.yml up -d     # Redis on localhost:6379
+SERVICE_MODE=local DIGITALKIN_REDIS_URL=redis://localhost:6379/0 python server.py
 ```
 
-### Common Tasks
-
-```bash
-task linter               # Format + lint (ruff) + type check (mypy)
-task check                # Linter + mypy + tests
-task run-tests            # Run pytest via Docker
-task build-package        # Build distribution
-task bump-version -- patch|minor|major
+The server listens on `SERVER_CHANNEL_HOST:SERVER_CHANNEL_PORT` (default `[::]:50055`).
+`SERVICE_MODE=local` (or `-d LOCAL`) uses in-process services; remote mode (default) needs the variables in `.env.minimum`.
+Consumers drive the module with the gateway RPCs described in the gateway protocol page.
 
-task docs-serve           # Serve docs locally (mkdocs)
-task docs-build           # Build docs
-
-task generate-certificates  # Generate mTLS certs for gRPC
-task clean                # Remove build artifacts + __pycache__
-task clean-all            # Above + remove .venv
-```
+<!-- --8<-- [end:intro] -->
 
-### Publishing Process
+## Documentation
 
-1. Update code and commit changes (following conventional branch/commit
-   standard).
-2. Use `task bump-version -- major|minor|patch` to commit the new version.
-3. Use GitHub "Create Release" workflow to publish the new version.
-4. Workflow automatically publishes to Test PyPI and PyPI.
+| Topic | Page |
+| --- | --- |
+| Architecture, data and signal flow | [docs/architecture.md](docs/architecture.md) |
+| Writing modules | [docs/module_authoring.md](docs/module_authoring.md) |
+| Wire contracts | [docs/gateway_protocol.md](docs/gateway_protocol.md), [docs/agui_events.md](docs/agui_events.md) (AG-UI) |
+| Deploy and tune | [docs/operations.md](docs/operations.md) |
+| Environment variables | [docs/environment_variables.md](docs/environment_variables.md) |
+| Testing | [docs/testing.md](docs/testing.md) |
+| Changelog | [docs/changelog/index.md](docs/changelog/index.md) |
+| API reference | [docs/api/api.md](docs/api/api.md) |
 
-## License
-
-This project is licensed under the terms specified in the LICENSE file.
+## Development
 
----
+```bash
+task setup-dev     # venv, dev/docs/tests deps, pre-commit hooks
+task check         # ruff format + lint, mypy, tests
+task run-tests     # full suite in docker compose (needs Redis)
+task docs-serve    # mkdocs serve
+```
 
-For more information, visit our [Documentation](https://digitalkin.com) or
-report issues on our
-[Issues page](https://github.com/DigitalKin-ai/digitalkin/issues).
+License: [CC BY-NC-SA 4.0](LICENSE).

digitalkin-1.0.5.dev7/README.md

@@ -0,0 +1,133 @@
+<!-- --8<-- [start:intro] -->
+# DigitalKin Python SDK
+
+`digitalkin` is the Python SDK for writing modules (tools and archetypes) of the DigitalKin agentic mesh.
+You declare typed Pydantic contracts and trigger handlers; the SDK serves them over gRPC.
+Each module process embeds a gateway that persists outputs to Redis Streams and pushes them to the consumer.
+
+**What it is not**
+
+- Not a standalone gateway process: the gateway runs inside each module server.
+- Not a registry server: modules register with an external registry in remote mode.
+- Not a client library: the consumer calls `StartStream` and hosts the dial-back `Stream` endpoint itself.
+- Not the wire contract: protos come from the `agentic-mesh-protocol` package.
+
+## Install
+
+`uv add digitalkin` or `pip install digitalkin` (Python 3.10+); add `[agno]` for the Agno integration.
+Extras: `agno` (Agno integration), `performance` (uvloop), `profiling`.
+
+## Quick start
+
+Files: `echo_module.py`, `server.py`, `triggers/__init__.py`, `triggers/echo_trigger.py` (discovery imports `*_trigger.py`).
+
+```python
+# echo_module.py
+from typing import Any, ClassVar, Literal
+from pydantic import BaseModel
+from digitalkin import ModuleContext, ToolModule
+from digitalkin.models.module import DataModel, DataTrigger, SetupModel
+from digitalkin.utils.package_discover import ModuleDiscoverer
+
+class EchoIn(DataTrigger):
+    protocol: Literal["message"] = "message"
+    text: str
+
+class EchoOut(DataTrigger):
+    protocol: Literal["message"] = "message"
+    reply: str
+
+class EchoInput(DataModel[EchoIn]):
+    root: EchoIn
+
+class EchoOutput(DataModel[EchoOut]):
+    root: EchoOut
+
+class EchoSetup(SetupModel):
+    prefix: str = "echo: "
+
+class EchoModule(ToolModule[EchoInput, EchoOutput, EchoSetup, BaseModel]):
+    name = "EchoModule"
+    description = "Echoes the input text."
+    input_format = EchoInput
+    output_format = EchoOutput
+    setup_format = EchoSetup
+    secret_format = BaseModel  # no secrets
+    metadata: ClassVar[dict[str, Any]] = {"module_id": "modules:echo", "version": "0.1.0"}
+    services_config_params: ClassVar[dict[str, Any]] = {"storage": {"config": {}}, "cost": {"config": {}}}
+    triggers_discoverer = ModuleDiscoverer(packages=["triggers"])
+
+    async def initialize(self, context: ModuleContext, setup_data: EchoSetup) -> None:
+        pass  # open clients, build agents
+
+    async def cleanup(self) -> None:
+        pass
+```
+
+```python
+# triggers/echo_trigger.py
+from digitalkin import ModuleContext, TriggerHandler
+from echo_module import EchoIn, EchoModule, EchoOut, EchoOutput, EchoSetup
+
+@EchoModule.register
+class EchoTrigger(TriggerHandler):
+    protocol = "message"
+    input_format = EchoIn
+    output_format = EchoOut
+
+    async def handle(self, input_data: EchoIn, setup_data: EchoSetup, context: ModuleContext) -> None:
+        reply = EchoOut(reply=setup_data.prefix + input_data.text)
+        await context.callbacks.send_message(EchoOutput(root=reply))
+```
+
+```python
+# server.py
+import asyncio
+from digitalkin.grpc_servers.module_server import ModuleServer
+from echo_module import EchoModule
+
+async def main() -> None:
+    server = ModuleServer(EchoModule)
+    await server.start_async()
+    try:
+        await server.await_termination()
+    finally:
+        await server.stop_async()
+
+asyncio.run(main())
+```
+
+```bash
+docker compose -f examples/redis_demo/docker-compose.yml up -d     # Redis on localhost:6379
+SERVICE_MODE=local DIGITALKIN_REDIS_URL=redis://localhost:6379/0 python server.py
+```
+
+The server listens on `SERVER_CHANNEL_HOST:SERVER_CHANNEL_PORT` (default `[::]:50055`).
+`SERVICE_MODE=local` (or `-d LOCAL`) uses in-process services; remote mode (default) needs the variables in `.env.minimum`.
+Consumers drive the module with the gateway RPCs described in the gateway protocol page.
+
+<!-- --8<-- [end:intro] -->
+
+## Documentation
+
+| Topic | Page |
+| --- | --- |
+| Architecture, data and signal flow | [docs/architecture.md](docs/architecture.md) |
+| Writing modules | [docs/module_authoring.md](docs/module_authoring.md) |
+| Wire contracts | [docs/gateway_protocol.md](docs/gateway_protocol.md), [docs/agui_events.md](docs/agui_events.md) (AG-UI) |
+| Deploy and tune | [docs/operations.md](docs/operations.md) |
+| Environment variables | [docs/environment_variables.md](docs/environment_variables.md) |
+| Testing | [docs/testing.md](docs/testing.md) |
+| Changelog | [docs/changelog/index.md](docs/changelog/index.md) |
+| API reference | [docs/api/api.md](docs/api/api.md) |
+
+## Development
+
+```bash
+task setup-dev     # venv, dev/docs/tests deps, pre-commit hooks
+task check         # ruff format + lint, mypy, tests
+task run-tests     # full suite in docker compose (needs Redis)
+task docs-serve    # mkdocs serve
+```
+
+License: [CC BY-NC-SA 4.0](LICENSE).

{digitalkin-1.0.5.dev6 → digitalkin-1.0.5.dev7}/examples/bench_module/triggers/message_trigger.py

@@ -5,7 +5,7 @@ from typing import ClassVar, Literal
 
 from echo_module import EchoToolModule
 from models.input import MessageInputPayload
-from models.output import MessageOutputPayload
+from models.output import EchoOutput, MessageOutputPayload
 from models.setup import EchoSetup
 
 from digitalkin.models.module import ModuleContext
@@ -56,8 +56,8 @@ class MessageTrigger(TriggerHandler):
                 result = f"{setup_data.prefix}{result}"
             chunk = f"[{i + 1}/{repeat}] {result}"
 
-            output = MessageOutputPayload(response=chunk)
-            await self.send_message(context, output)
+            output = EchoOutput(root=MessageOutputPayload(response=chunk))
+            await context.callbacks.send_message(output)
 
             if i < repeat - 1 and delay_s > 0:
                 await asyncio.sleep(delay_s)

{digitalkin-1.0.5.dev6 → digitalkin-1.0.5.dev7}/examples/modules/dynamic_setup_module.py

@@ -255,7 +255,10 @@ class DynamicSetupModule(
 
     # Services configuration (empty for this example)
     services_config_strategies: ClassVar[dict[str, ServicesStrategy | None]] = {}
-    services_config_params: ClassVar[dict[str, dict[str, Any | None] | None]] = {}
+    services_config_params: ClassVar[dict[str, dict[str, Any | None] | None]] = {
+        "storage": {"config": {}},
+        "cost": {"config": {}},
+    }
 
     async def initialize(self, context: ModuleContext, setup_data: DynamicAgentSetup) -> None:
         """Initialize the module with setup data.

{digitalkin-1.0.5.dev6 → digitalkin-1.0.5.dev7}/examples/redis_demo/echo_module.py

@@ -35,7 +35,10 @@ class EchoToolModule(ToolModule[EchoInput, EchoOutput, EchoSetup, EchoSecret]):
     }
 
     services_config_strategies: ClassVar[dict[str, Any]] = {}
-    services_config_params: ClassVar[dict[str, Any]] = {}
+    services_config_params: ClassVar[dict[str, Any]] = {
+        "storage": {"config": {}},
+        "cost": {"config": {}},
+    }
 
     triggers_discoverer = ModuleDiscoverer(packages=["triggers"])
 

{digitalkin-1.0.5.dev6 → digitalkin-1.0.5.dev7}/examples/redis_demo/server.py

@@ -1,6 +1,6 @@
 """EchoModule server — same pattern as template-tool.
 
-Uses ModuleServer with auto-embedded GatewayServicer (via DIGITALKIN_REDIS_URL).
+Uses ModuleServer, which embeds the GatewayServicer.
 Server config via env vars (ServerSettings from pydantic-settings):
   SERVER_CHANNEL_HOST, SERVER_CHANNEL_PORT, SERVER_CHANNEL_SECURITY, etc.
 
@@ -9,10 +9,9 @@ Usage:
     docker compose -f examples/redis_demo/docker-compose.yml up -d
 
     # 2. Set env and start:
-    DIGITALKIN_REDIS_URL=redis://localhost:6379/0 python examples/redis_demo/server.py
+    SERVICE_MODE=local DIGITALKIN_REDIS_URL=redis://localhost:6379/0 python examples/redis_demo/server.py
 
-    # 3. Test with the client:
-    python examples/redis_demo/client.py full --prompt "Hello world"
+A consumer then calls StartStream; see docs/gateway_protocol.md.
 """
 
 import asyncio

{digitalkin-1.0.5.dev6 → digitalkin-1.0.5.dev7}/examples/redis_demo/triggers/message_trigger.py

@@ -7,7 +7,7 @@ from digitalkin.models.module import ModuleContext
 from digitalkin.modules.trigger_handler import TriggerHandler
 
 from models.input import MessageInputPayload
-from models.output import MessageOutputPayload
+from models.output import EchoOutput, MessageOutputPayload
 from models.setup import EchoSetup
 from echo_module import EchoToolModule
 
@@ -56,8 +56,8 @@ class MessageTrigger(TriggerHandler):
                 result = f"{setup_data.prefix}{result}"
             chunk = f"[{i + 1}/{repeat}] {result}"
 
-            output = MessageOutputPayload(response=chunk)
-            await self.send_message(context, output)
+            output = EchoOutput(root=MessageOutputPayload(response=chunk))
+            await context.callbacks.send_message(output)
 
             if i < repeat - 1 and delay_s > 0:
                 await asyncio.sleep(delay_s)

{digitalkin-1.0.5.dev6 → digitalkin-1.0.5.dev7}/pyproject.toml

@@ -27,12 +27,8 @@
     ]
 
     dependencies = [
-        # Dev build: the only release carrying SUBAGENT_* events and event-level `metadata`,
-        # which is how a nested run's output is attributed to its author. Pinned exactly
-        # because it is a dated prerelease; move to >=0.1.21 once that ships stable.
-        "ag-ui-protocol==0.1.21.dev1787220377",
+        "ag-ui-protocol==1.0.0",
         "agentic-mesh-protocol==1.2.0",
-        "agno>=3.1.1,<4",
         "anyio>=4.13.0",
         "cel-python>=0.5",
         "grpcio-health-checking==1.82.1",
@@ -43,7 +39,7 @@
         "pydantic>=2.12.4",
         "redis[hiredis]>=7.4.0,<9",
     ]
-version = "1.0.5.dev6"
+version = "1.0.5.dev7"
 
     [project.optional-dependencies]
         agno = [ "agno>=3.1.1,<4" ]
@@ -64,6 +60,7 @@ version = "1.0.5.dev6"
 
 [dependency-groups]
     dev = [
+        "agno>=3.1.1,<4",
         "build>=1.5.0",
         "bump-my-version>=1.3.0",
         "cryptography>=48.0.0",
@@ -103,6 +100,7 @@ version = "1.0.5.dev6"
         "tomli>=2.4.1",
     ]
     tests = [
+        "agno>=3.1.1,<4",
         "fakeredis[lua]>=2.35.1",
         "freezegun>=1.5.5",
         "grpcio-testing>=1.81.0",

{digitalkin-1.0.5.dev6 → digitalkin-1.0.5.dev7}/src/digitalkin/__version__.py

@@ -5,4 +5,4 @@ from importlib.metadata import PackageNotFoundError, version
 try:
     __version__ = version("digitalkin")
 except PackageNotFoundError:
-    __version__ = "1.0.5.dev6"
+    __version__ = "1.0.5.dev7"

{digitalkin-1.0.5.dev6 → digitalkin-1.0.5.dev7}/src/digitalkin/community/agno/__init__.py

@@ -12,6 +12,8 @@ on top of the Agno agent framework. Exports:
   paused Agno run via the module's
   :class:`~digitalkin.services.storage.StorageStrategy` and resumes it
   when the front replies with a ``ToolMessage``.
+- :class:`AgUiStreamInput` — AG-UI 1.0 run input trigger; upgrades legacy
+  ``binary`` attachments to 1.0 media parts.
 - :class:`ToolCallMetadata`, :class:`ToolOutputMetadata` — metadata models
   for module tool calls made through the toolkit.
 
@@ -36,10 +38,18 @@ from digitalkin.community.agno.hitl import (
     PausedRunRecord,
     PausedRunStore,
 )
-from digitalkin.community.agno.models import PauseInfo, ToolCallMetadata, ToolOutputMetadata
+from digitalkin.community.agno.models import (
+    AgUiRunInput,
+    AgUiStreamInput,
+    PauseInfo,
+    ToolCallMetadata,
+    ToolOutputMetadata,
+)
 
 __all__ = [
     "HITL_STORAGE_CONFIG",
+    "AgUiRunInput",
+    "AgUiStreamInput",
     "AgnoHitlRunner",
     "AgnoStreamAdapter",
     "AguiTools",