mirror of
https://github.com/qdrant/fastembed.git
synced 2026-08-05 09:31:10 -05:00
bb815405aa
* new: Add mypy and pyright deps * chore: Add any to kwargs * chore: Add any to args * add missing kwargs --------- Co-authored-by: George Panchuk <george.panchuk@qdrant.tech>
25 lines
614 B
Python
25 lines
614 B
Python
from typing import Optional, Any
|
|
|
|
from loguru import logger
|
|
|
|
from fastembed import TextEmbedding
|
|
|
|
logger.warning(
|
|
"DefaultEmbedding, FlagEmbedding, JinaEmbedding are deprecated."
|
|
"Use from fastembed import TextEmbedding instead."
|
|
)
|
|
|
|
DefaultEmbedding = TextEmbedding
|
|
FlagEmbedding = TextEmbedding
|
|
|
|
|
|
class JinaEmbedding(TextEmbedding):
|
|
def __init__(
|
|
self,
|
|
model_name: str = "jinaai/jina-embeddings-v2-base-en",
|
|
cache_dir: Optional[str] = None,
|
|
threads: Optional[int] = None,
|
|
**kwargs: Any,
|
|
):
|
|
super().__init__(model_name, cache_dir, threads, **kwargs)
|