decision_tree.py
"""
Decision Tree HITL Example (AgentOS)
This example demonstrates multiple sequential human decisions in one workflow.
Each Condition pauses in AgentOS and the selected branch shapes the final output.
"""
from agno.os import AgentOS
from agno.workflow import OnReject
from agno.workflow.condition import Condition
from agno.workflow.step import Step
from agno.workflow.types import StepInput, StepOutput
from agno.workflow.workflow import Workflow
from workflow_db import db
def gather_requirements(step_input: StepInput) -> StepOutput:
topic = step_input.input or "Q4 sales performance"
return StepOutput(
content=f"Requirements gathered for '{topic}'.\n"
"Ready for analysis depth and output format decisions."
)
def detailed_analysis(step_input: StepInput) -> StepOutput:
return StepOutput(
content="Detailed analysis complete:\n"
"- Full statistical review\n"
"- Edge cases examined\n"
"- Confidence level: 95%"
)
def quick_summary(step_input: StepInput) -> StepOutput:
return StepOutput(
content="Quick summary complete:\n"
"- Key metrics computed\n"
"- Top highlights identified"
)
def formal_report(step_input: StepInput) -> StepOutput:
previous_content = step_input.previous_step_content or "No analysis"
return StepOutput(
content=f"Formal stakeholder report:\n\n{previous_content}\n\n"
"Formatted for presentation."
)
def internal_notes(step_input: StepInput) -> StepOutput:
previous_content = step_input.previous_step_content or "No analysis"
return StepOutput(
content=f"Internal team notes:\n\n{previous_content}\n\n"
"Saved as team reference material."
)
workflow = Workflow(
name="decision_tree_workflow",
description="Guide a workflow through sequential human decisions in AgentOS.",
db=db,
steps=[
Step(name="gather_requirements", executor=gather_requirements),
Condition(
name="analysis_depth",
requires_confirmation=True,
confirmation_message="Perform detailed analysis? Reject for quick summary.",
on_reject=OnReject.else_branch,
steps=[Step(name="detailed_analysis", executor=detailed_analysis)],
else_steps=[Step(name="quick_summary", executor=quick_summary)],
),
Condition(
name="output_format",
requires_confirmation=True,
confirmation_message="Generate a formal report? Reject for internal notes.",
on_reject=OnReject.else_branch,
steps=[Step(name="formal_report", executor=formal_report)],
else_steps=[Step(name="internal_notes", executor=internal_notes)],
),
],
)
workflow.id = "workflow-hitl-decision-tree"
agent_os = AgentOS(
name="Workflow Decision Tree HITL",
description="AgentOS workflow example for sequential human decisions.",
workflows=[workflow],
db=db,
)
app = agent_os.get_app()
if __name__ == "__main__":
agent_os.serve(app="decision_tree:app", reload=True)
workflow_db.py
from agno.db.postgres import PostgresDb
db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"
db = PostgresDb(db_url=db_url)
Run the Example
1
Set up your virtual environment
uv venv --python 3.12
source .venv/bin/activate
uv venv --python 3.12
.venv\Scripts\activate
2
Install dependencies
uv pip install -U "agno[os]" cel-python psycopg-binary
3
Run PgVector
docker run -d \
-e POSTGRES_DB=ai \
-e POSTGRES_USER=ai \
-e POSTGRES_PASSWORD=ai \
-e PGDATA=/var/lib/postgresql \
-v pgvolume:/var/lib/postgresql \
-p 5532:5432 \
--name pgvector \
agnohq/pgvector:18
4
Run the example
Save the code blocks above as
decision_tree.py and workflow_db.py in the same directory, then run:python decision_tree.py