diffusers-workflow 0.4.0a8__tar.gz → 0.4.0a12__tar.gz

{diffusers_workflow-0.4.0a8/diffusers_workflow.egg-info → diffusers_workflow-0.4.0a12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffusers-workflow
-Version: 0.4.0a8
+Version: 0.4.0a12
 Summary: Declarative workflow engine and web UI for Hugging Face Diffusers
 Author: dkackman
 License: Apache-2.0
@@ -57,6 +57,9 @@ Requires-Dist: packaging>=26.3
 Provides-Extra: server
 Requires-Dist: fastapi>=0.141.1; extra == "server"
 Requires-Dist: uvicorn[standard]>=0.52.4; extra == "server"
+Provides-Extra: mcp
+Requires-Dist: mcp>=2.1.1; extra == "mcp"
+Requires-Dist: httpx>=0.28.1; extra == "mcp"
 Provides-Extra: dev
 Requires-Dist: pytest>=9.1.1; extra == "dev"
 Requires-Dist: pytest-cov>=7.1.0; extra == "dev"
@@ -67,6 +70,7 @@ Requires-Dist: pytest-asyncio>=1.4.0; extra == "dev"
 Requires-Dist: httpx>=0.28.1; extra == "dev"
 Requires-Dist: black>=26.5.1; extra == "dev"
 Requires-Dist: build; extra == "dev"
+Requires-Dist: mcp>=2.1.1; extra == "dev"
 Dynamic: license-file
 
 [![CodeQL](https://github.com/dkackman/diffusers-workflow/actions/workflows/github-code-scanning/codeql/badge.svg)](https://github.com/dkackman/diffusers-workflow/actions/workflows/github-code-scanning/codeql)
@@ -75,6 +79,8 @@ Dynamic: license-file
 
 A declarative workflow engine and web UI for the [Hugging Face Diffusers library](https://github.com/huggingface/diffusers). Define image/video generation pipelines in JSON — with full access to the configuration diffusers exposes — and run them from the command line, an interactive REPL, or a browser.
 
+A workflow JSON file is portable and easy to hand off — but that same flexibility means loading one can execute arbitrary Python (dynamic imports are how it reaches any diffusers pipeline or quantization backend without a bespoke adapter for each). Treat a workflow file from someone else the way you'd treat a `.py` script: see [Trust model](docs/SECURITY.md#trust-model) before running one you didn't write.
+
 **Python 3.10-3.14 | CUDA (NVIDIA) | MPS (Apple Silicon) | CPU**
 
 <picture>
@@ -85,9 +91,10 @@ A declarative workflow engine and web UI for the [Hugging Face Diffusers library
 ## Features
 
 - **Web UI** — browse and run workflows, edit them in introspection-driven forms, watch jobs stream live progress, manage generated output and the models on disk. `python -m dw.serve` and open a browser. See [Server & Web UI](docs/SERVER.md).
+- **MCP server** — a stdio server that lets an MCP client (Claude Code first) author, validate, save, run and diagnose workflows against a running `dw.serve`. `dw-mcp`. See [MCP Server](docs/MCP.md).
 - **Declarative JSON workflows** with variable substitution and cross-step data flow
 - **Multi-step pipelines** — chain text-to-image, image-to-video, inpainting, ControlNet
-- **Reproducible by construction** — outputs embed their full workflow definition and seed; any image in the gallery reopens as the exact workflow that made it
+- **Reproducible by construction** — outputs embed their full workflow definition and seed; any image in the gallery reopens as the exact workflow that made it (see [Trust model](docs/SECURITY.md#trust-model) before reopening one someone else sent you)
 - **Long-video chaining** — run a video pipeline once per segment and stitch the segments into one clip, with audio-driven length and frame-to-frame continuity
 - **Quantization** — BitsAndBytes, TorchAO, GGUF, SDNQ, optimum-quanto
 - **Inference acceleration** — TeaCache, FirstBlockCache, FasterCache, MagCache, TaylorSeerCache
@@ -148,15 +155,26 @@ Jobs queue, stream progress live (per denoising step), cancel cooperatively, and
 
 ### Run a Workflow
 
+`workflows/sd15.json` is the smallest, fastest starting point — a small, ungated model and a literal prompt, so the first run needs no Hugging Face login and downloads only a few GB:
+
 ```bash
+python -m dw.run workflows/sd15.json
+python -m dw.run workflows/sd15.json prompt="a cat" num_images_per_prompt=4
+```
+
+Most of the workflows under `workflows/` (Flux, LTX-2, MiniMax...) use **gated** Hugging Face models — the repo owner has to approve your account before you can download them. Before running one of those, request access on the model's Hugging Face page (e.g. [black-forest-labs/FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev)) and log in locally:
+
+```bash
+huggingface-cli login
 python -m dw.run workflows/flux/FluxDev.json
-python -m dw.run workflows/flux/FluxDev.json prompt="a cat" num_images_per_prompt=4
 ```
 
+Without this, the run fails partway through with an HTTP 401/403 from the Hub.
+
 ### Validate a Workflow
 
 ```bash
-python -m dw.validate workflows/flux/FluxDev.json
+python -m dw.validate workflows/sd15.json
 ```
 
 ### Interactive REPL
@@ -323,6 +341,7 @@ See [workflows/](workflows/) for more workflow files.
 ### Guides
 
 - [Server & Web UI](docs/SERVER.md) — The web UI, jobs API, and introspection service
+- [MCP Server](docs/MCP.md) — Tool surface for MCP clients (Claude Code, Claude Desktop)
 - [Workflow Guide](docs/WORKFLOW_GUIDE.md) — JSON structure, variables, steps, data flow
 - [Quantization](docs/QUANTIZATION.md) — BitsAndBytes, TorchAO, GGUF, SDNQ
 - [Inference Acceleration](docs/ACCELERATION.md) — torch.compile, FirstBlockCache, MagCache, TaylorSeer, TeaCache

{diffusers_workflow-0.4.0a8 → diffusers_workflow-0.4.0a12}/README.md

@@ -4,6 +4,8 @@
 
 A declarative workflow engine and web UI for the [Hugging Face Diffusers library](https://github.com/huggingface/diffusers). Define image/video generation pipelines in JSON — with full access to the configuration diffusers exposes — and run them from the command line, an interactive REPL, or a browser.
 
+A workflow JSON file is portable and easy to hand off — but that same flexibility means loading one can execute arbitrary Python (dynamic imports are how it reaches any diffusers pipeline or quantization backend without a bespoke adapter for each). Treat a workflow file from someone else the way you'd treat a `.py` script: see [Trust model](docs/SECURITY.md#trust-model) before running one you didn't write.
+
 **Python 3.10-3.14 | CUDA (NVIDIA) | MPS (Apple Silicon) | CPU**
 
 <picture>
@@ -14,9 +16,10 @@ A declarative workflow engine and web UI for the [Hugging Face Diffusers library
 ## Features
 
 - **Web UI** — browse and run workflows, edit them in introspection-driven forms, watch jobs stream live progress, manage generated output and the models on disk. `python -m dw.serve` and open a browser. See [Server & Web UI](docs/SERVER.md).
+- **MCP server** — a stdio server that lets an MCP client (Claude Code first) author, validate, save, run and diagnose workflows against a running `dw.serve`. `dw-mcp`. See [MCP Server](docs/MCP.md).
 - **Declarative JSON workflows** with variable substitution and cross-step data flow
 - **Multi-step pipelines** — chain text-to-image, image-to-video, inpainting, ControlNet
-- **Reproducible by construction** — outputs embed their full workflow definition and seed; any image in the gallery reopens as the exact workflow that made it
+- **Reproducible by construction** — outputs embed their full workflow definition and seed; any image in the gallery reopens as the exact workflow that made it (see [Trust model](docs/SECURITY.md#trust-model) before reopening one someone else sent you)
 - **Long-video chaining** — run a video pipeline once per segment and stitch the segments into one clip, with audio-driven length and frame-to-frame continuity
 - **Quantization** — BitsAndBytes, TorchAO, GGUF, SDNQ, optimum-quanto
 - **Inference acceleration** — TeaCache, FirstBlockCache, FasterCache, MagCache, TaylorSeerCache
@@ -77,15 +80,26 @@ Jobs queue, stream progress live (per denoising step), cancel cooperatively, and
 
 ### Run a Workflow
 
+`workflows/sd15.json` is the smallest, fastest starting point — a small, ungated model and a literal prompt, so the first run needs no Hugging Face login and downloads only a few GB:
+
 ```bash
+python -m dw.run workflows/sd15.json
+python -m dw.run workflows/sd15.json prompt="a cat" num_images_per_prompt=4
+```
+
+Most of the workflows under `workflows/` (Flux, LTX-2, MiniMax...) use **gated** Hugging Face models — the repo owner has to approve your account before you can download them. Before running one of those, request access on the model's Hugging Face page (e.g. [black-forest-labs/FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev)) and log in locally:
+
+```bash
+huggingface-cli login
 python -m dw.run workflows/flux/FluxDev.json
-python -m dw.run workflows/flux/FluxDev.json prompt="a cat" num_images_per_prompt=4
 ```
 
+Without this, the run fails partway through with an HTTP 401/403 from the Hub.
+
 ### Validate a Workflow
 
 ```bash
-python -m dw.validate workflows/flux/FluxDev.json
+python -m dw.validate workflows/sd15.json
 ```
 
 ### Interactive REPL
@@ -252,6 +266,7 @@ See [workflows/](workflows/) for more workflow files.
 ### Guides
 
 - [Server & Web UI](docs/SERVER.md) — The web UI, jobs API, and introspection service
+- [MCP Server](docs/MCP.md) — Tool surface for MCP clients (Claude Code, Claude Desktop)
 - [Workflow Guide](docs/WORKFLOW_GUIDE.md) — JSON structure, variables, steps, data flow
 - [Quantization](docs/QUANTIZATION.md) — BitsAndBytes, TorchAO, GGUF, SDNQ
 - [Inference Acceleration](docs/ACCELERATION.md) — torch.compile, FirstBlockCache, MagCache, TaylorSeer, TeaCache

{diffusers_workflow-0.4.0a8 → diffusers_workflow-0.4.0a12/diffusers_workflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffusers-workflow
-Version: 0.4.0a8
+Version: 0.4.0a12
 Summary: Declarative workflow engine and web UI for Hugging Face Diffusers
 Author: dkackman
 License: Apache-2.0
@@ -57,6 +57,9 @@ Requires-Dist: packaging>=26.3
 Provides-Extra: server
 Requires-Dist: fastapi>=0.141.1; extra == "server"
 Requires-Dist: uvicorn[standard]>=0.52.4; extra == "server"
+Provides-Extra: mcp
+Requires-Dist: mcp>=2.1.1; extra == "mcp"
+Requires-Dist: httpx>=0.28.1; extra == "mcp"
 Provides-Extra: dev
 Requires-Dist: pytest>=9.1.1; extra == "dev"
 Requires-Dist: pytest-cov>=7.1.0; extra == "dev"
@@ -67,6 +70,7 @@ Requires-Dist: pytest-asyncio>=1.4.0; extra == "dev"
 Requires-Dist: httpx>=0.28.1; extra == "dev"
 Requires-Dist: black>=26.5.1; extra == "dev"
 Requires-Dist: build; extra == "dev"
+Requires-Dist: mcp>=2.1.1; extra == "dev"
 Dynamic: license-file
 
 [![CodeQL](https://github.com/dkackman/diffusers-workflow/actions/workflows/github-code-scanning/codeql/badge.svg)](https://github.com/dkackman/diffusers-workflow/actions/workflows/github-code-scanning/codeql)
@@ -75,6 +79,8 @@ Dynamic: license-file
 
 A declarative workflow engine and web UI for the [Hugging Face Diffusers library](https://github.com/huggingface/diffusers). Define image/video generation pipelines in JSON — with full access to the configuration diffusers exposes — and run them from the command line, an interactive REPL, or a browser.
 
+A workflow JSON file is portable and easy to hand off — but that same flexibility means loading one can execute arbitrary Python (dynamic imports are how it reaches any diffusers pipeline or quantization backend without a bespoke adapter for each). Treat a workflow file from someone else the way you'd treat a `.py` script: see [Trust model](docs/SECURITY.md#trust-model) before running one you didn't write.
+
 **Python 3.10-3.14 | CUDA (NVIDIA) | MPS (Apple Silicon) | CPU**
 
 <picture>
@@ -85,9 +91,10 @@ A declarative workflow engine and web UI for the [Hugging Face Diffusers library
 ## Features
 
 - **Web UI** — browse and run workflows, edit them in introspection-driven forms, watch jobs stream live progress, manage generated output and the models on disk. `python -m dw.serve` and open a browser. See [Server & Web UI](docs/SERVER.md).
+- **MCP server** — a stdio server that lets an MCP client (Claude Code first) author, validate, save, run and diagnose workflows against a running `dw.serve`. `dw-mcp`. See [MCP Server](docs/MCP.md).
 - **Declarative JSON workflows** with variable substitution and cross-step data flow
 - **Multi-step pipelines** — chain text-to-image, image-to-video, inpainting, ControlNet
-- **Reproducible by construction** — outputs embed their full workflow definition and seed; any image in the gallery reopens as the exact workflow that made it
+- **Reproducible by construction** — outputs embed their full workflow definition and seed; any image in the gallery reopens as the exact workflow that made it (see [Trust model](docs/SECURITY.md#trust-model) before reopening one someone else sent you)
 - **Long-video chaining** — run a video pipeline once per segment and stitch the segments into one clip, with audio-driven length and frame-to-frame continuity
 - **Quantization** — BitsAndBytes, TorchAO, GGUF, SDNQ, optimum-quanto
 - **Inference acceleration** — TeaCache, FirstBlockCache, FasterCache, MagCache, TaylorSeerCache
@@ -148,15 +155,26 @@ Jobs queue, stream progress live (per denoising step), cancel cooperatively, and
 
 ### Run a Workflow
 
+`workflows/sd15.json` is the smallest, fastest starting point — a small, ungated model and a literal prompt, so the first run needs no Hugging Face login and downloads only a few GB:
+
 ```bash
+python -m dw.run workflows/sd15.json
+python -m dw.run workflows/sd15.json prompt="a cat" num_images_per_prompt=4
+```
+
+Most of the workflows under `workflows/` (Flux, LTX-2, MiniMax...) use **gated** Hugging Face models — the repo owner has to approve your account before you can download them. Before running one of those, request access on the model's Hugging Face page (e.g. [black-forest-labs/FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev)) and log in locally:
+
+```bash
+huggingface-cli login
 python -m dw.run workflows/flux/FluxDev.json
-python -m dw.run workflows/flux/FluxDev.json prompt="a cat" num_images_per_prompt=4
 ```
 
+Without this, the run fails partway through with an HTTP 401/403 from the Hub.
+
 ### Validate a Workflow
 
 ```bash
-python -m dw.validate workflows/flux/FluxDev.json
+python -m dw.validate workflows/sd15.json
 ```
 
 ### Interactive REPL
@@ -323,6 +341,7 @@ See [workflows/](workflows/) for more workflow files.
 ### Guides
 
 - [Server & Web UI](docs/SERVER.md) — The web UI, jobs API, and introspection service
+- [MCP Server](docs/MCP.md) — Tool surface for MCP clients (Claude Code, Claude Desktop)
 - [Workflow Guide](docs/WORKFLOW_GUIDE.md) — JSON structure, variables, steps, data flow
 - [Quantization](docs/QUANTIZATION.md) — BitsAndBytes, TorchAO, GGUF, SDNQ
 - [Inference Acceleration](docs/ACCELERATION.md) — torch.compile, FirstBlockCache, MagCache, TaylorSeer, TeaCache

{diffusers_workflow-0.4.0a8 → diffusers_workflow-0.4.0a12}/diffusers_workflow.egg-info/SOURCES.txt

@@ -64,45 +64,45 @@ dw/server/ui/assets/csharp-BKxtCVv1.js
 dw/server/ui/assets/csp-bTuwJoIa.js
 dw/server/ui/assets/css-DIMkf-bt.js
 dw/server/ui/assets/css.worker-B3ciXF_0.js
-dw/server/ui/assets/cssMode-4JQhrTTm.js
+dw/server/ui/assets/cssMode-CRm5Fmld.js
 dw/server/ui/assets/cypher-CVaqCwHa.js
 dw/server/ui/assets/dart-onAF5SnQ.js
 dw/server/ui/assets/dockerfile-DZFCIeNp.js
 dw/server/ui/assets/ecl-D05T4iGw.js
 dw/server/ui/assets/editor-jjEx9u7D.css
-dw/server/ui/assets/editor.api-B_68laXH.js
+dw/server/ui/assets/editor.api-CArwgKCZ.js
 dw/server/ui/assets/editor.worker-q-txB4vs.js
 dw/server/ui/assets/elixir-6RTg0lbw.js
 dw/server/ui/assets/flow9-C5_-GSwl.js
-dw/server/ui/assets/freemarker2-C4IxK8gs.js
+dw/server/ui/assets/freemarker2-B42MpbWP.js
 dw/server/ui/assets/fsharp-C8Ef5oNN.js
 dw/server/ui/assets/go-C-y9NEjX.js
 dw/server/ui/assets/graphql-fmXr3nnJ.js
-dw/server/ui/assets/handlebars-BArDGhF4.js
+dw/server/ui/assets/handlebars-BhleeD_A.js
 dw/server/ui/assets/hcl-CpzslTdj.js
-dw/server/ui/assets/html-8yKFx0-x.js
+dw/server/ui/assets/html-Bi_v7qEe.js
 dw/server/ui/assets/html.worker-C93Ht9o9.js
-dw/server/ui/assets/htmlMode-BwXG6XFP.js
-dw/server/ui/assets/index-BY3Mtj4M.js
-dw/server/ui/assets/index-Bug26k2P.css
+dw/server/ui/assets/htmlMode-c39yjrZR.js
+dw/server/ui/assets/index-CKHZ9ZUD.css
+dw/server/ui/assets/index-Do_ZJ3yS.js
 dw/server/ui/assets/ini-sBoK_t0W.js
 dw/server/ui/assets/java-BEtHBSE6.js
-dw/server/ui/assets/javascript-B3mqpFQQ.js
+dw/server/ui/assets/javascript-BTuzJzdO.js
 dw/server/ui/assets/json.worker-B2V3pomh.js
-dw/server/ui/assets/jsonMode-C1m5Z8Ha.js
+dw/server/ui/assets/jsonMode-DY2FmMCP.js
 dw/server/ui/assets/julia-Bri6UV-V.js
 dw/server/ui/assets/kotlin-BOotOW0E.js
 dw/server/ui/assets/less-B9JPFI3C.js
 dw/server/ui/assets/lexon-CfSJPG6W.js
-dw/server/ui/assets/liquid-Cs5oVPJ1.js
-dw/server/ui/assets/lspLanguageFeatures-CnroSKaf.js
+dw/server/ui/assets/liquid-iXwDEnQP.js
+dw/server/ui/assets/lspLanguageFeatures-CVXyTq9V.js
 dw/server/ui/assets/lua-CsQS60Ue.js
 dw/server/ui/assets/m3-D-oSqn_W.js
 dw/server/ui/assets/markdown-Cimd5fb3.js
-dw/server/ui/assets/mdx-BGAIyUWE.js
+dw/server/ui/assets/mdx-QE70XlV0.js
 dw/server/ui/assets/mips-CIPQ_RoX.js
 dw/server/ui/assets/monaco--ixms01u.css
-dw/server/ui/assets/monaco-ab9gu2Wi.js
+dw/server/ui/assets/monaco-CzCru1EC.js
 dw/server/ui/assets/msdax-DauUninz.js
 dw/server/ui/assets/mysql-SOo6toE5.js
 dw/server/ui/assets/objective-c-FvmIjYaQ.js
@@ -117,10 +117,10 @@ dw/server/ui/assets/powerquery-D3hlyOfw.js
 dw/server/ui/assets/powershell-DmHpPYUd.js
 dw/server/ui/assets/protobuf-C531GsRP.js
 dw/server/ui/assets/pug-Z5eAx3Zn.js
-dw/server/ui/assets/python-XsfASqaE.js
+dw/server/ui/assets/python-DxHS-kHp.js
 dw/server/ui/assets/qsharp-DkqhCAOL.js
 dw/server/ui/assets/r-BwWrilGY.js
-dw/server/ui/assets/razor-C-1xVmcy.js
+dw/server/ui/assets/razor-CT5Qrhwg.js
 dw/server/ui/assets/redis-ClamHrr6.js
 dw/server/ui/assets/redshift-DT7zqm-g.js
 dw/server/ui/assets/restructuredtext-BYgofb2h.js
@@ -140,15 +140,15 @@ dw/server/ui/assets/swift-Bxkupp3x.js
 dw/server/ui/assets/systemverilog-Bz4Y3fRF.js
 dw/server/ui/assets/tcl-DISqw1ZD.js
 dw/server/ui/assets/ts.worker-D7T1-Ig5.js
-dw/server/ui/assets/tsMode-BDa9GeVq.js
+dw/server/ui/assets/tsMode-qXdDdnF1.js
 dw/server/ui/assets/twig-De2hgUGE.js
-dw/server/ui/assets/typescript-VE9GZNZf.js
+dw/server/ui/assets/typescript-BWfJagji.js
 dw/server/ui/assets/typespec-B8J7ngcE.js
 dw/server/ui/assets/vb-DV3o63ZY.js
 dw/server/ui/assets/wgsl-DpFanUEy.js
-dw/server/ui/assets/workers-C_2pIPh_.js
-dw/server/ui/assets/xml-BJp06vju.js
-dw/server/ui/assets/yaml-EI9SS79i.js
+dw/server/ui/assets/workers-C9vxrwoC.js
+dw/server/ui/assets/xml-DhJEwsDK.js
+dw/server/ui/assets/yaml-CiixMxAq.js
 dw/tasks/audio_utils.py
 dw/tasks/background_remover.py
 dw/tasks/borders.py
@@ -177,12 +177,23 @@ dw/workflows/augment_prompt.json
 dw/workflows/describe_image.json
 dw/workflows/h3_context_ir.json
 dw/workflows/test.json
+dw_mcp/__init__.py
+dw_mcp/__main__.py
+dw_mcp/authoring.py
+dw_mcp/catalog.py
+dw_mcp/client.py
+dw_mcp/diagnose.py
+dw_mcp/media.py
+dw_mcp/models.py
+dw_mcp/prompts.py
+dw_mcp/server.py
 tests/test_argument_updates.py
 tests/test_arguments.py
 tests/test_audio_utils.py
 tests/test_borders.py
 tests/test_cache_blocks.py
 tests/test_chain.py
+tests/test_cli.py
 tests/test_component_pruning.py
 tests/test_concat_videos.py
 tests/test_config_objects.py
@@ -190,6 +201,7 @@ tests/test_configuration_schema.py
 tests/test_depth_estimator.py
 tests/test_device.py
 tests/test_device_helpers.py
+tests/test_device_portability.py
 tests/test_diffusion_upscale.py
 tests/test_docs_links.py
 tests/test_events.py
@@ -202,9 +214,19 @@ tests/test_integration.py
 tests/test_interpolate_frames.py
 tests/test_introspection.py
 tests/test_list_images.py
+tests/test_mcp_authoring.py
+tests/test_mcp_catalog.py
+tests/test_mcp_client.py
+tests/test_mcp_diagnose.py
+tests/test_mcp_media.py
+tests/test_mcp_models.py
+tests/test_mcp_prompts.py
+tests/test_mcp_server.py
 tests/test_model_cache.py
 tests/test_modular_pipeline.py
+tests/test_offload_placement.py
 tests/test_pair_audio.py
+tests/test_phase_events.py
 tests/test_pipeline_caching.py
 tests/test_pipeline_components.py
 tests/test_previous_results.py
@@ -215,6 +237,7 @@ tests/test_repl_commands.py
 tests/test_repl_ergonomics.py
 tests/test_repl_hierarchical.py
 tests/test_repl_reorganization.py
+tests/test_repl_worker.py
 tests/test_resize_bucket.py
 tests/test_result.py
 tests/test_scheduler_config.py
@@ -231,8 +254,10 @@ tests/test_tensor_image.py
 tests/test_text_generation.py
 tests/test_text_sections.py
 tests/test_type_helpers.py
+tests/test_updater.py
 tests/test_variables.py
 tests/test_video_utils.py
 tests/test_worker.py
 tests/test_worker_execute.py
-tests/test_workflow.py
+tests/test_workflow.py
+tests/test_workflow_trust.py

{diffusers_workflow-0.4.0a8 → diffusers_workflow-0.4.0a12}/diffusers_workflow.egg-info/entry_points.txt

@@ -1,4 +1,5 @@
 [console_scripts]
+dw-mcp = dw_mcp.__main__:main
 dw-repl = dw.repl:main
 dw-run = dw.run:main
 dw-serve = dw.serve:main

{diffusers_workflow-0.4.0a8 → diffusers_workflow-0.4.0a12}/diffusers_workflow.egg-info/requires.txt

@@ -49,6 +49,11 @@ pytest-asyncio>=1.4.0
 httpx>=0.28.1
 black>=26.5.1
 build
+mcp>=2.1.1
+
+[mcp]
+mcp>=2.1.1
+httpx>=0.28.1
 
 [server]
 fastapi>=0.141.1

diffusers_workflow-0.4.0a12/diffusers_workflow.egg-info/top_level.txt

@@ -0,0 +1,2 @@
+dw
+dw_mcp

{diffusers_workflow-0.4.0a8 → diffusers_workflow-0.4.0a12}/dw/__init__.py

@@ -29,8 +29,20 @@ warnings.filterwarnings("ignore", category=FutureWarning)
 warnings.filterwarnings("ignore", category=UserWarning)
 warnings.filterwarnings("ignore", category=DeprecationWarning)
 
-# Lazy import torch to avoid import errors when torch isn't available
-# (e.g., when using system Python instead of venv)
+# Eager (but guarded) import of torch. This module's own top-level code needs
+# torch's exceptions/types unconditionally to detect its absence gracefully
+# (see _TORCH_AVAILABLE below) rather than deferring the try/except to first
+# use everywhere torch is touched. It would be nice for a torch-free path like
+# `python -m dw.validate` (pure JSON schema validation) to skip this multi-
+# second import, but that isn't achievable by changing dw/__init__.py alone:
+# dw/validate.py imports dw/workflow.py, which does its own unconditional
+# `import torch` at module scope (Workflow.run() uses torch.Generator() for
+# seeding) - so torch loads regardless of what this module does. Making the
+# import here lazy without also restructuring dw/workflow.py's random-seed
+# handling would just move the cost around, not remove it, and touching
+# dw/workflow.py's import structure is a larger, riskier change than this
+# fix pass is meant to make. Left eager on purpose; see the audit notes for
+# why a real fix needs to start in dw/workflow.py, not here.
 try:
     import torch
     import diffusers
@@ -138,6 +150,81 @@ def get_device_type(device=None):
     return torch.device(device if device is not None else get_device()).type
 
 
+def backend_available(device_type):
+    """
+    Whether this machine can actually run on a backend.
+
+    Only the backends dw knows how to probe get an answer. An unrecognized one is
+    reported available rather than guessed at, so a device torch understands and dw
+    does not is left alone instead of being rewritten.
+
+    Args:
+        device_type: Backend name ('cuda', 'mps', 'cpu', ...)
+
+    Returns:
+        bool: Whether the backend is usable here
+    """
+    if not _TORCH_AVAILABLE:
+        return device_type == "cpu"
+
+    if device_type == "cpu":
+        return True
+
+    if device_type == "cuda":
+        return torch.cuda.is_available()
+
+    if device_type == "mps":
+        return hasattr(torch.backends, "mps") and torch.backends.mps.is_available()
+
+    is_available = getattr(getattr(torch, device_type, None), "is_available", None)
+    return bool(is_available()) if callable(is_available) else True
+
+
+def resolve_device(requested):
+    """
+    Translate a device a workflow asked for into one this machine has.
+
+    A device named in a workflow is nearly always the author saying 'the accelerator'
+    rather than 'specifically NVIDIA', so a workflow written on a CUDA box runs on a
+    Mac and back again. The backend is what gets translated - an index is left alone
+    when the backend matches, since 'cuda:1' on a single-GPU CUDA box is a genuine
+    mistake and not a portability problem. A CPU device is never rewritten: pinning a
+    step to the CPU is how a GPU-specific problem gets ruled out.
+
+    Args:
+        requested: Device identifier from a workflow, or None for no override
+
+    Returns:
+        The device to use, or None if None was passed
+    """
+    if requested is None or not _TORCH_AVAILABLE:
+        return requested
+
+    try:
+        device = torch.device(requested)
+    except (RuntimeError, TypeError, ValueError):
+        # Not a device torch understands - let the failure surface where it is used
+        return requested
+
+    if device.type == "cpu" or backend_available(device.type):
+        return requested
+
+    target = get_device()
+    if device.index is not None:
+        logging.warning(
+            f"Workflow asks for '{requested}', which this machine has no {device.type} "
+            f"backend for - running on {target}. The device index is dropped, so a "
+            "workflow that meant to spread work across accelerators will not."
+        )
+    else:
+        logging.warning(
+            f"Workflow asks for '{requested}', which this machine has no {device.type} "
+            f"backend for - running on {target}"
+        )
+
+    return target
+
+
 def get_autocast_device_type():
     """
     Get the device type to use for torch.autocast.

{diffusers_workflow-0.4.0a8 → diffusers_workflow-0.4.0a12}/dw/events.py

@@ -27,6 +27,10 @@ class RunContext:
     def __init__(self, on_event=None):
         self._on_event = on_event
         self._cancel = threading.Event()
+        # The most recent phase this run reported (see emit_phase/PHASES) -
+        # cancel() reads it to tell a client whether the cancel takes effect
+        # right away or has to wait out an in-flight, non-interruptible phase
+        self._current_phase = None
         # Pipeline cache keys this run resolved - the worker evicts entries a
         # run no longer touches, so an edited workflow drops stale models
         self.touched_pipelines = set()
@@ -40,8 +44,18 @@ class RunContext:
             # A broken sink must not kill a run that is otherwise fine
             logger.warning(f"Progress event sink failed on '{event_type}': {e}")
 
+    def note_phase(self, phase):
+        """Record the run's latest phase - called by emit_phase, read by
+        cancel() to judge whether the current phase can be interrupted."""
+        self._current_phase = phase
+
     def cancel(self):
         self._cancel.set()
+        if self._current_phase in NON_INTERRUPTIBLE_PHASES:
+            # A model load or a task step has no checkpoint to catch this
+            # flag until it finishes - say so, or the client sees the cancel
+            # request go silent for however long that takes
+            self.emit("cancel_pending", phase=self._current_phase)
 
     @property
     def cancelled(self):
@@ -76,3 +90,27 @@ def activate_context(context):
 
 def deactivate_context(token):
     _active_context.reset(token)
+
+
+# The coarse states a run passes through. Small on purpose: a phase says what
+# the run is waiting on, not what any one library is doing internally
+PHASES = ("loading", "cached", "generating", "decoding", "saving", "task")
+
+# Phases with no checkpoint of their own: nothing inside a from_pretrained()
+# call or a task handler consults the cancel flag, so a cancel requested
+# during one of these can only take effect once the phase finishes on its own
+NON_INTERRUPTIBLE_PHASES = ("loading", "task")
+
+
+def emit_phase(phase, detail=None):
+    """Report a coarse phase change on the active run.
+
+    A step spends most of its wall clock outside the denoise loop - pulling
+    weights, decoding latents, encoding video - and a step counter says
+    nothing about any of that. These are the rest of the story. They are rare
+    enough (a handful per step) to carry a free-text detail alongside, which
+    is what makes 'loading' readable as 'which model'.
+    """
+    context = get_context()
+    context.note_phase(phase)
+    context.emit("phase", phase=phase, detail=detail)

{diffusers_workflow-0.4.0a8 → diffusers_workflow-0.4.0a12}/dw/hub_cache.py

@@ -150,6 +150,7 @@ class DownloadManager:
         except HFValidationError as e:
             raise ValueError(str(e))
 
+        cancel_event = threading.Event()
         with self._lock:
             for entry in self._downloads.values():
                 if entry["repo_id"] == repo_id and entry["status"] == "downloading":
@@ -163,12 +164,14 @@ class DownloadManager:
                 "error": None,
                 "started_at": time.time(),
                 "finished_at": None,
+                # Set before the entry is published: cancel() reaches for this
+                # the moment the id is visible, and assigning it after the
+                # lock left a window where cancelling raised KeyError
+                "_cancel": cancel_event,
             }
             self._downloads[entry["id"]] = entry
             self._prune()
 
-        cancel_event = threading.Event()
-        entry["_cancel"] = cancel_event
         thread = threading.Thread(
             target=self._run, args=(entry, cancel_event), daemon=True
         )