prefpo/__init__.py at main · DistylAI/prefpo · 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
"""PrefPO — Preference-based Prompt Optimization."""

from prefpo.config import (
    DiscriminatorConfig,
    ModelConfig,
    OptimizerConfig,
    PrefPOConfig,
)
from prefpo.generate import generate_outputs, generate_standalone
from prefpo.grading.base import GradeResult, Grader
from prefpo.judges.hack import judge_prompt_hack
from prefpo.judges.hygiene import judge_prompt_hygiene
from prefpo.llm.client import LLMResponse, call_llm, call_llm_json
from prefpo.optimize import (
    MultiTrialResult,
    OptimizationResult,
    optimize,
    optimize_async,
    optimize_multi_trial,
)
from prefpo.types import Prompt, PromptRole, Sample

__all__ = [
    "optimize",
    "optimize_async",
    "optimize_multi_trial",
    "PrefPOConfig",
    "ModelConfig",
    "DiscriminatorConfig",
    "OptimizerConfig",
    "OptimizationResult",
    "MultiTrialResult",
    "Prompt",
    "PromptRole",
    "Sample",
    "Grader",
    "GradeResult",
    "generate_outputs",
    "generate_standalone",
    "call_llm",
    "call_llm_json",
    "LLMResponse",
    "judge_prompt_hack",
    "judge_prompt_hygiene",
]