Class ResponseSchema

control_layer.py:127–148

Output contract the LLM response must satisfy.

class ResponseSchema(BaseModel):
    """Output contract the LLM response must satisfy."""
    required_keys:     List[str] = Field(default_factory=list)
    max_length:        Optional[int] = None
    min_length:        Optional[int] = None
    forbidden_phrases: List[str] = Field(default_factory=list)
    must_contain:      List[str] = Field(default_factory=list)
    must_be_json:      bool = False

    @field_validator("max_length")
    @classmethod
    def max_positive(cls, v):
        if v is not None and v <= 0:
            raise ValueError("max_length must be positive")
        return v

    @model_validator(mode="after")
    def min_lt_max(self):
        if self.min_length and self.max_length:
            if self.min_length >= self.max_length:
                raise ValueError("min_length must be < max_length")
        return self


# =============================================================================