stemtrace 0.2.0__tar.gz → 0.2.1__tar.gz

{stemtrace-0.2.0 → stemtrace-0.2.1}/.bumpversion.toml

@@ -1,5 +1,5 @@
 [tool.bumpversion]
-current_version = "0.2.0"
+current_version = "0.2.1"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
 serialize = ["{major}.{minor}.{patch}"]
 commit = true

{stemtrace-0.2.0 → stemtrace-0.2.1}/.github/workflows/ci.yml

@@ -3,6 +3,9 @@ name: CI
 on:
   pull_request:
     branches: [main]
+  push:
+    branches: [main]
+  workflow_dispatch:
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}
@@ -10,6 +13,7 @@ concurrency:
 
 jobs:
   lint:
+    if: github.event_name == 'pull_request'
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v6
@@ -34,6 +38,7 @@ jobs:
         run: uv run ruff format --check src/ tests/
 
   type-check:
+    if: github.event_name == 'pull_request'
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v6
@@ -55,6 +60,7 @@ jobs:
         run: uv run mypy src/ --strict
 
   test:
+    if: github.event_name == 'pull_request'
     runs-on: ubuntu-latest
     strategy:
       matrix:
@@ -87,6 +93,7 @@ jobs:
           fail_ci_if_error: false
 
   frontend:
+    if: github.event_name == 'pull_request'
     runs-on: ubuntu-latest
     defaults:
       run:
@@ -111,6 +118,7 @@ jobs:
         run: npm run typecheck
 
   build:
+    if: github.event_name == 'pull_request'
     runs-on: ubuntu-latest
     needs: [lint, type-check, test, frontend]
     steps:
@@ -148,3 +156,34 @@ jobs:
         with:
           name: dist
           path: dist/
+
+  coverage-main:
+    # Keep PR checks as the main gate, but ensure Codecov gets updated for the
+    # actual default-branch commit after merges (or direct pushes / releases).
+    if: github.ref == 'refs/heads/main' && (github.event_name == 'push' || github.event_name == 'workflow_dispatch')
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v6
+
+      - name: Set up Python
+        uses: actions/setup-python@v6
+        with:
+          python-version: "3.12"
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v7
+        with:
+          enable-cache: true
+
+      - name: Install dependencies
+        run: uv sync --all-extras
+
+      - name: Run tests (coverage only)
+        run: uv run pytest --cov=stemtrace --cov-branch --cov-report=xml --cov-fail-under=95
+
+      - name: Upload coverage (main)
+        uses: codecov/codecov-action@v5
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          files: ./coverage.xml
+          fail_ci_if_error: false

{stemtrace-0.2.0 → stemtrace-0.2.1}/CHANGELOG.md

@@ -7,6 +7,15 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.2.1] - 2026-01-03
+
+### Added
+- Task Registry now includes task definition metadata from workers (module, signature, docstring, bound)
+
+### Changed
+- README: updated PyPI badge styling
+- CI: upload Codecov coverage on `main`
+
 ## [0.2.0] - 2026-01-03
 
 ### Added
@@ -50,7 +59,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - **E2E test suite**: Docker API tests + Playwright browser tests
 - **Comprehensive test suite**: 350+ Python tests, 90%+ coverage
 
-[unreleased]: https://github.com/iansokolskyi/stemtrace/compare/v0.2.0...HEAD
+[unreleased]: https://github.com/iansokolskyi/stemtrace/compare/v0.2.1...HEAD
+[0.2.1]: https://github.com/iansokolskyi/stemtrace/compare/v0.2.0...v0.2.1
 [0.2.0]: https://github.com/iansokolskyi/stemtrace/compare/v0.1.1...v0.2.0
 [0.1.1]: https://github.com/iansokolskyi/stemtrace/compare/v0.1.0...v0.1.1
 [0.1.0]: https://github.com/iansokolskyi/stemtrace/releases/tag/v0.1.0

{stemtrace-0.2.0 → stemtrace-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stemtrace
-Version: 0.2.0
+Version: 0.2.1
 Summary: A lightweight Celery task flow visualizer
 Project-URL: Homepage, https://github.com/iansokolskyi/stemtrace
 Project-URL: Documentation, https://github.com/iansokolskyi/stemtrace#readme
@@ -55,7 +55,7 @@ Description-Content-Type: text/markdown
 
 **Zero-infrastructure Celery task flow visualizer**
 
-[![PyPI version](https://img.shields.io/pypi/v/stemtrace)](https://pypi.org/project/stemtrace/)
+[![PyPI version](https://img.shields.io/pypi/v/stemtrace?color=darklime)](https://pypi.org/project/stemtrace)
 [![Python](https://img.shields.io/pypi/pyversions/stemtrace.svg)](https://pypi.org/project/stemtrace/)
 [![CI](https://github.com/iansokolskyi/stemtrace/actions/workflows/ci.yml/badge.svg)](https://github.com/iansokolskyi/stemtrace/actions/workflows/ci.yml)
 [![codecov](https://codecov.io/gh/iansokolskyi/stemtrace/graph/badge.svg)](https://codecov.io/gh/iansokolskyi/stemtrace)

{stemtrace-0.2.0 → stemtrace-0.2.1}/README.md

@@ -2,7 +2,7 @@
 
 **Zero-infrastructure Celery task flow visualizer**
 
-[![PyPI version](https://img.shields.io/pypi/v/stemtrace)](https://pypi.org/project/stemtrace/)
+[![PyPI version](https://img.shields.io/pypi/v/stemtrace?color=darklime)](https://pypi.org/project/stemtrace)
 [![Python](https://img.shields.io/pypi/pyversions/stemtrace.svg)](https://pypi.org/project/stemtrace/)
 [![CI](https://github.com/iansokolskyi/stemtrace/actions/workflows/ci.yml/badge.svg)](https://github.com/iansokolskyi/stemtrace/actions/workflows/ci.yml)
 [![codecov](https://codecov.io/gh/iansokolskyi/stemtrace/graph/badge.svg)](https://codecov.io/gh/iansokolskyi/stemtrace)

{stemtrace-0.2.0 → stemtrace-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "stemtrace"
-version = "0.2.0"
+version = "0.2.1"
 description = "A lightweight Celery task flow visualizer"
 readme = "README.md"
 license = { text = "MIT" }

{stemtrace-0.2.0 → stemtrace-0.2.1}/src/stemtrace/__init__.py

@@ -38,7 +38,7 @@ if TYPE_CHECKING:
     from celery import Celery
     from fastapi import FastAPI
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 __all__ = [
     "ConfigurationError",
     "StemtraceConfig",

{stemtrace-0.2.0 → stemtrace-0.2.1}/src/stemtrace/core/events.py

@@ -4,7 +4,7 @@ from datetime import datetime
 from enum import Enum
 from typing import Any
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
 
 
 class TaskState(str, Enum):
@@ -27,6 +27,30 @@ class WorkerEventType(str, Enum):
     WORKER_SHUTDOWN = "worker_shutdown"
 
 
+class RegisteredTaskDefinition(BaseModel):
+    """Celery task definition metadata captured from workers.
+
+    This data is collected from a running worker's task registry and is used
+    to enrich the task registry API (`/api/tasks/registry`) with human-friendly
+    information like docstrings and signatures.
+
+    Attributes:
+        name: Fully qualified task name (e.g., "myapp.tasks.add").
+        module: Python module where the task implementation lives.
+        signature: Human-readable call signature (best-effort).
+        docstring: Task documentation (best-effort).
+        bound: Whether the task is bound (`bind=True`) and receives `self`.
+    """
+
+    model_config = ConfigDict(frozen=True)
+
+    name: str
+    module: str | None = None
+    signature: str | None = None
+    docstring: str | None = None
+    bound: bool = False
+
+
 class TaskEvent(BaseModel):
     """Immutable task lifecycle event.
 
@@ -95,11 +119,13 @@ class WorkerEvent(BaseModel):
     hostname: str
     pid: int
     timestamp: datetime
-    registered_tasks: list[str] = []
+    registered_tasks: list[str] = Field(default_factory=list)
+    task_definitions: dict[str, RegisteredTaskDefinition] = Field(default_factory=dict)
     shutdown_time: datetime | None = None
 
 
 __all__ = [
+    "RegisteredTaskDefinition",
     "TaskEvent",
     "TaskState",
     "WorkerEvent",

{stemtrace-0.2.0 → stemtrace-0.2.1}/src/stemtrace/library/signals.py

@@ -1,5 +1,6 @@
 """Celery signal handlers for task lifecycle events."""
 
+import inspect
 import logging
 import threading
 import traceback as tb_module
@@ -17,7 +18,7 @@ from celery.signals import (
     worker_shutdown,
 )
 
-from stemtrace.core.events import TaskEvent, TaskState
+from stemtrace.core.events import RegisteredTaskDefinition, TaskEvent, TaskState
 from stemtrace.core.ports import EventTransport
 from stemtrace.library.config import get_config
 from stemtrace.library.scrubbing import (
@@ -37,6 +38,9 @@ _transport: EventTransport | None = None
 _pending_emitted: set[str] = set()
 _pending_emitted_lock = threading.RLock()
 
+# Guard against overly-large docstrings bloating broker events.
+_MAX_DOCSTRING_CHARS = 4000
+
 
 def _extract_chord_info(chord_attr: Any) -> tuple[str | None, str | None]:
     """Extract chord group_id and callback task_id from chord attribute.
@@ -495,6 +499,74 @@ def _extract_registered_tasks(sender: Any) -> list[str]:
     return task_names
 
 
+def _extract_task_definitions(sender: Any) -> dict[str, RegisteredTaskDefinition]:
+    """Extract task metadata (docstring/signature/module/bound) from Celery app.
+
+    This runs on the worker in response to `worker_ready`, so we have access to
+    the actual Task objects. The server generally does not import user task code,
+    so this is the authoritative source for registry metadata.
+
+    Args:
+        sender: Worker instance (WorkController) from worker_ready signal.
+
+    Returns:
+        Mapping of task name to RegisteredTaskDefinition.
+    """
+    definitions: dict[str, RegisteredTaskDefinition] = {}
+
+    app = getattr(sender, "app", None)
+    if app is None:
+        return definitions
+
+    tasks_registry = getattr(app, "tasks", None)
+    if tasks_registry is None:
+        return definitions
+
+    # tasks_registry is dict-like: {name: Task}
+    for name, task_obj in getattr(tasks_registry, "items", lambda: [])():
+        if not isinstance(name, str) or name.startswith("celery."):
+            continue
+        if task_obj is None:
+            continue
+
+        run_callable = getattr(task_obj, "run", None)
+        module: str | None = None
+        docstring: str | None = None
+        signature: str | None = None
+
+        # Best-effort: prefer module/docstring from the task implementation (run).
+        if run_callable is not None:
+            module = getattr(run_callable, "__module__", None)
+            docstring = inspect.getdoc(run_callable)
+            try:
+                signature = str(inspect.signature(run_callable))
+            except (TypeError, ValueError):
+                signature = None
+
+        # Fall back to task object/module if needed.
+        if module is None:
+            module = getattr(task_obj, "__module__", None)
+
+        # Bound tasks are declared via @app.task(bind=True). Celery exposes this
+        # on the Task instance as `bind`. Only treat an actual bool as authoritative
+        # to avoid MagicMock truthiness in tests.
+        bind_attr = getattr(task_obj, "bind", None)
+        bound = bind_attr if isinstance(bind_attr, bool) else False
+
+        if docstring is not None and len(docstring) > _MAX_DOCSTRING_CHARS:
+            docstring = docstring[:_MAX_DOCSTRING_CHARS] + "\n\n[truncated]"
+
+        definitions[name] = RegisteredTaskDefinition(
+            name=name,
+            module=module,
+            signature=signature,
+            docstring=docstring,
+            bound=bound,
+        )
+
+    return definitions
+
+
 def on_worker_ready(sender: Any, **_: Any) -> None:
     """Handle worker_ready signal - publish worker registration event.
 
@@ -509,6 +581,7 @@ def on_worker_ready(sender: Any, **_: Any) -> None:
     try:
         hostname, pid = _get_hostname_and_pid(sender)
         registered_tasks = _extract_registered_tasks(sender)
+        task_definitions = _extract_task_definitions(sender)
 
         # Publish worker lifecycle event using the global transport
         # This goes to the same stream as task events for unified consumption
@@ -520,6 +593,7 @@ def on_worker_ready(sender: Any, **_: Any) -> None:
             pid=pid,
             timestamp=datetime.now(timezone.utc),
             registered_tasks=registered_tasks,
+            task_definitions=task_definitions,
         )
 
         if _transport is not None:

{stemtrace-0.2.0 → stemtrace-0.2.1}/src/stemtrace/server/api/routes.py

@@ -310,16 +310,25 @@ def create_api_router(
             if status and task_status.value != status:
                 continue
 
+            definition = (
+                worker_registry.get_task_definition(name) if worker_registry else None
+            )
+
             parts = name.rsplit(".", 1)
-            module = parts[0] if len(parts) > 1 else None
+            fallback_module = parts[0] if len(parts) > 1 else None
+            module = (
+                definition.module
+                if definition and definition.module
+                else fallback_module
+            )
 
             tasks.append(
                 RegisteredTaskResponse(
                     name=name,
                     module=module,
-                    signature=None,
-                    docstring=None,
-                    bound=False,
+                    signature=definition.signature if definition else None,
+                    docstring=definition.docstring if definition else None,
+                    bound=definition.bound if definition else False,
                     execution_count=execution_count,
                     registered_by=registered_by,
                     last_run=last_run,

{stemtrace-0.2.0 → stemtrace-0.2.1}/src/stemtrace/server/consumer.py

@@ -141,6 +141,7 @@ class EventConsumer:
                 hostname=event.hostname,
                 pid=event.pid,
                 tasks=event.registered_tasks,
+                task_definitions=event.task_definitions,
                 event_timestamp=event.timestamp,
             )
             logger.info(

{stemtrace-0.2.0 → stemtrace-0.2.1}/src/stemtrace/server/store.py

@@ -13,7 +13,7 @@ from stemtrace.core.graph import NodeType, TaskGraph, TaskNode
 from stemtrace.server.api.schemas import WorkerStatus
 
 if TYPE_CHECKING:
-    from stemtrace.core.events import TaskEvent
+    from stemtrace.core.events import RegisteredTaskDefinition, TaskEvent
 
 
 class WorkerInfo(BaseModel):
@@ -104,6 +104,7 @@ class WorkerRegistry:
     def __init__(self) -> None:
         """Initialize empty worker registry."""
         self._workers: dict[str, WorkerInfo] = {}
+        self._task_definitions_by_name: dict[str, RegisteredTaskDefinition] = {}
         self._lock = threading.RLock()
 
     def register_worker(
@@ -111,6 +112,7 @@ class WorkerRegistry:
         hostname: str,
         pid: int,
         tasks: list[str],
+        task_definitions: dict[str, RegisteredTaskDefinition] | None = None,
         event_timestamp: datetime | None = None,
     ) -> None:
         """Register a worker with its task list.
@@ -121,6 +123,9 @@ class WorkerRegistry:
             hostname: Worker hostname.
             pid: Worker process ID.
             tasks: List of registered task names.
+            task_definitions: Optional mapping of task name to task definition
+                metadata (docstring/signature/module/bound). When provided, this
+                is used to enrich `/api/tasks/registry`.
             event_timestamp: When the worker event occurred. Used for last_seen
                 to properly handle historical events from Redis replay.
         """
@@ -134,6 +139,10 @@ class WorkerRegistry:
             # Fall back to now() for direct calls (e.g., tests)
             timestamp = event_timestamp or datetime.now(timezone.utc)
 
+            # Update global task definitions map (best-effort, last-writer-wins).
+            if task_definitions:
+                self._task_definitions_by_name.update(task_definitions)
+
             # If worker already exists, update it (restart scenario)
             if worker_id in self._workers:
                 self._workers[worker_id].registered_tasks = tasks
@@ -151,6 +160,18 @@ class WorkerRegistry:
                     status=WorkerStatus.ONLINE,
                 )
 
+    def get_task_definition(self, name: str) -> RegisteredTaskDefinition | None:
+        """Get task definition metadata by task name.
+
+        Args:
+            name: Fully qualified task name.
+
+        Returns:
+            RegisteredTaskDefinition if known, otherwise None.
+        """
+        with self._lock:
+            return self._task_definitions_by_name.get(name)
+
     def mark_shutdown(self, hostname: str, pid: int) -> None:
         """Mark a worker as offline (shutdown).
 

{stemtrace-0.2.0 → stemtrace-0.2.1}/src/stemtrace/server/ui/frontend/package.json

@@ -1,7 +1,7 @@
 {
   "name": "stemtrace-ui",
   "private": true,
-  "version": "0.2.0",
+  "version": "0.2.1",
   "type": "module",
   "scripts": {
     "dev": "vite",

{stemtrace-0.2.0 → stemtrace-0.2.1}/tests/integration/test_fastapi_integration.py

@@ -8,12 +8,12 @@ import pytest
 from fastapi import FastAPI
 from fastapi.testclient import TestClient
 
-from stemtrace.core.events import TaskEvent, TaskState
+from stemtrace.core.events import RegisteredTaskDefinition, TaskEvent, TaskState
 from stemtrace.server.api.schemas import WorkerStatus
 from stemtrace.server.fastapi.auth import require_api_key, require_basic_auth
 from stemtrace.server.fastapi.extension import StemtraceExtension
 from stemtrace.server.fastapi.router import create_router
-from stemtrace.server.store import GraphStore
+from stemtrace.server.store import GraphStore, WorkerRegistry
 from stemtrace.server.websocket import WebSocketManager
 
 
@@ -621,8 +621,6 @@ class TestWorkersEndpoint:
 
     def test_workers_empty_registry(self) -> None:
         """Empty registry returns empty list."""
-        from stemtrace.server.store import WorkerRegistry
-
         store = GraphStore()
         worker_registry = WorkerRegistry()
         router = create_router(store=store, worker_registry=worker_registry)
@@ -639,8 +637,6 @@ class TestWorkersEndpoint:
 
     def test_workers_returns_registered_worker(self) -> None:
         """Registered worker appears in response."""
-        from stemtrace.server.store import WorkerRegistry
-
         store = GraphStore()
         worker_registry = WorkerRegistry()
         worker_registry.register_worker(
@@ -665,8 +661,6 @@ class TestWorkersEndpoint:
 
     def test_workers_by_hostname(self) -> None:
         """Filter workers by hostname."""
-        from stemtrace.server.store import WorkerRegistry
-
         store = GraphStore()
         worker_registry = WorkerRegistry()
         worker_registry.register_worker("worker-1", 12345, ["tasks.add"])
@@ -686,8 +680,6 @@ class TestWorkersEndpoint:
 
     def test_workers_shutdown_status(self) -> None:
         """Shutdown worker has offline status."""
-        from stemtrace.server.store import WorkerRegistry
-
         store = GraphStore()
         worker_registry = WorkerRegistry()
         worker_registry.register_worker("worker-1", 12345, ["tasks.add"])
@@ -712,8 +704,6 @@ class TestRegistryWithWorkers:
 
     def test_registry_shows_registered_by(self) -> None:
         """Tasks show which workers registered them."""
-        from stemtrace.server.store import WorkerRegistry
-
         store = GraphStore()
         worker_registry = WorkerRegistry()
         worker_registry.register_worker(
@@ -735,10 +725,54 @@ class TestRegistryWithWorkers:
         # Both workers should be listed (deduplicated and sorted)
         assert sorted(add_task["registered_by"]) == ["worker-1", "worker-2"]
 
+    def test_registry_includes_task_metadata(self) -> None:
+        """Registry includes docstring/signature/bound when provided by workers."""
+        store = GraphStore()
+        worker_registry = WorkerRegistry()
+
+        task_name = "tasks.add"
+        worker_registry.register_worker(
+            "worker-1",
+            12345,
+            [task_name],
+            task_definitions={
+                task_name: RegisteredTaskDefinition(
+                    name=task_name,
+                    module="tasks",
+                    signature="(x, y)",
+                    docstring="Add two numbers.",
+                    bound=True,
+                )
+            },
+        )
+
+        # Ensure the task is also observed (so it appears in the union set even if workers differ).
+        store.add_event(
+            TaskEvent(
+                task_id="task-1",
+                name=task_name,
+                state=TaskState.SUCCESS,
+                timestamp=datetime(2024, 1, 1, tzinfo=UTC),
+            )
+        )
+
+        router = create_router(store=store, worker_registry=worker_registry)
+        app = FastAPI()
+        app.include_router(router)
+        client = TestClient(app)
+
+        response = client.get("/api/tasks/registry")
+        assert response.status_code == 200
+        data = response.json()
+
+        add_task = next(t for t in data["tasks"] if t["name"] == task_name)
+        assert add_task["module"] == "tasks"
+        assert add_task["signature"] == "(x, y)"
+        assert add_task["docstring"] == "Add two numbers."
+        assert add_task["bound"] is True
+
     def test_registry_no_duplicate_hostnames(self) -> None:
         """Same hostname from multiple restarts appears only once."""
-        from stemtrace.server.store import WorkerRegistry
-
         store = GraphStore()
         worker_registry = WorkerRegistry()
         # Same hostname, different PIDs (simulating restarts)
@@ -760,8 +794,6 @@ class TestRegistryWithWorkers:
 
     def test_registry_never_run_status_filter(self) -> None:
         """status=never_run shows only registered but never executed tasks."""
-        from stemtrace.server.store import WorkerRegistry
-
         store = GraphStore()
         # Add an executed task (not registered) - status: not_registered
         store.add_event(

{stemtrace-0.2.0 → stemtrace-0.2.1}/tests/unit/test_routes.py

@@ -7,9 +7,9 @@ import pytest
 from fastapi import FastAPI
 from fastapi.testclient import TestClient
 
-from stemtrace.core.events import TaskEvent, TaskState
+from stemtrace.core.events import RegisteredTaskDefinition, TaskEvent, TaskState
 from stemtrace.server.api.routes import create_api_router
-from stemtrace.server.store import GraphStore
+from stemtrace.server.store import GraphStore, WorkerRegistry
 
 
 @pytest.fixture
@@ -431,8 +431,6 @@ class TestTaskRegistryWithWorkers:
         self, store: GraphStore, make_event: type
     ) -> None:
         """Task with executions AND registered by worker has 'active' status."""
-        from stemtrace.server.store import WorkerRegistry
-
         worker_registry = WorkerRegistry()
         worker_registry.register_worker(
             hostname="worker-1",
@@ -457,10 +455,46 @@ class TestTaskRegistryWithWorkers:
         assert task["status"] == "active"
         assert "worker-1" in task["registered_by"]
 
+    def test_registry_includes_task_metadata_from_worker_registry(
+        self, store: GraphStore, make_event: type
+    ) -> None:
+        """Registry includes docstring/signature/bound when provided by workers."""
+        worker_registry = WorkerRegistry()
+        task_name = "myapp.tasks.process"
+        worker_registry.register_worker(
+            hostname="worker-1",
+            pid=12345,
+            tasks=[task_name],
+            task_definitions={
+                task_name: RegisteredTaskDefinition(
+                    name=task_name,
+                    module="myapp.tasks",
+                    signature="(x, y)",
+                    docstring="Process data.",
+                    bound=True,
+                )
+            },
+        )
+
+        # Add execution so task is visible in registry
+        store.add_event(make_event.create("task-1", name=task_name))
+
+        app = FastAPI()
+        router = create_api_router(store, worker_registry=worker_registry)
+        app.include_router(router)
+        client = TestClient(app)
+
+        response = client.get("/api/tasks/registry")
+        data = response.json()
+
+        task = next(t for t in data["tasks"] if t["name"] == task_name)
+        assert task["module"] == "myapp.tasks"
+        assert task["signature"] == "(x, y)"
+        assert task["docstring"] == "Process data."
+        assert task["bound"] is True
+
     def test_registry_never_run_status_with_worker(self, store: GraphStore) -> None:
         """Task registered by worker but never executed has 'never_run' status."""
-        from stemtrace.server.store import WorkerRegistry
-
         worker_registry = WorkerRegistry()
         worker_registry.register_worker(
             hostname="worker-1",
@@ -488,8 +522,6 @@ class TestTaskRegistryWithWorkers:
         self, store: GraphStore, make_event: type
     ) -> None:
         """Filter to show only never_run tasks."""
-        from stemtrace.server.store import WorkerRegistry
-
         worker_registry = WorkerRegistry()
         worker_registry.register_worker(
             hostname="worker-1",
@@ -514,8 +546,6 @@ class TestTaskRegistryWithWorkers:
 
     def test_registry_mixed_statuses(self, store: GraphStore, make_event: type) -> None:
         """Registry shows all three status types correctly."""
-        from stemtrace.server.store import WorkerRegistry
-
         worker_registry = WorkerRegistry()
         worker_registry.register_worker(
             hostname="worker-1",

{stemtrace-0.2.0 → stemtrace-0.2.1}/tests/unit/test_signals.py

@@ -10,8 +10,10 @@ import pytest
 from stemtrace.core.events import TaskState, WorkerEvent, WorkerEventType
 from stemtrace.library.config import StemtraceConfig, set_config
 from stemtrace.library.signals import (
+    _MAX_DOCSTRING_CHARS,
     _extract_chord_info,
     _extract_registered_tasks,
+    _extract_task_definitions,
     _format_exception,
     _format_traceback,
     _get_hostname_and_pid,
@@ -211,6 +213,61 @@ class TestWorkerLifecycleSignals:
 
         assert len(MemoryTransport.events) == 0
 
+    def test_extract_task_definitions_empty_without_app(self) -> None:
+        """No sender.app returns empty task definitions mapping."""
+        sender = MagicMock()
+        sender.app = None
+        assert _extract_task_definitions(sender) == {}
+
+    def test_extract_task_definitions_empty_without_tasks_registry(self) -> None:
+        """No app.tasks returns empty task definitions mapping."""
+        sender = SimpleNamespace(app=SimpleNamespace(tasks=None))
+        assert _extract_task_definitions(sender) == {}
+
+    def test_extract_task_definitions_truncates_docstring(self) -> None:
+        """Docstrings longer than the limit are truncated with a marker."""
+
+        def run_impl() -> None:
+            pass
+
+        run_impl.__doc__ = "x" * (_MAX_DOCSTRING_CHARS + 50)
+
+        class TaskObj:
+            run = staticmethod(run_impl)
+            bind = True
+
+        sender = SimpleNamespace(app=SimpleNamespace(tasks={"my.task": TaskObj()}))
+        defs = _extract_task_definitions(sender)
+        assert "my.task" in defs
+        assert defs["my.task"].bound is True
+        assert defs["my.task"].docstring is not None
+        assert defs["my.task"].docstring.endswith("[truncated]")
+        assert len(defs["my.task"].docstring) > _MAX_DOCSTRING_CHARS
+
+    def test_extract_task_definitions_signature_best_effort(self) -> None:
+        """Non-callable run attribute yields signature=None without raising."""
+
+        class TaskObj:
+            run = 123  # inspect.signature will raise TypeError
+            bind = False
+
+        sender = SimpleNamespace(app=SimpleNamespace(tasks={"my.bad": TaskObj()}))
+        defs = _extract_task_definitions(sender)
+        assert defs["my.bad"].signature is None
+
+    def test_extract_task_definitions_module_fallback_and_bind_guard(self) -> None:
+        """Fallback module uses task_obj.__module__ and bind guard ignores non-bool."""
+
+        class TaskObj:
+            __module__ = "my.module"
+            run = None
+            bind = MagicMock()
+
+        sender = SimpleNamespace(app=SimpleNamespace(tasks={"my.task": TaskObj()}))
+        defs = _extract_task_definitions(sender)
+        assert defs["my.task"].module == "my.module"
+        assert defs["my.task"].bound is False
+
 
 class TestTaskPrerun:
     """Tests for task_prerun signal handler."""

{stemtrace-0.2.0 → stemtrace-0.2.1}/uv.lock

@@ -1595,7 +1595,7 @@ wheels = [
 
 [[package]]
 name = "stemtrace"
-version = "0.2.0"
+version = "0.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "celery" },