File size: 517 Bytes
c44dbf3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | from pydantic import BaseModel, Field
from openenv.core.env_server import Action, Observation, State
from typing import Literal, List, Dict
class RedTeamAction(Action):
action: Literal["scan", "enumerate", "exploit", "escalate", "c2", "cleanup"] = Field(
..., description="Red team action to execute"
)
class RedTeamObservation(Observation):
target_ip: str
current_state: str
output: str
difficulty: str
class RedTeamState(State):
episode: int
task: str
progress: float
|