44 lines
1.1 KiB
YAML
44 lines
1.1 KiB
YAML
|
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."
|