38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
|
# This file is auto-generated by alef — DO NOT EDIT.
|
||
|
|
# alef:hash:4e15143f4af1ae8bafbdb1506ef057da924484c66a19483966333558ad437e75
|
||
|
|
# To regenerate: alef generate
|
||
|
|
# To verify freshness: alef verify --exit-code
|
||
|
|
# Issues & docs: https://github.com/kreuzberg-dev/alef
|
||
|
|
|
||
|
|
"""E2e tests for category: embed_extra."""
|
||
|
|
|
||
|
|
import pytest # noqa: F401
|
||
|
|
from kreuzberg import embed_texts, ExtractionConfig
|
||
|
|
|
||
|
|
|
||
|
|
def _alef_e2e_text(value: object) -> str:
|
||
|
|
return "" if value is None else str(value)
|
||
|
|
|
||
|
|
|
||
|
|
def _alef_e2e_item_texts(item: object) -> tuple[str, ...]:
|
||
|
|
raw_items = getattr(item, "items", None)
|
||
|
|
items_text = " ".join(str(value) for value in raw_items) if isinstance(raw_items, list) else ""
|
||
|
|
return (
|
||
|
|
_alef_e2e_text(item),
|
||
|
|
_alef_e2e_text(getattr(item, "kind", None)),
|
||
|
|
_alef_e2e_text(getattr(item, "name", None)),
|
||
|
|
_alef_e2e_text(getattr(item, "source", None)),
|
||
|
|
_alef_e2e_text(getattr(item, "alias", None)),
|
||
|
|
_alef_e2e_text(getattr(item, "text", None)),
|
||
|
|
_alef_e2e_text(getattr(item, "signature", None)),
|
||
|
|
items_text,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_embed_texts_batch() -> None:
|
||
|
|
"""Batch embed texts."""
|
||
|
|
texts = ["Hello", "World"]
|
||
|
|
config = ExtractionConfig(model={"name": "balanced", "type": "preset"})
|
||
|
|
|
||
|
|
_ = embed_texts(texts, config)
|