b60439152b
- harness/es.py: append-only event store + проекции (образец — кооп-портал) - harness/levels.py: уровни 1с/1м/1ч/1д, контекст ∝ масштабу - harness/scheduler.py: уровневые потоки, идемпотентные тики, absorbed-стабы - harness/isolated_runner.py + _llm_worker.py: LLM-тик в сабпроцессе (глобальный asyncio-рантайм Strands не переживает 2+ create_harness) - cli.py: init/task/tasks/status/events + requeue (написан самим минутным агентом) - 9router: отдельный ключ sk-simbiot, модель auto (glm-5.3 первым)
101 lines
3.1 KiB
Python
101 lines
3.1 KiB
Python
"""Hook events emitted by bidirectional agents."""
|
|
|
|
from dataclasses import dataclass
|
|
from typing import TYPE_CHECKING, Literal
|
|
|
|
from ....hooks.registry import BaseHookEvent
|
|
|
|
if TYPE_CHECKING:
|
|
from ..agent.agent import BidiAgent
|
|
from ..models import ConnectionTimeoutError
|
|
|
|
|
|
@dataclass
|
|
class _HookEvent(BaseHookEvent):
|
|
"""Base class for BidiAgent hook events.
|
|
|
|
Attributes:
|
|
agent: The BidiAgent instance that triggered this event.
|
|
"""
|
|
|
|
agent: "BidiAgent"
|
|
|
|
|
|
@dataclass
|
|
class BidiAgentStopEvent(_HookEvent):
|
|
"""Event triggered after BidiAgent attempts to stop its streaming session.
|
|
|
|
This event is fired after background-task and model cleanup have been attempted,
|
|
including when cleanup raises an exception.
|
|
Hook providers can use this event for cleanup, logging, or state persistence.
|
|
|
|
Note: This event uses reverse callback ordering, meaning callbacks registered
|
|
later will be invoked first during cleanup.
|
|
|
|
This event is triggered at the end of agent.stop().
|
|
"""
|
|
|
|
@property
|
|
def should_reverse_callbacks(self) -> bool:
|
|
"""True to invoke callbacks in reverse order."""
|
|
return True
|
|
|
|
|
|
@dataclass
|
|
class BidiResponseStopEvent(_HookEvent):
|
|
"""Event triggered when the model reports that a response has ended.
|
|
|
|
A connection failure or shutdown without a model-reported completion does not
|
|
emit this event.
|
|
|
|
Attributes:
|
|
response_id: Identifier of the response that ended.
|
|
"""
|
|
|
|
response_id: str
|
|
|
|
|
|
@dataclass
|
|
class BidiBargeInEvent(_HookEvent):
|
|
"""Event triggered to stop current response generation or playback.
|
|
|
|
This event is fired when the user barges in (e.g., by speaking during the
|
|
assistant's response) or when an error stops output. This is
|
|
specific to a response and does not pause the bidirectional session.
|
|
|
|
Hook providers can use this event to log barge-ins, stop playback, or trigger cleanup.
|
|
|
|
Attributes:
|
|
reason: Why response output should stop ("user_speech" or "error").
|
|
"""
|
|
|
|
reason: Literal["user_speech", "error"]
|
|
|
|
|
|
@dataclass
|
|
class BidiBeforeConnectionRestartEvent(_HookEvent):
|
|
"""Event emitted before the agent restarts the model connection.
|
|
|
|
A restart is triggered either reactively, after the model reports a timeout, or
|
|
proactively, when the reconnect timer fires ahead of the provider's limit.
|
|
|
|
Attributes:
|
|
reason: What triggered the restart ("timeout" reactively, "scheduled" proactively).
|
|
timeout_error: The model's timeout error on the reactive path; None when scheduled.
|
|
"""
|
|
|
|
reason: Literal["timeout", "scheduled"]
|
|
timeout_error: "ConnectionTimeoutError | None" = None
|
|
|
|
|
|
@dataclass
|
|
class BidiAfterConnectionRestartEvent(_HookEvent):
|
|
"""Event emitted after the agent attempts to restart the model connection.
|
|
|
|
Attributes:
|
|
reason: What triggered the restart ("timeout" reactively, "scheduled" proactively).
|
|
exception: Populated if an exception was raised during the restart. None means success.
|
|
"""
|
|
|
|
reason: Literal["timeout", "scheduled"]
|
|
exception: Exception | None = None
|