j,TUddlmZddlmZmZddlmZddlmZ ddl m Z m Z m Z m Z mZddlmZmZn#e$rdZ dZ dZ dZ dZdZdZYnwxYwGdd ed ZGd d Ze rde jDniZeeeeejffed<e rde jDniZeeeeeffed<e rde jDn eZeeed<e rde jDniZ eeeeejffed<e rde jDniZ!eeeeejffed<erdejDniZ"eeeeejffed<dS))Any) BaseModelField) SparseVector)models) TextEmbeddingSparseTextEmbeddingImageEmbeddingLateInteractionTextEmbedding"LateInteractionMultimodalEmbedding) OnnxProvider ImageInputNceZdZUeezed<eedzed<edZ e dzed<e ee fed<eed<eed<dS) QueryResponseidN embedding)defaultsparse_embeddingmetadatadocumentscore) __name__ __module__ __qualname__strint__annotations__listfloatrrrdictrLC:\Projects\DSCons\.venv\Lib\site-packages\qdrant_client\fastembed_common.pyrrs{ c MMME{T!!!!,1E$,?,?,?lT)???38nMMM LLLLLr"rforbid)extraceZdZUdZeed<eZeeed<eZ eeed<eZ eeed<eZ eeed<eZ eeed<e defd Ze dd Ze deeeeejfffd Ze deeeeejfffd Ze deeeeejfffdZe deeeeejfffdZe deeeeefffdZe dedefdZe dedefdZe dedefdZe dedefdZe dedefdZd S) 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|jdksJt|jdksJt|jdksJt|jdksJt|jdksJd|_n#t$r d|_YnwxYw|jS)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_   $# # %              @*@BBCCaGGGG:}:<<==AAAA;~;==>>BBBBI3IKKLLqPPPPO9OQQRRUVVVVV#C   % % %$C    %sB?CC$#C$Nc2|jrdStd)Nzxfastembed is not installed. Please install it to compute embedding for document implicitly with `pip install fastembed`.)r(r3r4s r#import_fastembedzFastEmbedMisc.import_fastembedEs,    F l   r"cNtrdtjDniS)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. cNi|]"}|d|dtjjf#SmodeldimrDistanceCOSINE.0r<s r# z2FastEmbedMisc.list_text_models..Z<   guv/E F   r")rr2r7s r#list_text_modelszFastEmbedMisc.list_text_modelsPs@    *@BB       r"cNtrdtjDniS)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. cNi|]"}|d|dtjjf#Sr;r>rAs r#rCz3FastEmbedMisc.list_image_models..mrDr")r r2r7s r#list_image_modelszFastEmbedMisc.list_image_modelsbs@    +ACC       r"cNtrdtjDniS)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. cNi|]"}|d|dtjjf#Sr;r>rAs r#rCzCFastEmbedMisc.list_late_interaction_text_models..rDr")r r2r7s r#!list_late_interaction_text_modelsz/FastEmbedMisc.list_late_interaction_text_modelsus@ ,    9OQQ       r"cNtrdtjDniS)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. cNi|]"}|d|dtjjf#Sr;r>rAs r#rCzIFastEmbedMisc.list_late_interaction_multimodal_models..rDr")r r2r7s r#'list_late_interaction_multimodal_modelsz5FastEmbedMisc.list_late_interaction_multimodal_modelss@ 2    ?UWW       r"cti}tr.tjD]}|||d<|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. r<)r r2pop)r4 descriptions descriptions r#list_sparse_modelsz FastEmbedMisc.list_sparse_modelssK  E2HJJ E E 9D [__W5566r" model_namec||jvrdSd|D|_||jvrdSdS)Checks 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. Tc6h|]}|Sr!lowerrAs r# z8FastEmbedMisc.is_supported_text_model..s NNNeEKKMMNNNr"F)rYr)rEr4rTs r#is_supported_text_modelz%FastEmbedMisc.is_supported_text_modelsi     !1 1 14NNs7K7K7M7MNNN     !1 1 14ur"c||jvrdSd|D|_||jvrdSdS)rVTc6h|]}|Sr!rXrAs r#rZz9FastEmbedMisc.is_supported_image_model..s PPPuU[[]]PPPr"F)rYr*rHr[s r#is_supported_image_modelz&FastEmbedMisc.is_supported_image_modelsi     !2 2 24PP8M8M8O8OPPP     !2 2 24ur"c||jvrdSd|D|_||jvrdSdS)rVTc6h|]}|Sr!rXrAs r#rZzIFastEmbedMisc.is_supported_late_interaction_text_model..s-- - - #EKKMM- - - r"F)rYr+rKr[s r#(is_supported_late_interaction_text_modelz6FastEmbedMisc.is_supported_late_interaction_text_modelsv     !B B B4- - '*'L'L'N'N- - - )     !B B B4ur"c||jvrdSd|D|_||jvrdSdS)rVTc6h|]}|Sr!rXrAs r#rZzOFastEmbedMisc.is_supported_late_interaction_multimodal_model..s-3 3 3 #EKKMM3 3 3 r"F)rYr,rNr[s r#.is_supported_late_interaction_multimodal_modelz.s RRRekkmmRRRr"F)rYr-rSr[s r#is_supported_sparse_modelz'FastEmbedMisc.is_supported_sparse_modelsi     !3 3 34RR9O9O9Q9QRRR     !3 3 34ur")r.N) rrrr(boolrsetr)rr*r+r,r- classmethodr5r8r tuplerrr?rErHrKrNrrSr\r_rbrerhr!r"r#r'r'$sL$ SUUL#c("""!ceeM3s8###.1cee!3s833347CEE'S999"suuNCH$$$ T   [ 0   [  c5fo1E+F&F!G   [ " $sE#v2F,G'G"H   [ $ $sE#vBVrAs r#rCrC<  guv'=>r"SUPPORTED_EMBEDDING_MODELSc i|] }|d| S)r<r!rAs r#rCrCsTTTuU7^UTTTr"!SUPPORTED_SPARSE_EMBEDDING_MODELScJh|] }|dd|d!S) requires_idfNr<)get)rB model_configs r#rZrZ sC    ND 1 1Wr"IDF_EMBEDDING_MODELScNi|]"}|d|dtjjf#Sr;r>rAs r#rCrC*rnr""_LATE_INTERACTION_EMBEDDING_MODELScNi|]"}|d|dtjjf#Sr;r>rAs r#rCrC3rnr"_IMAGE_EMBEDDING_MODELScNi|]"}|d|dtjjf#Sr;r>rAs r#rCrC<rnr"-_LATE_INTERACTION_MULTIMODAL_EMBEDDING_MODELS)#typingrpydanticrr&qdrant_client.conversions.common_typesrqdrant_client.httprr0rr r r r fastembed.commonr rr3rr'r2ror rrlrr?rrqrjrvrxrzr|r!r"r#rs%%%%%%%%??????%%%%%%:99999999MN#' )-&LJJJIXffffffffb  8]8:: DeC,@&A!AB TT(Q(;(Q(S(STTTT "4T#s(^(;#< E/EGG c#h$  G1GII #DeC4H.I)I$J  9^9;; c5fo)=#>>?*  M7MOO .tCsFO?S9T4T/Us4A  A