wehub-resource-sync
6c9c7fe7f3
CI / integration tests (3.13) (push) Failing after 1s
Commit lint / pull request title (push) Has been skipped
Docs / links (push) Failing after 1s
CI / unit tests (3.13) (push) Failing after 1s
CI / lint (push) Failing after 1s
CI / integration tests (push) Failing after 1s
CI / package build (push) Failing after 1s
Commit lint / commit messages (push) Failing after 1s
CI / unit tests (push) Failing after 1s
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
"""``build_embedding_provider`` — settings validation + provider build."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
from pydantic import SecretStr
|
|
|
|
from everos.component.embedding import (
|
|
OpenAIEmbeddingProvider,
|
|
build_embedding_provider,
|
|
)
|
|
from everos.config.settings import EmbeddingSettings
|
|
|
|
|
|
def test_raises_when_model_missing() -> None:
|
|
s = EmbeddingSettings(model=None, api_key=SecretStr("k"), base_url="https://x")
|
|
with pytest.raises(ValueError, match="EVEROS_EMBEDDING__MODEL"):
|
|
build_embedding_provider(s)
|
|
|
|
|
|
def test_raises_when_api_key_missing() -> None:
|
|
s = EmbeddingSettings(model="m", api_key=None, base_url="https://x")
|
|
with pytest.raises(ValueError, match="EVEROS_EMBEDDING__API_KEY"):
|
|
build_embedding_provider(s)
|
|
|
|
|
|
def test_raises_when_base_url_missing() -> None:
|
|
s = EmbeddingSettings(model="m", api_key=SecretStr("k"), base_url=None)
|
|
with pytest.raises(ValueError, match="EVEROS_EMBEDDING__BASE_URL"):
|
|
build_embedding_provider(s)
|
|
|
|
|
|
def test_builds_openai_embedding_provider_with_default_dim() -> None:
|
|
s = EmbeddingSettings(model="m", api_key=SecretStr("k"), base_url="https://x")
|
|
p = build_embedding_provider(s)
|
|
assert isinstance(p, OpenAIEmbeddingProvider)
|
|
|
|
|
|
def test_custom_dim_passes_through() -> None:
|
|
s = EmbeddingSettings(model="m", api_key=SecretStr("k"), base_url="https://x")
|
|
p = build_embedding_provider(s, dim=512)
|
|
assert isinstance(p, OpenAIEmbeddingProvider)
|
|
# Provider stores dim on a private attr; assert via the public output shape
|
|
# only if straightforward. Skip introspection if attr name differs.
|
|
if hasattr(p, "_dim"):
|
|
assert p._dim == 512
|