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 первым)
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
"""Handle text input and output to and from bidi agent."""
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from prompt_toolkit import PromptSession
|
|
|
|
from ....types.content import TextBlock
|
|
from ..types.events import (
|
|
BidiBargeInEvent,
|
|
BidiConnectionStopEvent,
|
|
BidiOutputEvent,
|
|
BidiTranscriptDeltaEvent,
|
|
)
|
|
from ..types.io import InputStream, OutputStream
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class _ConsoleInputStream(InputStream):
|
|
"""Handle text input from user."""
|
|
|
|
def __init__(self, config: dict[str, Any]) -> None:
|
|
"""Extract configs and setup prompt session."""
|
|
prompt = config.get("input_prompt", "")
|
|
self._session: PromptSession = PromptSession(prompt)
|
|
|
|
async def __call__(self) -> TextBlock:
|
|
"""Read user input from stdin."""
|
|
text = await self._session.prompt_async()
|
|
return TextBlock(text.strip())
|
|
|
|
|
|
class _ConsoleOutputStream(OutputStream):
|
|
"""Handle text output from bidi agent."""
|
|
|
|
async def __call__(self, event: BidiOutputEvent) -> None:
|
|
"""Print text events to stdout."""
|
|
if isinstance(event, BidiBargeInEvent):
|
|
logger.debug("reason=<%s> | barge-in detected", event["reason"])
|
|
print("barge-in")
|
|
|
|
elif isinstance(event, BidiConnectionStopEvent):
|
|
if event.reason == "user_request":
|
|
print("user requested connection close using the stop tool.")
|
|
logger.debug("connection_id=<%s> | user requested connection close", event.connection_id)
|
|
elif isinstance(event, BidiTranscriptDeltaEvent):
|
|
logger.debug(
|
|
"role=<%s>, text_length=<%d> | text transcript received",
|
|
event.role,
|
|
len(event.delta),
|
|
)
|
|
print(event.delta)
|
|
|
|
|
|
class ConsoleIO:
|
|
"""Handle text input and output to and from bidi agent.
|
|
|
|
Accepts input from stdin and outputs to stdout.
|
|
"""
|
|
|
|
def __init__(self, **config: Any) -> None:
|
|
"""Initialize I/O.
|
|
|
|
Args:
|
|
**config: Optional I/O configurations.
|
|
|
|
- input_prompt (str): Input prompt to display on screen (default: blank)
|
|
"""
|
|
self._config = config
|
|
|
|
def input(self) -> _ConsoleInputStream:
|
|
"""Return the standard-input stream."""
|
|
return _ConsoleInputStream(self._config)
|
|
|
|
def output(self) -> _ConsoleOutputStream:
|
|
"""Return the standard-output stream."""
|
|
return _ConsoleOutputStream()
|