87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
from datetime import datetime
|
|
from pathlib import Path
|
|
from typing import Literal, Self
|
|
|
|
import yaml
|
|
from pydantic import BaseModel, Field, ConfigDict
|
|
|
|
|
|
# Stub locale di _YamlModel. In the reference implementation questa base vive in
|
|
# mschema/models.py; qui la si replica perche' SessionManifest ha bisogno di
|
|
# populate_by_name=True (db_schema usa l'alias "schema"), che la base mschema non
|
|
# imposta. Identici to_yaml/from_yaml della controparte mschema.
|
|
class _YamlModel(BaseModel):
|
|
model_config = ConfigDict(populate_by_name=True)
|
|
|
|
def to_yaml(self, path: Path) -> None:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
data = self.model_dump(by_alias=True, mode="json", exclude_defaults=False)
|
|
path.write_text(
|
|
yaml.safe_dump(data, sort_keys=False, allow_unicode=True, width=120)
|
|
)
|
|
|
|
@classmethod
|
|
def from_yaml(cls, path: Path) -> Self:
|
|
raw = yaml.safe_load(path.read_text())
|
|
return cls.model_validate(raw)
|
|
|
|
|
|
class SessionManifest(_YamlModel):
|
|
id: str
|
|
created_at: datetime
|
|
status: Literal["open", "closed", "failed", "finalized"] = "open"
|
|
question: str
|
|
database: str
|
|
db_schema: str = Field(alias="schema")
|
|
# D12/D15: autore della sessione (auth) e versione del workflow usato.
|
|
author: str | None = None
|
|
summary: str | None = None
|
|
updated_at: datetime | None = None
|
|
updated_by: str | None = None
|
|
schema_version: int | None = None
|
|
provider: str | None = None
|
|
model: str | None = None
|
|
thinking: str | None = None
|
|
name: str | None = None
|
|
archived: bool = False
|
|
group: str | None = None
|
|
|
|
|
|
class Candidate(BaseModel):
|
|
kind: Literal["table", "column"]
|
|
name: str
|
|
signals: dict = {}
|
|
evidence: list[str] = []
|
|
decision: Literal["promoted", "excluded", "pending"] = "pending"
|
|
decision_seq: int | None = None
|
|
# D14a: valori citati nella domanda ancorati a questa colonna/tabella.
|
|
grounded_values: list[dict] = []
|
|
|
|
|
|
class Join(BaseModel):
|
|
from_: str = Field(alias="from")
|
|
to: str
|
|
source: str = ""
|
|
decision: Literal["promoted", "excluded", "pending"] = "promoted"
|
|
decision_seq: int | None = None
|
|
|
|
model_config = {"populate_by_name": True}
|
|
|
|
|
|
class ExcludedItem(BaseModel):
|
|
kind: Literal["table", "column"]
|
|
name: str
|
|
decision_seq: int | None = None
|
|
|
|
|
|
class SchemaLinking(BaseModel):
|
|
question: str
|
|
candidates: list[Candidate] = []
|
|
joins: list[Join] = []
|
|
excluded: list[ExcludedItem] = []
|
|
open_questions: list[str] = []
|
|
# D14b: formule di concetto approvate, parte dello schema-linking.
|
|
concept_formulas: list[dict] = []
|
|
|
|
model_config = {"extra": "forbid"}
|