IgpBddlmZmZmZddlZddlmZGddeZy))IterableOptionalUnionN)ModelManagementc eZdZ ddedeedeefdZ ddeeeefdedeed ee jfd Z d eed ee jfd Z d eeeefd ee jfdZ y)TextEmbeddingBaseN model_name cache_dirthreadsc \||_||_||_|jdd|_y)Nlocal_files_onlyF)r r r pop_local_files_only)selfr r r kwargss _/var/www/html/answerous/venv/lib/python3.12/site-packages/fastembed/text/text_embedding_base.py__init__zTextEmbeddingBase.__init__ s-%" !',>!F documents batch_sizeparallelreturnc t)N)NotImplementedError)rrrrrs rembedzTextEmbeddingBase.embeds "##rtextsc+FK|j|fi|Ed{y7w)a+ Embeds a list of text passages into a list of embeddings. Args: texts (Iterable[str]): The list of texts to embed. **kwargs: Additional keyword argument to pass to the embed method. Yields: Iterable[np.ndarray]: The embeddings. N)r)rrrs r passage_embedzTextEmbeddingBase.passage_embeds!4::e.v...s !!queryc+Kt|tr|j|gfi|Ed{t|tr|j|fi|Ed{yy717w)z Embeds queries Args: query (Union[str, Iterable[str]]): The query to embed, or an iterable e.g. list of queries. Returns: Iterable[np.ndarray]: The embeddings. N) isinstancestrrr)rrrs r query_embedzTextEmbeddingBase.query_embed-s^ eS !!tzz5'4V4 4 4 eX &!tzz%262 2 2 ' 5 2s!(A A*A AA A )NN)N)__name__ __module__ __qualname__r"rintrrrnpndarrayrrr#rrrrs$(!% G GC= G# G"& $hsm+,$$3- $ "**  $ /8C= /x ?S /33 -.3 "** 3rr) typingrrrnumpyr)!fastembed.common.model_managementrrr+rrr/s,,=6363r