o :ib+@sUddlmZddlmZmZddlmZddlmZzddl m Z m Z m Z m Z mZddlmZmZWneyJdZ dZ dZ dZ dZdZdZYnwGdd d ed d ZGd d d Ze rgdde DniZeeeeejffed<e rdde DniZeeeeeffed<e rdde DneZeeed<e rdde DniZ eeeeejffed<e rdde DniZ!eeeeejffed<erddeDniZ"eeeeejffed<dS))Any) BaseModelField) SparseVector)models) TextEmbeddingSparseTextEmbeddingImageEmbeddingLateInteractionTextEmbedding"LateInteractionMultimodalEmbedding) OnnxProvider ImageInputNc@s`eZdZUeeBed<eedBed<eddZ e dBed<e ee fed<eed<eed<dS) QueryResponseidN embedding)defaultsparse_embeddingmetadatadocumentscore) __name__ __module__ __qualname__strint__annotations__listfloatrrrdictrrrk/home/aiteam/pcaa-dev/dashboard-backend/venv/lib/python3.10/site-packages/qdrant_client/fastembed_common.pyrs  rforbid)extrac@seZdZUdZeed<eZeeed<eZ eeed<eZ eeed<eZ eeed<eZ eeed<e defd d Ze d#d d Ze deeeeejfffddZe deeeeejfffddZe deeeeejfffddZe deeeeejfffddZe deeeeefffddZe dedefddZe dedefddZe dedefddZe dedefdd Ze dedefd!d"Zd S)$ FastEmbedMiscF IS_INSTALLED _TEXT_MODELS _IMAGE_MODELS_LATE_INTERACTION_TEXT_MODELS#_LATE_INTERACTION_MULTIMODAL_MODELS_SPARSE_MODELSreturncCs|jr|jSzGddlm}m}m}m}m}t|dksJt|dks)Jt|dks3Jt|dks=Jt|dksGJd|_W|jSt y[d|_Y|jSw)Nr)rrr r r TF) r$ fastembedrrr r r lenlist_supported_models ImportError)clsrrr r r rrr is_installed,s zFastEmbedMisc.is_installedNcCs|jrdStd)Nzjfastembed is not installed. Please install it to enable fast vector indexing with `pip install fastembed`.)r$r.r/rrr import_fastembedEs zFastEmbedMisc.import_fastembedcCtr ddtDSiS)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. cS"i|] }|d|dtjjfqSmodeldimrDistanceCOSINE.0r6rrr Zz2FastEmbedMisc.list_text_models..)rr-r1rrr list_text_modelsPszFastEmbedMisc.list_text_modelscCr3)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. cSr4r5r8r;rrr r=mr>z3FastEmbedMisc.list_image_models..)r r-r1rrr list_image_modelsbszFastEmbedMisc.list_image_modelscCr3)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. cSr4r5r8r;rrr r=r>zCFastEmbedMisc.list_late_interaction_text_models..)r r-r1rrr !list_late_interaction_text_modelsuz/FastEmbedMisc.list_late_interaction_text_modelscCr3)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. cSr4r5r8r;rrr r=r>zIFastEmbedMisc.list_late_interaction_multimodal_models..)r r-r1rrr 'list_late_interaction_multimodal_modelsrBz5FastEmbedMisc.list_late_interaction_multimodal_modelscCs(i}trtD] }|||d<q|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. r6)rr-pop)r/ descriptions descriptionrrr list_sparse_modelss   z FastEmbedMisc.list_sparse_models model_namecC<||jvr dSdd|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. TcSh|]}|qSrlowerr;rrr z8FastEmbedMisc.is_supported_text_model..F)rMr%r?r/rHrrr is_supported_text_model  z%FastEmbedMisc.is_supported_text_modelcCrI)rJTcSrKrrLr;rrr rNrOz9FastEmbedMisc.is_supported_image_model..F)rMr&r@rPrrr is_supported_image_modelrRz&FastEmbedMisc.is_supported_image_modelcCrI)rJTcSrKrrLr;rrr rNzIFastEmbedMisc.is_supported_late_interaction_text_model..F)rMr'rArPrrr (is_supported_late_interaction_text_model z6FastEmbedMisc.is_supported_late_interaction_text_modelcCrI)rJTcSrKrrLr;rrr rNrTzOFastEmbedMisc.is_supported_late_interaction_multimodal_model..F)rMr(rCrPrrr .is_supported_late_interaction_multimodal_modelrVz.F)rMr)rGrPrrr is_supported_sparse_modelrRz'FastEmbedMisc.is_supported_sparse_model)r*N) rrrr$boolrsetr%rr&r'r(r) classmethodr0r2rtuplerrr9r?r@rArCrrGrQrSrUrWrXrrrr r#$s>   """" r#cCr4r5r8r;rrr r=r>r=SUPPORTED_EMBEDDING_MODELScCsi|]}|d|qS)r6rr;rrr r=s!SUPPORTED_SPARSE_EMBEDDING_MODELScCs h|] }|ddr|dqS) requires_idfNr6)get)r< model_configrrr rN s  rNIDF_EMBEDDING_MODELScCr4r5r8r;rrr r=*r>"_LATE_INTERACTION_EMBEDDING_MODELScCr4r5r8r;rrr r=3r>_IMAGE_EMBEDDING_MODELScCr4r5r8r;rrr r=<r>-_LATE_INTERACTION_MULTIMODAL_EMBEDDING_MODELS)#typingrpydanticrr&qdrant_client.conversions.common_typesrqdrant_client.httprr+rrr r r fastembed.commonr r r.rr#r-r]rrr\rr9rr^rZrbrcrdrerrrr sj     r