digitalkin 0.3.2a2__tar.gz → 0.3.2.dev2__tar.gz

{digitalkin-0.3.2a2 → digitalkin-0.3.2.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.2a2
+Version: 0.3.2.dev2
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -434,32 +434,32 @@ License: Attribution-NonCommercial-ShareAlike 4.0 International
         Creative Commons may be contacted at creativecommons.org.
         https://creativecommons.org/licenses/by-nc-sa/4.0/
         
-Project-URL: Documentation, https://github.com/DigitalKin-ai/digitalkin
 Project-URL: Homepage, https://github.com/DigitalKin-ai/digitalkin
+Project-URL: Documentation, https://github.com/DigitalKin-ai/digitalkin
 Project-URL: Issues, https://github.com/DigitalKin-ai/digitalkin/issues
-Keywords: agent,digitalkin,gprc,kin,sdk
+Keywords: digitalkin,kin,agent,gprc,sdk
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
-Classifier: License :: Other/Proprietary License
 Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python
-Classifier: Topic :: Software Development :: Libraries
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: License :: Other/Proprietary License
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: agentic-mesh-protocol==0.2.1.dev1
+Requires-Dist: agentic-mesh-protocol==0.2.1.dev0
 Requires-Dist: grpcio-health-checking>=1.76.0
 Requires-Dist: grpcio-reflection>=1.76.0
 Requires-Dist: grpcio-status>=1.76.0
 Requires-Dist: pydantic>=2.12.5
 Requires-Dist: surrealdb>=1.0.7
 Provides-Extra: taskiq
-Requires-Dist: rstream>=0.40.1; extra == "taskiq"
+Requires-Dist: rstream>=0.40.0; extra == "taskiq"
 Requires-Dist: taskiq-aio-pika>=0.5.0; extra == "taskiq"
 Requires-Dist: taskiq-redis>=1.2.0; extra == "taskiq"
 Requires-Dist: taskiq[reload]>=0.12.1; extra == "taskiq"

{digitalkin-0.3.2a2 → digitalkin-0.3.2.dev2}/pyproject.toml

@@ -1,34 +1,34 @@
 [build-system]
-    build-backend = "setuptools.build_meta"
     requires      = [ "setuptools >= 75.8.2", "wheel" ]
+    build-backend = "setuptools.build_meta"
 
 [project]
 
-    description     = "SDK to build kin used in DigitalKin"
-    license         = { file = "LICENSE" }
     name            = "digitalkin"
+    description     = "SDK to build kin used in DigitalKin"
     readme          = "README.md"
     requires-python = ">=3.10"
+    license         = { file = "LICENSE" }
 
-    keywords = [ "agent", "digitalkin", "gprc", "kin", "sdk" ]
+    keywords = [ "digitalkin", "kin", "agent", "gprc", "sdk" ]
 
+    version = "0.3.2.dev2"
     classifiers = [
         "Development Status :: 3 - Alpha",
         "Intended Audience :: Developers",
-        "License :: Other/Proprietary License",
         "Operating System :: OS Independent",
-        "Programming Language :: Python :: 3 :: Only",
+        "Topic :: Software Development :: Libraries",
+        "Programming Language :: Python",
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
         "Programming Language :: Python :: 3.12",
         "Programming Language :: Python :: 3.13",
-        "Programming Language :: Python",
-        "Topic :: Software Development :: Libraries",
+        "Programming Language :: Python :: 3 :: Only",
+        "License :: Other/Proprietary License",
     ]
-    version = "0.3.2.a2"
 
     dependencies = [
-        "agentic-mesh-protocol==0.2.1.dev1",
+        "agentic-mesh-protocol==0.2.1.dev0",
         "grpcio-health-checking>=1.76.0",
         "grpcio-reflection>=1.76.0",
         "grpcio-status>=1.76.0",
@@ -38,81 +38,81 @@
 
     [project.optional-dependencies]
         taskiq = [
-            "rstream>=0.40.1",
+            "rstream>=0.40.0",
             "taskiq-aio-pika>=0.5.0",
             "taskiq-redis>=1.2.0",
             "taskiq[reload]>=0.12.1",
         ]
 
     [project.urls]
-        Documentation = "https://github.com/DigitalKin-ai/digitalkin"
         Homepage      = "https://github.com/DigitalKin-ai/digitalkin"
+        Documentation = "https://github.com/DigitalKin-ai/digitalkin"
         Issues        = "https://github.com/DigitalKin-ai/digitalkin/issues"
 
     [[project.authors]]
-        email = "contact@digitalkin.ai"
         name  = "DigitalKin.ai"
+        email = "contact@digitalkin.ai"
 
 [dependency-groups]
+    dev = [
+        "typos>=1.40.0",
+        "ruff>=0.14.8",
+        "mypy>=1.19.0",
+        "pyright>=1.1.407",
+        "pre-commit>=4.5.0",
+        "bump-my-version>=1.2.4",
+        "build>=1.3.0",
+        "twine>=6.2.0",
+        "cryptography>=46.0.3",
+    ]
     tests = [
         "freezegun>=1.5.5",
-        "grpcio-testing>=1.76.0",
         "hdrhistogram>=0.10.3",
+        "grpcio-testing>=1.76.0",
         "psutil>=7.1.3",
+        "pytest>=9.0.2",
         "pytest-asyncio>=1.3.0",
         "pytest-cov>=7.0.0",
-        "pytest-html==4.2.0",
+        "pytest-html==4.1.1",
         "pytest-json-report==1.5.0",
         "pytest-timeout>=2.4.0",
-        "pytest>=9.0.2",
-    ]
-    dev = [
-        "build>=1.3.0",
-        "bump-my-version>=1.2.5",
-        "cryptography>=46.0.4",
-        "mypy>=1.19.1",
-        "pre-commit>=4.5.1",
-        "pyright>=1.1.407",
-        "ruff>=0.14.14",
-        "twine>=6.2.0",
-        "typos>=1.42.3",
     ]
     docs = [
-        "griffe-inherited-docstrings>=1.1.2",
+        "mike>=2.1.3",
         "markdown-callouts>=0.4.0",
         "markdown-exec>=1.12.1",
-        "mike>=2.1.3",
+        "mkdocs>=1.6.1",
+        "mkdocs-coverage>=2.0.0",
+        "mkdocs-llmstxt>=0.5.0",
+        "mkdocs-redirects>=1.2.2",
+        "mkdocstrings>=1.0.0",
+        "griffe-inherited-docstrings>=1.1.2",
         "mkdocs-autorefs>=1.4.3",
         "mkdocs-awesome-pages-plugin>=2.10.1",
-        "mkdocs-coverage>=2.0.0",
         "mkdocs-git-committers-plugin-2>=2.5.0",
-        "mkdocs-git-revision-date-localized-plugin>=1.5.1",
+        "mkdocs-git-revision-date-localized-plugin>=1.5.0",
         "mkdocs-glightbox>=0.5.2",
-        "mkdocs-include-markdown-plugin>=7.2.1",
+        "mkdocs-include-markdown-plugin>=7.2.0",
         "mkdocs-literate-nav>=0.6.2",
-        "mkdocs-llmstxt>=0.5.0",
         "mkdocs-material[imaging]>=9.7.0",
         "mkdocs-minify-plugin>=0.8.0",
-        "mkdocs-open-in-new-tab>=1.0.8",
-        "mkdocs-redirects>=1.2.2",
         "mkdocs-section-index>=0.3.10",
-        "mkdocs>=1.6.1",
         "mkdocstrings-python>=2.0.1",
-        "mkdocstrings>=1.0.2",
+        "mkdocs-open-in-new-tab>=1.0.8",
         "tomli>=2.3.0",
     ]
 [tool.setuptools]
     package-dir = { "" = "src" }
 
     [tool.setuptools.packages.find]
-        where = [ "examples", "src" ]
-[tool.ruff]
+        where = [ "src", "examples" ]
 
-    indent-width   = 4
-    line-length    = 120
-    show-fixes     = true
+[tool.ruff]
     src            = [ "src" ]
     target-version = "py310"
+    line-length    = 120
+    indent-width   = 4
+    show-fixes     = true
     unsafe-fixes   = true
     # Enable preview features.
     exclude = [
@@ -135,59 +135,59 @@
         "buck-out",
         "build",
         "dist",
-        "docs/*",
-        "examples/*",
+        "node_modules",
+        "venv",
         "factory.py",
         "generate_certificates.py",
-        "node_modules",
+        "examples/*",
         "tests/*",
-        "venv",
+        "docs/*",
     ]
     preview = true
 
     [tool.ruff.lint]
         select = [
-            "A",     # flake8-builtins
             "ANN",   # flake8-annotations
-            "ARG",   # flake8-unused-arguments
             "ASYNC", # flake8-async
-            "C4",    # flake8-comprehensions
-            "C90",   # mccabe (complex-structure)
+            "S",     # flake8-bandit
+            "FBT",   # flake8-boolean-trap
+            "A",     # flake8-builtins
             "COM",   # flake8-commas
-            "D",     # pydocstyle
-            "DOC",   # pydoclint
+            "C4",    # flake8-comprehensions
             "DTZ",   # flake8-datetimez
-            "E",     # pycodestyle-errors
             "EM",    # flake8-errmsg
-            "ERA",   # eradicate
-            "F",     # pyflakes
-            "FBT",   # flake8-boolean-trap
-            "FLY",   # flynt
-            "FURB",  # Refurb
-            "G",     # flake8-logging-format
-            "I",     # isort
-            "ICN",   # flake8-import-conventions
             "ISC",   # flake8-implicit-str-concat
+            "ICN",   # flake8-import-conventions
+            "G",     # flake8-logging-format
             "LOG",   # flake8-logging
-            "N",     # pep8-naming
-            "PERF",  # Perflint
             "PIE",   # flake8-pie
-            "PL",    # Pylint
-            "PL",    # Pylint
-            "PT",    # flake8-pytest-style
+            "T20",   # flake8-print
             "PYI",   # flake8-pyi
+            "PT",    # flake8-pytest-style
             "Q",     # flake8-quotes
-            "RET",   # flake8-return
             "RSE",   # flake8-raise
-            "RUF",   # Ruff-specific rules
-            "S",     # flake8-bandit
-            "SIM",   # flake8-simplify
+            "RET",   # flake8-return
             "SLF",   # flake8-self
-            "T20",   # flake8-print
+            "SIM",   # flake8-simplify
             "TC",    # flake8-type-checking
+            "ARG",   # flake8-unused-arguments
+            "FLY",   # flynt
+            "C90",   # mccabe (complex-structure)
+            "PL",    # Pylint
             "TRY",   # tryceratops
-            "UP",    # pyupgrade
+            "PERF",  # Perflint
+            "E",     # pycodestyle-errors
             "W",     # pycodestyle-warnings
+            "F",     # pyflakes
+            "I",     # isort
+            "D",     # pydocstyle
+            "UP",    # pyupgrade
+            "N",     # pep8-naming
+            "ERA",   # eradicate
+            "RUF",   # Ruff-specific rules
+            "DOC",   # pydoclint
+            "FURB",  # Refurb
+            "PL",    # Pylint
         ]
 
         ignore = [
@@ -203,66 +203,66 @@
 
         [tool.ruff.lint.per-file-ignores]
             "tests/**/*.py" = [
+                "S101",
+                "SLF001",
+                "PLC2701",
+                "PLR6301",
+                "TRY002",
+                "FBT001",
+                "FBT002",
                 "ANN001",
-                "ANN002",
                 "ANN003",
-                "ANN201",
-                "ANN401",
-                "ARG001",
+                "ANN002",
                 "ARG002",
-                "C901",
-                "D100",
-                "D101",
-                "D102",
+                "PLR0914",
+                "PLR2004",
+                "PLR0915",
+                "PLR1702",
+                "D417",
+                "ARG001",
                 "D103",
+                "D102",
+                "D101",
                 "D107",
-                "D417",
+                "N802",
                 "DOC201",
                 "DOC501",
-                "E501",
-                "ERA001",
-                "FBT001",
-                "FBT002",
-                "G004",
-                "N802",
                 "PERF203",
-                "PLC2701",
+                "C901",
                 "PLR0911",
+                "G004",
+                "ERA001",
+                "E501",
+                "ANN401",
                 "PLR0912",
-                "PLR0914",
-                "PLR0915",
-                "PLR1702",
-                "PLR2004",
-                "PLR6301",
-                "S101",
-                "SLF001",
-                "TRY002",
+                "ANN201",
+                "D100",
             ]
 
         [tool.ruff.lint.pylint]
             max-args = 10
 
     [tool.ruff.format]
-        docstring-code-format     = true
+        quote-style               = "double"
         indent-style              = "space"
         line-ending               = "auto"
-        quote-style               = "double"
+        docstring-code-format     = true
         skip-magic-trailing-comma = false
 
 [tool.pyright]
-    exclude = [ "**/.venv", "**/__pycache__" ]
     include = [ "src" ]
+    exclude = [ "**/.venv", "**/__pycache__" ]
 
 [tool.mypy]
+    exclude = [ "tests", "examples" ]
 
-    exclude = [ "examples", "tests" ]
 [tool.pytest.ini_options]
-
-    asyncio_mode = "auto"
-    python_files = "test_*.py"
     testpaths    = [ "tests" ]
-    log_cli      = false
-    log_level    = "ERROR"
+    python_files = "test_*.py"
+    asyncio_mode = "auto"
+
+    log_cli   = false
+    log_level = "ERROR"
 
     # Default timeout for tests (in seconds)
     timeout        = 10
@@ -270,10 +270,10 @@
 
     # Custom markers
     markers = [
-        "edge_case: marks tests for boundary conditions and edge cases",
-        "grpc: marks tests for gRPC service functionality",
         "integration: marks tests that require real SurrealDB connection (deselect with '-m \"not integration\"')",
-        "regression: marks tests for previously fixed bugs",
+        "grpc: marks tests for gRPC service functionality",
         "smoke: marks critical path tests that should always pass",
+        "edge_case: marks tests for boundary conditions and edge cases",
         "validation: marks tests for input validation and schema checking",
+        "regression: marks tests for previously fixed bugs",
     ]

{digitalkin-0.3.2a2 → digitalkin-0.3.2.dev2}/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.dev2"

{digitalkin-0.3.2a2 → digitalkin-0.3.2.dev2}/src/digitalkin/core/job_manager/single_job_manager.py

@@ -5,7 +5,7 @@ import datetime
 import uuid
 from collections.abc import AsyncGenerator, AsyncIterator
 from contextlib import asynccontextmanager
-from typing import TYPE_CHECKING, Any
+from typing import Any
 
 import grpc
 
@@ -13,13 +13,10 @@ from digitalkin.core.common import ConnectionFactory, ModuleFactory
 from digitalkin.core.job_manager.base_job_manager import BaseJobManager
 from digitalkin.core.task_manager.local_task_manager import LocalTaskManager
 from digitalkin.core.task_manager.task_session import TaskSession
-
-if TYPE_CHECKING:
-    from digitalkin.core.task_manager.surrealdb_repository import SurrealDBConnection
 from digitalkin.logger import logger
 from digitalkin.models.core.task_monitor import TaskStatus
-from digitalkin.models.module.base_types import InputModelT, OutputModelT, SetupModelT
 from digitalkin.models.module.module import ModuleCodeModel
+from digitalkin.models.module.module_types import InputModelT, OutputModelT, SetupModelT
 from digitalkin.modules._base_module import BaseModule
 from digitalkin.services.services_models import ServicesMode
 
@@ -32,6 +29,10 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
     to handle their output data.
     """
 
+    async def start(self) -> None:
+        """Start manager."""
+        self.channel = await ConnectionFactory.create_surreal_connection("task_manager", datetime.timedelta(seconds=5))
+
     def __init__(
         self,
         module_class: type[BaseModule],
@@ -54,11 +55,6 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
         super().__init__(module_class, services_mode, task_manager)
 
         self._lock = asyncio.Lock()
-        self.channel: SurrealDBConnection | None = None
-
-    async def start(self) -> None:
-        """Start manager."""
-        self.channel = await ConnectionFactory.create_surreal_connection("task_manager", datetime.timedelta(seconds=5))
 
     async def generate_config_setup_module_response(self, job_id: str) -> SetupModelT | ModuleCodeModel:
         """Generate a stream consumer for a module's output data.
@@ -90,10 +86,7 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
                 message=f"Module {job_id} did not respond within 30 seconds",
             )
         finally:
-            logger.debug(
-                "Config setup response retrieved",
-                extra={"job_id": job_id, "queue_empty": session.queue.empty()},
-            )
+            logger.info(f"{job_id=}: {session.queue.empty()}")
 
     async def create_config_setup_instance_job(
         self,
@@ -117,14 +110,11 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
             str: The unique identifier (job ID) of the created job.
 
         Raises:
-            RuntimeError: If start() was not called before creating jobs.
             Exception: If the module fails to start.
         """
         job_id = str(uuid.uuid4())
+        # TODO: Ensure the job_id is unique.
         module = ModuleFactory.create_module_instance(self.module_class, job_id, mission_id, setup_id, setup_version_id)
-        if self.channel is None:
-            msg = "JobManager.start() must be called before creating jobs"
-            raise RuntimeError(msg)
         self.tasks_sessions[job_id] = TaskSession(job_id, mission_id, self.channel, module)
 
         try:
@@ -136,7 +126,7 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
         except Exception:
             # Remove the module from the manager in case of an error.
             del self.tasks_sessions[job_id]
-            logger.exception("Failed to start module", extra={"job_id": job_id})
+            logger.exception("Failed to start module %s: %s", job_id)
             raise
         else:
             return job_id
@@ -144,33 +134,13 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
     async def add_to_queue(self, job_id: str, output_data: OutputModelT | ModuleCodeModel) -> None:
         """Add output data to the queue for a specific job.
 
-        Uses timeout-based backpressure: if the queue is full after 5s,
-        drops the oldest message to make room for the new one.
-        Rejects writes after stream is closed to prevent message loss.
+        This method is used as a callback to handle output data generated by a module job.
 
         Args:
             job_id: The unique identifier of the job.
             output_data: The output data produced by the job.
         """
-        session = self.tasks_sessions.get(job_id)
-        if session is None:
-            logger.warning("Queue write rejected - session not found", extra={"job_id": job_id})
-            return
-
-        if session.stream_closed:
-            logger.debug("Queue write rejected - stream closed", extra={"job_id": job_id})
-            return
-
-        try:
-            await asyncio.wait_for(session.queue.put(output_data.model_dump()), timeout=5.0)
-        except asyncio.TimeoutError:
-            logger.warning("Queue full, dropping oldest message", extra={"job_id": job_id})
-            try:
-                session.queue.get_nowait()
-                session.queue.task_done()
-            except asyncio.QueueEmpty:
-                pass
-            session.queue.put_nowait(output_data.model_dump())
+        await self.tasks_sessions[job_id].queue.put(output_data.model_dump())
 
     @asynccontextmanager  # type: ignore
     async def generate_stream_consumer(self, job_id: str) -> AsyncIterator[AsyncGenerator[dict[str, Any], None]]:  # type: ignore
@@ -207,39 +177,42 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
         logger.debug("Session: %s with Module %s", job_id, session.module)
 
         async def _stream() -> AsyncGenerator[dict[str, Any], Any]:
-            """Stream output data from the module with bounded blocking.
+            """Stream output data from the module with simple blocking pattern.
 
-            Uses a 1-second timeout on queue.get() to periodically re-check
-            termination flags, preventing indefinite hangs when the task crashes
-            without producing output.
+            This implementation uses a simple one-item-at-a-time pattern optimized
+            for local execution where we have direct access to session status:
+            1. Block waiting for each item
+            2. Check termination conditions after each item
+            3. Clean shutdown when task completes
+
+            This pattern provides:
+            - Immediate termination when task completes
+            - Direct session status monitoring
+            - Simple, predictable behavior for local tasks
 
             Yields:
                 dict: Output data generated by the module.
             """
             while True:
-                if session.stream_closed or session.is_cancelled.is_set():
-                    logger.debug("Stream ending for job %s (pre-check)", job_id)
-                    break
-
-                try:
-                    msg = await asyncio.wait_for(session.queue.get(), timeout=1.0)
-                except asyncio.TimeoutError:
-                    continue
-
+                # Block for next item - if queue is empty but producer not finished yet
+                msg = await session.queue.get()
                 try:
                     yield msg
                 finally:
+                    # Always mark task as done, even if consumer raises exception
                     session.queue.task_done()
 
+                # Check termination conditions after each message
+                # This allows immediate shutdown when the task completes
                 if (
-                    session.stream_closed
-                    or session.is_cancelled.is_set()
+                    session.is_cancelled.is_set()
                     or (session.status is TaskStatus.COMPLETED and session.queue.empty())
                     or session.status is TaskStatus.FAILED
                 ):
                     logger.debug(
-                        "Stream ending for job %s: status=%s, queue_empty=%s",
+                        "Stream ending for job %s: cancelled=%s, status=%s, queue_empty=%s",
                         job_id,
+                        session.is_cancelled.is_set(),
                         session.status,
                         session.queue.empty(),
                     )
@@ -286,18 +259,6 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
         logger.info("Managed task started: '%s'", job_id, extra={"task_id": job_id})
         return job_id
 
-    async def clean_session(self, task_id: str, mission_id: str) -> bool:
-        """Clean a task's session.
-
-        Args:
-            task_id: Unique identifier for the task.
-            mission_id: Mission identifier.
-
-        Returns:
-            bool: True if the task was successfully cleaned, False otherwise.
-        """
-        return await self._task_manager.clean_session(task_id, mission_id)
-
     async def stop_module(self, job_id: str) -> bool:
         """Stop a running module job.
 
@@ -310,23 +271,20 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
         Raises:
             Exception: If an error occurs while stopping the module.
         """
-        logger.info("Stop module requested", extra={"job_id": job_id})
+        logger.info(f"STOP required for {job_id=}")
 
         async with self._lock:
             session = self.tasks_sessions.get(job_id)
 
             if not session:
-                logger.warning("Session not found", extra={"job_id": job_id})
+                logger.warning(f"session with id: {job_id} not found")
                 return False
             try:
                 await session.module.stop()
                 await self.cancel_task(job_id, session.mission_id)
-                logger.debug(
-                    "Module stopped successfully",
-                    extra={"job_id": job_id, "mission_id": session.mission_id},
-                )
-            except Exception:
-                logger.exception("Error stopping module", extra={"job_id": job_id})
+                logger.debug(f"session {job_id} ({session.module.name}) stopped successfully")
+            except Exception as e:
+                logger.error(f"Error while stopping module {job_id}: {e}")
                 raise
             else:
                 return True
@@ -373,7 +331,7 @@ class SingleJobManager(BaseJobManager[InputModelT, OutputModelT, SetupModelT]):
             await asyncio.gather(*stop_tasks, return_exceptions=True)
 
         # Close SurrealDB connection after stopping all modules
-        if self.channel is not None:
+        if hasattr(self, "channel"):
             try:
                 await self.channel.close()
                 logger.info("SingleJobManager: SurrealDB connection closed")