wehub-resource-sync
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
"""Progress events emitted by deep callees during a streaming orchestrator run.
|
|
|
|
Each subclass models one engine-side phase. The Java side forwards the JSON
|
|
verbatim into ``AiWorkflowProgressEvent.engineDetail``; the frontend switches
|
|
on ``phase`` and renders the typed fields.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Annotated, Literal
|
|
|
|
from pydantic import Field
|
|
|
|
from stirling.models import ApiModel
|
|
|
|
|
|
class WholeDocReadStarted(ApiModel):
|
|
phase: Literal["whole_doc_read_started"] = "whole_doc_read_started"
|
|
question: str
|
|
pages: int
|
|
slices: int
|
|
|
|
|
|
class WholeDocSliceDone(ApiModel):
|
|
"""Emitted as each chunked-reasoner worker completes.
|
|
|
|
``completed`` is a monotonically increasing counter (1..total) reflecting
|
|
the order in which workers finished, NOT the slice's position in the
|
|
document. Callers showing "Read X of Y" should use this directly so X
|
|
increments by one with each event.
|
|
"""
|
|
|
|
phase: Literal["whole_doc_slice_done"] = "whole_doc_slice_done"
|
|
completed: int
|
|
total: int
|
|
pages: str
|
|
duration_ms: int
|
|
excerpts: int
|
|
facts: int
|
|
|
|
|
|
class WholeDocCompressionRound(ApiModel):
|
|
"""Emitted when the gathered slice notes exceed the synthesis context
|
|
budget and the reasoner consolidates them with a fast-model fold pass.
|
|
|
|
Long documents (a 3000-page novel produces ~900k chars of raw notes)
|
|
would otherwise overflow the smart-model's prompt. ``notes_in`` is the
|
|
count entering the round; ``groups`` is the number of fold calls fired
|
|
(each producing one consolidated note). One or two rounds usually fit;
|
|
the event fires per round so callers can render "Consolidating notes
|
|
(round N)..." rather than going silent through the fold.
|
|
"""
|
|
|
|
phase: Literal["whole_doc_compression_round"] = "whole_doc_compression_round"
|
|
round_number: int
|
|
notes_in: int
|
|
groups: int
|
|
|
|
|
|
class WholeDocReadDone(ApiModel):
|
|
phase: Literal["whole_doc_read_done"] = "whole_doc_read_done"
|
|
completed: int
|
|
slices: int
|
|
duration_seconds: float
|
|
|
|
|
|
type ProgressEvent = Annotated[
|
|
WholeDocReadStarted | WholeDocSliceDone | WholeDocCompressionRound | WholeDocReadDone,
|
|
Field(discriminator="phase"),
|
|
]
|