wire orchestrator streaming context path and align event relay outputs

This commit is contained in:
CREDO23 2026-05-07 17:06:17 +02:00
parent 0f40279d95
commit a04b2e88bd
8 changed files with 94 additions and 109 deletions

View file

@ -16,7 +16,7 @@ from app.tasks.chat.streaming.handlers.custom_event_dispatch import (
)
from app.tasks.chat.streaming.handlers.tool_end import iter_tool_end_frames
from app.tasks.chat.streaming.handlers.tool_start import iter_tool_start_frames
from app.tasks.chat.streaming.orchestration.output import StreamOutput
from app.tasks.chat.streaming.orchestration.output import StreamingResult
from app.tasks.chat.streaming.relay.state import AgentEventRelayState
from app.tasks.chat.streaming.relay.thinking_step_completion import (
complete_active_thinking_step,
@ -52,7 +52,7 @@ class EventRelay:
events: AsyncIterator[dict[str, Any]],
*,
state: AgentEventRelayState,
result: StreamOutput,
result: StreamingResult,
step_prefix: str = "thinking",
content_builder: Any | None = None,
config: dict[str, Any] | None = None,