eval.embeddings

chunktuner.eval.embeddings

Embedding backends — dummy for tests, LiteLLM for live runs.

DummyEmbeddingFunction

DummyEmbeddingFunction(profile_name='dummy/test')

Deterministic pseudo-embeddings for unit tests (no network).

Source code in src/chunktuner/eval/embeddings.py
def __init__(self, profile_name: str = "dummy/test"):
    self.profile_name = profile_name

LiteLLMEmbeddingFunction

LiteLLMEmbeddingFunction(
    model, api_base=None, api_key=None
)

LiteLLM-backed embeddings (calls provider APIs).

Source code in src/chunktuner/eval/embeddings.py
def __init__(
    self,
    model: str,
    api_base: str | None = None,
    api_key: str | None = None,
):
    import litellm

    self._litellm = litellm
    self.model = model
    self.api_base = api_base
    self.api_key = api_key
    self.profile_name = model