Use prompt history for online mining records

This commit is contained in:
wuyang6
2026-05-14 17:42:08 +08:00
parent f1dc7e4b91
commit 87dd67c76a
10 changed files with 219 additions and 97 deletions
@@ -20,17 +20,28 @@ def main() -> int:
raise ValueError("request_ids is required")
sources = payload.get("sources") or ["main", "pre_processing"]
date = payload.get("date")
date_from = payload.get("date_from")
date_to = payload.get("date_to")
lookback_days = payload.get("lookback_days")
lookback_days = int(lookback_days) if lookback_days not in (None, "") else None
rows = []
for request_id in request_ids:
item = {"request_id": request_id}
for source in sources:
doc = fetch_one_by_request_id(str(source), request_id, date)
doc = fetch_one_by_request_id(
str(source),
request_id,
date,
date_from=date_from,
date_to=date_to,
lookback_days=lookback_days,
)
item[str(source)] = extract_case(str(source), doc) if doc else None
rows.append(item)
output_path = write_optional_jsonl(str(payload.get("output_path") or ""), rows)
emit_success(
{
"date": date or "past-48h",
"date": date or (f"{date_from}..{date_to}" if date_from and date_to else f"past-{lookback_days}d" if lookback_days else "past-48h"),
"count": len(rows),
"output_path": output_path,
"cases": [
@@ -58,4 +69,3 @@ def main() -> int:
if __name__ == "__main__":
raise SystemExit(main())