digitalkin 0.2.26__py3-none-any.whl → 0.3.0__py3-none-any.whl

digitalkin/modules/job_manager/task_session.py

@@ -0,0 +1,276 @@
+"""."""
+
+import asyncio
+import datetime
+from collections.abc import AsyncGenerator
+
+from digitalkin.logger import logger
+from digitalkin.models.module.task_monitor import HeartbeatMessage, SignalMessage, SignalType, TaskStatus
+from digitalkin.modules._base_module import BaseModule
+from digitalkin.modules.job_manager.surrealdb_repository import SurrealDBConnection
+
+
+class TaskSession:
+    """Task Session with lifecycle management.
+
+    The Session defined the whole lifecycle of a task as an epheneral context.
+    """
+
+    db: SurrealDBConnection
+    module: BaseModule
+
+    status: TaskStatus
+    signal_queue: AsyncGenerator | None
+
+    task_id: str
+    signal_record_id: str | None
+    heartbeat_record_id: str | None
+
+    started_at: datetime.datetime | None
+    completed_at: datetime.datetime | None
+
+    is_cancelled: asyncio.Event
+    _paused: asyncio.Event
+    _heartbeat_interval: datetime.timedelta
+    _last_heartbeat: datetime.datetime
+
+    def __init__(
+        self,
+        task_id: str,
+        db: SurrealDBConnection,
+        module: BaseModule,
+        heartbeat_interval: datetime.timedelta = datetime.timedelta(seconds=2),
+    ) -> None:
+        """."""
+        self.db = db
+        self.module = module
+
+        self.status = TaskStatus.PENDING
+        self.queue: asyncio.Queue = asyncio.Queue()
+
+        self.task_id = task_id
+        self.heartbeat = None
+        self.started_at = None
+        self.completed_at = None
+
+        self.signal_record_id = None
+        self.heartbeat_record_id = None
+
+        self.is_cancelled = asyncio.Event()
+        self._paused = asyncio.Event()
+        self._heartbeat_interval = heartbeat_interval
+
+        logger.info(
+            "TaskContext initialized for task: '%s'",
+            task_id,
+            extra={"task_id": task_id, "heartbeat_interval": heartbeat_interval},
+        )
+
+    @property
+    def cancelled(self) -> bool:
+        """."""
+        return self.is_cancelled.is_set()
+
+    @property
+    def paused(self) -> bool:
+        """."""
+        return self._paused.is_set()
+
+    async def send_heartbeat(self) -> bool:
+        """Rate-limited heartbeat with connection resilience.
+
+        Returns:
+            bool: True if heartbeat was successful, False otherwise
+        """
+        heartbeat = HeartbeatMessage(
+            task_id=self.task_id,
+            timestamp=datetime.datetime.now(datetime.timezone.utc),
+        )
+
+        if self.heartbeat_record_id is None:
+            try:
+                success = await self.db.create("heartbeats", heartbeat.model_dump())
+                logger.critical(f"{success=} | {'code' not in success}")
+                if "code" not in success:
+                    self.heartbeat_record_id = success.get("id")  # type: ignore
+                    self._last_heartbeat = heartbeat.timestamp
+                    return True
+            except Exception as e:
+                logger.error(
+                    "Heartbeat exception for task: '%s'",
+                    self.task_id,
+                    extra={"task_id": self.task_id, "error": str(e)},
+                    exc_info=True,
+                )
+            logger.error(
+                "Initial heartbeat failed for task: '%s'",
+                self.task_id,
+                extra={"task_id": self.task_id},
+            )
+            return False
+
+        if (heartbeat.timestamp - self._last_heartbeat) < self._heartbeat_interval:
+            logger.debug(
+                "Heartbeat skipped due to rate limiting for task: '%s' | delta=%s",
+                self.task_id,
+                heartbeat.timestamp - self._last_heartbeat,
+            )
+            return True
+
+        try:
+            success = await self.db.merge("heartbeats", self.heartbeat_record_id, heartbeat.model_dump())
+            if "code" not in success:
+                self._last_heartbeat = heartbeat.timestamp
+                return True
+        except Exception as e:
+            logger.error(
+                "Heartbeat exception for task: '%s'",
+                self.task_id,
+                extra={"task_id": self.task_id, "error": str(e)},
+                exc_info=True,
+            )
+        logger.warning(
+            "Heartbeat failed for task: '%s'",
+            self.task_id,
+            extra={"task_id": self.task_id},
+        )
+        return False
+
+    async def generate_heartbeats(self) -> None:
+        """Periodic heartbeat generator with cancellation support."""
+        logger.critical("Heartbeat started")
+        while not self.cancelled:
+            logger.debug(f"Heartbeat tick for task: '{self.task_id}' | {self.cancelled=}")
+            success = await self.send_heartbeat()
+            if not success:
+                logger.error("Heartbeat failed, cancelling task: '%s'", self.task_id, extra={"task_id": self.task_id})
+                await self._handle_cancel()
+                break
+            await asyncio.sleep(self._heartbeat_interval.total_seconds())
+
+    async def wait_if_paused(self) -> None:
+        """Block execution if task is paused."""
+        if self._paused.is_set():
+            logger.info("Task paused, waiting for resume: '%s'", self.task_id, extra={"task_id": self.task_id})
+            await self._paused.wait()
+
+    async def listen_signals(self) -> None:  # noqa: C901
+        """Enhanced signal listener with comprehensive handling.
+
+        Raises:
+            CancelledError: Asyncio when task cancelling
+        """
+        logger.info("Signal listener started for task: '%s'", self.task_id, extra={"task_id": self.task_id})
+        if self.signal_record_id is None:
+            self.signal_record_id = (await self.db.select_by_task_id("tasks", self.task_id)).get("id")
+
+        live_id, live_signals = await self.db.start_live("tasks")
+        try:
+            async for signal in live_signals:
+                logger.critical("Signal received for task '%s': %s", self.task_id, signal)
+                if self.cancelled:
+                    break
+
+                if signal is None or signal["id"] == self.signal_record_id or "payload" not in signal:
+                    continue
+
+                if signal["action"] == "cancel":
+                    await self._handle_cancel()
+                elif signal["action"] == "pause":
+                    await self._handle_pause()
+                elif signal["action"] == "resume":
+                    await self._handle_resume()
+                elif signal["action"] == "status":
+                    await self._handle_status_request()
+
+        except asyncio.CancelledError:
+            logger.debug("Signal listener cancelled for task: '%s'", self.task_id, extra={"task_id": self.task_id})
+            raise
+        except Exception as e:
+            logger.error(
+                "Signal listener fatal error for task: '%s'",
+                self.task_id,
+                extra={"task_id": self.task_id, "error": str(e)},
+                exc_info=True,
+            )
+        finally:
+            await self.db.stop_live(live_id)
+            logger.info("Signal listener stopped for task: '%s'", self.task_id, extra={"task_id": self.task_id})
+
+    async def _handle_cancel(self) -> None:
+        """Idempotent cancellation with acknowledgment."""
+        logger.critical("Handle cancel called")
+        if self.is_cancelled.is_set():
+            logger.debug(
+                "Cancel signal ignored - task already cancelled: '%s'", self.task_id, extra={"task_id": self.task_id}
+            )
+            return
+
+        logger.info("Cancelling task: '%s'", self.task_id, extra={"task_id": self.task_id})
+
+        self.status = TaskStatus.CANCELLED
+        self.is_cancelled.set()
+
+        # Resume if paused so cancellation can proceed
+        if self._paused.is_set():
+            self._paused.set()
+
+        await self.db.update(
+            "tasks",
+            self.signal_record_id,  # type: ignore
+            SignalMessage(
+                task_id=self.task_id,
+                action=SignalType.ACK_CANCEL,
+                status=self.status,
+            ).model_dump(),
+        )
+
+    async def _handle_pause(self) -> None:
+        """Pause task execution."""
+        if not self._paused.is_set():
+            logger.info("Pausing task: '%s'", self.task_id, extra={"task_id": self.task_id})
+            self._paused.set()
+
+        await self.db.update(
+            "tasks",
+            self.signal_record_id,  # type: ignore
+            SignalMessage(
+                task_id=self.task_id,
+                action=SignalType.ACK_PAUSE,
+                status=self.status,
+            ).model_dump(),
+        )
+
+    async def _handle_resume(self) -> None:
+        """Resume paused task."""
+        if self._paused.is_set():
+            logger.info("Resuming task: '%s'", self.task_id, extra={"task_id": self.task_id})
+            self._paused.clear()
+
+        await self.db.update(
+            "tasks",
+            self.signal_record_id,  # type: ignore
+            SignalMessage(
+                task_id=self.task_id,
+                action=SignalType.ACK_RESUME,
+                status=self.status,
+            ).model_dump(),
+        )
+
+    async def _handle_status_request(self) -> None:
+        """Send current task status."""
+        await self.db.update(
+            "tasks",
+            self.signal_record_id,  # type: ignore
+            SignalMessage(
+                action=SignalType.ACK_STATUS,
+                task_id=self.task_id,
+                status=self.status,
+            ).model_dump(),
+        )
+
+        logger.debug(
+            "Status report sent for task: '%s'",
+            self.task_id,
+            extra={"task_id": self.task_id},
+        )

digitalkin/modules/job_manager/taskiq_job_manager.py

@@ -19,7 +19,7 @@ from rstream import Consumer, ConsumerOffsetSpecification, MessageContext, Offse
 
 from digitalkin.logger import logger
 from digitalkin.models.module import InputModelT, SetupModelT
-from digitalkin.models.module.module import ModuleStatus
+from digitalkin.models.module.task_monitor import TaskStatus
 from digitalkin.modules._base_module import BaseModule
 from digitalkin.modules.job_manager.base_job_manager import BaseJobManager
 from digitalkin.modules.job_manager.taskiq_broker import STREAM, STREAM_RETENTION, TASKIQ_BROKER
@@ -279,7 +279,7 @@ class TaskiqJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
         msg = "stop_all_modules not implemented in TaskiqJobManager"
         raise NotImplementedError(msg)
 
-    async def get_module_status(self, job_id: str) -> ModuleStatus | None:
+    async def get_module_status(self, job_id: str) -> TaskStatus:
         """Query a module status."""
         msg = "get_module_status not implemented in TaskiqJobManager"
         raise NotImplementedError(msg)

digitalkin/modules/tool_module.py

@@ -3,8 +3,16 @@
 from abc import ABC
 
 from digitalkin.models.module import InputModelT, OutputModelT, SecretModelT, SetupModelT
-from digitalkin.modules._base_module import BaseModule # type: ignore
+from digitalkin.modules._base_module import BaseModule  # type: ignore
 
 
-class ToolModule(BaseModule[InputModelT, OutputModelT, SetupModelT, SecretModelT,], ABC):
+class ToolModule(
+    BaseModule[
+        InputModelT,
+        OutputModelT,
+        SetupModelT,
+        SecretModelT,
+    ],
+    ABC,
+):
     """ToolModule extends BaseModule to implement specific module types."""

digitalkin/modules/trigger_handler.py

@@ -1,14 +1,14 @@
 """Definition of the Trigger type."""
 
 from abc import ABC, abstractmethod
-from collections.abc import Callable, Coroutine
-from typing import Any, ClassVar, Generic
+from typing import ClassVar, Generic
 
+from digitalkin.mixins import BaseMixin
+from digitalkin.models.module.module_context import ModuleContext
 from digitalkin.models.module.module_types import InputModelT, OutputModelT, SetupModelT
-from digitalkin.modules._base_module import ModuleContext
 
 
-class TriggerHandler(ABC, Generic[InputModelT, SetupModelT, OutputModelT]):
+class TriggerHandler(ABC, BaseMixin, Generic[InputModelT, SetupModelT, OutputModelT]):
     """Base class for all input-trigger handlers.
 
     Each handler declares:
@@ -28,7 +28,6 @@ class TriggerHandler(ABC, Generic[InputModelT, SetupModelT, OutputModelT]):
         self,
         input_data: InputModelT,
         setup_data: SetupModelT,
-        callback: Callable[[Any], Coroutine[Any, Any, None]],
         context: ModuleContext,
     ) -> None:
         """Asynchronously processes the input data specific to Handler and streams results via the provided callback.
@@ -36,12 +35,14 @@ class TriggerHandler(ABC, Generic[InputModelT, SetupModelT, OutputModelT]):
         Args:
             input_data (InputModelT): The input data to be processed by the handler.
             setup_data (SetupModelT): The setup or configuration data required for processing.
-            callback (Callable[[Any], Coroutine[Any, Any, None]]): callback that stream results.
             context (ModuleContext): The context object containing module-specific information and resources.
 
         Returns:
             Any: The result of the processing, if applicable.
 
         Note:
+            self.send_message: : callback used to stream results.
+                (Callable[[OutputMdodelT], Coroutine[Any, Any, None]])
+
             The callback must be awaited to ensure results are streamed correctly during processing.
         """

digitalkin/services/cost/__init__.py

@@ -1,7 +1,14 @@
 """This module is responsible for handling the cost services."""
 
-from digitalkin.services.cost.cost_strategy import CostStrategy
+from digitalkin.services.cost.cost_strategy import CostConfig, CostData, CostStrategy, CostType
 from digitalkin.services.cost.default_cost import DefaultCost
 from digitalkin.services.cost.grpc_cost import GrpcCost
 
-__all__ = ["CostStrategy", "DefaultCost", "GrpcCost"]
+__all__ = [
+    "CostConfig",
+    "CostData",
+    "CostStrategy",
+    "CostType",
+    "DefaultCost",
+    "GrpcCost",
+]

digitalkin/services/storage/grpc_storage.py

@@ -128,7 +128,7 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
             resp = self.exec_grpc_query("UpdateRecord", req)
             return self._build_record_from_proto(resp.stored_data)
         except Exception:
-            logger.exception("gRPC UpdateRecord failed for %s:%s", collection, record_id)
+            logger.warning("gRPC UpdateRecord failed for %s:%s", collection, record_id)
             return None
 
     def _remove(self, collection: str, record_id: str) -> bool:

{digitalkin-0.2.26.dist-info → digitalkin-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.2.26
+Version: 0.3.0
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -456,32 +456,32 @@ Requires-Dist: digitalkin-proto>=0.1.16
 Requires-Dist: grpcio-health-checking>=1.71.0
 Requires-Dist: grpcio-reflection>=1.71.0
 Requires-Dist: grpcio-status>=1.71.0
-Requires-Dist: pydantic>=2.11.5
+Requires-Dist: pydantic>=2.11.10
+Requires-Dist: surrealdb>=1.0.6
+Provides-Extra: taskiq
+Requires-Dist: rstream>=0.31.0; extra == "taskiq"
+Requires-Dist: taskiq-aio-pika>=0.4.3; extra == "taskiq"
+Requires-Dist: taskiq-redis>=1.1.2; extra == "taskiq"
+Requires-Dist: taskiq[reload]>=0.11.18; extra == "taskiq"
 Provides-Extra: dev
-Requires-Dist: typos>=1.34.0; extra == "dev"
-Requires-Dist: ruff>=0.12.5; extra == "dev"
-Requires-Dist: mypy>=1.17.0; extra == "dev"
-Requires-Dist: pyright>=1.1.403; extra == "dev"
+Requires-Dist: typos>=1.38.0; extra == "dev"
+Requires-Dist: ruff>=0.13.3; extra == "dev"
+Requires-Dist: mypy>=1.18.2; extra == "dev"
+Requires-Dist: pyright>=1.1.406; extra == "dev"
 Requires-Dist: pre-commit>=4.3.0; extra == "dev"
 Requires-Dist: bump2version>=1.0.1; extra == "dev"
 Requires-Dist: build>=1.3.0; extra == "dev"
-Requires-Dist: twine>=6.1.0; extra == "dev"
-Requires-Dist: cryptography>=45.0.5; extra == "dev"
-Provides-Extra: examples
-Requires-Dist: openai>=1.75.0; extra == "examples"
+Requires-Dist: twine>=6.2.0; extra == "dev"
+Requires-Dist: cryptography>=46.0.2; extra == "dev"
 Provides-Extra: tests
 Requires-Dist: freezegun>=1.5.3; extra == "tests"
 Requires-Dist: hdrhistogram>=0.10.3; extra == "tests"
 Requires-Dist: grpcio-testing>=1.71.0; extra == "tests"
 Requires-Dist: psutil>=7.0.0; extra == "tests"
-Requires-Dist: pytest>=8.4.0; extra == "tests"
-Requires-Dist: pytest-asyncio>=1.1.0; extra == "tests"
-Requires-Dist: pytest-cov>=6.1.0; extra == "tests"
-Provides-Extra: taskiq
-Requires-Dist: rstream>=0.30.0; extra == "taskiq"
-Requires-Dist: taskiq-aio-pika>=0.4.3; extra == "taskiq"
-Requires-Dist: taskiq-redis>=1.1.0; extra == "taskiq"
-Requires-Dist: taskiq[reload]>=0.11.18; extra == "taskiq"
+Requires-Dist: pytest>=8.4.2; extra == "tests"
+Requires-Dist: pytest-asyncio>=1.2.0; extra == "tests"
+Requires-Dist: pytest-cov>=7.0.0; extra == "tests"
+Requires-Dist: hypothesis>=6.140.3; extra == "tests"
 Dynamic: license-file
 
 # DigitalKin Python SDK

{digitalkin-0.2.26.dist-info → digitalkin-0.3.0.dist-info}/RECORD

@@ -7,38 +7,51 @@ 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=BUsHyIFlBw1wdLzO9qTJ6akGPKXQMBxDjRgtc6eTS4E,191
-digitalkin/logger.py,sha256=eskzs365CjEhLZe4OrhilEY77Tf377VQi7EtGVGax5c,5007
+digitalkin/__version__.py,sha256=1cAQQLeheVwVxIu7yYP-6PyMVzQpLtDNXsgt9VoOgTQ,190
+digitalkin/logger.py,sha256=8ze_tjt2G6mDTuQcsf7-UTXWP3UHZ7LZVSs_iqF4rX4,4685
 digitalkin/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 digitalkin/grpc_servers/__init__.py,sha256=0cJBlwipSmFdXkyH3T0i6OJ1WpAtNsZgYX7JaSnkbtg,804
 digitalkin/grpc_servers/_base_server.py,sha256=32yrMCFAFE7FmdDL1WbuUlDuxRkaUJ_P7Dc-3_b40qs,18678
-digitalkin/grpc_servers/module_server.py,sha256=UDlUUEYquf0Fuq4tzvA9xs1L1PstlL5_9Da3UoQRzAc,10500
-digitalkin/grpc_servers/module_servicer.py,sha256=ZP2-HAS61rjAnosvW44bMRPCrd_ECZ5J31T_hBVgz0g,19132
+digitalkin/grpc_servers/module_server.py,sha256=bi1XreL5zaPxbaDfsFN7-qtEbo0sKFNeyyLJjWdYMck,10212
+digitalkin/grpc_servers/module_servicer.py,sha256=Df2mQYrGngGYHC6fPV3Othc5hCR2g8IuLxJx9IhzDAc,19345
 digitalkin/grpc_servers/registry_server.py,sha256=StY18DKYoPKQIU1SIzgito6D4_QA1aMVddZ8O2WGlHY,2223
 digitalkin/grpc_servers/registry_servicer.py,sha256=dqsKGHZ0LnaIvGt4ipaAuigd37sbJBndT4MAT029GsY,16471
 digitalkin/grpc_servers/utils/exceptions.py,sha256=SyOgvjggaUECYmSiqy8KJLHwHVt5IClSTxslHM-IZzI,931
 digitalkin/grpc_servers/utils/factory.py,sha256=jm6rFjiqmtSv7BIHNAOxsG9xXtSvWpx9TfzSQiX97MQ,5899
 digitalkin/grpc_servers/utils/grpc_client_wrapper.py,sha256=Vtt05KdePSFHq8FZUJ-bJo8l_d5vu7gdOv5UBtsgScg,2752
-digitalkin/grpc_servers/utils/models.py,sha256=hiwiGHy3sEBBFdOUBCoC1n_7GVeJhjRmi0sl4tCEJy0,8965
+digitalkin/grpc_servers/utils/models.py,sha256=ZQ1Gk4zmLRX_ueuwFOwXgtyG9OEPSzFa_91I_CXZmZ0,8957
 digitalkin/grpc_servers/utils/types.py,sha256=rQ78s4nAet2jy-NIDj_PUWriT0kuGHr_w6ELjmjgBao,539
+digitalkin/mixins/__init__.py,sha256=d6ljaoyJZJT9XxOrXZG5FVNvbLURb3_CZrkp4GPZWYM,590
+digitalkin/mixins/base_mixin.py,sha256=uLkg6MbDtVc9DysjdfNIGKahxQLnnjuL3DYpuyNLbk8,486
+digitalkin/mixins/callback_mixin.py,sha256=90nHm9-pbKT14GAy3CB3fsBtpYu5IH0woOQdNLM2e_Y,836
+digitalkin/mixins/chat_history_mixin.py,sha256=cGLJb1KWY1WgC5akV6WQfk2a0blN-RQmU5TDs_ppUGM,4111
+digitalkin/mixins/cost_mixin.py,sha256=nFqhLsRHdXAt3GOH6qI8sqX9KW3leTPW3gaclensBVw,2274
+digitalkin/mixins/file_history_mixin.py,sha256=4rDpm-TVcD9ihZGPuqr73y1_4FnUzBlWUSoje-pQIIU,3759
+digitalkin/mixins/filesystem_mixin.py,sha256=myCUmocdumG7XSN1acnn4hw1e1QbkngbnBtQtrVZQHU,1604
+digitalkin/mixins/logger_mixin.py,sha256=FdVIlPXOQ7eo8DUlMmsygO9L8bis-Jhj_zw83RiE6wo,2150
+digitalkin/mixins/storage_mixin.py,sha256=ptZ4a2bydIa48q0V9e395vWHTu7yw4A6rI4jwKY6gwI,2392
 digitalkin/models/__init__.py,sha256=hDHtUfswaNh8wo4NZaBItg9JqC0uNSRqXArNWSrGynY,163
-digitalkin/models/module/__init__.py,sha256=fgTVbsNmLZgM43Vy-Ea5-g0EG3pncmAmJl9nk-OQdzo,561
-digitalkin/models/module/module.py,sha256=NWOwCeD_NH0NL89AF_NJ0SwE7G1n-HTcRalodLCDmpM,749
-digitalkin/models/module/module_context.py,sha256=AL6bzwjYfYPLHb1Hu5kKBjaIOxCpBCPMvxEnapdyrX4,1003
-digitalkin/models/module/module_types.py,sha256=s8qpIr1dwsQSq13iP3v22u2OdZNKbj4Y6fGQXDcO124,3539
-digitalkin/models/services/__init__.py,sha256=HsW7MUGFPvH7Ri28WN4BHHBfEQk5dzU_9FOWAc-0lSE,56
+digitalkin/models/module/__init__.py,sha256=nFPHp0JcAZq9ISkcSu1y2zeNyXH3V4j10oMjHzMBYBU,599
+digitalkin/models/module/module.py,sha256=k0W8vfJJFth8XdDzkHm32SyTuSf3h2qF0hSrxAfGF1s,956
+digitalkin/models/module/module_context.py,sha256=nhZ_Pw9lWcq5B2PPy-Rf3N0S7MXz9Ehh5Nb-4tEOmmM,3764
+digitalkin/models/module/module_types.py,sha256=kkOst1OWeQVzLdf0P8JBlZqf2LLWj14-bQHyYvwUN8Y,3562
+digitalkin/models/module/task_monitor.py,sha256=sEzpfmTg1cXC1ieD72YL3zwkSE6D5CIHWRi2CgK3GeQ,1288
+digitalkin/models/services/__init__.py,sha256=jhfVw6egq0OcHmos_fypH9XFehbHTBw09wluVFVFEyw,226
 digitalkin/models/services/cost.py,sha256=QTEuFD6xz62nob0z4ksE-INJWcZ-iFiuNW5mvXhpFes,1599
-digitalkin/models/services/storage.py,sha256=cYTVIriGKiprF9OerhSxmc_jM6fUTVwmeon1yQCinkE,143
+digitalkin/models/services/storage.py,sha256=k0lQUfTY9qP-PPxC1piuOhw-DfH7ao-yNvykRDJWdko,1145
 digitalkin/modules/__init__.py,sha256=VwVbKok81NGyPIBZgEj_SR-59G8tTlSb4eBJI9W6Vx4,281
-digitalkin/modules/_base_module.py,sha256=AavwuKQo-oKzJ_xEznmzV1fPeoW8P7wP3MW39_MVsMY,19098
+digitalkin/modules/_base_module.py,sha256=y_Pw2twqlAcsJlUUa-8QKdITSrZv6KpiwVw2X7mNads,18392
 digitalkin/modules/archetype_module.py,sha256=lOe3yYufwfylZR_VGy1w-zqdqVaMI_JANfKkbH9eODE,471
-digitalkin/modules/tool_module.py,sha256=h9oo2vrFJdiBLW2qL_m79a9hEzfXV0L6bZd4KD5OSQs,422
-digitalkin/modules/trigger_handler.py,sha256=TpxFc_xkYia9B9rAvHkj02e818W08JOfIqssQqbsCpM,1785
-digitalkin/modules/job_manager/base_job_manager.py,sha256=4qrCw68ZAleZp8vnpd1qYH320aQO0ApH2cZrKM87Grk,6121
+digitalkin/modules/tool_module.py,sha256=rLJfdYMWddiUgPLBDGG_UcHnRCHEeaqX6Fdp4DOUaCs,472
+digitalkin/modules/trigger_handler.py,sha256=qPNMi-8NHqscOxciHeaXtpwjXApT3YzjMF23zQAjaZY,1770
+digitalkin/modules/job_manager/base_job_manager.py,sha256=iDv9caispLmM7mqPgG_g7oaACMvNW7i7MTzBeSlmNp0,6355
 digitalkin/modules/job_manager/job_manager_models.py,sha256=onHy-DfInLZQveniMIWIKwTKSQjojz500JvHB54x93c,1129
-digitalkin/modules/job_manager/single_job_manager.py,sha256=nMC1BX6G62Ehmam7aVSAlKW9ShJDB1ntuMq7KbD_sSI,10564
+digitalkin/modules/job_manager/single_job_manager.py,sha256=Fw64gGRdQuTLZO-HChBTbBlZvOPyNaeb_p-HhcQpGJE,11177
+digitalkin/modules/job_manager/surrealdb_repository.py,sha256=3Kaz-PqfaMzSpX2RmbgbiO2pQcLVQ1tcspdCMPm7VTI,7712
+digitalkin/modules/job_manager/task_manager.py,sha256=VpRoscXkNf8LBWw3jCE6MwPvtUb2Q8V_dGi059fVyVw,14402
+digitalkin/modules/job_manager/task_session.py,sha256=rDKWgXf0sq7gBd6KODLPYjov3GIIyHAmFxAKTI-n1J0,9655
 digitalkin/modules/job_manager/taskiq_broker.py,sha256=4q3U03SNlb1ypup0VOa73KHLKJ9peuZUPCR5eFLMNAk,7498
-digitalkin/modules/job_manager/taskiq_job_manager.py,sha256=JxvNS95J_kEuRP08PvC3pQIO3DL8V1GRv7u6WUG59xg,10254
+digitalkin/modules/job_manager/taskiq_job_manager.py,sha256=NHSGsqqKCWQQqx0FqM3c4tf011vFlqX1npHkcv0kEOU,10249
 digitalkin/services/__init__.py,sha256=LqGk_5DJy8Bzz62ajIq9jCeYNKQUIgtSCpafZk15FLc,910
 digitalkin/services/base_strategy.py,sha256=yA9KUJGRKuuaxA6l3GcMv8zKfWoIsW03UxJT80Yea2I,766
 digitalkin/services/services_config.py,sha256=JnyzZcG7OYBelwgn-wdVgY2n3yFTEkwLPHRZB8Tjw10,7468
@@ -46,7 +59,7 @@ digitalkin/services/services_models.py,sha256=5zXkWcfKnXGwQi9sN4OAL3XrgqOcmsTl8a
 digitalkin/services/agent/__init__.py,sha256=vJc8JN0pdtA8ecypLBeHrwAUIW6H2C8NyW-dk24rTpk,244
 digitalkin/services/agent/agent_strategy.py,sha256=42Q9RciHX6tg3CgDQkbrlIx4h_TX0WIuSpLmCjitVmA,492
 digitalkin/services/agent/default_agent.py,sha256=4N_E_eQxJGOx1KVUUg5jNOje-3ncMxF3ePB-uDuGrJc,345
-digitalkin/services/cost/__init__.py,sha256=Wi9ZB4LSXFsUYgkX-V1UJQkVXYDNDpp8q2dXccR2uRM,303
+digitalkin/services/cost/__init__.py,sha256=sD_a5LrnLluASOC5m5vgIqjaco-MzZJd6XhillIBHr0,400
 digitalkin/services/cost/cost_strategy.py,sha256=MpPX33P_S5b2by6F4zT-rcyeRuh2V4NYPZe05VpDOGQ,2649
 digitalkin/services/cost/default_cost.py,sha256=XE7kNFde8NmbulU9m1lc3mi-vHFkbaJf0XHUc0D4UHE,3945
 digitalkin/services/cost/grpc_cost.py,sha256=cGtb0atPXSEEOrNIWee-o3ScfNRSAFXJGDu0vcWT6zg,6295
@@ -69,21 +82,21 @@ digitalkin/services/snapshot/default_snapshot.py,sha256=Mb8QwWRsHh9I_tN0ln_ZiFa1
 digitalkin/services/snapshot/snapshot_strategy.py,sha256=B1TU3V_k9A-OdqBkdyc41-ihnrW5Btcwd1KyQdHT46A,898
 digitalkin/services/storage/__init__.py,sha256=T-ocYLLphudkQgzvG47jBOm5GQsRFRIGA88y7Ur4akg,341
 digitalkin/services/storage/default_storage.py,sha256=D8e-UYUkb2GvDEHMWcN3EkcIKXWA8DrsaQsXVjoXAYQ,7975
-digitalkin/services/storage/grpc_storage.py,sha256=3ZHGq3wnbF8dZjJcYN1EGTEUXA4rur0XEGtTBwAcGB4,7206
+digitalkin/services/storage/grpc_storage.py,sha256=7RPeqkjG4JdW1lhjGpKx1qF5fp_LE-sST7fp5oT6uD0,7204
 digitalkin/services/storage/storage_strategy.py,sha256=sERF5tIJnzpb1iNqTXic9xRkGaXMifo6kb709ubB-Yo,8848
 digitalkin/utils/__init__.py,sha256=sJnY-ZUgsjMfojAjONC1VN14mhgIDnzyOlGkw21rRnM,28
 digitalkin/utils/arg_parser.py,sha256=nvjI1pKDY1HfS0oGcMQPtdTQcggXLtpxXMbnMxNEKRU,3109
 digitalkin/utils/development_mode_action.py,sha256=TqRuAF_A7bDD4twRB4PnZcRoNeaiAnEdxM5kvy4aoaA,1511
 digitalkin/utils/llm_ready_schema.py,sha256=JjMug_lrQllqFoanaC091VgOqwAd-_YzcpqFlS7p778,2375
 digitalkin/utils/package_discover.py,sha256=3e9-6Vf3yAAv2VkHHVK4QVqHJBxQqg3d8uuDTsXph24,13471
-digitalkin-0.2.26.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
+digitalkin-0.3.0.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
 modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 modules/cpu_intensive_module.py,sha256=ejB9XPnFfA0uCuFUQbM3fy5UYfqqAlF36rv_P5Ri8ho,8363
 modules/minimal_llm_module.py,sha256=Ijld__ZnhzfLwpXD1XVkLZ7jyKZKyOFZczOpiPttJZc,11216
 modules/text_transform_module.py,sha256=bwPSnEUthZQyfLwcTLo52iAxItAoknkLh8Y3m5aywaY,7251
 services/filesystem_module.py,sha256=71Mcja8jCQqiqFHPdsIXplFIHTvgkxRhp0TRXuCfgkk,7430
 services/storage_module.py,sha256=ybTMqmvGaTrR8PqJ4FU0cwxaDjT36TskVrGoetTGmno,6955
-digitalkin-0.2.26.dist-info/METADATA,sha256=43WocXOsMwRLYxm4eKVI8li_OiQub-hLnlrKnfKtny0,30579
-digitalkin-0.2.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-digitalkin-0.2.26.dist-info/top_level.txt,sha256=gcjqlyrZuLjIyxrOIavCQM_olpr6ND5kPKkZd2j0xGo,40
-digitalkin-0.2.26.dist-info/RECORD,,
+digitalkin-0.3.0.dist-info/METADATA,sha256=_6KDoMRgjmgnAXmPLUVpNY1EcyN86SE5c_zvtNo-K8w,30588
+digitalkin-0.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+digitalkin-0.3.0.dist-info/top_level.txt,sha256=gcjqlyrZuLjIyxrOIavCQM_olpr6ND5kPKkZd2j0xGo,40
+digitalkin-0.3.0.dist-info/RECORD,,