| from __future__ import annotations |
|
|
| from conftest import ( |
| call_with_supported_kwargs, |
| install_fake_adapter, |
| maybe_import, |
| resolve_callable, |
| sqlite_contains_text, |
| sqlite_user_tables, |
| text_blob, |
| unload_promptboost_modules, |
| ) |
|
|
|
|
| BOOST_MODULES = ( |
| "promptboost", |
| "promptboost.core", |
| "promptboost.boost", |
| ) |
|
|
| BOOST_FUNCS = ( |
| "boost", |
| "boost_prompt", |
| ) |
|
|
|
|
| def _patch_guards_to_pass(monkeypatch): |
| for module_name in ( |
| "promptboost.guards", |
| "promptboost.guardrails", |
| "promptboost.rules.guards", |
| "promptboost.rules", |
| "promptboost.core", |
| "promptboost.boost", |
| ): |
| module = maybe_import(module_name) |
| if module is None: |
| continue |
| for attr_name in ( |
| "ensure_no_new_facts", |
| "check_no_new_facts", |
| "validate_no_new_facts", |
| "guard_no_new_facts", |
| ): |
| monkeypatch.setattr(module, attr_name, lambda *_args, **_kwargs: True, raising=False) |
|
|
|
|
| def _call_boost(raw_prompt: str, target_harness: str, target_model: str): |
| boost = resolve_callable(BOOST_MODULES, BOOST_FUNCS) |
| return call_with_supported_kwargs( |
| boost, |
| raw_prompt=raw_prompt, |
| prompt=raw_prompt, |
| text=raw_prompt, |
| target_harness=target_harness, |
| harness=target_harness, |
| target_model=target_model, |
| model=target_model, |
| ) |
|
|
|
|
| def test_boost_persists_run_to_local_sqlite(promptboost_db_path, monkeypatch): |
| unload_promptboost_modules() |
| raw_prompt = ( |
| "Create a PromptBoost test prompt about FastAPI, local SQLite, and no paid API calls." |
| ) |
| boosted_prompt = ( |
| "Create a focused PromptBoost test prompt about FastAPI, local SQLite, " |
| "and no paid API calls." |
| ) |
|
|
| resolve_callable(BOOST_MODULES, BOOST_FUNCS) |
| install_fake_adapter(monkeypatch, boosted_prompt) |
| _patch_guards_to_pass(monkeypatch) |
|
|
| result = _call_boost(raw_prompt, target_harness="codex", target_model="test-model") |
|
|
| assert boosted_prompt.lower() in text_blob(result) |
| assert promptboost_db_path.exists() |
| assert sqlite_user_tables(promptboost_db_path) |
| assert sqlite_contains_text(promptboost_db_path, raw_prompt) |
| assert sqlite_contains_text(promptboost_db_path, boosted_prompt) |
| assert sqlite_contains_text(promptboost_db_path, "codex") |
| assert sqlite_contains_text(promptboost_db_path, "test-model") |
|
|
