File size: 2,352 Bytes
ad6248e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | from __future__ import annotations
from pydantic import BaseModel, Field
from typing import Dict, List, Optional, Any
class ServiceMetrics(BaseModel):
cpu_percent: float
memory_mb: float
error_rate: float # errors per second
p99_latency_ms: float
request_queue: int
last_deploy: str
status: str # healthy | degraded | critical | down
class Alert(BaseModel):
alert_id: str
severity: str # P0 | P1 | P2
service: str
message: str
triggered_at_step: int
is_red_herring: bool = False
class FileView(BaseModel):
service: str
filename: str
content: str
total_lines: int
class EditRecord(BaseModel):
step: int
service: str
filename: str
line_number: int
old_code: str
new_code: str
class TestResult(BaseModel):
suite: str # unit | integration | load | security
passed: bool
output: str
errors: List[str] = Field(default_factory=list)
class Observation(BaseModel):
step: int
incident_id: str
system_metrics: Dict[str, ServiceMetrics]
active_alerts: List[Alert]
open_file: Optional[FileView] = None
terminal_output: str
git_diff: Optional[str] = None
dependency_graph: Dict[str, List[str]]
sre_memory: List[str] = Field(default_factory=list)
budget_remaining: int
task_id: int
task_description: str
class ActionRequest(BaseModel):
tool: str
params: Dict[str, Any] = Field(default_factory=dict)
class ActionResult(BaseModel):
tool: str
output: Any
reward_delta: float
done: bool
observation: Observation
class EpisodeResult(BaseModel):
incident_id: str
task_id: int
steps_taken: int
total_reward: float
normalized_score: float # 0.0 – 1.0
tests_passed: bool
incident_report_accuracy: float
fixed_within_sla: bool
tool_call_log: List[Dict[str, Any]]
weakness_tags: List[str] # for DifficultyController
class IncidentReport(BaseModel):
root_cause: str
fix_applied: str
services_affected: List[str]
severity_classification: str # P0 | P1 | P2
class DifficultyState(BaseModel):
async_bugs: float = 0.5
data_corruption: float = 0.5
security_bugs: float = 0.5
cascading_failures: float = 0.5
red_herrings: float = 0.5
|