athena-sdk-lite

A Pythonic, single-import SDK for building Athena workflows.

from athena_sdk_lite import Workflow
from athena_sdk_lite.nodes import postgres, ai_tagging, branch, output

with Workflow("triage") as wf:
    rows = postgres("load", operation="select", query="SELECT ... FROM events", connection={...})
    tagged = ai_tagging("classify", inputs=rows, agent_url="https://...")
    gate = branch("is-critical", inputs=tagged, condition="$input.data.get('severity') == 'critical'")
    output("alerts", inputs=gate.out("true"), format="json")
    output("queue",  inputs=gate.out("false"), format="json")

print(wf.visualize())
issues = wf.validate()

That is the entire surface a normal user sees.

---

Where to go next

• Getting started

  ---

  Start here if you've never built a workflow with this SDK.

  Building a workflow → How nodes connect →

• Vibe coding

  ---

  Use AI agents (Claude Code, Cursor) to author workflows with this SDK.

  Using AI agents → AGENTS.md context file →

• Reference (v0.1.0)

  ---

  The full API surface, architecture diagrams, and conventions.

  Overview → Architecture → Technical →

• Worked examples

  ---

  Eleven runnable scripts under examples/, covering linear pipelines, fan-out, branching, custom transforms, and wrapper authoring.

  See examples/01_pubmed_to_ai.py through examples/11_triage_pipeline.py.

What this is, in one paragraph

A small Python library for building data + AI workflows as DAGs. You import it, declare nodes (Postgres read, AI classification, transform, branch), wire them with inputs=, and the library produces a workflow object you can validate, visualize, and run locally, in-process. No backend. No API key. No service to deploy. The vendored engine does the actual work; this package is a thin, obvious surface on top.

When to reach for this

Use this when	Reach for something heavier when
Workflow runs in one Python process	You need distributed execution across machines
You want stakeholders to read the workflow code	You need non-engineers to author workflows
The 11 helpers + escape hatch cover your nodes	You need first-class support for many bespoke node types
Local-first development and testing	You need a UI / registry / scheduler