IgQ pddlmZddlmZmZmZmZddlZddl m Z eGddZ Gdde Z y) ) dataclass)DictIterableOptionalUnionN)ModelManagementceZdZUejed<ejed<deeejffdZdee e ffdZ e dee e fddfdZ y) SparseEmbeddingvaluesindicesreturnc4|j|jdS)Nr r r)selfs c/var/www/html/answerous/venv/lib/python3.12/site-packages/fastembed/sparse/sparse_embedding_base.py as_objectzSparseEmbedding.as_objectskk||  cvt|j|jDcic]\}}|| c}}Scc}}wN)zipr r )rivs ras_dictzSparseEmbedding.as_dicts-!$T\\4;;!?@A1@@@s 5datact|dk(r0|tjgtjgSt|j \}}|tj|tj|S)Nrr)lennparrayritems)clsrr r s r from_dictzSparseEmbedding.from_dicts] t9>bhhrlBHHRLA Atzz|,"((6*BHHW4EFFrN)__name__ __module__ __qualname__rndarray__annotations__rstrrintfloatr classmethodr!rrr r sv JJ ZZ 4RZZ0 Ac5j)AGT#u*-G2CGGrr c eZdZ ddedeedeefdZ ddeeeefdedeed ee fd Z d eed ee fd Z d eeeefd ee fdZ y)SparseTextEmbeddingBaseN model_name cache_dirthreadsc \||_||_||_|jdd|_y)Nlocal_files_onlyF)r.r/r0pop_local_files_only)rr.r/r0kwargss r__init__z SparseTextEmbeddingBase.__init__ s-%" !',>!Fr documents batch_sizeparallelr c tr)NotImplementedError)rr7r8r9r5s rembedzSparseTextEmbeddingBase.embed,s "##rtextsc+FK|j|fi|Ed{y7w)a7 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[SparseEmbedding]: The sparse embeddings. N)r<)rr=r5s r passage_embedz%SparseTextEmbeddingBase.passage_embed5s!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[SparseEmbedding]: The sparse embeddings. N) isinstancer'r<r)rr@r5s r query_embedz#SparseTextEmbeddingBase.query_embedFs^ eS !!tzz5'4V4 4 4 eX &!tzz%262 2 2 ' 5 2s!(A A*A AA A )NN)N) r"r#r$r'rr(r6rrr r<r?rCr+rrr-r-s$(!% G GC= G# G"& $hsm+,$$3- $ / " $/c]/ / "/"33 -.3 / "3rr-) dataclassesrtypingrrrrnumpyr!fastembed.common.model_managementrr r-r+rrrIs<!22= GG G*83o83r