digitalkin 0.3.5.dev11__tar.gz → 0.3.5.dev13__tar.gz

{digitalkin-0.3.5.dev11 → digitalkin-0.3.5.dev13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.5.dev11
+Version: 0.3.5.dev13
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -454,7 +454,7 @@ 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.12.1
+Requires-Dist: anyio==4.13.0
 Requires-Dist: grpcio-health-checking==1.78.0
 Requires-Dist: grpcio-reflection==1.78.0
 Requires-Dist: grpcio-status==1.78.0
@@ -463,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.3; extra == "profiling"
+Requires-Dist: yappi==1.7.6; 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.dev11 → digitalkin-0.3.5.dev13}/examples/modules/archetype_with_tools_module.py

@@ -5,10 +5,11 @@ from typing import Any, ClassVar, Literal
 
 from pydantic import BaseModel, Field
 
-from digitalkin.models.grpc_servers.models import ClientConfig, SecurityMode, ServerMode
+from digitalkin.models.grpc_servers.models import ClientConfig
 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
 
@@ -98,7 +99,7 @@ class ArchetypeSecret(BaseModel):
 client_config = ClientConfig(
     host="[::]",
     port=50152,
-    mode=ServerMode.ASYNC,
+    mode=ControlFlow.ASYNC,
     security=SecurityMode.INSECURE,
     credentials=None,
 )

{digitalkin-0.3.5.dev11 → digitalkin-0.3.5.dev13}/pyproject.toml

@@ -29,20 +29,20 @@
     dependencies = [
         "ag-ui-protocol>=0.1.14",
         "agentic-mesh-protocol==0.2.3",
-        "anyio==4.12.1",
+        "anyio==4.13.0",
         "grpcio-health-checking==1.78.0",
         "grpcio-reflection==1.78.0",
         "grpcio-status==1.78.0",
         "pydantic==2.12.5",
-]
-    version = "0.3.5.dev11"
+    ]
+    version = "0.3.5.dev13"
 
     [project.optional-dependencies]
         profiling = [
             "asyncio-inspector==0.1.0",
             "pyinstrument==5.1.2",
             "viztracer==1.1.1",
-            "yappi==1.7.3",
+            "yappi==1.7.6",
         ]
         taskiq = [
             "rstream==1.0.0",
@@ -61,13 +61,13 @@
 
 [dependency-groups]
     dev = [
-        "build==1.4.0",
+        "build==1.4.2",
         "bump-my-version==1.2.7",
-        "cryptography==46.0.5",
+        "cryptography==46.0.6",
         "mypy==1.19.1",
         "pre-commit==4.5.1",
         "pyright==1.1.408",
-        "ruff==0.15.5",
+        "ruff==0.15.8",
         "twine==6.2.0",
         "types-grpcio-health-checking==1.0.0.20250506",
         "types-grpcio-reflection==1.0.0.20250506",
@@ -87,17 +87,17 @@
         "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.2",
+        "mkdocs-literate-nav==0.6.3",
         "mkdocs-llmstxt==0.5.0",
-        "mkdocs-material[imaging]==9.7.5",
+        "mkdocs-material[imaging]==9.7.6",
         "mkdocs-minify-plugin==0.8.0",
         "mkdocs-open-in-new-tab==1.0.8",
         "mkdocs-redirects==1.2.2",
-        "mkdocs-section-index==0.3.10",
+        "mkdocs-section-index==0.3.11",
         "mkdocs==1.6.1",
         "mkdocstrings-python==2.0.3",
         "mkdocstrings==1.0.3",
-        "tomli==2.4.0",
+        "tomli==2.4.1",
     ]
     tests = [
         "freezegun==1.5.5",
@@ -105,7 +105,7 @@
         "hdrhistogram==0.10.3",
         "psutil==7.2.2",
         "pytest-asyncio==1.3.0",
-        "pytest-cov==7.0.0",
+        "pytest-cov==7.1.0",
         "pytest-html==4.2.0",
         "pytest-json-report==1.5.0",
         "pytest-timeout==2.4.0",

{digitalkin-0.3.5.dev11 → digitalkin-0.3.5.dev13}/src/digitalkin/__version__.py

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

{digitalkin-0.3.5.dev11 → digitalkin-0.3.5.dev13}/src/digitalkin/community/agno/agno_adapter.py

@@ -63,6 +63,9 @@ class AgnoStreamAdapter:
 
         self._closed_tool_call_ids: set[str] = set()
 
+        self._active_run_id: str | None = None
+        self._completed_run_ids: set[str] = set()
+
     def to_digitalkin_events(self, agno_event: Any) -> list[BaseAgentRunEvent]:  # noqa: C901, PLR0911, PLR0912, PLR0915
         """Convert one Agno event into one or more DigitalKin events.
 
@@ -83,17 +86,25 @@ class AgnoStreamAdapter:
 
         event_type = agno_event.event
 
-        logger.info("[DK STREAM-DEBUG => agno_adapter] Converting Agno event type: %s", event_type)
+        logger.debug("Converting Agno event: %s", event_type)
 
         timestamp = getattr(agno_event, "timestamp", None)
 
         # ── Run Lifecycle ────────────────────────────────────────────────
 
         if event_type == RunEvent.run_started:
+            run_id = getattr(agno_event, "run_id", None)
+
+            # Skip duplicate RunStarted for the same run_id
+            if run_id and run_id == self._active_run_id:
+                logger.debug("Skipping duplicate RunStarted for run_id=%s", run_id)
+                return []
+
+            self._active_run_id = run_id
             return [
                 RunStartedEvent(
                     event=AgentRunEvent.RUN_STARTED,
-                    run_id=getattr(agno_event, "run_id", None),
+                    run_id=run_id,
                     thread_id=getattr(agno_event, "thread_id", None),
                     timestamp=timestamp,
                     metadata=None,
@@ -104,6 +115,13 @@ class AgnoStreamAdapter:
             return self._handle_run_content(agno_event, timestamp)
 
         if event_type == RunEvent.run_completed:
+            run_id = getattr(agno_event, "run_id", None)
+
+            # Skip if this run was already completed and not reopened
+            if run_id and run_id in self._completed_run_ids and run_id != self._active_run_id:
+                logger.debug("Skipping duplicate RunCompleted for run_id=%s", run_id)
+                return []
+
             events: list[BaseAgentRunEvent] = []
 
             # Close any open sequences before completing the run
@@ -112,10 +130,14 @@ class AgnoStreamAdapter:
             if self._reasoning_active:
                 events.extend(self._close_reasoning(timestamp))
 
+            if run_id:
+                self._completed_run_ids.add(run_id)
+            self._active_run_id = None
+
             events.append(
                 RunCompletedEvent(
                     event=AgentRunEvent.RUN_COMPLETED,
-                    run_id=getattr(agno_event, "run_id", None),
+                    run_id=run_id,
                     final_content=str(agno_event.content) if getattr(agno_event, "content", None) else None,
                     usage=None,
                     timestamp=timestamp,
@@ -147,10 +169,7 @@ class AgnoStreamAdapter:
 
             self._current_reasoning_id = str(uuid.uuid4())
             self._reasoning_active = True
-            logger.info(
-                "[DK STREAM-DEBUG => agno_adapter] Reasoning started (explicit), id=%s",
-                self._current_reasoning_id,
-            )
+            logger.debug("Reasoning started, id=%s", self._current_reasoning_id)
             events.append(
                 ReasoningStartedEvent(
                     event=AgentRunEvent.REASONING_STARTED,
@@ -186,7 +205,7 @@ class AgnoStreamAdapter:
             ]
 
         if event_type == RunEvent.reasoning_completed:
-            logger.info("[DK STREAM-DEBUG => agno_adapter] Reasoning completed (explicit)")
+            logger.debug("Reasoning completed")
             return self._close_reasoning(timestamp)
 
         # ── Tool Call Events ─────────────────────────────────────────────
@@ -196,7 +215,7 @@ class AgnoStreamAdapter:
 
             # Close both reasoning and content before tool calls
             if self._reasoning_active:
-                logger.info("[DK STREAM-DEBUG => agno_adapter] Reasoning auto-completed (tool call started)")
+                logger.debug("Reasoning auto-completed (tool call started)")
                 events.extend(self._close_reasoning(timestamp))
             if self._content_active:
                 events.extend(self._close_content(timestamp))
@@ -253,10 +272,7 @@ class AgnoStreamAdapter:
 
             # Skip if already closed by ToolCallCompleted (Agno emits both for the same tool_call_id)
             if tool_call_id and tool_call_id in self._closed_tool_call_ids:
-                logger.info(
-                    "[DK STREAM-DEBUG => agno_adapter] Skipping duplicate ToolCallError for already-closed tool %s",
-                    tool_call_id,
-                )
+                logger.debug("Skipping duplicate ToolCallError for tool %s", tool_call_id)
                 return []
 
             tool_info = None
@@ -293,10 +309,10 @@ class AgnoStreamAdapter:
         """
         events: list[BaseAgentRunEvent] = []
         if self._content_active:
-            logger.info("[DK STREAM-DEBUG => agno_adapter] Flushing: closing active content")
+            logger.debug("Flushing active content sequence")
             events.extend(self._close_content(None))
         if self._reasoning_active:
-            logger.info("[DK STREAM-DEBUG => agno_adapter] Flushing: closing active reasoning")
+            logger.debug("Flushing active reasoning sequence")
             events.extend(self._close_reasoning(None))
         return events
 
@@ -372,7 +388,7 @@ class AgnoStreamAdapter:
 
         # Edge case: neither reasoning_content nor content
         if reasoning_content is None and content is None:
-            logger.debug("[DK STREAM-DEBUG => agno_adapter] run_content with no content, skipping")
+            logger.debug("run_content with no content, skipping")
 
         return events
 
@@ -398,10 +414,7 @@ class AgnoStreamAdapter:
         # Auto-open reasoning on first chunk
         if not self._reasoning_active:
             self._current_reasoning_id = str(uuid.uuid4())
-            logger.info(
-                "[DK STREAM-DEBUG => agno_adapter] Reasoning auto-started from run_content, id=%s",
-                self._current_reasoning_id,
-            )
+            logger.debug("Reasoning auto-started, id=%s", self._current_reasoning_id)
             events.append(
                 ReasoningStartedEvent(
                     event=AgentRunEvent.REASONING_STARTED,
@@ -412,10 +425,6 @@ class AgnoStreamAdapter:
             )
             self._reasoning_active = True
 
-        logger.info(
-            "[DK STREAM-DEBUG => agno_adapter] Reasoning content delta (len=%d)",
-            len(reasoning_content),
-        )
         events.append(
             ReasoningContentDeltaEvent(
                 event=AgentRunEvent.REASONING_CONTENT_DELTA,
@@ -444,7 +453,7 @@ class AgnoStreamAdapter:
         # Non-empty string → text data
         # Close reasoning if transitioning from reasoning to content
         if self._reasoning_active:
-            logger.info("[DK STREAM-DEBUG => agno_adapter] Reasoning auto-completed (text content arrived)")
+            logger.debug("Reasoning auto-completed (text content arrived)")
             events.extend(self._close_reasoning(timestamp))
 
         # Auto-open text message on first chunk

{digitalkin-0.3.5.dev11 → digitalkin-0.3.5.dev13}/src/digitalkin/core/job_manager/base_job_manager.py

@@ -130,6 +130,12 @@ 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.dev11 → digitalkin-0.3.5.dev13}/src/digitalkin/core/job_manager/single_job_manager.py

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