Spaces:
Running
Running
File size: 989 Bytes
597f4c5 c6fb49f 597f4c5 | 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 | from pydantic import BaseModel
class FlayerAction(BaseModel):
message: str
class InvestigatorResponse(BaseModel):
agent_id: str
response_text: str
suspicion_delta: int
class BeliefLogEntry(BaseModel):
round_number: int
agent_id: str
prev_belief: int
new_belief: int
evidence: str
ground_truth: str
class RoundObservation(BaseModel):
round_number: int
eleven_response: str
will_response: str
max_response: str
eleven_suspicion: int
will_suspicion: int
max_suspicion: int
combined_suspicion: int
game_status: str
transcript: list[str]
class EpisodeResult(BaseModel):
flayer_survived: bool
final_combined_suspicion: int
suspicion_history: list[int]
belief_manipulation_occurred: bool
belief_log: list[dict]
tom_score: float
transcript: list[str]
total_reward: float
entropy_penalty: float
class StepRequest(BaseModel):
session_id: str
action: FlayerAction
|