from dataclasses import dataclass, field
from typing import List, Optional
from StepTrace import StepTrace



@dataclass
class RunTrace:
    run_id: str
    task_id: str
    dataset: str
    instruction: str

    model_name: Optional[str]

    started_at: float
    finished_at: Optional[float]

    success: bool
    final_url: Optional[str]
    final_answer: Optional[str]

    error: Optional[str]
    fail_reason: Optional[str]

    total_steps: int
    total_tokens: Optional[int]
    total_latency_sec: Optional[float]

    screenshots_dir: Optional[str]

    steps: List[StepTrace] = field(default_factory=list)