stageflow-framework 0.1.6__tar.gz → 0.10.0__tar.gz

stageflow_framework-0.10.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025-2026 лень
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

stageflow_framework-0.10.0/PKG-INFO

@@ -0,0 +1,186 @@
+Metadata-Version: 2.4
+Name: stageflow-framework
+Version: 0.10.0
+Summary: StageFlow: pipeline framework for stages
+Author-email: лень <pzrnqt1vrss@protonmail.com>
+License-Expression: MIT
+Project-URL: Repository, https://github.com/leo-need-more-coffee/stageflow
+Project-URL: Issues, https://github.com/leo-need-more-coffee/stageflow/issues
+Keywords: pipeline,workflow,orchestration,state-machine,cel
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
+Classifier: Typing :: Typed
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pyyaml
+Requires-Dist: jsonschema>=4.0.0
+Requires-Dist: immutables>=0.20
+Requires-Dist: common-expression-language>=0.9
+Provides-Extra: dev
+Requires-Dist: flake8; extra == "dev"
+Requires-Dist: build; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: mkdocs-material; extra == "docs"
+Requires-Dist: mkdocs-static-i18n; extra == "docs"
+Dynamic: license-file
+
+<div align="center">
+
+# StageFlow
+
+**Pipelines described in JSON, executed in Python — and debuggable node by node.**
+
+[![tests](https://github.com/leo-need-more-coffee/stageflow/actions/workflows/tests.yml/badge.svg)](https://github.com/leo-need-more-coffee/stageflow/actions/workflows/tests.yml)
+[![PyPI](https://img.shields.io/pypi/v/stageflow-framework)](https://pypi.org/project/stageflow-framework/)
+[![Python](https://img.shields.io/pypi/pyversions/stageflow-framework)](https://pypi.org/project/stageflow-framework/)
+[![docs](https://img.shields.io/badge/docs-stageflow-blue)](https://leo-need-more-coffee.github.io/stageflow/)
+[![license](https://img.shields.io/badge/license-MIT-green)](LICENSE)
+
+[Documentation](https://leo-need-more-coffee.github.io/stageflow/) ·
+[Tutorial](https://leo-need-more-coffee.github.io/stageflow/tutorial/) ·
+[Документация на русском](https://leo-need-more-coffee.github.io/stageflow/ru/)
+
+</div>
+
+A pipeline is a graph of nodes in JSON. The work happens in stages — Python
+classes you write. Between them travels one immutable frame of variables, and
+everything the graph does with that frame is visible in the JSON: branching,
+retries, error handling, concurrency, nested graphs.
+
+Because the pipeline is data, it can be stored, diffed, generated, validated
+before it runs — and drawn:
+
+![A pipeline open in the StageFlow editor](docs/img/tut-final.png)
+
+That is the [editor](https://github.com/leo-need-more-coffee/stageflow-ui): a
+separate static page that draws and debugs a graph while this core executes it.
+
+## Install
+
+```bash
+pip install stageflow-framework
+```
+
+Python 3.11+.
+
+## A pipeline in 30 seconds
+
+```python
+import asyncio
+
+from stageflow import BaseStage, Pipeline, Session, register_stage
+
+
+@register_stage("HelloStage")
+class HelloStage(BaseStage):
+    """
+    description: "Greets whoever the pipeline points at"
+    arguments:
+      name: string
+    outputs:
+      greeting: string
+    """
+
+    async def run(self):
+        name = self.get_arguments().get("name", "world")
+        self.set_outputs({"greeting": f"Hello, {name}!"})
+
+
+pipeline = Pipeline.from_dict({
+    "nodes": [
+        {"id": "start", "type": "entry",
+         "variables": {"user_name": "Alice"}, "next": "hello"},
+        {"id": "hello", "type": "stage", "stage": "HelloStage",
+         "arguments": {"vars": {"name": "user_name"}},
+         "outputs": {"greeting": "greeting"}, "next": "finish"},
+        {"id": "finish", "type": "terminal",
+         "result": {"status": "ok"}, "artifacts": ["greeting"]},
+    ],
+})
+pipeline.validate()
+
+result = asyncio.run(Session(id="demo", pipeline=pipeline).run())
+print(result.result)     # {'status': 'ok'}
+print(result.artifacts)  # {'greeting': 'Hello, Alice!'}
+```
+
+The docstring is the stage's specification. `validate()` checks the graph
+against it, so a wrong argument name is an error before anything runs.
+
+## What the graph can do
+
+| | |
+|---|---|
+| **Nine node types** | `entry`, `stage`, `condition`, `switch`, `parallel`, `try`, `map`, `subpipeline`, `terminal` |
+| **One immutable frame** | variables travel along the path; a write produces a new frame, so branches never collide |
+| **CEL expressions** | in conditions, in switch cases, and in any argument or output through the `.$` suffix |
+| **Errors as roads** | `retry` on a node, `try`/`except` over a region of the graph derived from its shape |
+| **Real concurrency** | `parallel` branches with their own frames and explicit merge rules |
+| **Loops over data** | `map` runs a region of the graph once per element, sequentially or at once |
+| **Nested graphs** | a `subpipeline` starts with a fresh frame and returns artifacts |
+| **Gradual typing** | declare the variables that matter; checked at validation and on every write |
+| **Step debugging** | stop between nodes, read and edit the frame, replay the event stream |
+
+## Documentation
+
+The [tutorial](https://leo-need-more-coffee.github.io/stageflow/tutorial/)
+builds one working pipeline step by step, with screenshots from the editor.
+The reference covers the rest:
+
+[Quick start](https://leo-need-more-coffee.github.io/stageflow/quick-start/) ·
+[Node types](https://leo-need-more-coffee.github.io/stageflow/node-types/) ·
+[Data model](https://leo-need-more-coffee.github.io/stageflow/data-model/) ·
+[Expressions](https://leo-need-more-coffee.github.io/stageflow/expressions/) ·
+[Errors](https://leo-need-more-coffee.github.io/stageflow/errors/) ·
+[Variable typing](https://leo-need-more-coffee.github.io/stageflow/variable-typing/) ·
+[Session control](https://leo-need-more-coffee.github.io/stageflow/session-control/) ·
+[Step debugging](https://leo-need-more-coffee.github.io/stageflow/step-debugging/)
+
+One page per node type:
+[entry](https://leo-need-more-coffee.github.io/stageflow/entry-node/) ·
+[stage](https://leo-need-more-coffee.github.io/stageflow/stage-node/) ·
+[condition](https://leo-need-more-coffee.github.io/stageflow/condition-node/) ·
+[switch](https://leo-need-more-coffee.github.io/stageflow/switch-node/) ·
+[parallel](https://leo-need-more-coffee.github.io/stageflow/parallel-branches/) ·
+[try](https://leo-need-more-coffee.github.io/stageflow/try-node/) ·
+[map](https://leo-need-more-coffee.github.io/stageflow/map-node/) ·
+[subpipeline](https://leo-need-more-coffee.github.io/stageflow/subpipeline-node/) ·
+[terminal](https://leo-need-more-coffee.github.io/stageflow/terminal-node/)
+
+Sources are in [`docs/`](docs) (`*.md` English, `*.ru.md` Russian) and publish
+themselves on every push to `main`.
+
+## The rest of the project
+
+| Repository | What it is |
+|---|---|
+| **stageflow** | this one: the core that runs the pipelines |
+| [stageflow-ui](https://github.com/leo-need-more-coffee/stageflow-ui) | the editor: a static page that draws and debugs a graph |
+| [stageflow-example](https://github.com/leo-need-more-coffee/stageflow-example) | a working backend for the editor: a support bot in four pipelines |
+
+## Development
+
+```bash
+pip install -e ".[dev]"
+python -m unittest discover -s tests
+```
+
+Pipelines can be tested declaratively:
+
+```python
+from stageflow.testing import PipelineTestSpec, run_pipeline_test
+```
+
+Releases are tag-driven: the tag has to match `project.version` in
+`pyproject.toml`, and the workflow publishes to PyPI.
+
+## License
+
+MIT — see [LICENSE](LICENSE).

stageflow_framework-0.10.0/README.md

@@ -0,0 +1,153 @@
+<div align="center">
+
+# StageFlow
+
+**Pipelines described in JSON, executed in Python — and debuggable node by node.**
+
+[![tests](https://github.com/leo-need-more-coffee/stageflow/actions/workflows/tests.yml/badge.svg)](https://github.com/leo-need-more-coffee/stageflow/actions/workflows/tests.yml)
+[![PyPI](https://img.shields.io/pypi/v/stageflow-framework)](https://pypi.org/project/stageflow-framework/)
+[![Python](https://img.shields.io/pypi/pyversions/stageflow-framework)](https://pypi.org/project/stageflow-framework/)
+[![docs](https://img.shields.io/badge/docs-stageflow-blue)](https://leo-need-more-coffee.github.io/stageflow/)
+[![license](https://img.shields.io/badge/license-MIT-green)](LICENSE)
+
+[Documentation](https://leo-need-more-coffee.github.io/stageflow/) ·
+[Tutorial](https://leo-need-more-coffee.github.io/stageflow/tutorial/) ·
+[Документация на русском](https://leo-need-more-coffee.github.io/stageflow/ru/)
+
+</div>
+
+A pipeline is a graph of nodes in JSON. The work happens in stages — Python
+classes you write. Between them travels one immutable frame of variables, and
+everything the graph does with that frame is visible in the JSON: branching,
+retries, error handling, concurrency, nested graphs.
+
+Because the pipeline is data, it can be stored, diffed, generated, validated
+before it runs — and drawn:
+
+![A pipeline open in the StageFlow editor](docs/img/tut-final.png)
+
+That is the [editor](https://github.com/leo-need-more-coffee/stageflow-ui): a
+separate static page that draws and debugs a graph while this core executes it.
+
+## Install
+
+```bash
+pip install stageflow-framework
+```
+
+Python 3.11+.
+
+## A pipeline in 30 seconds
+
+```python
+import asyncio
+
+from stageflow import BaseStage, Pipeline, Session, register_stage
+
+
+@register_stage("HelloStage")
+class HelloStage(BaseStage):
+    """
+    description: "Greets whoever the pipeline points at"
+    arguments:
+      name: string
+    outputs:
+      greeting: string
+    """
+
+    async def run(self):
+        name = self.get_arguments().get("name", "world")
+        self.set_outputs({"greeting": f"Hello, {name}!"})
+
+
+pipeline = Pipeline.from_dict({
+    "nodes": [
+        {"id": "start", "type": "entry",
+         "variables": {"user_name": "Alice"}, "next": "hello"},
+        {"id": "hello", "type": "stage", "stage": "HelloStage",
+         "arguments": {"vars": {"name": "user_name"}},
+         "outputs": {"greeting": "greeting"}, "next": "finish"},
+        {"id": "finish", "type": "terminal",
+         "result": {"status": "ok"}, "artifacts": ["greeting"]},
+    ],
+})
+pipeline.validate()
+
+result = asyncio.run(Session(id="demo", pipeline=pipeline).run())
+print(result.result)     # {'status': 'ok'}
+print(result.artifacts)  # {'greeting': 'Hello, Alice!'}
+```
+
+The docstring is the stage's specification. `validate()` checks the graph
+against it, so a wrong argument name is an error before anything runs.
+
+## What the graph can do
+
+| | |
+|---|---|
+| **Nine node types** | `entry`, `stage`, `condition`, `switch`, `parallel`, `try`, `map`, `subpipeline`, `terminal` |
+| **One immutable frame** | variables travel along the path; a write produces a new frame, so branches never collide |
+| **CEL expressions** | in conditions, in switch cases, and in any argument or output through the `.$` suffix |
+| **Errors as roads** | `retry` on a node, `try`/`except` over a region of the graph derived from its shape |
+| **Real concurrency** | `parallel` branches with their own frames and explicit merge rules |
+| **Loops over data** | `map` runs a region of the graph once per element, sequentially or at once |
+| **Nested graphs** | a `subpipeline` starts with a fresh frame and returns artifacts |
+| **Gradual typing** | declare the variables that matter; checked at validation and on every write |
+| **Step debugging** | stop between nodes, read and edit the frame, replay the event stream |
+
+## Documentation
+
+The [tutorial](https://leo-need-more-coffee.github.io/stageflow/tutorial/)
+builds one working pipeline step by step, with screenshots from the editor.
+The reference covers the rest:
+
+[Quick start](https://leo-need-more-coffee.github.io/stageflow/quick-start/) ·
+[Node types](https://leo-need-more-coffee.github.io/stageflow/node-types/) ·
+[Data model](https://leo-need-more-coffee.github.io/stageflow/data-model/) ·
+[Expressions](https://leo-need-more-coffee.github.io/stageflow/expressions/) ·
+[Errors](https://leo-need-more-coffee.github.io/stageflow/errors/) ·
+[Variable typing](https://leo-need-more-coffee.github.io/stageflow/variable-typing/) ·
+[Session control](https://leo-need-more-coffee.github.io/stageflow/session-control/) ·
+[Step debugging](https://leo-need-more-coffee.github.io/stageflow/step-debugging/)
+
+One page per node type:
+[entry](https://leo-need-more-coffee.github.io/stageflow/entry-node/) ·
+[stage](https://leo-need-more-coffee.github.io/stageflow/stage-node/) ·
+[condition](https://leo-need-more-coffee.github.io/stageflow/condition-node/) ·
+[switch](https://leo-need-more-coffee.github.io/stageflow/switch-node/) ·
+[parallel](https://leo-need-more-coffee.github.io/stageflow/parallel-branches/) ·
+[try](https://leo-need-more-coffee.github.io/stageflow/try-node/) ·
+[map](https://leo-need-more-coffee.github.io/stageflow/map-node/) ·
+[subpipeline](https://leo-need-more-coffee.github.io/stageflow/subpipeline-node/) ·
+[terminal](https://leo-need-more-coffee.github.io/stageflow/terminal-node/)
+
+Sources are in [`docs/`](docs) (`*.md` English, `*.ru.md` Russian) and publish
+themselves on every push to `main`.
+
+## The rest of the project
+
+| Repository | What it is |
+|---|---|
+| **stageflow** | this one: the core that runs the pipelines |
+| [stageflow-ui](https://github.com/leo-need-more-coffee/stageflow-ui) | the editor: a static page that draws and debugs a graph |
+| [stageflow-example](https://github.com/leo-need-more-coffee/stageflow-example) | a working backend for the editor: a support bot in four pipelines |
+
+## Development
+
+```bash
+pip install -e ".[dev]"
+python -m unittest discover -s tests
+```
+
+Pipelines can be tested declaratively:
+
+```python
+from stageflow.testing import PipelineTestSpec, run_pipeline_test
+```
+
+Releases are tag-driven: the tag has to match `project.version` in
+`pyproject.toml`, and the workflow publishes to PyPI.
+
+## License
+
+MIT — see [LICENSE](LICENSE).

stageflow_framework-0.10.0/pyproject.toml

@@ -0,0 +1,54 @@
+[build-system]
+requires = ["setuptools>=77.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "stageflow-framework"
+version = "0.10.0"
+description = "StageFlow: pipeline framework for stages"
+readme = "README.md"
+license = "MIT"
+license-files = ["LICENSE"]
+authors = [
+    { name = "лень", email = "pzrnqt1vrss@protonmail.com" }
+]
+dependencies = [
+    "pyyaml",
+    "jsonschema>=4.0.0",
+    "immutables>=0.20",
+    "common-expression-language>=0.9",
+]
+requires-python = ">=3.11"
+keywords = ["pipeline", "workflow", "orchestration", "state-machine", "cel"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+    "Topic :: Software Development :: Libraries :: Application Frameworks",
+    "Typing :: Typed",
+]
+
+[project.urls]
+Repository = "https://github.com/leo-need-more-coffee/stageflow"
+Issues = "https://github.com/leo-need-more-coffee/stageflow/issues"
+
+[project.optional-dependencies]
+dev = [
+    "flake8",
+    "build",
+    "twine",
+]
+docs = [
+    "mkdocs-material",
+    "mkdocs-static-i18n",
+]
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["stageflow*"]
+
+[tool.setuptools.package-data]
+"stageflow" = ["py.typed", "docs/schemas/pipeline.json"]

stageflow_framework-0.10.0/stageflow/__init__.py

@@ -0,0 +1,99 @@
+from . import builtins  # noqa: F401
+from .core import (
+    BaseStage,
+    CelEngine,
+    CelError,
+    ConditionNode,
+    Context,
+    EntryNode,
+    Event,
+    EventSpec,
+    InputSpec,
+    MapNode,
+    Node,
+    ParallelNode,
+    Pipeline,
+    Retrier,
+    Session,
+    SessionResult,
+    StepDebugger,
+    StageNode,
+    SubPipelineNode,
+    SwitchNode,
+    TerminalNode,
+    TryNode,
+    ExceptHandler,
+    TypeRegistry,
+    TypeSystem,
+    get_node_types,
+    get_stage,
+    get_stages,
+    get_stages_by_category,
+    parse_type,
+    register_node,
+    register_stage,
+)
+from .docs import generate_stages_json, generate_stages_yaml
+from .exceptions import (
+    ArtifactNotFoundError,
+    BranchError,
+    ExpressionError,
+    PayloadValidationError,
+    PipelineDefinitionError,
+    PipelineValidationError,
+    RegistryError,
+    StageContractError,
+    StageFlowError,
+    StageOutputError,
+    TypeCheckError,
+    TypeDeclarationError,
+)
+
+__all__ = [
+    "Pipeline",
+    "Session",
+    "SessionResult",
+    "StepDebugger",
+    "Context",
+    "BaseStage",
+    "register_stage",
+    "get_stage",
+    "get_stages",
+    "get_stages_by_category",
+    "Node",
+    "register_node",
+    "get_node_types",
+    "EntryNode",
+    "StageNode",
+    "ConditionNode",
+    "SwitchNode",
+    "ParallelNode",
+    "MapNode",
+    "SubPipelineNode",
+    "TerminalNode",
+    "Retrier",
+    "TryNode",
+    "ExceptHandler",
+    "Event",
+    "EventSpec",
+    "InputSpec",
+    "CelEngine",
+    "CelError",
+    "TypeSystem",
+    "TypeRegistry",
+    "parse_type",
+    "generate_stages_yaml",
+    "generate_stages_json",
+    "StageFlowError",
+    "RegistryError",
+    "PipelineDefinitionError",
+    "PipelineValidationError",
+    "ExpressionError",
+    "PayloadValidationError",
+    "StageContractError",
+    "StageOutputError",
+    "ArtifactNotFoundError",
+    "BranchError",
+    "TypeDeclarationError",
+    "TypeCheckError",
+]

{stageflow_framework-0.1.6 → stageflow_framework-0.10.0}/stageflow/builtins/__init__.py

@@ -1,17 +1,14 @@
-from .vars import (
-    SetValueStage,
-    CopyValueStage,
-    IncrementStage,
-    MergeDictStage,
-)
+from .dicts import DropKeysStage, PickKeysStage
 from .lists import (
     AppendListStage,
     ExtendListStage,
+    FilterListStage,
+    PopListStage,
+    UniqueListStage,
 )
 from .logic import AssertStage, FailStage, LogStage, SleepStage
 from .strings import ConcatStage, TemplateStage
-from .dicts import PickKeysStage, DropKeysStage
-from .lists_extra import FilterListStage, UniqueListStage, PopListStage
+from .vars import CopyValueStage, IncrementStage, MergeDictStage, SetValueStage
 
 __all__ = [
     "SetValueStage",
@@ -20,15 +17,15 @@ __all__ = [
     "MergeDictStage",
     "AppendListStage",
     "ExtendListStage",
+    "FilterListStage",
+    "UniqueListStage",
+    "PopListStage",
+    "PickKeysStage",
+    "DropKeysStage",
+    "ConcatStage",
+    "TemplateStage",
     "AssertStage",
     "FailStage",
     "LogStage",
     "SleepStage",
-    "ConcatStage",
-    "TemplateStage",
-    "PickKeysStage",
-    "DropKeysStage",
-    "FilterListStage",
-    "UniqueListStage",
-    "PopListStage",
 ]

stageflow_framework-0.10.0/stageflow/builtins/_args.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+from typing import Any
+
+from ..exceptions import StageContractError
+
+
+def require_list(stage_name: str, field: str, value: Any) -> list:
+    if not isinstance(value, list):
+        raise StageContractError(
+            f"{stage_name}: '{field}' must be a list, got {type(value).__name__}"
+        )
+    return value
+
+
+def require_dict(stage_name: str, field: str, value: Any) -> dict:
+    if not isinstance(value, dict):
+        raise StageContractError(
+            f"{stage_name}: '{field}' must be an object, got {type(value).__name__}"
+        )
+    return value
+
+
+def require_number(stage_name: str, field: str, value: Any) -> int | float:
+    if isinstance(value, bool) or not isinstance(value, (int, float)):
+        raise StageContractError(
+            f"{stage_name}: '{field}' must be a number, got {type(value).__name__}"
+        )
+    return value
+
+
+def require_present(stage_name: str, field: str, value: Any) -> Any:
+    if value is None:
+        raise StageContractError(f"{stage_name}: argument '{field}' is required")
+    return value

{stageflow_framework-0.1.6 → stageflow_framework-0.10.0}/stageflow/builtins/dicts.py

@@ -1,15 +1,16 @@
-from stageflow.core.stage import BaseStage, register_stage
+from ..core.stage import BaseStage, register_stage
+from ._args import require_dict, require_list
 
 
 @register_stage("PickKeysStage")
 class PickKeysStage(BaseStage):
     """
     description: "Pick only specified keys from a dict and return new object"
+    icon: "⊂"
     arguments:
       src:
         type: object
         description: "Source dict"
-    config:
       keys:
         type: list
         description: "Keys to keep in the result"
@@ -18,27 +19,25 @@ class PickKeysStage(BaseStage):
         type: object
         description: "Dict containing only picked keys"
     """
+
     category = "builtin.dicts"
 
     async def run(self):
         args = self.get_arguments()
-        src_val = args.get("src") or {}
-        keys = self.config.get("keys", [])
-        if not isinstance(src_val, dict):
-            raise ValueError("PickKeysStage expects dict in arguments.src")
-        picked = {k: src_val[k] for k in keys if k in src_val}
-        self.set_outputs({"result": picked})
+        src = require_dict("PickKeysStage", "src", args.get("src") or {})
+        keys = require_list("PickKeysStage", "keys", args.get("keys") or [])
+        self.set_outputs({"result": {k: src[k] for k in keys if k in src}})
 
 
 @register_stage("DropKeysStage")
 class DropKeysStage(BaseStage):
     """
     description: "Remove specified keys from dict and return cleaned object"
+    icon: "⊘"
     arguments:
       src:
         type: object
         description: "Source dict"
-    config:
       keys:
         type: list
         description: "Keys to remove from the dict"
@@ -47,13 +46,11 @@ class DropKeysStage(BaseStage):
         type: object
         description: "Dict without removed keys"
     """
+
     category = "builtin.dicts"
 
     async def run(self):
         args = self.get_arguments()
-        keys = set(self.config.get("keys", []))
-        src_val = args.get("src") or {}
-        if not isinstance(src_val, dict):
-            raise ValueError("DropKeysStage expects dict in arguments.src")
-        cleaned = {k: v for k, v in src_val.items() if k not in keys}
-        self.set_outputs({"result": cleaned})
+        src = require_dict("DropKeysStage", "src", args.get("src") or {})
+        drop = set(require_list("DropKeysStage", "keys", args.get("keys") or []))
+        self.set_outputs({"result": {k: v for k, v in src.items() if k not in drop}})