Add dbos plugin

Author: adrianlyjak

AGENTS.md

@@ -74,3 +74,13 @@ Linting, typechecking, and formatting:
 ```bash
 uv run --directory packages/llama-index-workflows pre-commit run -a
 ```
+
+## Testing Patterns
+
+We use **pytest** with idiomatic pytest patterns. Follow these guidelines:
+
+- **No Test Classes**: Do not use test classes to organize tests. Write tests as standalone functions. Achieve organization through descriptive function names (e.g., `test_create_job_with_invalid_input_raises_error`) or by splitting into separate test files.
+- **Pytest Fixtures**: Use fixtures for setup/teardown and shared test dependencies. Prefer fixtures over manual setup code repeated across tests.
+- **Prefer Real Objects Over Mocks**: Use simple dataclasses and real objects directly when available rather than mocking them. Only mock external dependencies or things that are truly difficult to instantiate.
+- **DRY Test Setup**: Do not repeat patches or setup code. Create reusable abstractions—fixtures, helper functions, or module-level constants—that can be shared across tests. Tests can easily be overwhelmed with setup; start from a rich suite of testing utilities to enable many small, expressive tests.
+- **Simple Testing Utilities**: Testing utilities should be basic—just functions, fixtures, and global variables. Avoid over-engineering test infrastructure.

packages/llama-index-utils-workflow/pyproject.toml

@@ -19,7 +19,7 @@ authors = [{name = "Adrian Lyjak", email = "adrianlyjak@gmail.com"}]
 requires-python = ">=3.9"
 dependencies = [
   "llama-index-core>=0.14,<0.15.0",
-  "llama-index-workflows>=2.12.0,<3.0.0",
+  "llama-index-workflows>=2.13.0,<3.0.0",
   "pyvis>=0.3.2"
 ]
 

packages/llama-index-utils-workflow/src/llama_index/utils/workflow/__init__.py

@@ -12,6 +12,7 @@
 from llama_index.core.tools import AsyncBaseTool, BaseTool
 from pyvis.network import Network
 from workflows import Workflow
+from workflows.context.external_context import ExternalContext
 from workflows.events import (
     Event,
     StartEvent,
@@ -46,15 +47,15 @@ def _get_node_color(node: WorkflowGraphNode) -> str:
     elif node.node_type == "resource":
         return "#DDA0DD"  # Plum/light purple for resources
     elif node.node_type == "event":
-        if node.is_subclass_of("StartEvent"):
+        if node.is_subclass_of("StartEvent"):  # type: ignore[possibly-missing-attribute]
             return "#E27AFF"  # Pink for start events
-        elif node.is_subclass_of("StopEvent"):
+        elif node.is_subclass_of("StopEvent"):  # type: ignore[possibly-missing-attribute]
             return "#FFA07A"  # Orange for stop events
         return "#90EE90"  # Light green for other events
     elif node.node_type == "agent":
-        if node.is_subclass_of("ReActAgent"):
+        if node.is_subclass_of("ReActAgent"):  # type: ignore[possibly-missing-attribute]
             return "#E27AFF"
-        elif node.is_subclass_of("CodeActAgent"):
+        elif node.is_subclass_of("CodeActAgent"):  # type: ignore[possibly-missing-attribute]
             return "#66ccff"
         return "#90EE90"
     elif node.node_type == "tool":
@@ -172,15 +173,15 @@ def _get_mermaid_css_class(node: WorkflowGraphNode) -> str:
     elif node.node_type == "resource":
         return "resourceStyle"
     elif node.node_type == "event":
-        if node.is_subclass_of("StartEvent"):
+        if node.is_subclass_of("StartEvent"):  # type: ignore[possibly-missing-attribute]
             return "startEventStyle"
-        elif node.is_subclass_of("StopEvent"):
+        elif node.is_subclass_of("StopEvent"):  # type: ignore[possibly-missing-attribute]
             return "stopEventStyle"
         return "defaultEventStyle"
     elif node.node_type == "agent":
-        if node.is_subclass_of("ReActAgent"):
+        if node.is_subclass_of("ReActAgent"):  # type: ignore[possibly-missing-attribute]
             return "reactAgentStyle"
-        elif node.is_subclass_of("CodeActAgent"):
+        elif node.is_subclass_of("CodeActAgent"):  # type: ignore[possibly-missing-attribute]
             return "codeActAgentStyle"
         return "defaultAgentStyle"
     elif node.node_type == "tool":
@@ -435,10 +436,12 @@ def _extract_execution_graph(
     handler: WorkflowHandler, max_label_length: int | None = None
 ) -> Tuple[Dict[str, Tuple[str, str, type | None]], List[Tuple[str, str]]]:
     """Helper to extract nodes and edges from the workflow handler's tick log."""
-    if handler.ctx is None or handler.ctx._broker_run is None:
-        raise ValueError("No context/run info in this handler. Has it been run yet?")
 
-    ticks: List[WorkflowTick] = handler.ctx._broker_run._tick_log
+    ticks: List[WorkflowTick] = []
+    if handler.ctx is not None:
+        face = handler.ctx._face
+        if isinstance(face, ExternalContext):
+            ticks = face._tick_log
     nodes: Dict[str, Tuple[str, str, type | None]] = {}
     edges: List[Tuple[str, str]] = []
     event_node_by_identity: Dict[int, str] = {}

packages/llama-index-workflows/pyproject.toml

@@ -37,6 +37,7 @@ dependencies = [
 [project.optional-dependencies]
 server = ["starlette>=0.39.0", "uvicorn>=0.32.0"]
 client = ["httpx>=0.28.1,<1"]
+dbos = ["dbos>=1.14.0"]
 
 [tool.hatch.envs.server]
 features = ["server"]