wehub-resource-sync
e4dcfc49aa
Tests / Import Check (Python 3.13) (push) Has been cancelled
Tests / Import Check (Python 3.14) (push) Has been cancelled
Tests / Python Tests (Python 3.11) (push) Has been cancelled
Tests / Python Tests (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.14) (push) Has been cancelled
Tests / Test Summary (push) Has been cancelled
Tests / Lint and Format (push) Has been cancelled
Tests / Web Node Tests (push) Has been cancelled
Tests / Import Check (Python 3.11) (push) Has been cancelled
Tests / Import Check (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.13) (push) Has been cancelled
73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
"""
|
|
Stream Event Protocol
|
|
=====================
|
|
|
|
Defines the unified streaming event format used by all tools, capabilities,
|
|
and plugins to communicate progress and results to consumers (CLI, WebSocket, SDK).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass, field
|
|
from enum import Enum
|
|
import time
|
|
from typing import Any
|
|
|
|
|
|
class StreamEventType(str, Enum):
|
|
"""All possible event types in a streaming session."""
|
|
|
|
STAGE_START = "stage_start"
|
|
STAGE_END = "stage_end"
|
|
THINKING = "thinking"
|
|
OBSERVATION = "observation"
|
|
CONTENT = "content"
|
|
TOOL_CALL = "tool_call"
|
|
TOOL_RESULT = "tool_result"
|
|
PROGRESS = "progress"
|
|
SOURCES = "sources"
|
|
RESULT = "result"
|
|
ERROR = "error"
|
|
SESSION = "session"
|
|
SESSION_META = "session_meta"
|
|
DONE = "done"
|
|
WAIT_FOR_INPUT = "wait_for_input"
|
|
|
|
|
|
@dataclass
|
|
class StreamEvent:
|
|
"""
|
|
A single streaming event emitted during a chat turn.
|
|
|
|
Attributes:
|
|
type: The semantic kind of this event.
|
|
source: Which tool / capability / plugin produced it (e.g. "deep_solve").
|
|
stage: Current stage within the source (e.g. "planning").
|
|
content: Human-readable text payload.
|
|
metadata: Arbitrary structured data (tool args, sources, metrics, …).
|
|
timestamp: Unix epoch seconds when the event was created.
|
|
"""
|
|
|
|
type: StreamEventType
|
|
source: str = ""
|
|
stage: str = ""
|
|
content: str = ""
|
|
metadata: dict[str, Any] = field(default_factory=dict)
|
|
session_id: str = ""
|
|
turn_id: str = ""
|
|
seq: int = 0
|
|
timestamp: float = field(default_factory=time.time)
|
|
|
|
def to_dict(self) -> dict[str, Any]:
|
|
return {
|
|
"type": self.type.value,
|
|
"source": self.source,
|
|
"stage": self.stage,
|
|
"content": self.content,
|
|
"metadata": self.metadata,
|
|
"session_id": self.session_id,
|
|
"turn_id": self.turn_id,
|
|
"seq": self.seq,
|
|
"timestamp": self.timestamp,
|
|
}
|