42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
"""Backend protocol.
|
|
|
|
A backend turns an Anthropic-style turn (``messages`` + agent definition)
|
|
into a stream of :class:`~beaver_gateway.events.stream.MessageStreamEvent`
|
|
records; provider quirks are translated here, not in the frontend.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, Any, Protocol
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import AsyncIterator, Iterable
|
|
|
|
from anthropic.types import MessageParam
|
|
|
|
from beaver_gateway.agents.base import BaseAgent
|
|
from beaver_gateway.events.stream import MessageStreamEvent
|
|
|
|
|
|
class Backend(Protocol):
|
|
"""Single-method protocol; ``complete`` returns an async iterator of events."""
|
|
|
|
def complete(
|
|
self,
|
|
*,
|
|
agent: BaseAgent,
|
|
messages: Iterable[MessageParam],
|
|
system: str | None = None,
|
|
**options: Any,
|
|
) -> AsyncIterator[MessageStreamEvent]:
|
|
"""Yield Anthropic stream events for one turn against ``agent``.
|
|
|
|
``**options`` keys are optional and ignored by backends that don't
|
|
keep state: ``conversation_id`` (stable id the backend may pin a
|
|
live session to), ``session_id`` (backend session to resume when
|
|
nothing is live), ``capture`` (a
|
|
:class:`~beaver_gateway.backends.capture.TurnCapture` the backend
|
|
fills after the stream closes).
|
|
"""
|
|
...
|