149 lines
4.8 KiB
Python
149 lines
4.8 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
查看 smoke 跑出来的 coding_memory.db 的全部内容（不截断），经验层 + 事实层。
|
||
|
||
用法:
|
||
python scripts/dump_coding_db.py /path/to/coding_memory.db
|
||
python scripts/dump_coding_db.py /path/to/coding_memory.db --json
|
||
python scripts/dump_coding_db.py /path/to/coding_memory.db --id <experience_id>
|
||
python scripts/dump_coding_db.py /path/to/coding_memory.db --facts
|
||
"""
|
||
import argparse
|
||
import json
|
||
import sqlite3
|
||
import sys
|
||
|
||
|
||
def _decode(row, json_fields):
|
||
d = dict(row)
|
||
for fld in json_fields:
|
||
if d.get(fld):
|
||
try:
|
||
d[fld] = json.loads(d[fld])
|
||
except Exception:
|
||
pass
|
||
return d
|
||
|
||
|
||
def fetch(db_path: str, experience_id: str | None = None):
|
||
conn = sqlite3.connect(db_path)
|
||
conn.row_factory = sqlite3.Row
|
||
try:
|
||
if experience_id:
|
||
cur = conn.execute(
|
||
"SELECT * FROM coding_memory_meta WHERE experience_id=?",
|
||
(experience_id,),
|
||
)
|
||
else:
|
||
cur = conn.execute(
|
||
"SELECT * FROM coding_memory_meta ORDER BY created_at ASC"
|
||
)
|
||
return [_decode(r, ("search_keys", "files", "source_fact_ids")) for r in cur]
|
||
finally:
|
||
conn.close()
|
||
|
||
|
||
def fetch_facts(db_path: str):
|
||
conn = sqlite3.connect(db_path)
|
||
conn.row_factory = sqlite3.Row
|
||
try:
|
||
cur = conn.execute("SELECT * FROM working_fact ORDER BY created_at ASC")
|
||
return [_decode(r, ("files", "derived_experience_ids")) for r in cur]
|
||
except sqlite3.OperationalError:
|
||
# 老库还没建事实表
|
||
return []
|
||
finally:
|
||
conn.close()
|
||
|
||
|
||
def render_text(rows):
|
||
for k, r in enumerate(rows, 1):
|
||
print("=" * 80)
|
||
print(f"#{k} experience_id={r['experience_id']}")
|
||
print(f" user_id : {r['user_id']}")
|
||
print(f" agent_id : {r['agent_id']}")
|
||
print(f" workspace: {r['workspace_id']}")
|
||
print(f" branch : {r['branch']}")
|
||
print(f" confidence: {r['confidence']}")
|
||
print(f" created : {r['created_at']}")
|
||
print(f" updated : {r['updated_at']}")
|
||
print(f"\n TASK:\n {r['task']}")
|
||
print(f"\n SEARCH_KEYS:")
|
||
for sk in r["search_keys"] or []:
|
||
print(f" - {sk}")
|
||
print(f"\n SOLUTION:")
|
||
for line in (r["solution"] or "").splitlines():
|
||
print(f" {line}")
|
||
if r.get("boundary_envs"):
|
||
print(f"\n BOUNDARY_ENVS:")
|
||
for line in r["boundary_envs"].splitlines():
|
||
print(f" {line}")
|
||
if r.get("files"):
|
||
print(f"\n FILES:")
|
||
for f in r["files"]:
|
||
print(f" - {f}")
|
||
if r.get("source_fact_ids"):
|
||
print(f"\n SOURCE_FACT_IDS:")
|
||
for fid in r["source_fact_ids"]:
|
||
print(f" - {fid}")
|
||
print()
|
||
print(f"\n[total] {len(rows)} experiences")
|
||
|
||
|
||
def render_facts(rows):
|
||
for k, r in enumerate(rows, 1):
|
||
print("-" * 80)
|
||
status = r["status"]
|
||
if r.get("fact_link"):
|
||
status += f" → {r['fact_link']}"
|
||
print(f"#{k} fact_id={r['fact_id']} [{status}]")
|
||
print(f" user_id : {r['user_id']}")
|
||
print(f" workspace: {r['workspace_id']}")
|
||
print(f" created : {r['created_at']}")
|
||
print(f"\n STATEMENT:\n {r['statement']}")
|
||
if r.get("files"):
|
||
print(f"\n FILES:")
|
||
for f in r["files"]:
|
||
print(f" - {f}")
|
||
if r.get("derived_experience_ids"):
|
||
print(f"\n DERIVED_EXPERIENCE_IDS:")
|
||
for eid in r["derived_experience_ids"]:
|
||
print(f" - {eid}")
|
||
print()
|
||
active = sum(1 for r in rows if r["status"] == "active")
|
||
print(f"\n[total] {len(rows)} facts ({active} active)")
|
||
|
||
|
||
def main():
|
||
ap = argparse.ArgumentParser()
|
||
ap.add_argument("db", help="path to coding_memory.db")
|
||
ap.add_argument("--id", help="只看某条经验（experience_id）")
|
||
ap.add_argument("--facts", action="store_true", help="只看事实层")
|
||
ap.add_argument("--json", action="store_true", help="输出 JSON 而非可读文本")
|
||
args = ap.parse_args()
|
||
|
||
facts = fetch_facts(args.db)
|
||
if args.facts:
|
||
if args.json:
|
||
print(json.dumps(facts, ensure_ascii=False, indent=2, default=str))
|
||
else:
|
||
render_facts(facts)
|
||
return
|
||
|
||
rows = fetch(args.db, args.id)
|
||
if args.json:
|
||
print(json.dumps(
|
||
{"experiences": rows, "facts": facts},
|
||
ensure_ascii=False, indent=2, default=str,
|
||
))
|
||
else:
|
||
render_text(rows)
|
||
if facts:
|
||
print()
|
||
render_facts(facts)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|