File size: 471 Bytes
d9194e2
5b93dcc
 
0f55283
 
 
 
d9194e2
5b93dcc
d9194e2
5b93dcc
0f55283
 
 
 
5b93dcc
0f55283
 
 
d9194e2
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 typing import List, Optional, Dict, Any


class Observation(BaseModel):
    failed_logins: int
    port_scans: int
    suspicious_ips: List[str] = Field(default_factory=list)
    total_requests: Optional[int] = 0
    attack_types: List[str] = Field(default_factory=list)


class Action(BaseModel):
    action: str


class Reward(BaseModel):
    reward: float
    done: bool
    info: Dict[str, Any] = Field(default_factory=dict)