(no commit message)

2026-04-02 18:15:47 -07:00
parent 2b40f813a0
commit 051376c286
2 changed files with 4 additions and 2 deletions
--- a/optimize.py
+++ b/optimize.py
@@ -33,6 +33,7 @@ def classification_metric(


 if __name__ == "__main__":
+    program = DocExtract.from_precompiled("tyrin/doc-extract")
    dspy.configure(lm=dspy.LM("openai/gpt-4.1-mini", temperature=0.0))

    optimized_program = dspy.GEPA(
@@ -40,7 +41,7 @@ if __name__ == "__main__":
        reflection_lm=dspy.LM("openai/gpt-4.1-mini", temperature=1.0, max_tokens=8000),
        auto="light",
    ).compile(
-        DocExtract(),
+        program,
        trainset=load_dataset("data/train.jsonl"),
        valset=load_dataset("data/val.jsonl"),
    )