Use prompt history for online mining records
This commit is contained in:
@@ -58,17 +58,35 @@ def main() -> int:
|
||||
if not request_ids:
|
||||
raise ValueError("request_ids, cases or cases_path is required")
|
||||
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:
|
||||
main_doc = fetch_one_by_request_id("main", request_id, date)
|
||||
pre_doc = fetch_one_by_request_id("pre_processing", request_id, date)
|
||||
main_doc = fetch_one_by_request_id(
|
||||
"main",
|
||||
request_id,
|
||||
date,
|
||||
date_from=date_from,
|
||||
date_to=date_to,
|
||||
lookback_days=lookback_days,
|
||||
)
|
||||
pre_doc = fetch_one_by_request_id(
|
||||
"pre_processing",
|
||||
request_id,
|
||||
date,
|
||||
date_from=date_from,
|
||||
date_to=date_to,
|
||||
lookback_days=lookback_days,
|
||||
)
|
||||
main_case = extract_case("main", main_doc) if main_doc else None
|
||||
pre_case = extract_case("pre_processing", pre_doc) if pre_doc else None
|
||||
rows.append(merge_case(request_id, main_case, pre_case))
|
||||
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": [
|
||||
@@ -95,4 +113,3 @@ def main() -> int:
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
|
||||
|
||||
Reference in New Issue
Block a user