54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
|
"""LLM JSON 边界契约（Memoir agents）。"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from pydantic import AliasChoices, BaseModel, ConfigDict, Field
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ClassificationOutput(BaseModel):
|
|||
|
|
category: str = ""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class MemoirTitleOutput(BaseModel):
|
|||
|
|
title: str = ""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class FidelityOutput(BaseModel):
|
|||
|
|
model_config = ConfigDict(populate_by_name=True)
|
|||
|
|
|
|||
|
|
pass_: bool = Field(default=True, alias="pass")
|
|||
|
|
reason: str | None = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
class StateExtractionOutput(BaseModel):
|
|||
|
|
detected_stage: str = ""
|
|||
|
|
slots: dict[str, str] = Field(default_factory=dict)
|
|||
|
|
emotion: str | None = None
|
|||
|
|
is_new_chapter: bool | None = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
class BatchPhase1SegmentRowOut(BaseModel):
|
|||
|
|
id: str
|
|||
|
|
detected_stage: str = ""
|
|||
|
|
slots: dict[str, str] = Field(default_factory=dict)
|
|||
|
|
chapter_category: str = Field(
|
|||
|
|
default="",
|
|||
|
|
validation_alias=AliasChoices("chapter_category", "category"),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
model_config = ConfigDict(extra="ignore", populate_by_name=True)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class BatchPhase1LLMOutput(BaseModel):
|
|||
|
|
segments: list[BatchPhase1SegmentRowOut]
|
|||
|
|
|
|||
|
|
|
|||
|
|
__all__ = [
|
|||
|
|
"BatchPhase1LLMOutput",
|
|||
|
|
"BatchPhase1SegmentRowOut",
|
|||
|
|
"ClassificationOutput",
|
|||
|
|
"FidelityOutput",
|
|||
|
|
"MemoirTitleOutput",
|
|||
|
|
"StateExtractionOutput",
|
|||
|
|
]
|