ddeutil-workflow 0.0.74__tar.gz → 0.0.76__tar.gz

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ddeutil-workflow
-Version: 0.0.74
+Version: 0.0.76
 Summary: Lightweight workflow orchestration with YAML template
 Author-email: ddeutils <korawich.anu@gmail.com>
 License: MIT
@@ -68,7 +68,7 @@ by a `.yaml` template.
 3. All parallel tasks inside workflow core engine use **Multi-Threading** pool
    (Python 3.13 unlock GIL 🐍🔓)
 4. Recommend to pass a **Secret Value** with environment variable in YAML template 🔐
-5. Any datatime value convert to **No Timezone**
+5. Any datatime value convert to **UTC Timezone** 🌐
 
 ---
 
@@ -288,10 +288,10 @@ it will use default value and do not raise any error to you.
 | **REGISTRY_CALLER**         |   CORE    | `.`                                                                                                                             | List of importable string for the call stage.                                          |
 | **REGISTRY_FILTER**         |   CORE    | `ddeutil.workflow.templates`                                                                                                    | List of importable string for the filter template.                                     |
 | **CONF_PATH**               |   CORE    | `./conf`                                                                                                                        | The config path that keep all template `.yaml` files.                                  |
-| **TIMEZONE**                |   CORE    | `Asia/Bangkok`                                                                                                                  | A Timezone string value that will pass to `ZoneInfo` object.                           |
 | **STAGE_DEFAULT_ID**        |   CORE    | `false`                                                                                                                         | A flag that enable default stage ID that use for catch an execution output.            |
 | **GENERATE_ID_SIMPLE_MODE** |   CORE    | `true`                                                                                                                          | A flog that enable generating ID with `md5` algorithm.                                 |
 | **DEBUG_MODE**              |    LOG    | `true`                                                                                                                          | A flag that enable logging with debug level mode.                                      |
+| **TIMEZONE**                |    LOG    | `Asia/Bangkok`                                                                                                                  | A Timezone string value that will pass to `ZoneInfo` object.                           |
 | **FORMAT**                  |    LOG    | `%(asctime)s.%(msecs)03d (%(name)-10s, %(process)-5d,%(thread)-5d) [%(levelname)-7s] %(message)-120s (%(filename)s:%(lineno)s)` | A trace message console format.                                                        |
 | **FORMAT_FILE**             |    LOG    | `{datetime} ({process:5d}, {thread:5d}) {message:120s} ({filename}:{lineno})`                                                   | A trace message format that use to write to target pointer.                            |
 | **DATETIME_FORMAT**         |    LOG    | `%Y-%m-%d %H:%M:%S`                                                                                                             | A datetime format of the trace log.                                                    |

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/README.md

@@ -26,7 +26,7 @@ by a `.yaml` template.
 3. All parallel tasks inside workflow core engine use **Multi-Threading** pool
    (Python 3.13 unlock GIL 🐍🔓)
 4. Recommend to pass a **Secret Value** with environment variable in YAML template 🔐
-5. Any datatime value convert to **No Timezone**
+5. Any datatime value convert to **UTC Timezone** 🌐
 
 ---
 
@@ -246,10 +246,10 @@ it will use default value and do not raise any error to you.
 | **REGISTRY_CALLER**         |   CORE    | `.`                                                                                                                             | List of importable string for the call stage.                                          |
 | **REGISTRY_FILTER**         |   CORE    | `ddeutil.workflow.templates`                                                                                                    | List of importable string for the filter template.                                     |
 | **CONF_PATH**               |   CORE    | `./conf`                                                                                                                        | The config path that keep all template `.yaml` files.                                  |
-| **TIMEZONE**                |   CORE    | `Asia/Bangkok`                                                                                                                  | A Timezone string value that will pass to `ZoneInfo` object.                           |
 | **STAGE_DEFAULT_ID**        |   CORE    | `false`                                                                                                                         | A flag that enable default stage ID that use for catch an execution output.            |
 | **GENERATE_ID_SIMPLE_MODE** |   CORE    | `true`                                                                                                                          | A flog that enable generating ID with `md5` algorithm.                                 |
 | **DEBUG_MODE**              |    LOG    | `true`                                                                                                                          | A flag that enable logging with debug level mode.                                      |
+| **TIMEZONE**                |    LOG    | `Asia/Bangkok`                                                                                                                  | A Timezone string value that will pass to `ZoneInfo` object.                           |
 | **FORMAT**                  |    LOG    | `%(asctime)s.%(msecs)03d (%(name)-10s, %(process)-5d,%(thread)-5d) [%(levelname)-7s] %(message)-120s (%(filename)s:%(lineno)s)` | A trace message console format.                                                        |
 | **FORMAT_FILE**             |    LOG    | `{datetime} ({process:5d}, {thread:5d}) {message:120s} ({filename}:{lineno})`                                                   | A trace message format that use to write to target pointer.                            |
 | **DATETIME_FORMAT**         |    LOG    | `%Y-%m-%d %H:%M:%S`                                                                                                             | A datetime format of the trace log.                                                    |

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/pyproject.toml

@@ -64,6 +64,7 @@ where = ["src"]
 [tool.shelf.version]
 version = "./src/ddeutil/workflow/__about__.py"
 changelog = "CHANGELOG.md"
+files = ["json-schema.json"]
 commit_msg_format = "- {subject}"
 
 [tool.shelf.git]

ddeutil_workflow-0.0.76/src/ddeutil/workflow/__about__.py

@@ -0,0 +1 @@
+__version__: str = "0.0.76"

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/__cron.py

@@ -793,10 +793,11 @@ class CronRunner:
                     "Invalid type of `tz` parameter, it should be str or "
                     "ZoneInfo instance."
                 )
-            try:
-                self.tz = ZoneInfo(tz)
-            except ZoneInfoNotFoundError as err:
-                raise ValueError(f"Invalid timezone: {tz}") from err
+            else:
+                try:
+                    self.tz = ZoneInfo(tz)
+                except ZoneInfoNotFoundError as err:
+                    raise ValueError(f"Invalid timezone: {tz}") from err
 
         # NOTE: Prepare date
         if date:
@@ -807,6 +808,7 @@ class CronRunner:
             if tz is not None:
                 self.date: datetime = date.astimezone(self.tz)
             else:
+                self.tz = date.tzinfo
                 self.date: datetime = date
         else:
             self.date: datetime = datetime.now(tz=self.tz)
@@ -841,7 +843,11 @@ class CronRunner:
 
     @property
     def next(self) -> datetime:
-        """Returns the next time of the schedule."""
+        """Returns the next time of the schedule.
+
+        Returns:
+            datetime: A next datetime from the current with shifting step.
+        """
         self.date = (
             self.date
             if self.reset_flag
@@ -858,7 +864,11 @@ class CronRunner:
     def find_date(self, reverse: bool = False) -> datetime:
         """Returns the time the schedule would run by `next` or `prev` methods.
 
-        :param reverse: A reverse flag.
+        Args:
+            reverse: A reverse flag.
+
+        Returns:
+            datetime: A next datetime from shifting step.
         """
         # NOTE: Set reset flag to false if start any action.
         self.reset_flag: bool = False
@@ -868,7 +878,8 @@ class CronRunner:
             max(self.shift_limit, 100) if self.is_year else self.shift_limit
         ):
 
-            # NOTE: Shift the date
+            # NOTE: Shift the date from year to minute.
+            mode: DatetimeMode  # noqa: F842
             if all(
                 not self.__shift_date(mode, reverse)
                 for mode in ("year", "month", "day", "hour", "minute")

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/__init__.py

@@ -113,6 +113,7 @@ from .result import (
     WAIT,
     Result,
     Status,
+    get_status_from_error,
 )
 from .reusables import *
 from .stages import (
@@ -131,7 +132,7 @@ from .stages import (
     VirtualPyStage,
 )
 from .traces import (
-    ConsoleTrace,
+    BaseTrace,
     FileTrace,
     Trace,
     TraceData,

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/audits.py

@@ -79,7 +79,10 @@ class BaseAudit(BaseModel, ABC):
         default=None, description="A parent running ID."
     )
     run_id: str = Field(description="A running ID")
-    execution_time: float = Field(default=0, description="An execution time.")
+    runs_metadata: DictData = Field(
+        default_factory=dict,
+        description="A runs metadata that will use to tracking this audit log.",
+    )
 
     @model_validator(mode="after")
     def __model_action(self) -> Self:
@@ -296,20 +299,22 @@ class FileAudit(BaseAudit):
 
 
 class SQLiteAudit(BaseAudit):  # pragma: no cov
-    """SQLite Audit Pydantic Model."""
+    """SQLite Audit model."""
 
     table_name: ClassVar[str] = "audits"
     schemas: ClassVar[
         str
     ] = """
-        workflow        str,
-        release         int,
-        type            str,
-        context         json,
-        parent_run_id   int,
-        run_id          int,
-        update          datetime
-        primary key ( run_id )
+        workflow          str
+        , release         int
+        , type            str
+        , context         JSON
+        , parent_run_id   int
+        , run_id          int
+        , metadata        JSON
+        , created_at      datetime
+        , updated_at      datetime
+        primary key ( workflow, release )
         """
 
     @classmethod

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/cli.py

@@ -8,6 +8,7 @@ from __future__ import annotations
 import json
 from pathlib import Path
 from platform import python_version
+from textwrap import dedent
 from typing import Annotated, Any, Literal, Optional, Union
 
 import typer
@@ -15,15 +16,13 @@ from pydantic import Field, TypeAdapter
 
 from .__about__ import __version__
 from .__types import DictData
+from .conf import config
 from .errors import JobError
 from .job import Job
 from .params import Param
-from .result import Result
 from .workflow import Workflow
 
-app = typer.Typer(
-    pretty_exceptions_enable=True,
-)
+app = typer.Typer(pretty_exceptions_enable=True)
 
 
 @app.callback()
@@ -41,12 +40,70 @@ def version() -> None:
     typer.echo(f"python-version=={python_version()}")
 
 
+@app.command()
+def init() -> None:
+    """Initialize a Workflow structure on the current context."""
+    config.conf_path.mkdir(exist_ok=True)
+    (config.conf_path / ".confignore").touch()
+
+    conf_example_path: Path = config.conf_path / "examples"
+    conf_example_path.mkdir(exist_ok=True)
+
+    example_template: Path = conf_example_path / "wf_examples.yml"
+    example_template.write_text(
+        dedent(
+            """
+        # Example workflow template.
+        wf-example:
+          type: Workflow
+          desc: |
+            An example workflow template.
+          params:
+            name:
+                type: str
+                default: "World"
+          jobs:
+            first-job:
+              stages:
+                - name: "Call tasks"
+                  uses: tasks/say-hello-func@example
+                  with:
+                    name: ${{ params.name }}
+        """
+        ).lstrip("\n")
+    )
+
+    if "." in config.registry_caller:
+        task_path = Path("./tasks")
+        task_path.mkdir(exist_ok=True)
+
+        dummy_tasks_path = task_path / "example.py"
+        dummy_tasks_path.write_text(
+            dedent(
+                """
+            from ddeutil.workflow import Result, tag
+
+            @tag(name="example", alias="say-hello-func")
+            def hello_world_task(name: str, rs: Result) -> dict[str, str]:
+                \"\"\"Logging hello task function\"\"\"
+                rs.trace.info(f"Hello, {name}")
+                return {"name": name}
+            """
+            ).lstrip("\n")
+        )
+
+        init_path = task_path / "__init__.py"
+        init_path.write_text("from .example import hello_world_task\n")
+    typer.echo(
+        "Starter command: `workflow-cli workflows execute --name=wf-example`"
+    )
+
+
 @app.command(name="job")
 def execute_job(
     params: Annotated[str, typer.Option(help="A job execute parameters")],
     job: Annotated[str, typer.Option(help="A job model")],
-    parent_run_id: Annotated[str, typer.Option(help="A parent running ID")],
-    run_id: Annotated[Optional[str], typer.Option(help="A running ID")] = None,
+    run_id: Annotated[str, typer.Option(help="A running ID")],
 ) -> None:
     """Job execution on the local.
 
@@ -62,26 +119,19 @@ def execute_job(
         job_dict: dict[str, Any] = json.loads(job)
         _job: Job = Job.model_validate(obj=job_dict)
     except json.JSONDecodeError as e:
-        raise ValueError(f"Params does not support format: {params!r}.") from e
+        raise ValueError(f"Jobs does not support format: {job!r}.") from e
 
     typer.echo(f"Job params: {params_dict}")
-    rs: Result = Result(
-        run_id=run_id,
-        parent_run_id=parent_run_id,
-    )
-
     context: DictData = {}
     try:
         _job.set_outputs(
-            _job.execute(
-                params=params_dict,
-                run_id=rs.run_id,
-                parent_run_id=rs.parent_run_id,
-            ).context,
+            _job.execute(params=params_dict, run_id=run_id).context,
             to=context,
         )
+        typer.echo("[JOB]: Context result:")
+        typer.echo(json.dumps(context, default=str, indent=0))
     except JobError as err:
-        rs.trace.error(f"[JOB]: {err.__class__.__name__}: {err}")
+        typer.echo(f"[JOB]: {err.__class__.__name__}: {err}")
 
 
 @app.command()
@@ -136,8 +186,24 @@ def workflow_callback():
 
 
 @workflow_app.command(name="execute")
-def workflow_execute():
-    """"""
+def workflow_execute(
+    name: Annotated[
+        str,
+        typer.Option(help="A name of workflow template."),
+    ],
+    params: Annotated[
+        str,
+        typer.Option(help="A workflow execute parameters"),
+    ] = "{}",
+):
+    """Execute workflow by passing a workflow template name."""
+    try:
+        params_dict: dict[str, Any] = json.loads(params)
+    except json.JSONDecodeError as e:
+        raise ValueError(f"Params does not support format: {params!r}.") from e
+
+    typer.echo(f"Start execute workflow template: {name}")
+    typer.echo(f"... with params: {params_dict}")
 
 
 WORKFLOW_TYPE = Literal["Workflow"]
@@ -167,7 +233,7 @@ def workflow_json_schema(
     template_schema: dict[str, str] = {
         "$schema": "http://json-schema.org/draft-07/schema#",
         "title": "Workflow Configuration Schema",
-        "version": "1.0.0",
+        "version": __version__,
     }
     with open(output, mode="w", encoding="utf-8") as f:
         json.dump(template_schema | json_schema, f, indent=2)

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/conf.py

@@ -89,16 +89,6 @@ class Config:  # pragma: no cov
         """
         return Path(env("CORE_CONF_PATH", "./conf"))
 
-    @property
-    def tz(self) -> ZoneInfo:
-        """Timezone value that return with the `ZoneInfo` object and use for all
-        datetime object in this workflow engine.
-
-        Returns:
-            ZoneInfo: The timezone configuration for the workflow engine.
-        """
-        return ZoneInfo(env("CORE_TIMEZONE", "UTC"))
-
     @property
     def generate_id_simple_mode(self) -> bool:
         """Flag for generate running ID with simple mode. That does not use
@@ -143,6 +133,16 @@ class Config:  # pragma: no cov
         """
         return str2bool(env("LOG_DEBUG_MODE", "true"))
 
+    @property
+    def log_tz(self) -> ZoneInfo:
+        """Timezone value that return with the `ZoneInfo` object and use for all
+        datetime object in this workflow engine.
+
+        Returns:
+            ZoneInfo: The timezone configuration for the workflow engine.
+        """
+        return ZoneInfo(env("LOG_TIMEZONE", "UTC"))
+
     @property
     def log_format(self) -> str:
         return env(

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/errors.py

@@ -136,16 +136,14 @@ class BaseError(Exception):
             ErrorData or dict: Exception data, optionally mapped by reference ID
 
         Example:
-            ```python
-            error = BaseError("Something failed", refs="stage-1")
-
-            # Simple format
-            data = error.to_dict()
-            # Returns: {"name": "BaseError", "message": "Something failed"}
-
-            # With reference mapping
-            ref_data = error.to_dict(with_refs=True)
-            # Returns: {"stage-1": {"name": "BaseError", "message": "Something failed"}}
+            >>> error = BaseError("Something failed", refs="stage-1")
+            >>> # Simple format
+            >>> error.to_dict()
+            >>> # Returns: {"name": "BaseError", "message": "Something failed"}
+
+            >>> # With reference mapping
+            >>> error.to_dict(with_refs=True)
+            >>> # Returns: {"stage-1": {"name": "BaseError", "message": "Something failed"}}
             ```
         """
         data: ErrorData = to_dict(self)

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/job.py

@@ -656,6 +656,7 @@ class Job(BaseModel):
         to: DictData,
         *,
         job_id: StrOrNone = None,
+        **kwargs,
     ) -> DictData:
         """Set an outputs from execution result context to the received context
         with a `to` input parameter. The result context from job strategy
@@ -693,12 +694,15 @@ class Job(BaseModel):
         :raise JobError: If the job's ID does not set and the setting
             default job ID flag does not set.
 
-        :param output: (DictData) A result data context that want to extract
-            and transfer to the `strategies` key in receive context.
-        :param to: (DictData) A received context data.
-        :param job_id: (StrOrNone) A job ID if the `id` field does not set.
+        Args:
+            output: (DictData) A result data context that want to extract
+                and transfer to the `strategies` key in receive context.
+            to: (DictData) A received context data.
+            job_id: (StrOrNone) A job ID if the `id` field does not set.
+            kwargs: Any values that want to add to the target context.
 
-        :rtype: DictData
+        Returns:
+            DictData: Return updated the target context with a result context.
         """
         if "jobs" not in to:
             to["jobs"] = {}
@@ -716,8 +720,9 @@ class Job(BaseModel):
         status: dict[str, Status] = (
             {"status": output.pop("status")} if "status" in output else {}
         )
+        kwargs: DictData = kwargs or {}
         if self.strategy.is_set():
-            to["jobs"][_id] = {"strategies": output} | errors | status
+            to["jobs"][_id] = {"strategies": output} | errors | status | kwargs
         elif len(k := output.keys()) > 1:  # pragma: no cov
             raise JobError(
                 "Strategy output from execution return more than one ID while "
@@ -726,7 +731,7 @@ class Job(BaseModel):
         else:
             _output: DictData = {} if len(k) == 0 else output[list(k)[0]]
             _output.pop("matrix", {})
-            to["jobs"][_id] = _output | errors | status
+            to["jobs"][_id] = _output | errors | status | kwargs
         return to
 
     def get_outputs(
@@ -800,8 +805,7 @@ class Job(BaseModel):
             return docker_execution(
                 self,
                 params,
-                run_id=run_id,
-                parent_run_id=parent_run_id,
+                run_id=parent_run_id,
                 event=event,
             ).make_info({"execution_time": time.monotonic() - ts})
 
@@ -1294,7 +1298,6 @@ def docker_execution(
     params: DictData,
     *,
     run_id: StrOrNone = None,
-    parent_run_id: StrOrNone = None,
     event: Optional[Event] = None,
 ):  # pragma: no cov
     """Docker job execution.

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/params.py

@@ -52,7 +52,7 @@ from pydantic import BaseModel, Field
 
 from .__types import StrOrInt
 from .errors import ParamError
-from .utils import get_d_now, get_dt_now
+from .utils import UTC, get_d_now, get_dt_now
 
 T = TypeVar("T")
 
@@ -169,16 +169,18 @@ class DatetimeParam(DefaultParam):
             return self.default
 
         if isinstance(value, datetime):
-            return value
+            if value.tzinfo is None:
+                return value.replace(tzinfo=UTC)
+            return value.astimezone(UTC)
         elif isinstance(value, date):
-            return datetime(value.year, value.month, value.day)
+            return datetime(value.year, value.month, value.day, tzinfo=UTC)
         elif not isinstance(value, str):
             raise ParamError(
                 f"Value that want to convert to datetime does not support for "
                 f"type: {type(value)}"
             )
         try:
-            return datetime.fromisoformat(value)
+            return datetime.fromisoformat(value).replace(tzinfo=UTC)
         except ValueError:
             raise ParamError(
                 f"Invalid the ISO format string for datetime: {value!r}"

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/result.py

@@ -16,7 +16,6 @@ Classes:
 Functions:
     validate_statuses: Determine final status from multiple status values
     get_status_from_error: Convert exception types to appropriate status
-    get_dt_tznow: Get current datetime with timezone configuration
 """
 from __future__ import annotations
 
@@ -43,7 +42,7 @@ from . import (
 from .__types import DictData
 from .audits import Trace, get_trace
 from .errors import ResultError
-from .utils import default_gen_id, get_dt_ntz_now
+from .utils import default_gen_id, get_dt_now
 
 
 class Status(str, Enum):
@@ -89,6 +88,7 @@ class Status(str, Enum):
         return self.name
 
     def is_result(self) -> bool:
+        """Return True if this status is the status for result object."""
         return self in ResultStatuses
 
 
@@ -115,15 +115,13 @@ def validate_statuses(statuses: list[Status]) -> Status:
         Status: Final consolidated status based on workflow logic
 
     Example:
-        ```python
-        # Mixed statuses - FAILED takes priority
-        result = validate_statuses([SUCCESS, FAILED, SUCCESS])
-        # Returns: FAILED
-
-        # All same status
-        result = validate_statuses([SUCCESS, SUCCESS, SUCCESS])
-        # Returns: SUCCESS
-        ```
+        >>> # Mixed statuses - FAILED takes priority
+        >>> validate_statuses([SUCCESS, FAILED, SUCCESS])
+        >>> # Returns: FAILED
+
+        >>> # All same status
+        >>> validate_statuses([SUCCESS, SUCCESS, SUCCESS])
+        >>> # Returns: SUCCESS
     """
     if any(s == CANCEL for s in statuses):
         return CANCEL
@@ -153,6 +151,9 @@ def get_status_from_error(
 ) -> Status:
     """Get the Status from the error object.
 
+    Args:
+        error: An error object.
+
     Returns:
         Status: The status from the specific exception class.
     """
@@ -189,8 +190,8 @@ class Result:
     context: DictData = field(default_factory=default_context)
     info: DictData = field(default_factory=dict)
     run_id: Optional[str] = field(default_factory=default_gen_id)
-    parent_run_id: Optional[str] = field(default=None, compare=False)
-    ts: datetime = field(default_factory=get_dt_ntz_now, compare=False)
+    parent_run_id: Optional[str] = field(default=None)
+    ts: datetime = field(default_factory=get_dt_now, compare=False)
     trace: Optional[Trace] = field(default=None, compare=False, repr=False)
     extras: DictData = field(default_factory=dict, compare=False, repr=False)
 
@@ -266,7 +267,7 @@ class Result:
 
         :rtype: float
         """
-        return (get_dt_ntz_now() - self.ts).total_seconds()
+        return (get_dt_now() - self.ts).total_seconds()
 
 
 def catch(

{ddeutil_workflow-0.0.74 → ddeutil_workflow-0.0.76}/src/ddeutil/workflow/stages.py

@@ -295,7 +295,7 @@ class BaseStage(BaseModel, ABC):
         ts: float = time.monotonic()
         parent_run_id: str = run_id
         run_id: str = run_id or gen_id(self.iden, unique=True)
-        context: DictData = {}
+        context: DictData = {"status": WAIT}
         trace: Trace = get_trace(
             run_id, parent_run_id=parent_run_id, extras=self.extras
         )
@@ -413,7 +413,7 @@ class BaseStage(BaseModel, ABC):
         self,
         output: DictData,
         to: DictData,
-        info: Optional[DictData] = None,
+        **kwargs,
     ) -> DictData:
         """Set an outputs from execution result context to the received context
         with a `to` input parameter. The result context from stage execution
@@ -447,12 +447,14 @@ class BaseStage(BaseModel, ABC):
         to the `to` argument. The result context was soft copied before set
         output step.
 
-        :param output: (DictData) A result data context that want to extract
-            and transfer to the `outputs` key in receive context.
-        :param to: (DictData) A received context data.
-        :param info: (DictData)
+        Args:
+            output: (DictData) A result data context that want to extract
+                and transfer to the `outputs` key in receive context.
+            to: (DictData) A received context data.
+            kwargs: Any values that want to add to the target context.
 
-        :rtype: DictData
+        Returns:
+            DictData: Return updated the target context with a result context.
         """
         if "stages" not in to:
             to["stages"] = {}
@@ -470,8 +472,8 @@ class BaseStage(BaseModel, ABC):
         status: dict[str, Status] = (
             {"status": output.pop("status")} if "status" in output else {}
         )
-        info: DictData = {"info": info} if info else {}
-        to["stages"][_id] = {"outputs": output} | errors | status | info
+        kwargs: DictData = kwargs or {}
+        to["stages"][_id] = {"outputs": output} | errors | status | kwargs
         return to
 
     def get_outputs(self, output: DictData) -> DictData: