Augments LabsAugments ADK

API Reference

The public surface of the framework's core modules, grouped by theme. Each page lists the exported symbols; their signatures and docstrings are authoritative in the source. Usage walkthroughs live under Guides and the per-topic sections linked from each page.

Core

  • Agent β€” Configuration-only object. Name, instructions, tools, handoffs, guardrails.
  • Runner β€” Execution entry-points: arun, arun_streamed, arun_graph, arun_swarm.
  • LLM β€” Framework-owned LLM abstract base class.
  • FunctionTool β€” The Tool ABC + decorator + types.

Orchestration

  • Swarms β€” Iterative multi-agent collaboration with explicit termination and pluggable routing.
  • Graphs β€” State-machine orchestration with checkpointing, interrupts, and streaming events.
  • Flows β€” Decorator-driven multi-step orchestration over typed shared state.
  • Tasks β€” Declarative units of work composed into pipelines and groups.

State and persistence

  • Memory β€” Extracted, searchable knowledge carried across sessions.
  • Session β€” Conversation persistence for agent runs.

Safety and protocols

  • Guardrails β€” Built-in PII, prompt-injection, and wrong-language guardrails.
  • MCP β€” Model Context Protocol servers, lifecycle, and tool filters.
  • A2A β€” Agent-to-Agent protocol client and server surfaces.

Foundations

  • Exceptions β€” The framework exception hierarchy rooted at AugmentsError.
  • Types β€” Provider-agnostic wire and history types.