From 99d00c412436ef43fa09eb6cad2a35a378689726 Mon Sep 17 00:00:00 2001
From: 732642856 <732642856@qq.com>
Date: Wed, 8 Jul 2026 20:06:06 +0800
Subject: [PATCH] feat: archive VedAstro official raw responses

---
 scripts/vedastro_gateway.py    | 36 +++++++++++++++++++++++++++++-----
 tests/test_vedastro_gateway.py | 25 +++++++++++++++++++++++
 2 files changed, 56 insertions(+), 5 deletions(-)

diff --git a/scripts/vedastro_gateway.py b/scripts/vedastro_gateway.py
index c3b964e1..5e82b688 100644
--- a/scripts/vedastro_gateway.py
+++ b/scripts/vedastro_gateway.py
@@ -128,6 +128,36 @@ def get_gateway_job(job_id: str) -> dict[str, Any] | None:
     return json.loads(path.read_text(encoding="utf-8"))
 
 
+def _official_raw_response(result: dict[str, Any]) -> Any:
+    raw = result.get("official_raw_response")
+    if raw:
+        return raw
+    raw = result.get("raw_response")
+    if isinstance(raw, dict) and raw.get("source") == "vedastro_official":
+        return raw
+    return None
+
+
+def _raw_response_archive(job_id: str, result: dict[str, Any]) -> dict[str, Any]:
+    raw = _official_raw_response(result)
+    if not raw:
+        return {
+            "status": "stored_gateway_packet_not_official_raw",
+            "official_raw_response_available": False,
+            "boundary": "Gateway packet was archived; VedAstro official raw response is still separate evidence.",
+        }
+    archive_rel = f"{job_id}.official_raw_response.json"
+    archive_path = _queue_dir() / archive_rel
+    archive_path.parent.mkdir(parents=True, exist_ok=True)
+    archive_path.write_text(json.dumps(raw, ensure_ascii=False, indent=2, sort_keys=True), encoding="utf-8")
+    return {
+        "status": "official_raw_response_archived",
+        "official_raw_response_available": True,
+        "official_raw_response_path": archive_rel,
+        "boundary": "VedAstro official raw response archived separately from the gateway summary packet.",
+    }
+
+
 def complete_gateway_job(job_id: str, result: dict[str, Any]) -> dict[str, Any]:
     job = get_gateway_job(job_id)
     if job is None:
@@ -135,11 +165,7 @@ def complete_gateway_job(job_id: str, result: dict[str, Any]) -> dict[str, Any]:
     job["status"] = "completed"
     job["updated_at"] = _now_iso()
     job["result"] = dict(result or {})
-    job["raw_response_archive"] = {
-        "status": "stored_gateway_packet_not_official_raw",
-        "official_raw_response_available": False,
-        "boundary": "Gateway packet was archived; VedAstro official raw response is still separate evidence.",
-    }
+    job["raw_response_archive"] = _raw_response_archive(job_id, job["result"])
     return _write_job(job)
 
 
diff --git a/tests/test_vedastro_gateway.py b/tests/test_vedastro_gateway.py
index 3613901b..3e23c046 100644
--- a/tests/test_vedastro_gateway.py
+++ b/tests/test_vedastro_gateway.py
@@ -113,6 +113,31 @@ def test_gateway_queue_lifecycle_uses_file_job_store(monkeypatch, tmp_path):
     assert polled["raw_response_archive"]["status"] == "stored_gateway_packet_not_official_raw"
 
 
+def test_gateway_completion_archives_official_raw_response(monkeypatch, tmp_path):
+    from scripts import vedastro_gateway
+
+    monkeypatch.setenv("VEDASTRO_GATEWAY_QUEUE_DIR", str(tmp_path))
+    job = vedastro_gateway.enqueue_gateway_job({"year": 1955}, question="x")
+    completed = vedastro_gateway.complete_gateway_job(
+        job["job_id"],
+        {
+            "scope": "vedastro_gateway_run",
+            "status": "ok",
+            "official_raw_response": {
+                "source": "vedastro_official",
+                "payload": {"Status": "Pass"},
+            },
+        },
+    )
+
+    archive = completed["raw_response_archive"]
+    assert archive["status"] == "official_raw_response_archived"
+    assert archive["official_raw_response_available"] is True
+    path = tmp_path / archive["official_raw_response_path"]
+    assert path.exists()
+    assert '"vedastro_official"' in path.read_text(encoding="utf-8")
+
+
 def test_gateway_run_job_executes_queued_request(monkeypatch, tmp_path):
     from scripts import vedastro_gateway