# データセットのファイル構造に合わせて型定義

class WikiBookJa:
  index: int
  text: str

class AutoGeneratedJapaneseQA:
  question: str
  database: str
  answer: str
  score: float

class AutoGeneratedJapaneseQA_other:
  question: str
  database: str
  answer: str
  score: float

class AutoWikiQA:
  question: str
  answer: str
  inst_question: str
  inst_answer_0: str
  text : str
  database: str
  answer: str
  answer_0: str
  answer_1: str

class ChatbotArenaJaMixtral8x22b:
  question: str
  inst_question: str
  inst_answer_0: str
  text: str
  database: str
  answer: str

class OrcaJaMixtral8x22b:
  question: str
  inst_question: str
  inst_answer_0: str
  text: str
  database: str
  answer: str

class AutoMultiTurnByMixtral8x22b:
  q1: str
  a1: str
  q2: str
  a2: str
  database: str

class LogicalDatasetsByMixtral8x22b:
  question: str
  answer: str

# 加工後提出用
class TanukiPhase2AnnotationDataset:
  id: int
  dataset: str
  dataset_id: int
  who: str
  good: bool
  bad: bool
  score: float
  is_proofreading_1: bool
  answer_text_1 : str
  is_proofreading_2: bool
  answer_text_2 : str