Add training and planning eval exports

2026-05-11 14:50:00 +08:00
parent 8cca3793e0
commit be731caed7
15 changed files with 1074 additions and 4 deletions
@@ -0,0 +1,36 @@
+from __future__ import annotations
+
+from product_data_portable import (
+    DEFAULT_SYSTEM_PROMPT,
+    default_training_output_path,
+    emit_error,
+    emit_success,
+    export_training_jsonl,
+    load_json_payload,
+    load_records,
+)
+
+
+def main() -> int:
+    try:
+        payload = load_json_payload()
+        records = load_records(payload)
+        result = export_training_jsonl(
+            records,
+            output_path=default_training_output_path(payload),
+            session_num=int(payload.get("session_num", 5)),
+            session_time_minutes=int(payload.get("session_time_minutes", 5)),
+            context_fields=payload.get("context_fields"),
+            system_prompt=str(payload.get("system_prompt") or DEFAULT_SYSTEM_PROMPT),
+            require_validation_ok=bool(payload.get("require_validation_ok", True)),
+            overwrite=bool(payload.get("overwrite", True)),
+        )
+        emit_success(result)
+        return 0
+    except Exception as exc:  # noqa: BLE001 - CLI 需要把错误稳定转成 JSON
+        emit_error(exc)
+        return 1
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())