"""Public API surface for codeflash-python."""

from __future__ import annotations

from importlib.metadata import version as _get_version

from ._model import FunctionParent, FunctionSource, FunctionToOptimize
from .api import OptimizationConfig, OptimizationSession, optimize_function

try:
    __version__: str = _get_version("codeflash-python")
except Exception:  # noqa: BLE001
    __version__ = "0.0.0"

__all__ = [
    "FunctionParent",
    "FunctionSource",
    "FunctionToOptimize",
    "OptimizationConfig",
    "OptimizationSession",
    "__version__",
    "optimize_function",
]
