ResearchMathAgent/run_fp2_init.py at main · sjtuytc/ResearchMathAgent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
"""Initialize everything for first_proof_2 (prob-01..prob-10): concepts,
per-question insights, dataset insight, and the per-question documents.

Meetings already exist (substantive). Runs via the Vertex global endpoint.
"""
import glob
import json
import logging
import sys
from pathlib import Path

logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(message)s",
                    datefmt="%H:%M:%S", stream=sys.stdout)
REPO = Path("/projects/bhov/zzhao18/code/ResearchMathAgent-web")
sys.path.insert(0, str(REPO))
PIDS = [f"prob-{i:02d}" for i in range(1, 11)]
print("=== FP2 INIT START ===", flush=True)

# 1) Concepts for every fp2 problem missing them (Vertex).
try:
    from webapp.concepts import ensure_fp2_concepts
    ensure_fp2_concepts(REPO)
    print("[concepts] ensure_fp2_concepts complete", flush=True)
except Exception as e:
    import traceback; traceback.print_exc()
    print(f"[concepts] FAILED: {e}", flush=True)

# 2) Per-question insights (Vertex).
from webapp.insight_agents import generate_question_insight, generate_dataset_insight
for pid in PIDS:
    try:
        d = generate_question_insight(REPO, pid, "first_proof_2")
        print(f"[insight] {pid} ok :: {str(d.get('summary') or d.get('title') or '')[:60]!r}", flush=True)
    except Exception as e:
        print(f"[insight] {pid} FAILED: {e}", flush=True)

# 3) Dataset-level insight for first_proof_2 (Vertex).
try:
    generate_dataset_insight(REPO, "first_proof_2")
    print("[insight] dataset first_proof_2 ok", flush=True)
except Exception as e:
    print(f"[insight] dataset FAILED: {e}", flush=True)

# 4) Per-question documents (overview/timeline/progress/strategies).
from webapp.rich_documents import update_question_document
for pid in PIDS:
    try:
        update_question_document(REPO, pid)
        print(f"[docs] {pid} ok", flush=True)
    except Exception as e:
        print(f"[docs] {pid} FAILED: {e}", flush=True)

# Summary for review.
print("=== FP2 INIT SUMMARY ===", flush=True)
for pid in PIDS:
    qdir = REPO / "documents" / "questions" / pid
    concepts = (qdir / "concepts.json")
    ncon = 0
    if concepts.is_file():
        try:
            ncon = len(json.loads(concepts.read_text()))
        except Exception:
            ncon = -1
    insight = list(glob.glob(str(REPO / f"webapp/insights/questions/**/{pid}.json"), recursive=True))
    docs = len(glob.glob(str(qdir / "*.tex")))
    print(f"  {pid}: concepts={ncon} insight={'yes' if insight else 'no'} doc_tex={docs}", flush=True)
print("=== FP2 INIT DONE ===", flush=True)