name: ProfanityPipeline
version: "1.0.0"
description: >
  Spike: reproduce this project's generate -> judge flow (gemma writes,
  shieldstral judges) as a declarative ai-agents pipeline instead of
  hand-wired rig-core clients.

system_prompt: "Run the generate/judge pipeline."

llms:
  default:
    provider: ollama
    model: gemma4-e4b:latest

# Pre-spawn both sub-agents; each carries its own provider (ollama vs.
# openai-compatible against local llama-server), so shared_llms stays off.
spawner:
  shared_llms: false
  auto_spawn:
    - id: writer
      agent: agents/writer.yaml
    - id: judge
      agent: agents/judge.yaml

states:
  initial: process
  states:
    process:
      pipeline:
        stages:
          - writer
          - id: judge
            input: |
              <Instruct>Flag content containing profanity or hostile language.</Instruct>
              <Query>Does this content violate the policy?</Query>
              <Document>{{ stages.writer }}</Document>
        timeout_ms: 60000
      transitions:
        - to: done
          when: "Pipeline complete"

    done:
      prompt: "Pipeline complete."