buildathena-sdk 0.1.21

Athena Python SDK for block development

athena-sdk

Python SDK for building blocks and workflows on the Athena ML orchestration platform.

Installation

pip install buildathena-sdk

Quick Start

Define a Block

from athena import block, BlockContext

@block(name="TrainModel", outputs=["checkpoint"])
async def train_model(ctx: BlockContext, epochs: int = 100):
    for epoch in range(epochs):
        loss = train_epoch()
        await ctx.emit_metric("loss", loss, epoch=epoch)

    artifact_id = ctx.artifacts.register("model.pt", schema="checkpoint")
    return {"checkpoint": artifact_id}

Define an Entrypoint

from athena import entrypoint

@entrypoint()
async def main():
    # Entrypoint is the DAG root that triggers block execution
    pass

Use Credentials

Declare required secrets in the @block decorator and access via ctx.secrets:

from athena import block, BlockContext

@block(name="MyBlock", secrets=["OPENAI_API_KEY"])
async def my_block(ctx: BlockContext, config):
    key = ctx.secrets["OPENAI_API_KEY"]

Features

• Block Decorators: @block, @entrypoint for defining workflow nodes
• BlockContext: Emit metrics, progress, logs, and register artifacts
• Config System: YAML with $include, $extends, ${ref} substitution
• Discovery: AST-based scanner for workspace block/entrypoint detection
• Credentials: Secure secret management with @block(secrets=...) and ctx.secrets

License

Proprietary - Copyright (c) 2026 Athena Technologies. All rights reserved.