codekingpro/portable-devtools

1from typing import Any2 3from loguru import logger4 5from fastembed import TextEmbedding6 7logger.warning(8    "DefaultEmbedding, FlagEmbedding, JinaEmbedding are deprecated."9    "Use from fastembed import TextEmbedding instead."10)11 12DefaultEmbedding = TextEmbedding13FlagEmbedding = TextEmbedding14 15 16class JinaEmbedding(TextEmbedding):17    def __init__(18        self,19        model_name: str = "jinaai/jina-embeddings-v2-base-en",20        cache_dir: str | None = None,21        threads: int | None = None,22        **kwargs: Any,23    ):24        super().__init__(model_name, cache_dir, threads, **kwargs)25