| """Regenera docs/interfaces/*.schema.json a partir de src/ragsolver/rag/schema.py. | |
| Uso: python scripts/export_schemas.py | |
| Correr después de cualquier cambio a rag/schema.py (RNF-4: los contratos | |
| JSON Schema en docs/ deben reflejar siempre el código, no al revés). | |
| """ | |
| from __future__ import annotations | |
| import json | |
| from pathlib import Path | |
| from ragsolver.agents import schemas as a | |
| from ragsolver.rag import schema as s | |
| MODELS = [ | |
| s.Document, | |
| s.ChunkBase, | |
| s.InformalChunk, | |
| s.MathlibChunk, | |
| s.Figure, | |
| s.Citation, | |
| s.TraceEvent, | |
| s.RetrievalRequest, | |
| s.RetrievalResult, | |
| # Fase 4 — contratos de los agentes (RNF-4) | |
| a.Problem, | |
| a.PlannerOutput, | |
| a.GeneratorOutput, | |
| a.FormalizerOutput, | |
| a.CriticOutput, | |
| a.BaselineAOutput, | |
| a.BaselineBOutput, | |
| a.Verdict, | |
| a.SolutionReport, | |
| ] | |
| OUT_DIR = Path(__file__).resolve().parent.parent / "docs" / "interfaces" | |
| def main() -> None: | |
| OUT_DIR.mkdir(parents=True, exist_ok=True) | |
| for model in MODELS: | |
| out_path = OUT_DIR / f"{model.__name__}.schema.json" | |
| out_path.write_text(json.dumps(model.model_json_schema(), indent=2, ensure_ascii=False) + "\n") | |
| print(f"wrote {out_path.relative_to(OUT_DIR.parent.parent)}") | |
| if __name__ == "__main__": | |
| main() | |