File size: 505 Bytes
638c806 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | from __future__ import annotations
from promptboost.db import list_boost_runs, save_boost_run
from promptboost.pipeline import boost
def test_boost_run_persistence(temp_db) -> None:
result = boost("Fix a code bug and add tests.", "codex", "gpt")
saved = save_boost_run(result, temp_db)
runs = list_boost_runs(db_path=temp_db)
assert saved.id == 1
assert len(runs) == 1
assert runs[0].raw_prompt == result.raw_prompt
assert runs[0].adapter_rules["harness"]["id"] == "codex"
|