smlflg's picture
Initial public upload from Projekte/PromptBoost (part 5)
638c806 verified
Raw History Blame Contribute Delete
833 Bytes
"""Compatibility facade for the PromptBoost core pipeline."""
from __future__ import annotations
from typing import Any
from promptboost.adapters import select_adapter
from promptboost.db import save_boost_run
from promptboost.pipeline import boost as pipeline_boost
from promptboost.pipeline import classify_task
def boost(raw_prompt: str, target_harness: str = "codex", target_model: str = "gpt", **_: Any):
adapter = select_adapter(target_harness, target_model)
result = pipeline_boost(
raw_prompt,
adapter["normalized_harness"],
adapter["normalized_model"],
)
saved = save_boost_run(result)
data = result.model_dump()
data["run_id"] = saved.id
return data
boost_prompt = boost
classify_prompt = classify_task
classify = classify_task
classify_raw_prompt = classify_task