Context Editing

from agno.agent import Agent
from agno.db.sqlite import SqliteDb
from agno.models.anthropic import Claude
from agno.tools.websearch import WebSearchTools

agent = Agent(
    model=Claude(
        id="claude-sonnet-4-5-20250929",
        betas=["context-management-2025-06-27"],
        context_management={
            "edits": [
                {
                    "type": "clear_tool_uses_20250919",
                    "trigger": {"type": "tool_uses", "value": 2},
                    "keep": {"type": "tool_uses", "value": 1},
                }
            ]
        },
    ),
    db=SqliteDb(db_file="tmp/context_management.db"),
    instructions="Use web search to answer with current, sourced information.",
    tools=[WebSearchTools()],
    session_id="context-editing",
    add_history_to_context=True,
    markdown=True,
)

if __name__ == "__main__":
    agent.print_response(
        "Search for AI regulation in the US. Make at least three searches, then summarize the latest developments."
    )

    run = agent.get_last_run_output()
    if run and run.metrics:
        print(f"Input tokens: {run.metrics.input_tokens:,}")

    if run and run.messages:
        for message in reversed(run.messages):
            context_data = (message.provider_data or {}).get("context_management")
            if context_data:
                edits = context_data.get("applied_edits", [])
                if edits:
                    latest_edit = edits[-1]
                    print(
                        f"Cleared tool uses: {latest_edit.get('cleared_tool_uses', 0)}"
                    )
                    print(
                        f"Cleared input tokens: {latest_edit.get('cleared_input_tokens', 0):,}"
                    )
                    break