Bip+UddlmZddlmZmZddlmZddlmZ ddl m Z m Z m Z m Z mZddlmZmZGdd ed ZGd d Ze r<e j*Dcic]!}|d|dej,j.f#c}niZeeeeej,ffed<e r#e j*Dcic]}|d| c}niZeeeeeffed<e r4e j*Dchc]}|j?ddr|dc}ne Z!e eed<e r<e j*Dcic]!}|d|dej,j.f#c}niZ"eeeeej,ffed<e r<e j*Dcic]!}|d|dej,j.f#c}niZ#eeeeej,ffed<er<ej*Dcic]!}|d|dej,j.f#c}niZ$eeeeej,ffed<y#e$rdZ dZ dZ dZ dZdZdZY wxYwcc}wcc}wcc}wcc}wcc}wcc}w))Any) BaseModelField) SparseVector)models) TextEmbeddingSparseTextEmbeddingImageEmbeddingLateInteractionTextEmbedding"LateInteractionMultimodalEmbedding) OnnxProvider ImageInputNc~eZdZUeezed<eedzed<edZ e dzed<e ee fed<eed<eed<y) QueryResponseidN embedding)defaultsparse_embeddingmetadatadocumentscore) __name__ __module__ __qualname__strint__annotations__listfloatrrrdictr`/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/qdrant_client/fastembed_common.pyrrsG c ME{T!!,1$,?lT)?38nM Lr"rforbid)extrac~eZdZUdZeed<eZeeed<eZ eeed<eZ eeed<eZ eeed<eZ eeed<e defd Ze dd Ze deeeeej(fffd Ze deeeeej(fffd Ze deeeeej(fffdZe deeeeej(fffdZe deeeeefffdZe dedefdZe dedefdZe dedefdZe dedefdZe dedefdZy ) FastEmbedMiscF IS_INSTALLED _TEXT_MODELS _IMAGE_MODELS_LATE_INTERACTION_TEXT_MODELS#_LATE_INTERACTION_MULTIMODAL_MODELS_SPARSE_MODELSreturnc|jr |jS ddlm}m}m}m}m}t|jdkDsJt|jdkDsJt|jdkDsJt|jdkDsJt|jdkDsJd|_|jS#t$rd|_Y|jSwxYw)Nr)r rr r r TF) r( fastembedr rr r r lenlist_supported_models ImportError)clsr rr r r s r# is_installedzFastEmbedMisc.is_installed,s   ## # %  @*@@BCaG GG:}::<=A AA;~;;=>B BBI3IIKLqP PPO9OOQRUVV VV#C  %$C  %sB0CC43C4Nc2|jrytd)Nzxfastembed is not installed. Please install it to compute embedding for document implicitly with `pip install fastembed`.)r(r3)r4s r#import_fastembedzFastEmbedMisc.import_fastembedEs#     l  r"ctrCtjDcic]%}|d|dtjjf'c}SiScc}w)aLists the supported dense text models. Requires invocation of TextEmbedding.list_supported_models() to support custom models. Returns: dict[str, tuple[int, models.Distance]]: A dict of model names, their dimensions and distance metrics. modeldim)rr2rDistanceCOSINEr4r9s r#list_text_modelszFastEmbedMisc.list_text_modelsPsX+@@B guv/E/E FF     *A ctrCtjDcic]%}|d|dtjjf'c}SiScc}w)atLists the supported image dense models. Custom image models are not supported yet, but calls to ImageEmbedding.list_supported_models() is done each time in order for preserving the same style as with TextEmbedding. Returns: dict[str, tuple[int, models.Distance]]: A dict of model names, their dimensions and distance metrics. r9r:)r r2rr;r<r=s r#list_image_modelszFastEmbedMisc.list_image_modelsbsX,AAC guv/E/E FF     r?ctrCtjDcic]%}|d|dtjjf'c}SiScc}w)aLists the supported late interaction text models. Custom late interaction models are not supported yet, but calls to LateInteractionTextEmbedding.list_supported_models() is done each time in order for preserving the same style as with TextEmbedding. Returns: dict[str, tuple[int, models.Distance]]: A dict of model names, their dimensions and distance metrics. r9r:)r r2rr;r<r=s r#!list_late_interaction_text_modelsz/FastEmbedMisc.list_late_interaction_text_modelsusX ,:OOQ guv/E/E FF     r?ctrCtjDcic]%}|d|dtjjf'c}SiScc}w)aLists the supported late interaction multimodal models. Custom late interaction multimodal models are not supported yet, but calls to LateInteractionMultimodalEmbedding.list_supported_models() is done each time in order for preserving the same style as with TextEmbedding. Returns: dict[str, tuple[int, models.Distance]]: A dict of model names, their dimensions and distance metrics. r9r:)r r2rr;r<r=s r#'list_late_interaction_multimodal_modelsz5FastEmbedMisc.list_late_interaction_multimodal_modelssY 2@UUW guv/E/E FF     r?cpi}tr-tjD]}|||jd<|S)a`Lists the supported sparse models. Custom sparse models are not supported yet, but calls to SparseTextEmbedding.list_supported_models() is done each time in order for preserving the same style as with TextEmbedding. Returns: dict[str, dict[str, Any]]: A dict of model names and their descriptions. r9)r r2pop)r4 descriptions descriptions r#list_sparse_modelsz FastEmbedMisc.list_sparse_modelss? 2HHJ E 9D [__W56 Er" model_namec|j|jvry|jDchc]}|jc}|_|j|jvryycc}wzChecks if the model is supported by fastembed. Args: model_name (str): The name of the model to check. Returns: bool: True if the model is supported, False otherwise. TF)lowerr)r>r4rKr9s r#is_supported_text_modelz%FastEmbedMisc.is_supported_text_modelsc    !1!1 17:7K7K7MNeEKKMN    !1!1 1OA-c|j|jvry|jDchc]}|jc}|_|j|jvryycc}wrM)rNr*rArOs r#is_supported_image_modelz&FastEmbedMisc.is_supported_image_modelsc    !2!2 28;8M8M8OPuU[[]P    !2!2 2QrQc|j|jvry|jDchc]}|jc}|_|j|jvryycc}wrM)rNr+rCrOs r#(is_supported_late_interaction_text_modelz6FastEmbedMisc.is_supported_late_interaction_text_modelsk    !B!B B(+'L'L'N- #EKKM- )    !B!B B - rQc|j|jvry|jDchc]}|jc}|_|j|jvryycc}wrM)rNr,rErOs r#.is_supported_late_interaction_multimodal_modelzrArCrErrJrPrSrUrWrYr!r"r#r'r'$s/L$ UL#c("!eM3s8#.1e!3s8347E'S9"uNCH$ T  0   c5foo1E+F&F!G  " $sE#v2F,G'G"H  $ $sE#vBVrls%?%:IXffb9]88:  guv'='=>> DeC,@&A!AB)R(;(Q(Q(STuU7^UT "4T#s(^(;#<F/EEG    ND 1 W c#h$H1GGI  guv'='=>> #DeC4H.I)I$J:^99;  guv'='=>> c5foo)=#>>?*N7MMO  guv'='=>> .tCsFOO?S9T4T/UU MN#' )-&LJ@U s5H*&I8 I .I3&I &I%&I*II