digitalkin 0.3.2a2__py3-none-any.whl → 0.3.2a3__py3-none-any.whl

digitalkin/__version__.py

@@ -5,4 +5,4 @@ from importlib.metadata import PackageNotFoundError, version
 try:
     __version__ = version("digitalkin")
 except PackageNotFoundError:
-    __version__ = "0.3.2.a2"
+    __version__ = "0.3.2.a3"

digitalkin/core/task_manager/surrealdb_repository.py

@@ -145,10 +145,26 @@ class SurrealDBConnection(Generic[TSurreal]):
 
         Returns:
             Dict[str, Any]: The created record as returned by the database
+
+        Raises:
+            RuntimeError: If the database returns an error response
         """
         logger.debug("Creating record in %s with data: %s", table_name, data)
         result = await self.db.create(table_name, data)
         logger.debug("create result: %s", result)
+
+        # Check for error response from SurrealDB
+        if isinstance(result, dict) and "code" in result:
+            error_msg = result.get("message", result.get("information", "Unknown error"))
+            logger.error(
+                "SurrealDB create failed: %s (code: %s)",
+                error_msg,
+                result.get("code"),
+                extra={"table": table_name, "error": result},
+            )
+            msg = f"SurrealDB create failed in '{table_name}': {error_msg}"
+            raise RuntimeError(msg)
+
         return cast("list[dict[str, Any]] | dict[str, Any]", result)
 
     async def merge(

digitalkin/core/task_manager/task_executor.py

@@ -55,7 +55,9 @@ class TaskExecutor:
         async def signal_wrapper() -> None:
             """Create initial signal record and listen for signals."""
             try:
-                await channel.create(
+                # Create task record and capture the record ID directly
+                # This avoids a race condition where SELECT might run before CREATE completes
+                result = await channel.create(
                     "tasks",
                     SignalMessage(
                         task_id=task_id,
@@ -66,7 +68,22 @@ class TaskExecutor:
                         action=SignalType.START,
                     ).model_dump(),
                 )
-                await session.listen_signals()
+                # Store the record ID in session - required before starting live query
+                if isinstance(result, dict) and "id" in result:
+                    session.signal_record_id = result["id"]
+                    logger.debug(
+                        "Task signal record created",
+                        extra={"mission_id": mission_id, "task_id": task_id, "record_id": result["id"]},
+                    )
+                    # Only start listening if we have a valid record ID
+                    await session.listen_signals()
+                else:
+                    # Create failed - wait for cancellation instead of listening
+                    logger.error(
+                        "Failed to get record ID from task creation, waiting for cancellation",
+                        extra={"mission_id": mission_id, "task_id": task_id, "result": result},
+                    )
+                    await session.is_cancelled.wait()
             except asyncio.CancelledError:
                 logger.debug("Signal listener cancelled", extra={"mission_id": mission_id, "task_id": task_id})
             finally:

digitalkin/core/task_manager/task_session.py

@@ -229,17 +229,25 @@ class TaskSession:
         """Enhanced signal listener with comprehensive handling.
 
         Raises:
-            CancelledError: Asyncio when task cancelling
+            CancelledError: If task is cancelled during signal listening.
         """
         logger.info("Signal listener started", extra=self.session_ids)
+
+        # signal_record_id must be set by TaskExecutor before calling this method.
+        # If not set, we cannot filter signals correctly - abort early.
         if self.signal_record_id is None:
-            self.signal_record_id = (await self.db.select_by_task_id("tasks", self.task_id)).get("id")
+            logger.error(
+                "signal_record_id not set - cannot start signal listener without valid record ID",
+                extra=self.session_ids,
+            )
+            return
 
         live_id, live_signals = await self.db.start_live("tasks")
         try:
             async for signal in live_signals:
                 logger.debug("Signal received", extra={**self.session_ids, "signal": signal})
-                if self.cancelled:
+                # Check both cancelled and stream_closed to ensure clean shutdown
+                if self.cancelled or self.stream_closed:
                     break
 
                 if signal is None or signal["id"] == self.signal_record_id or "payload" not in signal:

{digitalkin-0.3.2a2.dist-info → digitalkin-0.3.2a3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.2a2
+Version: 0.3.2a3
 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.2a2.dist-info → digitalkin-0.3.2a3.dist-info}/RECORD

@@ -7,7 +7,7 @@ base_server/mock/__init__.py,sha256=YZFT-F1l_TpvJYuIPX-7kTeE1CfOjhx9YmNRXVoi-jQ,
 base_server/mock/mock_pb2.py,sha256=sETakcS3PAAm4E-hTCV1jIVaQTPEAIoVVHupB8Z_k7Y,1843
 base_server/mock/mock_pb2_grpc.py,sha256=BbOT70H6q3laKgkHfOx1QdfmCS_HxCY4wCOX84YAdG4,3180
 digitalkin/__init__.py,sha256=7LLBAba0th-3SGqcpqFO-lopWdUkVLKzLZiMtB-mW3M,162
-digitalkin/__version__.py,sha256=Ax_aYd7TkL4irPiANTsxbqr0Wwn-cJMP8BvBEqQjCzE,193
+digitalkin/__version__.py,sha256=9skx1vQVp8GGSDC8y9Ukcx2HKYinpCJ_xSH3ODFDgXs,193
 digitalkin/logger.py,sha256=8ze_tjt2G6mDTuQcsf7-UTXWP3UHZ7LZVSs_iqF4rX4,4685
 digitalkin/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 digitalkin/core/__init__.py,sha256=FJRcJ-B1Viyn-38L8XpOpZ8KOnf1I7PCDOAmKXLQhqc,71
@@ -22,9 +22,9 @@ digitalkin/core/task_manager/__init__.py,sha256=k9i-qIoee_1yXogyQolaVFDUQBIZU3EN
 digitalkin/core/task_manager/base_task_manager.py,sha256=9njUerhlJ7s3KA7IKgpVC3DYxq_HMHSqOcDhtZKkDiU,19439
 digitalkin/core/task_manager/local_task_manager.py,sha256=Z1gv4dCGD32LBSfMZJ4dGyYDe80lZRAyowTgGC6E4Vk,3534
 digitalkin/core/task_manager/remote_task_manager.py,sha256=zgccmnwwtB0nyeIZlL5Ji8SY1J89z_vjA4JD9ur7HzY,3082
-digitalkin/core/task_manager/surrealdb_repository.py,sha256=rUNPGK_flDdOl3-ritZif97zYHxQ7SIt_Aq_P2dymTQ,10092
-digitalkin/core/task_manager/task_executor.py,sha256=ByON-CxRBMbaDUgL6z5XYjbrUwV9X4DGktUBa0jqIPE,12005
-digitalkin/core/task_manager/task_session.py,sha256=iujisx6SHTP7qqQxZbMHG9xQTiDsGOsblg35zsejt30,14739
+digitalkin/core/task_manager/surrealdb_repository.py,sha256=VhgQBCvcclUmx9jyr6WCklf9iLXT2YfQN-YlKpApWWc,10712
+digitalkin/core/task_manager/task_executor.py,sha256=rPimQaGQTRwuRs9fUq8lySyZHlpXB-nEJDfYizXgm8w,13087
+digitalkin/core/task_manager/task_session.py,sha256=FWlFsS1we9-WVtT4_upOXj5l1pZcboKLpG2N35m56eA,15111
 digitalkin/grpc_servers/__init__.py,sha256=ZIRMJ1Lcas8yQ106GCup6hn2UBOsx1sNk8ap0lpEDnY,72
 digitalkin/grpc_servers/_base_server.py,sha256=ZVeCDwI7w7fFbPTXPkeJb_SOuLfd2T7za3T4oCu2UWY,18680
 digitalkin/grpc_servers/module_server.py,sha256=JE1hn1JTUPFakQOmhcTLTohC9n5-kVYUZD6ZKZKQBx8,7035
@@ -123,7 +123,7 @@ digitalkin/utils/dynamic_schema.py,sha256=y5csxjuqVHjWDpnTUzxbcUuI_wou9-ibRVHQlB
 digitalkin/utils/llm_ready_schema.py,sha256=JjMug_lrQllqFoanaC091VgOqwAd-_YzcpqFlS7p778,2375
 digitalkin/utils/package_discover.py,sha256=sa6Zp5Kape1Zr4iYiNrnZxiHDnqM06ODk6yfWHom53w,13465
 digitalkin/utils/schema_splitter.py,sha256=lMaReyLD4kGAZSQ9MSxpNZ4D4luWANVArFqNWV_bsko,14040
-digitalkin-0.3.2a2.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
+digitalkin-0.3.2a3.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
 modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 modules/archetype_with_tools_module.py,sha256=kJkVhAFWG0aDDqzupOXOnV3l8j3z5bEdWos_6Z9rUP8,7303
 modules/cpu_intensive_module.py,sha256=GZlirQDZdYuXrI46sv1q4RNAHZjL4EptHVQTvgK9zz8,8363
@@ -138,7 +138,7 @@ monitoring/digitalkin_observability/prometheus.py,sha256=gDmM9ySaVwPAe7Yg84pLxmE
 monitoring/tests/test_metrics.py,sha256=ugnYfAwqBPO6zA8z4afKTlyBWECTivacYSN-URQCn2E,5856
 services/filesystem_module.py,sha256=U4dgqtuDadaXz8PJ1d_uQ_1EPncBqudAQCLUICF9yL4,7421
 services/storage_module.py,sha256=Wz2MzLvqs2D_bnBBgtnujYcAKK2V2KFMk8K21RoepSE,6972
-digitalkin-0.3.2a2.dist-info/METADATA,sha256=H_Vj7SYn9cFu8jpO3-zwsiQFB_NC4x_xsQT1I746tYU,29721
-digitalkin-0.3.2a2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-digitalkin-0.3.2a2.dist-info/top_level.txt,sha256=AYVIesKrO0jnedQ-Muog9JBehG81WeTCNeOFoJgwsgE,51
-digitalkin-0.3.2a2.dist-info/RECORD,,
+digitalkin-0.3.2a3.dist-info/METADATA,sha256=GYGoIv2BFuAYj8A2nSXB2366r0zSznUar8x6JRHEVzc,29721
+digitalkin-0.3.2a3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+digitalkin-0.3.2a3.dist-info/top_level.txt,sha256=AYVIesKrO0jnedQ-Muog9JBehG81WeTCNeOFoJgwsgE,51
+digitalkin-0.3.2a3.dist-info/RECORD,,