digitalkin 1.0.0.dev4__tar.gz → 1.0.0.dev5__tar.gz

{digitalkin-1.0.0.dev4 → digitalkin-1.0.0.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 1.0.0.dev4
+Version: 1.0.0.dev5
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -458,7 +458,7 @@ Requires-Dist: anyio>=4.13.0
 Requires-Dist: grpcio-health-checking==1.80.0
 Requires-Dist: grpcio-reflection==1.80.0
 Requires-Dist: grpcio-status==1.80.0
-Requires-Dist: pydantic>=2.13.4
+Requires-Dist: pydantic>=2.12.4
 Requires-Dist: pydantic-settings>=2.14.0
 Requires-Dist: redis[hiredis]<8,>=7.4.0
 Provides-Extra: performance

{digitalkin-1.0.0.dev4 → digitalkin-1.0.0.dev5}/pyproject.toml

@@ -33,11 +33,11 @@
         "grpcio-health-checking==1.80.0",
         "grpcio-reflection==1.80.0",
         "grpcio-status==1.80.0",
-        "pydantic>=2.13.4",
+        "pydantic>=2.12.4",
         "pydantic-settings>=2.14.0",
         "redis[hiredis]>=7.4.0,<8",
     ]
-    version = "1.0.0.dev4"
+    version = "1.0.0.dev5"
 
     [project.optional-dependencies]
         performance = [ "uvloop>=0.21" ]

{digitalkin-1.0.0.dev4 → digitalkin-1.0.0.dev5}/src/digitalkin/__version__.py

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

{digitalkin-1.0.0.dev4 → digitalkin-1.0.0.dev5}/src/digitalkin/core/task_manager/module_runner.py

@@ -115,8 +115,16 @@ class ModuleRunner:
                 nonlocal first_logged
                 data = output_data.model_dump(mode="json")
                 if data.get("root", {}).get("protocol") == "stream.end":
+                    t_eos_write_start = time.perf_counter_ns()
                     await self._redis_client.xadd(stream_key, {"eos": b"true"})
                     await self._redis_client.expire(stream_key, 60)
+                    t_eos_write_end = time.perf_counter_ns()
+                    logger.info(
+                        "[close-debug] producer_eos_write: xadd_expire=%.2fms t_done_ns=%d task_id=%s",
+                        (t_eos_write_end - t_eos_write_start) / 1e6,
+                        t_eos_write_end,
+                        task_id,
+                    )
                     return
                 s = struct_pb2.Struct()
                 s.update(data)

{digitalkin-1.0.0.dev4 → digitalkin-1.0.0.dev5}/src/digitalkin/core/task_manager/redis/proto_streams.py

@@ -369,11 +369,13 @@ class ProtoStreamReader:
         """
         entries_since_save = 0
         while True:
+            t_xread_start = time.perf_counter_ns()
             result = await self._redis_client.xread(
                 {self._stream_key: self._last_id},
                 count=count,
                 block=block_ms,
             )
+            t_xread_end = time.perf_counter_ns()
             if not result:
                 continue
 
@@ -383,6 +385,13 @@ class ProtoStreamReader:
 
                     eos = fields.get(b"eos", b"")
                     if eos == b"true":
+                        logger.info(
+                            "[close-debug] reader_saw_eos: last_xread_block=%.2fms "
+                            "t_seen_ns=%d task_id=%s",
+                            (t_xread_end - t_xread_start) / 1e6,
+                            t_xread_end,
+                            self._task_id,
+                        )
                         await self._save_cursor()
                         return
 

{digitalkin-1.0.0.dev4 → digitalkin-1.0.0.dev5}/src/digitalkin/grpc_servers/gateway_servicer.py

@@ -555,8 +555,17 @@ class GatewayServicer:
         # consumed silently by ProtoStreamReader). Emit an explicit stream.end
         # sentinel so the wire contract is uniform: every stream ends with
         # exactly one stream.end entry, regardless of how it concluded.
+        t_after_reader = time.perf_counter_ns()
         seq += 1
         yield self._sentinel(seq, task_id, "stream.end")
+        t_after_yield = time.perf_counter_ns()
+        logger.info(
+            "[close-debug] gateway_stream_end: reader_to_yield=%.2fms "
+            "t_yielded_ns=%d task_id=%s",
+            (t_after_yield - t_after_reader) / 1e6,
+            t_after_yield,
+            task_id,
+        )
 
     async def _dial_consumer(  # noqa: C901, PLR0912, PLR0915
         self,

{digitalkin-1.0.0.dev4 → digitalkin-1.0.0.dev5}/src/digitalkin/modules/_base_module.py

@@ -695,11 +695,14 @@ class BaseModule(  # Module SDK base class requires many public methods # noqa:
             t3 = _t.perf_counter_ns()
             self._status = ModuleStatus.STOPPED
             logger.info(
-                "module.stop: cleanup=%.1fms flush=%.1fms eos=%.1fms total=%.1fms",
+                "[close-debug] module.stop: cleanup=%.2fms flush=%.2fms eos=%.2fms "
+                "total=%.2fms t_done_ns=%d task_id=%s",
                 (t1 - t0) / 1e6,
                 (t2 - t1) / 1e6,
                 (t3 - t2) / 1e6,
                 (t3 - t0) / 1e6,
+                t3,
+                self.context.session.current_ids().get("task_id", ""),
             )
         except Exception:
             self._status = ModuleStatus.FAILED

{digitalkin-1.0.0.dev4 → digitalkin-1.0.0.dev5}/src/digitalkin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 1.0.0.dev4
+Version: 1.0.0.dev5
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -458,7 +458,7 @@ Requires-Dist: anyio>=4.13.0
 Requires-Dist: grpcio-health-checking==1.80.0
 Requires-Dist: grpcio-reflection==1.80.0
 Requires-Dist: grpcio-status==1.80.0
-Requires-Dist: pydantic>=2.13.4
+Requires-Dist: pydantic>=2.12.4
 Requires-Dist: pydantic-settings>=2.14.0
 Requires-Dist: redis[hiredis]<8,>=7.4.0
 Provides-Extra: performance

{digitalkin-1.0.0.dev4 → digitalkin-1.0.0.dev5}/src/digitalkin.egg-info/requires.txt

@@ -4,7 +4,7 @@ anyio>=4.13.0
 grpcio-health-checking==1.80.0
 grpcio-reflection==1.80.0
 grpcio-status==1.80.0
-pydantic>=2.13.4
+pydantic>=2.12.4
 pydantic-settings>=2.14.0
 redis[hiredis]<8,>=7.4.0