from dataclasses import Field
from typing import List

from openai import BaseModel


class QAPair(BaseModel):
    question: str = Field(description="从聊天记录中提取或生成的问题")
    answer: str = Field(description="问题的答案，如果没有明确答案则留空")

class QAList(BaseModel):
    qa_pairs: List[QAPair] = Field(description="生成的QA对列表，1-2个")