digitalkin 0.3.5.dev0__tar.gz → 0.3.5.dev1__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.5.dev0
+Version: 0.3.5.dev1
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -452,8 +452,9 @@ Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: ag-ui-protocol>=0.1.14
 Requires-Dist: agentic-mesh-protocol==0.2.3
-Requires-Dist: anyio==4.13.0
+Requires-Dist: anyio==4.12.1
 Requires-Dist: grpcio-health-checking==1.78.0
 Requires-Dist: grpcio-reflection==1.78.0
 Requires-Dist: grpcio-status==1.78.0
@@ -462,7 +463,7 @@ Provides-Extra: profiling
 Requires-Dist: asyncio-inspector==0.1.0; extra == "profiling"
 Requires-Dist: pyinstrument==5.1.2; extra == "profiling"
 Requires-Dist: viztracer==1.1.1; extra == "profiling"
-Requires-Dist: yappi==1.7.6; extra == "profiling"
+Requires-Dist: yappi==1.7.3; extra == "profiling"
 Provides-Extra: taskiq
 Requires-Dist: rstream==1.0.0; extra == "taskiq"
 Requires-Dist: taskiq-aio-pika==0.6.0; extra == "taskiq"

{digitalkin-0.3.5.dev0 → digitalkin-0.3.5.dev1}/examples/modules/archetype_with_tools_module.py

@@ -5,11 +5,10 @@ from typing import Any, ClassVar, Literal
 
 from pydantic import BaseModel, Field
 
-from digitalkin.models.grpc_servers.models import ClientConfig
+from digitalkin.models.grpc_servers.models import ClientConfig, SecurityMode, ServerMode
 from digitalkin.models.module.module_context import ModuleContext
 from digitalkin.models.module.setup_types import SetupModel
 from digitalkin.models.module.tool_reference import ToolReference
-from digitalkin.models.settings.utils.channel import ControlFlow, SecurityMode
 from digitalkin.modules._base_module import BaseModule  # noqa: PLC2701
 from digitalkin.services.services_models import ServicesStrategy
 
@@ -99,7 +98,7 @@ class ArchetypeSecret(BaseModel):
 client_config = ClientConfig(
     host="[::]",
     port=50152,
-    mode=ControlFlow.ASYNC,
+    mode=ServerMode.ASYNC,
     security=SecurityMode.INSECURE,
     credentials=None,
 )

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

@@ -27,21 +27,22 @@
     ]
 
     dependencies = [
+        "ag-ui-protocol>=0.1.14",
         "agentic-mesh-protocol==0.2.3",
-        "anyio==4.13.0",
+        "anyio==4.12.1",
         "grpcio-health-checking==1.78.0",
         "grpcio-reflection==1.78.0",
         "grpcio-status==1.78.0",
         "pydantic==2.12.5",
-    ]
-    version = "0.3.5.dev0"
+]
+    version = "0.3.5.dev1"
 
     [project.optional-dependencies]
         profiling = [
             "asyncio-inspector==0.1.0",
             "pyinstrument==5.1.2",
             "viztracer==1.1.1",
-            "yappi==1.7.6",
+            "yappi==1.7.3",
         ]
         taskiq = [
             "rstream==1.0.0",
@@ -60,13 +61,13 @@
 
 [dependency-groups]
     dev = [
-        "build==1.4.1",
+        "build==1.4.0",
         "bump-my-version==1.2.7",
         "cryptography==46.0.5",
         "mypy==1.19.1",
         "pre-commit==4.5.1",
         "pyright==1.1.408",
-        "ruff==0.15.7",
+        "ruff==0.15.5",
         "twine==6.2.0",
         "types-grpcio-health-checking==1.0.0.20250506",
         "types-grpcio-reflection==1.0.0.20250506",
@@ -86,13 +87,13 @@
         "mkdocs-git-revision-date-localized-plugin==1.5.1",
         "mkdocs-glightbox==0.5.2",
         "mkdocs-include-markdown-plugin==7.2.1",
-        "mkdocs-literate-nav==0.6.3",
+        "mkdocs-literate-nav==0.6.2",
         "mkdocs-llmstxt==0.5.0",
-        "mkdocs-material[imaging]==9.7.6",
+        "mkdocs-material[imaging]==9.7.5",
         "mkdocs-minify-plugin==0.8.0",
         "mkdocs-open-in-new-tab==1.0.8",
         "mkdocs-redirects==1.2.2",
-        "mkdocs-section-index==0.3.11",
+        "mkdocs-section-index==0.3.10",
         "mkdocs==1.6.1",
         "mkdocstrings-python==2.0.3",
         "mkdocstrings==1.0.3",
@@ -104,7 +105,7 @@
         "hdrhistogram==0.10.3",
         "psutil==7.2.2",
         "pytest-asyncio==1.3.0",
-        "pytest-cov==7.1.0",
+        "pytest-cov==7.0.0",
         "pytest-html==4.2.0",
         "pytest-json-report==1.5.0",
         "pytest-timeout==2.4.0",

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

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

digitalkin-0.3.5.dev1/src/digitalkin/community/__init__.py

@@ -0,0 +1,5 @@
+"""Community integrations for DigitalKin.
+
+This package contains community-contributed integrations with various AI frameworks
+and tools (Agno, LangChain, etc.).
+"""

digitalkin-0.3.5.dev1/src/digitalkin/community/agno/__init__.py

@@ -0,0 +1,5 @@
+"""Agno framework integration for DigitalKin.
+
+This module provides utilities and adapters for integrating Agno agents
+with the DigitalKin SDK.
+"""

digitalkin-0.3.5.dev1/src/digitalkin/community/agno/agno_adapter.py

@@ -0,0 +1,183 @@
+"""Adapter to convert Agno events to DigitalKin framework-agnostic events.
+
+This adapter bridges Agno-specific events to the DigitalKin event model,
+allowing the core DigitalKin SDK to remain independent of Agno.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+from digitalkin.models.events import (
+    AgentRunEvent,
+    BaseAgentRunEvent,
+    ReasoningCompletedEvent,
+    ReasoningContentDeltaEvent,
+    ReasoningStartedEvent,
+    ReasoningStepEvent,
+    RunCompletedEvent,
+    RunContentEvent,
+    RunErrorEvent,
+    RunStartedEvent,
+    ToolCallCompletedEvent,
+    ToolCallErrorEvent,
+    ToolCallStartedEvent,
+    ToolInfo,
+)
+
+
+def to_digitalkin_event(agno_event: Any) -> BaseAgentRunEvent:  # noqa: PLR0911, C901, PLR0912
+    """Convert an Agno event to a DigitalKin framework-agnostic event.
+
+    Args:
+        agno_event: Event from Agno's streaming API.
+
+    Returns:
+        Corresponding DigitalKin BaseAgentRunEvent.
+
+    Raises:
+        ImportError: If the optional 'agno' dependency is not installed.
+        ValueError: If the event type is not recognized.
+    """
+    try:
+        from agno.run.agent import RunEvent  # pylint: disable=C0415 # pyright: ignore[reportMissingImports]
+    except ImportError as exc:
+        message = "The 'agno' package is required to use agno_to_digitalkin_event. \
+Install it with: pip install agno"
+        raise ImportError(message) from exc
+
+    event_type = agno_event.event
+
+    # Import timestamp if available
+    timestamp = getattr(agno_event, "timestamp", None)
+
+    if event_type == RunEvent.run_started:
+        return RunStartedEvent(
+            event=AgentRunEvent.RUN_STARTED,
+            run_id=getattr(agno_event, "run_id", None),
+            thread_id=getattr(agno_event, "thread_id", None),
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.run_content:
+        return RunContentEvent(
+            event=AgentRunEvent.RUN_CONTENT,
+            content=str(agno_event.content) if agno_event.content else None,
+            reasoning_content=getattr(agno_event, "reasoning_content", None),
+            content_type=None,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.run_completed:
+        return RunCompletedEvent(
+            event=AgentRunEvent.RUN_COMPLETED,
+            run_id=getattr(agno_event, "run_id", None),
+            final_content=str(agno_event.content) if getattr(agno_event, "content", None) else None,
+            usage=None,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.run_error:
+        return RunErrorEvent(
+            event=AgentRunEvent.RUN_ERROR,
+            error_type=getattr(agno_event, "error_type", None),
+            content=str(agno_event.content) if getattr(agno_event, "content", None) else None,
+            error_details=None,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.reasoning_started:
+        return ReasoningStartedEvent(
+            event=AgentRunEvent.REASONING_STARTED,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.reasoning_content_delta:
+        reasoning_content = getattr(agno_event, "reasoning_content", "")
+        return ReasoningContentDeltaEvent(
+            event=AgentRunEvent.REASONING_CONTENT_DELTA,
+            delta=reasoning_content,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.reasoning_step:
+        reasoning_content = getattr(agno_event, "reasoning_content", "")
+        return ReasoningStepEvent(
+            event=AgentRunEvent.REASONING_STEP,
+            delta=reasoning_content,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.reasoning_completed:
+        return ReasoningCompletedEvent(
+            event=AgentRunEvent.REASONING_COMPLETED,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.tool_call_started:
+        tool = getattr(agno_event, "tool", None)
+        tool_info = None
+        if tool:
+            tool_info = ToolInfo(
+                tool_call_id=getattr(tool, "tool_call_id", None),
+                tool_name=getattr(tool, "tool_name", None),
+                tool_args=getattr(tool, "tool_args", None),
+                result=None,
+            )
+        return ToolCallStartedEvent(
+            event=AgentRunEvent.TOOL_CALL_STARTED,
+            tool=tool_info,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.tool_call_completed:
+        tool = getattr(agno_event, "tool", None)
+        tool_info = None
+        if tool:
+            tool_info = ToolInfo(
+                tool_call_id=getattr(tool, "tool_call_id", None),
+                tool_name=getattr(tool, "tool_name", None),
+                tool_args=getattr(tool, "tool_args", None),
+                result=getattr(tool, "result", None),
+            )
+        return ToolCallCompletedEvent(
+            event=AgentRunEvent.TOOL_CALL_COMPLETED,
+            tool=tool_info,
+            content=str(agno_event.content) if getattr(agno_event, "content", None) else None,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    if event_type == RunEvent.tool_call_error:
+        tool = getattr(agno_event, "tool", None)
+        tool_info = None
+        if tool:
+            tool_info = ToolInfo(
+                tool_call_id=getattr(tool, "tool_call_id", None),
+                tool_name=getattr(tool, "tool_name", None),
+                tool_args=None,
+                result=None,
+            )
+        return ToolCallErrorEvent(
+            event=AgentRunEvent.TOOL_CALL_ERROR,
+            tool=tool_info,
+            error_message=str(agno_event.content) if getattr(agno_event, "content", None) else None,
+            timestamp=timestamp,
+            metadata=None,
+        )
+
+    # Unknown event - create a generic base event
+    return BaseAgentRunEvent(
+        event=AgentRunEvent.RUN_CONTENT,  # Default fallback
+        timestamp=timestamp,
+        metadata={"original_event_type": str(event_type)},
+    )

{digitalkin-0.3.5.dev0 → digitalkin-0.3.5.dev1}/src/digitalkin/core/job_manager/base_job_manager.py

@@ -130,12 +130,6 @@ class BaseJobManager(abc.ABC, Generic[InputModelT, OutputModelT, SetupModelT]):
         required for the job manager to function.
         """
 
-    async def stop(self) -> None:
-        """Stop the job manager and clean up resources.
-
-        Default no-op. Subclasses with external connections override.
-        """
-
     @staticmethod
     async def job_specific_callback(
         callback: Callable[[str, DataModel | ModuleCodeModel], Coroutine[Any, Any, None]],

{digitalkin-0.3.5.dev0 → digitalkin-0.3.5.dev1}/src/digitalkin/core/job_manager/single_job_manager.py

@@ -285,7 +285,10 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
                 if session.cancelled:
                     break
 
-        yield _stream()
+        try:
+            yield _stream()
+        finally:
+            session.close_stream()
 
     async def create_module_instance_job(
         self,