digitalkin 1.0.0.dev22__tar.gz → 1.0.0.dev23__tar.gz

{digitalkin-1.0.0.dev22 → digitalkin-1.0.0.dev23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 1.0.0.dev22
+Version: 1.0.0.dev23
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -453,7 +453,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ag-ui-protocol>=0.1.18
-Requires-Dist: agentic-mesh-protocol==1.0.0a1
+Requires-Dist: agentic-mesh-protocol==1.0.0a2
 Requires-Dist: anyio>=4.13.0
 Requires-Dist: grpcio-health-checking==1.81.0
 Requires-Dist: grpcio-reflection==1.81.0

{digitalkin-1.0.0.dev22 → digitalkin-1.0.0.dev23}/pyproject.toml

@@ -28,7 +28,7 @@
 
     dependencies = [
         "ag-ui-protocol>=0.1.18",
-        "agentic-mesh-protocol==1.0.0a1",
+        "agentic-mesh-protocol==1.0.0a2",
         "anyio>=4.13.0",
         "grpcio-health-checking==1.81.0",
         "grpcio-reflection==1.81.0",
@@ -37,7 +37,7 @@
         "pydantic-settings>=2.14.1",
         "redis[hiredis]>=7.4.0,<9",
     ]
-    version = "1.0.0.dev22"
+    version = "1.0.0.dev23"
 
     [project.optional-dependencies]
         performance = [ "uvloop>=0.21" ]

{digitalkin-1.0.0.dev22 → digitalkin-1.0.0.dev23}/src/digitalkin/__version__.py

@@ -5,4 +5,4 @@ from importlib.metadata import PackageNotFoundError, version
 try:
     __version__ = version("digitalkin")
 except PackageNotFoundError:
-    __version__ = "1.0.0.dev22"
+    __version__ = "1.0.0.dev23"

{digitalkin-1.0.0.dev22 → digitalkin-1.0.0.dev23}/src/digitalkin/grpc_servers/gateway_servicer.py

@@ -7,6 +7,7 @@ import contextlib
 import json
 import random
 import time
+import uuid
 from datetime import datetime, timezone
 from typing import TYPE_CHECKING, Any
 
@@ -158,6 +159,43 @@ class GatewayServicer:
         await self._m2m.stop()
         await self._registry.shutdown()
 
+    async def AssociateTask(
+        self,
+        request: Any,
+        context: grpc.aio.ServicerContext,  # noqa: ARG002
+    ) -> Any:
+        """Mint a sub-task id linked to a running parent task.
+
+        Args:
+            request: AssociateTaskRequest proto.
+            context: gRPC service context.
+
+        Returns:
+            AssociateTaskResponse(task_id, parent_task_id); empty ``task_id`` on rejection.
+        """
+        parent_task_id = request.parent_task_id
+        err = GatewayValidator.validate_id(parent_task_id, "parent_task_id")
+        if err is not None:
+            logger.warning("AssociateTask rejected: %s", err)
+            return gateway_pb2.AssociateTaskResponse(task_id="", parent_task_id=parent_task_id)
+
+        child_task_id = str(uuid.uuid4())
+        ttl = get_gateway_settings().stream.redis_stream_initial_ttl
+        try:
+            await self._redis_client.sadd(f"task:{parent_task_id}:children", child_task_id)
+            await self._redis_client.expire(f"task:{parent_task_id}:children", ttl)
+            await self._redis_client.set(f"task:{child_task_id}:parent", parent_task_id, ex=ttl)
+        except RedisError:
+            logger.warning(
+                "[VALIDATE AT1] AssociateTask persist failed parent=%s → task_id=''",
+                parent_task_id,
+                exc_info=True,
+            )  # TODO(validate): remove after prod validation
+            return gateway_pb2.AssociateTaskResponse(task_id="", parent_task_id=parent_task_id)
+
+        logger.info("AssociateTask: parent=%s child=%s", parent_task_id, child_task_id)
+        return gateway_pb2.AssociateTaskResponse(task_id=child_task_id, parent_task_id=parent_task_id)
+
     async def StartStream(  # noqa: PLR0911, PLR0915
         self,
         request: Any,

{digitalkin-1.0.0.dev22 → digitalkin-1.0.0.dev23}/src/digitalkin/services/communication/grpc_communication.py

@@ -4,7 +4,6 @@ from __future__ import annotations
 
 import asyncio
 import time
-import uuid
 from typing import TYPE_CHECKING, Any
 
 import grpc.aio
@@ -16,6 +15,7 @@ from agentic_mesh_protocol.module.v1 import (
 from google.protobuf import json_format, struct_pb2
 
 from digitalkin.core.profiling.step_timer import StepTimer
+from digitalkin.grpc_servers.interceptors.request_ids import RequestContext
 from digitalkin.grpc_servers.utils.grpc_client_wrapper import GrpcClientWrapper
 from digitalkin.grpc_servers.utils.validators import GatewayValidator
 from digitalkin.logger import logger
@@ -315,8 +315,7 @@ class GrpcCommunication(CommunicationStrategy, GrpcClientWrapper):
             "target_key": target_key,
         }
 
-        task_id = str(uuid.uuid4())
-        log_extra["task_id"] = task_id
+        task_id = ""
         breaker = m2m.breaker_for(target_key)
 
         last_mark = "init"
@@ -342,6 +341,38 @@ class GrpcCommunication(CommunicationStrategy, GrpcClientWrapper):
             timer.mark("acquire_slot")
             last_mark = "acquire_slot"
 
+            self._get_or_create_channel(module_address, module_port)
+            timer.mark("channel_create")
+            last_mark = "channel_create"
+            stub = self._get_or_create_stub(gateway_service_pb2_grpc.GatewayServiceStub)
+            timer.mark("stub_create")
+            last_mark = "stub_create"
+
+            # The gateway mints the sub-task id and links it to the running parent task.
+            try:
+                assoc = await stub.AssociateTask(
+                    gateway_pb2.AssociateTaskRequest(
+                        parent_task_id=RequestContext.current().get("task_id", ""),
+                    ),
+                )
+            except grpc.aio.AioRpcError as exc:
+                breaker.record_failure()
+                logger.warning(
+                    "[m2m] AssociateTask failed: [%s] %s",
+                    exc.code().name,
+                    exc.details() or "",
+                    extra=log_extra,
+                )
+                raise
+            task_id = assoc.task_id
+            if not task_id:
+                breaker.record_failure()
+                msg = f"target {target_key} returned no task_id from AssociateTask"
+                raise RuntimeError(msg)  # noqa: TRY301
+            log_extra["task_id"] = task_id
+            timer.mark("associate_task")
+            last_mark = "associate_task"
+
             output_queue: asyncio.Queue[struct_pb2.Struct | None] = asyncio.Queue(
                 maxsize=m2m_settings.call_queue_maxsize,
             )
@@ -361,13 +392,6 @@ class GrpcCommunication(CommunicationStrategy, GrpcClientWrapper):
             timer.mark("register")
             last_mark = "register"
 
-            self._get_or_create_channel(module_address, module_port)
-            timer.mark("channel_create")
-            last_mark = "channel_create"
-            stub = self._get_or_create_stub(gateway_service_pb2_grpc.GatewayServiceStub)
-            timer.mark("stub_create")
-            last_mark = "stub_create"
-
             try:  # noqa: PLW0717
                 grpc_metadata: list[tuple[str, str]] = []
                 if metadata:
@@ -467,7 +491,7 @@ class GrpcCommunication(CommunicationStrategy, GrpcClientWrapper):
                 cancelled = True
                 raise
             finally:
-                if cancelled and stub is not None:
+                if cancelled and stub is not None and task_id:
                     sig_t0 = time.perf_counter_ns()
                     sig_failure = ""
                     try:

{digitalkin-1.0.0.dev22 → digitalkin-1.0.0.dev23}/src/digitalkin/services/services_config.py

@@ -78,6 +78,12 @@ class ServicesConfig(BaseModel):
             self._strategies[name] = override if override is not None else defaults[name]
             self._configs[name] = services_config_params.get(name) or {}
 
+        # The secret service is backed by the UserProfileService — reuse the
+        # user_profile client_config (same host/port) when no dedicated secret
+        # config is registered, so GrpcSecret can build its channel.
+        if not self._configs.get("secret"):
+            self._configs["secret"] = self._configs.get("user_profile") or {}
+
     @classmethod
     def valid_strategy_names(cls) -> set[str]:
         """Get the list of valid strategy names.

{digitalkin-1.0.0.dev22 → digitalkin-1.0.0.dev23}/src/digitalkin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 1.0.0.dev22
+Version: 1.0.0.dev23
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -453,7 +453,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ag-ui-protocol>=0.1.18
-Requires-Dist: agentic-mesh-protocol==1.0.0a1
+Requires-Dist: agentic-mesh-protocol==1.0.0a2
 Requires-Dist: anyio>=4.13.0
 Requires-Dist: grpcio-health-checking==1.81.0
 Requires-Dist: grpcio-reflection==1.81.0

{digitalkin-1.0.0.dev22 → digitalkin-1.0.0.dev23}/src/digitalkin.egg-info/requires.txt

@@ -1,5 +1,5 @@
 ag-ui-protocol>=0.1.18
-agentic-mesh-protocol==1.0.0a1
+agentic-mesh-protocol==1.0.0a2
 anyio>=4.13.0
 grpcio-health-checking==1.81.0
 grpcio-reflection==1.81.0