chunker ui redo
This commit is contained in:
@@ -41,6 +41,13 @@ class CancelRequest:
|
||||
job_id: str
|
||||
|
||||
|
||||
@dataclass
|
||||
class ChunkStreamRequest:
|
||||
"""Request to stream chunk pipeline events."""
|
||||
|
||||
job_id: str
|
||||
|
||||
|
||||
@dataclass
|
||||
class Empty:
|
||||
"""Empty message for requests with no parameters."""
|
||||
@@ -94,6 +101,26 @@ class WorkerStatus:
|
||||
gpu_available: bool
|
||||
|
||||
|
||||
@dataclass
|
||||
class ChunkPipelineEvent:
|
||||
"""Streaming chunk pipeline event."""
|
||||
|
||||
job_id: str
|
||||
event_type: str # pipeline_start, chunk_queued, chunk_done, etc.
|
||||
sequence: int = 0
|
||||
worker_id: str = ""
|
||||
state: str = ""
|
||||
queue_size: int = 0
|
||||
elapsed: float = 0.0
|
||||
throughput_mbps: float = 0.0
|
||||
total_chunks: int = 0
|
||||
processed_chunks: int = 0
|
||||
failed_chunks: int = 0
|
||||
error: str = ""
|
||||
processing_time: float = 0.0
|
||||
retries: int = 0
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Service Definition (for documentation, generator uses this)
|
||||
# -----------------------------------------------------------------------------
|
||||
@@ -126,5 +153,11 @@ GRPC_SERVICE = {
|
||||
"response": WorkerStatus,
|
||||
"stream_response": False,
|
||||
},
|
||||
{
|
||||
"name": "StreamChunkPipeline",
|
||||
"request": ChunkStreamRequest,
|
||||
"response": ChunkPipelineEvent,
|
||||
"stream_response": True, # Server streaming
|
||||
},
|
||||
],
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user