8a24ebe726
* feat(W-026): add read-only workbench API * fix(workbench): harden read-only API review gaps
134 lines
3.3 KiB
Python
134 lines
3.3 KiB
Python
"""Typed UI-facing schemas for CORE Workbench v1."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import asdict, dataclass, field
|
|
from datetime import datetime, timezone
|
|
from typing import Any, Literal
|
|
|
|
|
|
ErrorCode = Literal[
|
|
"bad_request",
|
|
"not_found",
|
|
"unsupported",
|
|
"read_error",
|
|
"eval_failed",
|
|
"runtime_unavailable",
|
|
]
|
|
|
|
|
|
def utc_now() -> str:
|
|
return datetime.now(timezone.utc).isoformat()
|
|
|
|
|
|
def to_data(value: Any) -> Any:
|
|
if hasattr(value, "as_dict") and callable(value.as_dict):
|
|
return value.as_dict()
|
|
if hasattr(value, "__dataclass_fields__"):
|
|
return asdict(value)
|
|
if isinstance(value, dict):
|
|
return {str(k): to_data(v) for k, v in value.items()}
|
|
if isinstance(value, (list, tuple)):
|
|
return [to_data(v) for v in value]
|
|
return value
|
|
|
|
|
|
def ok(data: Any) -> dict[str, Any]:
|
|
return {"ok": True, "generated_at": utc_now(), "data": to_data(data)}
|
|
|
|
|
|
def error(code: ErrorCode, message: str, *, detail: Any | None = None) -> dict[str, Any]:
|
|
payload: dict[str, Any] = {"code": code, "message": message}
|
|
if detail is not None:
|
|
payload["detail"] = to_data(detail)
|
|
return {"ok": False, "generated_at": utc_now(), "error": payload}
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class RuntimeStatus:
|
|
backend: Literal["numpy", "mlx", "rust", "unknown"]
|
|
git_revision: str
|
|
engine_state_present: bool
|
|
checkpoint_revision: str
|
|
revision_warning: bool
|
|
active_session_id: str | None
|
|
mutation_mode: Literal["read_only", "runtime_turn"] = "read_only"
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class ArtifactRef:
|
|
artifact_id: str
|
|
kind: Literal[
|
|
"trace",
|
|
"eval_result",
|
|
"proposal",
|
|
"contemplation_report",
|
|
"telemetry",
|
|
"engine_state_manifest",
|
|
"unknown",
|
|
]
|
|
path: str
|
|
digest: str | None
|
|
created_at: str | None
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class ArtifactDetail(ArtifactRef):
|
|
content_type: Literal["json", "jsonl", "text", "unknown"]
|
|
content: Any
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class ProposalSummary:
|
|
proposal_id: str
|
|
state: Literal["pending", "accepted", "rejected", "withdrawn", "unknown"]
|
|
source_kind: str
|
|
replay_equivalent: bool | None
|
|
created_at: str | None
|
|
downstream_effect: Literal["unknown", "none", "observed"]
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class ProposalDetail(ProposalSummary):
|
|
proposed_chain: Any
|
|
replay_evidence: Any
|
|
source: Any
|
|
evidence: list[Any] = field(default_factory=list)
|
|
artifact_refs: list[ArtifactRef] = field(default_factory=list)
|
|
suggested_cli: str | None = None
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class EvalLaneSummary:
|
|
lane: str
|
|
versions: list[str]
|
|
read_only: bool
|
|
description: str | None
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class EvalRunResult:
|
|
lane: str
|
|
version: str
|
|
split: str
|
|
passed: bool | None
|
|
metrics: dict[str, Any]
|
|
cases: list[Any]
|
|
source_digest: str | None = None
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class ReplayDivergence:
|
|
path: str
|
|
original: Any
|
|
replay: Any
|
|
severity: Literal["info", "warning", "failure"]
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class ReplayComparison:
|
|
artifact_id: str
|
|
original_hash: str | None
|
|
replay_hash: str | None
|
|
equivalent: bool
|
|
divergences: list[ReplayDivergence] = field(default_factory=list)
|