ace/protocols/agent.py

"""Protocol defining what steps need from an Agent implementation."""

from __future__ import annotations

from typing import Any, Optional, Protocol, runtime_checkable

from ..core.outputs import AgentOutput


@runtime_checkable
class AgentLike(Protocol):
    """Structural interface for Agent-like objects.

    Any object with a matching ``generate`` method satisfies this —
    ``ace.roles.Agent`` and ``ace.roles.ReplayAgent`` both do.
    """

    def generate(
        self,
        *,
        question: str,
        context: Optional[str],
        skillbook: Any,
        reflection: Optional[str] = ...,
        **kwargs: Any,
    ) -> AgentOutput: ...