Python

Augments ADK

A framework to orchestrate a complex system of agents that performs real-world actions.

agent.py

1import asyncio

3from augments.adk import (

4 Agent,

5 PermissionResultAllow,

6 PermissionResultDeny,

7 Runner,

10async def require_admin(tool_name, args, ctx):

11 if tool_name.startswith("delete_"):

12 context = ctx.run_context.context

13 if not context.get("is_admin", False):

14 return PermissionResultDeny(

15 message="Admin required for deletes",

16 interrupt=False,

17 )

18 return PermissionResultAllow()

20agent = Agent(

21 name="SecureAgent",

22 can_use_tool=require_admin,

23)

25result = asyncio.run(Runner.arun(

26 agent,

27 "Delete the staging database",

28 context={"is_admin": False},

29))

30print(result.final_output)

Augments ADK asks your permission callback before a tool runs. The Runner carries the context it reads, so a non-admin run never deletes anything.

Quickstart

Install with uv, or with plain pip

uv add augments-adk  # or: pip install augments-adk