digitalkin 0.3.5.dev1__tar.gz → 0.3.5.dev3__tar.gz

{digitalkin-0.3.5.dev1 → digitalkin-0.3.5.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.5.dev1
+Version: 0.3.5.dev3
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International

{digitalkin-0.3.5.dev1 → digitalkin-0.3.5.dev3}/pyproject.toml

@@ -35,7 +35,7 @@
         "grpcio-status==1.78.0",
         "pydantic==2.12.5",
 ]
-    version = "0.3.5.dev1"
+    version = "0.3.5.dev3"
 
     [project.optional-dependencies]
         profiling = [

{digitalkin-0.3.5.dev1 → digitalkin-0.3.5.dev3}/src/digitalkin/__version__.py

@@ -5,4 +5,4 @@ from importlib.metadata import PackageNotFoundError, version
 try:
     __version__ = version("digitalkin")
 except PackageNotFoundError:
-    __version__ = "0.3.5.dev1"
+    __version__ = "0.3.5.dev3"

{digitalkin-0.3.5.dev1 → digitalkin-0.3.5.dev3}/src/digitalkin/community/agno/__init__.py

@@ -3,3 +3,7 @@
 This module provides utilities and adapters for integrating Agno agents
 with the DigitalKin SDK.
 """
+
+from digitalkin.community.agno.agno_adapter import to_digitalkin_event
+
+__all__ = ["to_digitalkin_event"]

{digitalkin-0.3.5.dev1 → digitalkin-0.3.5.dev3}/src/digitalkin/mixins/agui_mixin.py

@@ -25,25 +25,25 @@ from digitalkin.models.events import (
     ToolCallErrorEvent,
     ToolCallStartedEvent,
 )
-from digitalkin.models.module.ag_ui import (
-    AgUiReasoningEndOutput,
-    AgUiReasoningMessageContentOutput,
-    AgUiReasoningMessageEndOutput,
-    AgUiReasoningMessageStartOutput,
-    AgUiReasoningStartOutput,
-    AgUiRunErrorOutput,
-    AgUiRunFinishedOutput,
-    AgUiRunStartedOutput,
-    AgUiTextMessageContentOutput,
-    AgUiTextMessageEndOutput,
-    AgUiTextMessageStartOutput,
-    AgUiToolCallArgsOutput,
-    AgUiToolCallEndOutput,
-    AgUiToolCallResultOutput,
-    AgUiToolCallStartOutput,
-)
 
 if TYPE_CHECKING:
+    from digitalkin.models.module.ag_ui import (
+        AgUiReasoningEndOutput,
+        AgUiReasoningMessageContentOutput,
+        AgUiReasoningMessageEndOutput,
+        AgUiReasoningMessageStartOutput,
+        AgUiReasoningStartOutput,
+        AgUiRunErrorOutput,
+        AgUiRunFinishedOutput,
+        AgUiRunStartedOutput,
+        AgUiTextMessageContentOutput,
+        AgUiTextMessageEndOutput,
+        AgUiTextMessageStartOutput,
+        AgUiToolCallArgsOutput,
+        AgUiToolCallEndOutput,
+        AgUiToolCallResultOutput,
+        AgUiToolCallStartOutput,
+    )
     from digitalkin.models.module.module_context import ModuleContext
 
 
@@ -78,7 +78,7 @@ class AgUiMixin:
         self._text_started: bool = False
         self._reasoning_started: bool = False
 
-    async def send_message(  # noqa: C901
+    async def agui_send_message(  # noqa: C901
         self,
         context: ModuleContext,
         event: BaseAgentRunEvent,
@@ -153,6 +153,8 @@ class AgUiMixin:
         """Handle run started event - emit AG-UI RunStarted."""
         from ag_ui.core.events import RunStartedEvent as AgUiRunStartedEvent  # pylint: disable=C0415
 
+        from digitalkin.models.module.ag_ui import AgUiRunStartedOutput  # pylint: disable=C0415
+
         self._run_id = event.run_id or str(uuid.uuid4())
         if event.thread_id:
             self._thread_id = event.thread_id
@@ -178,6 +180,11 @@ class AgUiMixin:
             TextMessageStartEvent as AgUiTextMessageStartEvent,
         )
 
+        from digitalkin.models.module.ag_ui import (  # pylint: disable=C0415
+            AgUiTextMessageContentOutput,
+            AgUiTextMessageStartOutput,
+        )
+
         content = event.content
         if not content:
             return
@@ -211,6 +218,11 @@ class AgUiMixin:
         from ag_ui.core.events import RunFinishedEvent as AgUiRunFinishedEvent  # pylint: disable=C0415
         from ag_ui.core.events import TextMessageEndEvent as AgUiTextMessageEndEvent  # pylint: disable=C0415
 
+        from digitalkin.models.module.ag_ui import (  # pylint: disable=C0415
+            AgUiRunFinishedOutput,
+            AgUiTextMessageEndOutput,
+        )
+
         # Close any open text message
         if self._text_started:
             end_output = AgUiTextMessageEndOutput(
@@ -240,6 +252,8 @@ class AgUiMixin:
         """Handle run error event - emit AG-UI RunError."""
         from ag_ui.core.events import RunErrorEvent as AgUiRunErrorEvent  # pylint: disable=C0415
 
+        from digitalkin.models.module.ag_ui import AgUiRunErrorOutput  # pylint: disable=C0415
+
         error_msg = event.content or "Agent run failed"
         output = AgUiRunErrorOutput(
             event=AgUiRunErrorEvent(
@@ -260,6 +274,12 @@ class AgUiMixin:
         from ag_ui.core.events import ToolCallArgsEvent as AgUiToolCallArgsEvent  # pylint: disable=C0415
         from ag_ui.core.events import ToolCallStartEvent as AgUiToolCallStartEvent  # pylint: disable=C0415
 
+        from digitalkin.models.module.ag_ui import (  # pylint: disable=C0415
+            AgUiTextMessageEndOutput,
+            AgUiToolCallArgsOutput,
+            AgUiToolCallStartOutput,
+        )
+
         tool = event.tool
         if not tool or not tool.tool_name:
             return
@@ -308,6 +328,11 @@ class AgUiMixin:
         from ag_ui.core.events import ToolCallEndEvent as AgUiToolCallEndEvent  # pylint: disable=C0415
         from ag_ui.core.events import ToolCallResultEvent as AgUiToolCallResultEvent  # pylint: disable=C0415
 
+        from digitalkin.models.module.ag_ui import (  # pylint: disable=C0415
+            AgUiToolCallEndOutput,
+            AgUiToolCallResultOutput,
+        )
+
         tool = event.tool
         if not tool:
             return
@@ -340,6 +365,8 @@ class AgUiMixin:
         """Handle tool call error event - emit AG-UI ToolCallEnd."""
         from ag_ui.core.events import ToolCallEndEvent as AgUiToolCallEndEvent  # pylint: disable=C0415
 
+        from digitalkin.models.module.ag_ui import AgUiToolCallEndOutput  # pylint: disable=C0415
+
         tool = event.tool
         if not tool:
             return
@@ -357,8 +384,13 @@ class AgUiMixin:
         from ag_ui.core.events import (  # pylint: disable=import-outside-toplevel
             ReasoningMessageStartEvent as AgUiReasoningMessageStartEvent,
         )
-        from ag_ui.core.events import (
-            ReasoningStartEvent as AgUiReasoningStartEvent,  # pylint: disable=import-outside-toplevel
+        from ag_ui.core.events import (  # pylint: disable=import-outside-toplevel
+            ReasoningStartEvent as AgUiReasoningStartEvent,
+        )
+
+        from digitalkin.models.module.ag_ui import (  # pylint: disable=C0415
+            AgUiReasoningMessageStartOutput,
+            AgUiReasoningStartOutput,
         )
 
         if self._reasoning_started:
@@ -388,6 +420,8 @@ class AgUiMixin:
             ReasoningMessageContentEvent as AgUiReasoningMessageContentEvent,
         )
 
+        from digitalkin.models.module.ag_ui import AgUiReasoningMessageContentOutput  # pylint: disable=C0415
+
         delta = event.delta
         if not delta:
             return
@@ -417,6 +451,8 @@ class AgUiMixin:
             ReasoningMessageContentEvent as AgUiReasoningMessageContentEvent,
         )
 
+        from digitalkin.models.module.ag_ui import AgUiReasoningMessageContentOutput  # pylint: disable=C0415
+
         delta = event.delta
         if not delta:
             return
@@ -453,6 +489,11 @@ class AgUiMixin:
             ReasoningMessageEndEvent as AgUiReasoningMessageEndEvent,
         )
 
+        from digitalkin.models.module.ag_ui import (  # pylint: disable=C0415
+            AgUiReasoningEndOutput,
+            AgUiReasoningMessageEndOutput,
+        )
+
         if not self._reasoning_started:
             return
 
@@ -463,7 +504,9 @@ class AgUiMixin:
         await context.callbacks.send_message(message_end_output)
 
         # Emit ReasoningEnd
-        end_output = AgUiReasoningEndOutput(event=AgUiReasoningEndEvent(message_id=self._reasoning_id))
+        end_output = AgUiReasoningEndOutput(
+            event=AgUiReasoningEndEvent(message_id=self._reasoning_id),
+        )
         await context.callbacks.send_message(end_output)
 
         self._reasoning_started = False

{digitalkin-0.3.5.dev1 → digitalkin-0.3.5.dev3}/src/digitalkin/models/module/__init__.py

@@ -1,6 +1,8 @@
 """This module contains the models for the modules."""
 
-from digitalkin.models.module.ag_ui import AgUiEventOutput, AgUiOutput
+# Import module_types first to avoid circular import with ag_ui
+# Note: AgUiEventOutput and AgUiOutput are not imported here to avoid circular imports.
+# Import them directly from digitalkin.models.module.ag_ui if needed.
 from digitalkin.models.module.module_context import ModuleContext
 from digitalkin.models.module.module_types import (
     DataModel,
@@ -28,8 +30,8 @@ from digitalkin.models.module.utility import (
 )
 
 __all__ = [
-    "AgUiEventOutput",
-    "AgUiOutput",
+    # Note: AgUiEventOutput and AgUiOutput removed to avoid circular imports
+    # Import them directly from digitalkin.models.module.ag_ui if needed
     "DataModel",
     "DataTrigger",
     "EndOfStreamOutput",

{digitalkin-0.3.5.dev1 → digitalkin-0.3.5.dev3}/src/digitalkin/models/module/ag_ui.py

@@ -41,7 +41,7 @@ from ag_ui.core.events import (
 from pydantic import ConfigDict, Field
 from pydantic.alias_generators import to_camel
 
-from digitalkin.models.module import DataModel, DataTrigger
+from digitalkin.models.module.module_types import DataModel, DataTrigger
 
 # ── AG-UI base class with camelCase aliases ──────────────────────────────────
 

{digitalkin-0.3.5.dev1 → digitalkin-0.3.5.dev3}/src/digitalkin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.5.dev1
+Version: 0.3.5.dev3
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International