Retry Tool Call

Raise RetryAgentRun after adding an item to ask the model for more items.

retry_tool_call.py
"""
Retry Tool Call
=============================

Demonstrates retry tool call.
"""

from agno.agent import Agent
from agno.db.sqlite import SqliteDb
from agno.exceptions import RetryAgentRun
from agno.models.openai import OpenAIChat
from agno.run import RunContext
from agno.utils.log import logger

# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------


def add_item(run_context: RunContext, item: str) -> str:
    """Add an item to the shopping list."""
    if run_context.session_state is None:
        run_context.session_state = {}

    if "shopping_list" not in run_context.session_state:
        run_context.session_state["shopping_list"] = []

    run_context.session_state["shopping_list"].append(item)
    len_shopping_list = len(run_context.session_state["shopping_list"])

    if len_shopping_list < 3:
        logger.info(
            f"Asking the model to add {3 - len_shopping_list} more items to the shopping list."
        )
        raise RetryAgentRun(
            f"Shopping list is: {run_context.session_state['shopping_list']}. Minimum 3 items in the shopping list. "
            + f"Add {3 - len_shopping_list} more items.",
        )

    logger.info(
        f"The shopping list is now: {run_context.session_state.get('shopping_list')}"
    )  # type: ignore
    return f"The shopping list is now: {run_context.session_state.get('shopping_list')}"  # type: ignore


agent = Agent(
    model=OpenAIChat(id="gpt-5.2"),
    session_id="retry_tool_call_session",
    db=SqliteDb(
        session_table="retry_tool_call_session",
        db_file="tmp/retry_tool_call.db",
    ),
    # Initialize the session state with empty shopping list
    session_state={"shopping_list": []},
    tools=[add_item],
    markdown=True,
)

# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
    agent.print_response("Add milk", stream=True)
    print(
        f"Final session state: {agent.get_session_state(session_id='retry_tool_call_session')}"
    )

RetryAgentRun sends feedback to the model for its next decision; it does not replay the same call through a fixed retry/backoff loop. The tool appends the item before raising, and that state change remains. The model can add another item or repeat one; this example does not enforce uniqueness.

The fixed session ID and SQLite file retain the shopping list between executions. Use a new session ID to demonstrate the empty-list case again. A prompt asking for milk alone does not guarantee that the model will supply two additional items.

Run the Example

Set up your virtual environment

uv venv --python 3.12
source .venv/bin/activate

Install dependencies

uv pip install -U agno openai sqlalchemy

Export your OpenAI API key

export OPENAI_API_KEY="your_openai_api_key_here"

Run the example

Save the code above as retry_tool_call.py, then run:

python retry_tool_call.py

Full source: cookbook/91_tools/exceptions/retry_tool_call.py