diffusers-workflow 0.4.0a4__tar.gz → 0.4.0a10__tar.gz

{diffusers_workflow-0.4.0a4/diffusers_workflow.egg-info → diffusers_workflow-0.4.0a10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffusers-workflow
-Version: 0.4.0a4
+Version: 0.4.0a10
 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)
@@ -85,6 +89,7 @@ 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
@@ -92,6 +97,7 @@ A declarative workflow engine and web UI for the [Hugging Face Diffusers library
 - **Quantization** — BitsAndBytes, TorchAO, GGUF, SDNQ, optimum-quanto
 - **Inference acceleration** — TeaCache, FirstBlockCache, FasterCache, MagCache, TaylorSeerCache
 - **Prompt weighting** — A1111-style `(word:1.5)` syntax with long prompt support
+- **Prompt library** — store prompts once in `prompts/` and reference them from any workflow as `prompt:name` or `prompt:folder/name`, with a web UI for browsing, editing, and AI-enhancing them
 - **LoRA and IP-Adapter** support
 - **Composable workflows** from multiple JSON files with `builtin:` references
 - **Utility tasks** — upscaling, face restoration, segmentation, captioning, frame interpolation, QR codes, and more
@@ -135,6 +141,8 @@ Everything the engine does, in a browser backed by a persistent GPU worker — m
 
 ![The gallery with generated images and videos](docs/img/ui-gallery.jpg)
 
+**A prompt library shared by every workflow.** Store a prompt once, reference it anywhere as `prompt:name` — the Prompts page browses, edits, and filters the library, and an *Enhance with AI* panel expands an idea into a full prompt with a local language model.
+
 **A model manager for the disk your models actually consume.** The Hugging Face hub cache, inventoried: sizes, revisions, last-used dates, free space — download new models by id with live progress, delete with one click.
 
 ![The model manager listing cached models with sizes](docs/img/ui-models.png)
@@ -146,14 +154,14 @@ Jobs queue, stream progress live (per denoising step), cancel cooperatively, and
 ### Run a Workflow
 
 ```bash
-python -m dw.run examples/flux/FluxDev.json
-python -m dw.run examples/flux/FluxDev.json prompt="a cat" num_images_per_prompt=4
+python -m dw.run workflows/flux/FluxDev.json
+python -m dw.run workflows/flux/FluxDev.json prompt="a cat" num_images_per_prompt=4
 ```
 
 ### Validate a Workflow
 
 ```bash
-python -m dw.validate examples/flux/FluxDev.json
+python -m dw.validate workflows/flux/FluxDev.json
 ```
 
 ### Interactive REPL
@@ -313,13 +321,14 @@ a (photorealistic:1.4) portrait with (bright red hair:1.3) and [freckles]
 
 **Interactive schema browser:** [View Schema](https://json-schema.app/view/%23?url=https%3A%2F%2Fraw.githubusercontent.com%2Fdkackman%2Fdiffusers-workflow%2Frefs%2Fheads%2Fmaster%2Fdw%2Fworkflow_schema.json)
 
-See [examples/](examples/) for more workflow files.
+See [workflows/](workflows/) for more workflow files.
 
 ## Documentation
 
 ### 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
@@ -327,6 +336,7 @@ See [examples/](examples/) for more workflow files.
 - [LoRA](docs/LORAS.md) — Loading and stacking LoRA adapters
 - [IP-Adapter](docs/IP_ADAPTER.md) — Image-prompt conditioning
 - [Prompt Weighting](docs/PROMPT_WEIGHTING.md) — A1111-style syntax
+- [Prompt References](docs/WORKFLOW_GUIDE.md#prompt-references) — The stored prompt library and `prompt:` references
 - [Tasks](docs/TASKS.md) — Image processing, ControlNet preprocessors, utilities
 
 ### Reference

{diffusers_workflow-0.4.0a4 → diffusers_workflow-0.4.0a10}/README.md

@@ -14,6 +14,7 @@ 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
@@ -21,6 +22,7 @@ A declarative workflow engine and web UI for the [Hugging Face Diffusers library
 - **Quantization** — BitsAndBytes, TorchAO, GGUF, SDNQ, optimum-quanto
 - **Inference acceleration** — TeaCache, FirstBlockCache, FasterCache, MagCache, TaylorSeerCache
 - **Prompt weighting** — A1111-style `(word:1.5)` syntax with long prompt support
+- **Prompt library** — store prompts once in `prompts/` and reference them from any workflow as `prompt:name` or `prompt:folder/name`, with a web UI for browsing, editing, and AI-enhancing them
 - **LoRA and IP-Adapter** support
 - **Composable workflows** from multiple JSON files with `builtin:` references
 - **Utility tasks** — upscaling, face restoration, segmentation, captioning, frame interpolation, QR codes, and more
@@ -64,6 +66,8 @@ Everything the engine does, in a browser backed by a persistent GPU worker — m
 
 ![The gallery with generated images and videos](docs/img/ui-gallery.jpg)
 
+**A prompt library shared by every workflow.** Store a prompt once, reference it anywhere as `prompt:name` — the Prompts page browses, edits, and filters the library, and an *Enhance with AI* panel expands an idea into a full prompt with a local language model.
+
 **A model manager for the disk your models actually consume.** The Hugging Face hub cache, inventoried: sizes, revisions, last-used dates, free space — download new models by id with live progress, delete with one click.
 
 ![The model manager listing cached models with sizes](docs/img/ui-models.png)
@@ -75,14 +79,14 @@ Jobs queue, stream progress live (per denoising step), cancel cooperatively, and
 ### Run a Workflow
 
 ```bash
-python -m dw.run examples/flux/FluxDev.json
-python -m dw.run examples/flux/FluxDev.json prompt="a cat" num_images_per_prompt=4
+python -m dw.run workflows/flux/FluxDev.json
+python -m dw.run workflows/flux/FluxDev.json prompt="a cat" num_images_per_prompt=4
 ```
 
 ### Validate a Workflow
 
 ```bash
-python -m dw.validate examples/flux/FluxDev.json
+python -m dw.validate workflows/flux/FluxDev.json
 ```
 
 ### Interactive REPL
@@ -242,13 +246,14 @@ a (photorealistic:1.4) portrait with (bright red hair:1.3) and [freckles]
 
 **Interactive schema browser:** [View Schema](https://json-schema.app/view/%23?url=https%3A%2F%2Fraw.githubusercontent.com%2Fdkackman%2Fdiffusers-workflow%2Frefs%2Fheads%2Fmaster%2Fdw%2Fworkflow_schema.json)
 
-See [examples/](examples/) for more workflow files.
+See [workflows/](workflows/) for more workflow files.
 
 ## Documentation
 
 ### 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
@@ -256,6 +261,7 @@ See [examples/](examples/) for more workflow files.
 - [LoRA](docs/LORAS.md) — Loading and stacking LoRA adapters
 - [IP-Adapter](docs/IP_ADAPTER.md) — Image-prompt conditioning
 - [Prompt Weighting](docs/PROMPT_WEIGHTING.md) — A1111-style syntax
+- [Prompt References](docs/WORKFLOW_GUIDE.md#prompt-references) — The stored prompt library and `prompt:` references
 - [Tasks](docs/TASKS.md) — Image processing, ControlNet preprocessors, utilities
 
 ### Reference

{diffusers_workflow-0.4.0a4 → diffusers_workflow-0.4.0a10/diffusers_workflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffusers-workflow
-Version: 0.4.0a4
+Version: 0.4.0a10
 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)
@@ -85,6 +89,7 @@ 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
@@ -92,6 +97,7 @@ A declarative workflow engine and web UI for the [Hugging Face Diffusers library
 - **Quantization** — BitsAndBytes, TorchAO, GGUF, SDNQ, optimum-quanto
 - **Inference acceleration** — TeaCache, FirstBlockCache, FasterCache, MagCache, TaylorSeerCache
 - **Prompt weighting** — A1111-style `(word:1.5)` syntax with long prompt support
+- **Prompt library** — store prompts once in `prompts/` and reference them from any workflow as `prompt:name` or `prompt:folder/name`, with a web UI for browsing, editing, and AI-enhancing them
 - **LoRA and IP-Adapter** support
 - **Composable workflows** from multiple JSON files with `builtin:` references
 - **Utility tasks** — upscaling, face restoration, segmentation, captioning, frame interpolation, QR codes, and more
@@ -135,6 +141,8 @@ Everything the engine does, in a browser backed by a persistent GPU worker — m
 
 ![The gallery with generated images and videos](docs/img/ui-gallery.jpg)
 
+**A prompt library shared by every workflow.** Store a prompt once, reference it anywhere as `prompt:name` — the Prompts page browses, edits, and filters the library, and an *Enhance with AI* panel expands an idea into a full prompt with a local language model.
+
 **A model manager for the disk your models actually consume.** The Hugging Face hub cache, inventoried: sizes, revisions, last-used dates, free space — download new models by id with live progress, delete with one click.
 
 ![The model manager listing cached models with sizes](docs/img/ui-models.png)
@@ -146,14 +154,14 @@ Jobs queue, stream progress live (per denoising step), cancel cooperatively, and
 ### Run a Workflow
 
 ```bash
-python -m dw.run examples/flux/FluxDev.json
-python -m dw.run examples/flux/FluxDev.json prompt="a cat" num_images_per_prompt=4
+python -m dw.run workflows/flux/FluxDev.json
+python -m dw.run workflows/flux/FluxDev.json prompt="a cat" num_images_per_prompt=4
 ```
 
 ### Validate a Workflow
 
 ```bash
-python -m dw.validate examples/flux/FluxDev.json
+python -m dw.validate workflows/flux/FluxDev.json
 ```
 
 ### Interactive REPL
@@ -313,13 +321,14 @@ a (photorealistic:1.4) portrait with (bright red hair:1.3) and [freckles]
 
 **Interactive schema browser:** [View Schema](https://json-schema.app/view/%23?url=https%3A%2F%2Fraw.githubusercontent.com%2Fdkackman%2Fdiffusers-workflow%2Frefs%2Fheads%2Fmaster%2Fdw%2Fworkflow_schema.json)
 
-See [examples/](examples/) for more workflow files.
+See [workflows/](workflows/) for more workflow files.
 
 ## Documentation
 
 ### 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
@@ -327,6 +336,7 @@ See [examples/](examples/) for more workflow files.
 - [LoRA](docs/LORAS.md) — Loading and stacking LoRA adapters
 - [IP-Adapter](docs/IP_ADAPTER.md) — Image-prompt conditioning
 - [Prompt Weighting](docs/PROMPT_WEIGHTING.md) — A1111-style syntax
+- [Prompt References](docs/WORKFLOW_GUIDE.md#prompt-references) — The stored prompt library and `prompt:` references
 - [Tasks](docs/TASKS.md) — Image processing, ControlNet preprocessors, utilities
 
 ### Reference

{diffusers_workflow-0.4.0a4 → diffusers_workflow-0.4.0a10}/diffusers_workflow.egg-info/SOURCES.txt

@@ -17,7 +17,9 @@ dw/hub_cache.py
 dw/introspection.py
 dw/log_setup.py
 dw/previous_results.py
+dw/prompt_schema.json
 dw/prompt_weighting.py
+dw/prompts.py
 dw/repl.py
 dw/repl_commands.py
 dw/repl_worker.py
@@ -44,6 +46,7 @@ dw/pipeline_processors/pipeline.py
 dw/pipeline_processors/remote.py
 dw/server/__init__.py
 dw/server/app.py
+dw/server/enhancers.py
 dw/server/jobs.py
 dw/server/updater.py
 dw/server/ui/index.html
@@ -61,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-DklhHuQC.js
+dw/server/ui/assets/cssMode-DBf-T_da.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-DCWJa9Fw.js
+dw/server/ui/assets/editor.api-B0D5Ygq7.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-CVxDKoFq.js
+dw/server/ui/assets/freemarker2-Br8g4G1T.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-DK-ehq_7.js
+dw/server/ui/assets/handlebars-BKo1TaOF.js
 dw/server/ui/assets/hcl-CpzslTdj.js
-dw/server/ui/assets/html-CrtXvEnP.js
+dw/server/ui/assets/html-Bis1seyM.js
 dw/server/ui/assets/html.worker-C93Ht9o9.js
-dw/server/ui/assets/htmlMode-D0r9dgS3.js
-dw/server/ui/assets/index-SQcvGbXy.css
-dw/server/ui/assets/index-qLIa9n7C.js
+dw/server/ui/assets/htmlMode-D2SOD2MF.js
+dw/server/ui/assets/index-D7eWnTCB.css
+dw/server/ui/assets/index-JyFxgIR9.js
 dw/server/ui/assets/ini-sBoK_t0W.js
 dw/server/ui/assets/java-BEtHBSE6.js
-dw/server/ui/assets/javascript-rD650v5-.js
+dw/server/ui/assets/javascript-CkEbVb78.js
 dw/server/ui/assets/json.worker-B2V3pomh.js
-dw/server/ui/assets/jsonMode-DvWKiZ5V.js
+dw/server/ui/assets/jsonMode-dVJbYwxg.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-DvjV9-yQ.js
-dw/server/ui/assets/lspLanguageFeatures-Bv4c-HOB.js
+dw/server/ui/assets/liquid-Btak-6uF.js
+dw/server/ui/assets/lspLanguageFeatures-BfeZZQbb.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-CmyDcP1k.js
+dw/server/ui/assets/mdx-D9KXK7-s.js
 dw/server/ui/assets/mips-CIPQ_RoX.js
 dw/server/ui/assets/monaco--ixms01u.css
-dw/server/ui/assets/monaco-CFDZdQeR.js
+dw/server/ui/assets/monaco-ZMCk_MWf.js
 dw/server/ui/assets/msdax-DauUninz.js
 dw/server/ui/assets/mysql-SOo6toE5.js
 dw/server/ui/assets/objective-c-FvmIjYaQ.js
@@ -114,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-BbNG9-ih.js
+dw/server/ui/assets/python-0FnO4xGM.js
 dw/server/ui/assets/qsharp-DkqhCAOL.js
 dw/server/ui/assets/r-BwWrilGY.js
-dw/server/ui/assets/razor-BruFjATi.js
+dw/server/ui/assets/razor-B6PMzeY4.js
 dw/server/ui/assets/redis-ClamHrr6.js
 dw/server/ui/assets/redshift-DT7zqm-g.js
 dw/server/ui/assets/restructuredtext-BYgofb2h.js
@@ -137,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-DhwsPBWM.js
+dw/server/ui/assets/tsMode-nxyvZH1t.js
 dw/server/ui/assets/twig-De2hgUGE.js
-dw/server/ui/assets/typescript-DLBTIt_O.js
+dw/server/ui/assets/typescript-BhBe8aGf.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-BD2vxpDW.js
-dw/server/ui/assets/xml-BCRgtEff.js
-dw/server/ui/assets/yaml-B9W8qjO1.js
+dw/server/ui/assets/workers-ChAf-t8A.js
+dw/server/ui/assets/xml-z2r9dHSO.js
+dw/server/ui/assets/yaml-DrcHrz7f.js
 dw/tasks/audio_utils.py
 dw/tasks/background_remover.py
 dw/tasks/borders.py
@@ -174,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
@@ -187,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
@@ -199,18 +214,30 @@ 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
 tests/test_prompt_parser.py
 tests/test_prompt_weighting.py
+tests/test_prompts.py
 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

{diffusers_workflow-0.4.0a4 → diffusers_workflow-0.4.0a10}/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.0a4 → diffusers_workflow-0.4.0a10}/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.0a10/diffusers_workflow.egg-info/top_level.txt

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

{diffusers_workflow-0.4.0a4 → diffusers_workflow-0.4.0a10}/dw/__init__.py

@@ -138,6 +138,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.0a4 → diffusers_workflow-0.4.0a10}/dw/arguments.py

@@ -3,6 +3,7 @@ import copy
 import logging
 from inspect import Parameter, signature
 from .type_helpers import load_type_from_name, load_constant_from_name, has_method
+from .prompts import PROMPT_PREFIX, fetch_prompt
 from diffusers.utils import load_image, load_video
 from .security import (
     validate_path,
@@ -92,6 +93,11 @@ def realize_args(arg, base_dir=None):
             # conventions below - what it holds is the value, not a file to load
             if is_constant_reference(v):
                 arg[k] = fetch_constant(v)
+            # A stored prompt resolves the same way - to its text, under any
+            # name, before the media conventions could mistake it for a file.
+            # The workflow's directory anchors prompt-library discovery
+            elif is_prompt_reference(v):
+                arg[k] = fetch_prompt(v, base_dir=base_dir)
             # An explicit media reference loads under any argument name - the
             # key conventions below only cover arguments named like their media
             elif is_media_reference(v):
@@ -140,6 +146,9 @@ def realize_args(arg, base_dir=None):
             if is_constant_reference(item):
                 arg[i] = fetch_constant(item)
                 continue
+            if is_prompt_reference(item):
+                arg[i] = fetch_prompt(item, base_dir=base_dir)
+                continue
             if is_media_reference(item):
                 arg[i] = fetch_media(item, base_dir)
                 continue
@@ -152,6 +161,11 @@ def is_constant_reference(value):
     return isinstance(value, str) and value.startswith(CONSTANT_PREFIX)
 
 
+def is_prompt_reference(value):
+    """Whether a value references a stored prompt in the prompt library."""
+    return isinstance(value, str) and value.startswith(PROMPT_PREFIX)
+
+
 def fetch_constant(reference):
     """Read the value a 'constant:' reference names.
 
@@ -703,12 +717,22 @@ def media_arguments(object_type, artifact):
             arguments["sample_rate"] = sample_rate
         return arguments
 
+    if audio is None and hasattr(artifact, "ndim") and artifact.ndim <= 3:
+        # A step that generates audio alone - a music pipeline, a slice_audio
+        # task - produces the waveform itself rather than an AudioVideo. It
+        # carries no rate of its own, so the reference declares 'sample_rate'
+        # alongside 'from_previous_result'
+        audio = torch.as_tensor(as_channels_samples(artifact))
+
     if audio is None:
         raise ValueError(
             f"{object_type.__name__} holds audio, but the step it names produced "
             f"none - reference a step that generates a soundtrack"
         )
-    return {"audio": audio, "sample_rate": sample_rate}
+    arguments = {"audio": audio}
+    if sample_rate is not None:
+        arguments["sample_rate"] = sample_rate
+    return arguments
 
 
 def validate_media_location(location, base_dir=None):

{diffusers_workflow-0.4.0a4 → diffusers_workflow-0.4.0a10}/dw/events.py

@@ -76,3 +76,20 @@ 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")
+
+
+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'.
+    """
+    get_context().emit("phase", phase=phase, detail=detail)