digitalkin 0.3.2a2__tar.gz → 0.3.2a3__tar.gz

{digitalkin-0.3.2a2 → digitalkin-0.3.2a3}/PKG-INFO

@@ -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 → digitalkin-0.3.2a3}/pyproject.toml

@@ -25,7 +25,7 @@
         "Programming Language :: Python",
         "Topic :: Software Development :: Libraries",
     ]
-    version = "0.3.2.a2"
+    version = "0.3.2.a3"
 
     dependencies = [
         "agentic-mesh-protocol==0.2.1.dev1",

{digitalkin-0.3.2a2 → digitalkin-0.3.2a3}/src/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-0.3.2a2 → digitalkin-0.3.2a3}/src/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-0.3.2a2 → digitalkin-0.3.2a3}/src/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-0.3.2a2 → digitalkin-0.3.2a3}/src/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 → digitalkin-0.3.2a3}/src/digitalkin.egg-info/PKG-INFO

@@ -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