Foundations of Agentic AI with the Claude Agent SDK

# the loop, tools, streaming, cost — built in
import asyncio
from claude_agent_sdk import query

async def run(prompt, options):
    async for m in query(prompt, options):
        print(m)

# In real use, call asyncio.run(run(prompt, options))

import asyncio
import os
from claude_agent_sdk import (
    query,
    ClaudeAgentOptions,
    AssistantMessage,
    TextBlock,
)

# Configure the agent
options = ClaudeAgentOptions(
    env={"ANTHROPIC_API_KEY": os.getenv("ANTHROPIC_API_KEY")},
    model="claude-sonnet-4-6",
    allowed_tools=["Bash"],  # safety whitelist
    permission_mode="acceptEdits",
)

# A simple plain-English task
prompt = "Use the Bash tool to run `aws sts get-caller-identity` and show me the AWS account information."

async def main():
    # query is asynchronous and streams structured messages
    async for message in query(prompt, options):
        # Filter for the assistant's content and print text blocks
        if isinstance(message, AssistantMessage):
            for block in message.content:
                if isinstance(block, TextBlock):
                    print(block.text)

# Run the async main function
asyncio.run(main())