name: hello
kind: horus_workflow

tasks:
  - id: greet
    name: Greet
    kind: horus_task
    target: { kind: local, working_directory: "./horus-work" }
    runtime:
      kind: command
      command: "echo 'Hello from Horus!' > $message\n"
    executor:
      kind: shell
    outputs:
      - { kind: file, id: message, path: "./horus-out/message.txt" }

horus run hello.yaml

horus run pipeline.yaml

# pipeline.yaml: a multi-task workflow.
#
#   horus run pipeline.yaml
#
# ingest -> validate -> { text features, image features } -> train -> report
#
# It fans out into two parallel feature stages and merges back at training, so
# the dependency graph is diamond-shaped. The `sleep`s let you watch the live
# TUI advance. Artifacts are passed between tasks with `$id` substitution (the
# on-target path), and the `edges` at the bottom declare both the ordering and
# the artifact routing. `skip_if_complete: false` makes every run actually
# execute (handy for a demo).
name: pipeline
kind: horus_workflow

tasks:
  - id: ingest
    name: Ingest data
    kind: horus_task
    skip_if_complete: false
    target: { kind: local, working_directory: "./horus-work" }
    runtime:
      kind: command
      command: "sleep 1\necho 'rows: 1000' > $raw\n"
    executor:
      kind: shell
    outputs:
      - { kind: file, id: raw, path: "./horus-out/raw.txt" }

  - id: validate
    name: Validate schema
    kind: horus_task
    skip_if_complete: false
    target: { kind: local, working_directory: "./horus-work" }
    runtime:
      kind: command
      command: "sleep 1\ncat $raw > $valid\necho 'validated: true' >> $valid\n"
    executor:
      kind: shell
    inputs:
      - { kind: file, id: raw, path: "./horus-out/raw.txt" }
    outputs:
      - { kind: file, id: valid, path: "./horus-out/valid.txt" }

  - id: features_text
    name: Text features
    kind: horus_task
    skip_if_complete: false
    target: { kind: local, working_directory: "./horus-work" }
    runtime:
      kind: command
      command: "sleep 2\ncat $valid > $feat_text\necho 'text_features: 256' >> $feat_text\n"
    executor:
      kind: shell
    inputs:
      - { kind: file, id: valid, path: "./horus-out/valid.txt" }
    outputs:
      - { kind: file, id: feat_text, path: "./horus-out/feat_text.txt" }

  - id: features_image
    name: Image features
    kind: horus_task
    skip_if_complete: false
    target: { kind: local, working_directory: "./horus-work" }
    runtime:
      kind: command
      command: "sleep 2\ncat $valid > $feat_image\necho 'image_features: 512' >> $feat_image\n"
    executor:
      kind: shell
    inputs:
      - { kind: file, id: valid, path: "./horus-out/valid.txt" }
    outputs:
      - { kind: file, id: feat_image, path: "./horus-out/feat_image.txt" }

  - id: train
    name: Train model
    kind: horus_task
    skip_if_complete: false
    target: { kind: local, working_directory: "./horus-work" }
    resources:
      cpus: 8
      gpus: 2
      memory_gb: 32
      walltime: "01:00:00"
    runtime:
      kind: command
      command: "sleep 3\ncat $feat_text $feat_image > $model\necho 'model: trained' >> $model\n"
    executor:
      kind: shell
    inputs:
      - { kind: file, id: feat_text, path: "./horus-out/feat_text.txt" }
      - { kind: file, id: feat_image, path: "./horus-out/feat_image.txt" }
    outputs:
      - { kind: file, id: model, path: "./horus-out/model.txt" }

  - id: report
    name: Write report
    kind: horus_task
    skip_if_complete: false
    target: { kind: local, working_directory: "./horus-work" }
    runtime:
      kind: command
      command: "sleep 1\necho '=== REPORT ===' > $report\ncat $model >> $report\n"
    executor:
      kind: shell
    inputs:
      - { kind: file, id: model, path: "./horus-out/model.txt" }
    outputs:
      - { kind: file, id: report, path: "./horus-out/report.txt" }

# Edges are the sole source of truth for ordering and the dependency DAG.
edges:
  - { source: ingest,         source_output: raw,        target: validate,       target_input: raw }
  - { source: validate,       source_output: valid,      target: features_text,  target_input: valid }
  - { source: validate,       source_output: valid,      target: features_image, target_input: valid }
  - { source: features_text,  source_output: feat_text,  target: train,          target_input: feat_text }
  - { source: features_image, source_output: feat_image, target: train,          target_input: feat_image }
  - { source: train,          source_output: model,      target: report,         target_input: model }

python hello_workflow.py

"""A minimal Python-defined Horus workflow.

  python hello_workflow.py

Two tasks wired by an edge: ``make_greeting`` writes a file, ``shout_it`` reads
it and writes an uppercased copy. ``render_workflow`` runs the workflow under
the same live TUI as ``horus run``. Reach for a Python workflow when a task
needs real Python logic, such as reading or processing files, calling a
library, or prompting the user (see interactive_workflow.py).
"""

from pathlib import Path

from horus_builtin.artifact.file import FileArtifact
from horus_builtin.event.tui_subscriber import render_workflow
from horus_builtin.target.local import LocalTarget
from horus_builtin.task.function import FunctionTask
from horus_builtin.workflow.horus_workflow import HorusWorkflow
from horus_runtime.context import HorusContext
from horus_runtime.core.artifact.base import BaseArtifact
from horus_runtime.core.workflow.edge import WorkflowEdge

# Boot the runtime once before building or running anything.
HorusContext.boot()

wf = HorusWorkflow(name="hello_python")

# Per-task scratch lives under ./horus-work; outputs under ./horus-out.
work = LocalTarget(working_directory="./horus-work")
greeting = FileArtifact(id="greeting", path=Path("horus-out/greeting.txt"))
shout = FileArtifact(id="shout", path=Path("horus-out/shout.txt"))


# The task id and name default to the function name ("make_greeting"). The
# parameters after `task` are the task's artifacts, matched by their id.
@FunctionTask.task(wf, outputs=[greeting], target=work)
async def make_greeting(
  task: FunctionTask, greeting: FileArtifact
) -> list[BaseArtifact]:
  """Write a greeting to the output artifact."""
  greeting.write("Hello from a Python workflow!")
  return []


@FunctionTask.task(wf, inputs=[greeting], outputs=[shout], target=work)
async def shout_it(
  task: FunctionTask, greeting: FileArtifact, shout: FileArtifact
) -> list[BaseArtifact]:
  """Read the greeting and write an uppercased version."""
  shout.write(greeting.read().upper())
  return []


# Wire make_greeting's output into shout_it's input: this declares the ordering.
wf.edges.append(
  WorkflowEdge(
    source="make_greeting",
    source_output="greeting",
    target="shout_it",
    target_input="greeting",
  )
)

if __name__ == "__main__":
  # Trigger from the first task; the edge pulls shout_it in as a descendant.
  render_workflow(wf, trigger_id="make_greeting")

python interactive_workflow.py

"""A Python workflow that prompts the user.

  python interactive_workflow.py

A ``FunctionTask`` can ask the user questions through ``task.interaction``. When
running under the live TUI, the dashboard pauses while the prompt is shown and
resumes once you answer. YAML workflows cannot do this; it needs real Python.
"""

from pathlib import Path

from horus_builtin.artifact.file import FileArtifact
from horus_builtin.event.tui_subscriber import render_workflow
from horus_builtin.interaction.common.confirm import ConfirmInteraction
from horus_builtin.interaction.common.string import StringInteraction
from horus_builtin.target.local import LocalTarget
from horus_builtin.task.function import FunctionTask
from horus_builtin.workflow.horus_workflow import HorusWorkflow
from horus_runtime.context import HorusContext
from horus_runtime.core.artifact.base import BaseArtifact

HorusContext.boot()

wf = HorusWorkflow(name="interactive")

work = LocalTarget(working_directory="./horus-work")
greeting = FileArtifact(id="greeting", path=Path("horus-out/greeting.txt"))


@FunctionTask.task(wf, outputs=[greeting], target=work)
async def greet_user(
  task: FunctionTask, greeting: FileArtifact
) -> list[BaseArtifact]:
  """Ask the user for their name, confirm, then write a greeting."""
  name = await task.interaction.ask(
    StringInteraction(
      value_key="name",
      prompt="What is your name?",
      default="World",
    )
  )

  shout = await task.interaction.ask(
    ConfirmInteraction(
      value_key="shout",
      prompt="Shout the greeting?",
      default=False,
    )
  )

  message = f"Hello, {name}!"
  greeting.write(message.upper() if shout else message)
  return []


if __name__ == "__main__":
  render_workflow(wf, trigger_id="greet_user")