mirror of
https://github.com/qdrant/fastembed.git
synced 2026-07-23 11:20:51 -05:00
b389798d8e
* Migration of models to dataclasses * Model description file * Test fix * kw_only support * Multitask embeddings test fix * list_supported_models type fix * Dim fix for sparsemodels * Dim fix for sparsemodels * Dim fix for sparsemodels (x2) * Model management type fix * Interface docstring fixes * Mypy fixes * Typing fix again * Typing fix again * Special cast to SparseModelDescription * Special cast to SparseModelDescription * Special cast to SparseModelDescription * typing fix for colpali * typing fix for colpali * typing fix for colpali * typing fix for colpali * Let's try generic typing for ModelManagment * wip: dataclass idea, small fixes (#475) * wip: dataclass idea, small fixes * fix: fix exception message in base model description * remove custom model descriptions * make license, description and size in gb mandatory in model description * fix: introduce _list_supported_models which returns model description objects * test: add test for list supported models * fix: fix list supported models usage in tests --------- Co-authored-by: George <george.panchuk@qdrant.tech>
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from fastembed import (
|
|
TextEmbedding,
|
|
SparseTextEmbedding,
|
|
ImageEmbedding,
|
|
LateInteractionMultimodalEmbedding,
|
|
LateInteractionTextEmbedding,
|
|
)
|
|
|
|
|
|
def test_text_list_supported_models():
|
|
for model_type in [
|
|
TextEmbedding,
|
|
SparseTextEmbedding,
|
|
ImageEmbedding,
|
|
LateInteractionMultimodalEmbedding,
|
|
LateInteractionTextEmbedding,
|
|
]:
|
|
supported_models = model_type.list_supported_models()
|
|
assert isinstance(supported_models, list)
|
|
description = supported_models[0]
|
|
assert isinstance(description, dict)
|
|
|
|
assert "model" in description and description["model"]
|
|
if model_type != SparseTextEmbedding:
|
|
assert "dim" in description and description["dim"]
|
|
assert "license" in description and description["license"]
|
|
assert "size_in_GB" in description and description["size_in_GB"]
|
|
assert "model_file" in description and description["model_file"]
|
|
assert "sources" in description and description["sources"]
|
|
assert "hf" in description["sources"] or "url" in description["sources"]
|