Stop After Tool Call Dual Inheritance

Confirm stop_after_tool_call_tools flags still apply when a Toolkit subclass also inherits from a non-Toolkit base class.

Example showing stop_after_tool_call_tools with dual inheritance.

stop_after_tool_call_dual_inheritance.py

"""
Example showing stop_after_tool_call_tools with dual inheritance.

This demonstrates that stop_after_tool_call_tools works correctly even when
the Toolkit class uses multiple inheritance.
"""

from agno.agent import Agent
from agno.tools import Toolkit

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


class BaseConfig:
    """Base configuration class for dual inheritance."""

    def __init__(self, render_type: str = "OBJECT"):
        self._render_type = render_type


class DualInheritanceToolkit(Toolkit, BaseConfig):
    """Toolkit with dual inheritance - simulating the user's case."""

    def __init__(self, render_type: str = "JSON"):
        # Initialize base class first
        BaseConfig.__init__(self, render_type)

        # Then initialize Toolkit
        Toolkit.__init__(
            self,
            name="dual_inheritance_toolkit",
            tools=[self.filter_changed, self.get_render_type],
            stop_after_tool_call_tools=["filter_changed"],
        )

    def filter_changed(self, session_state) -> str:
        """
        Handle filter change event. Should stop after this tool call.

        Args:
            session_state: The session state (injected automatically)

        Returns:
            Message indicating filter changed
        """
        return (
            f"Filter changed! Render type: {self._render_type}. Agent should stop here!"
        )

    def get_render_type(self) -> str:
        """
        Get the current render type.

        Returns:
            Current render type
        """
        return f"Current render type: {self._render_type}"


# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------

if __name__ == "__main__":
    toolkit = DualInheritanceToolkit(render_type="CUSTOM")

    print("Registered functions:")
    for name, func in toolkit.functions.items():
        print(f"  {name}:")
        print(f"    stop_after_tool_call = {func.stop_after_tool_call}")
        print(f"    show_result = {func.show_result}")

    # Verify the flag is set correctly
    assert toolkit.functions["filter_changed"].stop_after_tool_call is True
    assert toolkit.functions["get_render_type"].stop_after_tool_call is False

    agent = Agent(
        tools=[toolkit],
        markdown=True,
    )

    agent.print_response("Call the filter_changed tool.")

Run the Example

1. Set up your virtual environment

   Mac

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

   uv venv --python 3.12
   .venv\Scripts\activate
   
2. Install dependencies

   uv pip install -U agno openai
   
3. Export your OpenAI API key

   Mac/Linux

   export OPENAI_API_KEY="your_openai_api_key_here"
   

   $Env:OPENAI_API_KEY="your_openai_api_key_here"
   
4. Run the example

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

   python stop_after_tool_call_dual_inheritance.py
   

Full source: cookbook/91_tools/other/stop_after_tool_call_dual_inheritance.py