50 lines
2.0 KiB
Python
50 lines
2.0 KiB
Python
"""答题 / 练习记录，以及 AI 调用审计。"""
|
|
import enum
|
|
|
|
from sqlalchemy import Boolean, ForeignKey, Integer, String, Text
|
|
from sqlalchemy import Enum as SAEnum
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
from sqlalchemy.types import JSON
|
|
|
|
from app.models.base import new_uuid
|
|
from app.db import Base, TimestampMixin
|
|
|
|
|
|
class JudgedBy(str, enum.Enum):
|
|
auto = "auto" # 客观题本地判分
|
|
ai = "ai" # 主观题 AI 判分
|
|
self_ = "self" # 用户自评
|
|
|
|
|
|
class PracticeRecord(Base, TimestampMixin):
|
|
__tablename__ = "practice_records"
|
|
|
|
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
|
|
user_id: Mapped[str] = mapped_column(
|
|
ForeignKey("users.id", ondelete="CASCADE"), index=True
|
|
)
|
|
question_id: Mapped[str] = mapped_column(
|
|
ForeignKey("questions.id", ondelete="CASCADE"), index=True
|
|
)
|
|
user_answer: Mapped[list | None] = mapped_column(JSON, nullable=True)
|
|
is_correct: Mapped[bool | None] = mapped_column(Boolean, nullable=True)
|
|
judged_by: Mapped[JudgedBy] = mapped_column(SAEnum(JudgedBy), default=JudgedBy.auto)
|
|
ai_feedback_markdown: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
|
|
class AiGeneration(Base, TimestampMixin):
|
|
"""记录每次 AI 调用，便于复用缓存、成本观察、排查。"""
|
|
|
|
__tablename__ = "ai_generations"
|
|
|
|
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=new_uuid)
|
|
user_id: Mapped[str] = mapped_column(
|
|
ForeignKey("users.id", ondelete="CASCADE"), index=True
|
|
)
|
|
question_id: Mapped[str | None] = mapped_column(String(36), nullable=True)
|
|
task: Mapped[str] = mapped_column(String(32)) # extract/judge/summarize/explain
|
|
model: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
request_tokens: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
response_tokens: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
content_markdown: Mapped[str | None] = mapped_column(Text, nullable=True)
|