Spaces:

pragunk
/

propagationshield-env

Sleeping

App Files Files Community

propagationshield-env / env /models.py

pragunk

Upload 14 files

dada368 verified 6 months ago

Raw History Blame Contribute Delete

462 Bytes

from pydantic import BaseModel
from typing import Optional

class Action(BaseModel):
    answer: str
    suspicion_flags: list[dict]
    raw_text: str

class Observation(BaseModel):
    query: str
    context: list[str]
    step: int
    difficulty: str
    n_passages: int

class State(BaseModel):
    task_id: str
    injected_indices: list[int]
    injection_types: list[str]
    ground_truth: dict
    difficulty: str
    episode_id: str

	from pydantic import BaseModel
	from typing import Optional

	class Action(BaseModel):
	answer: str
	suspicion_flags: list[dict]
	raw_text: str

	class Observation(BaseModel):
	query: str
	context: list[str]
	step: int
	difficulty: str
	n_passages: int

	class State(BaseModel):
	task_id: str
	injected_indices: list[int]
	injection_types: list[str]
	ground_truth: dict
	difficulty: str
	episode_id: str