^Nj  UddlmZmZddlmZddlmZmZddlm Z m Z e ddddd e d d d g gZ e e e d<GddeZGddeZy))AnyType) NumpyArray)ColbertColbertEmbeddingWorker)DenseModelDescription ModelSourcezjinaai/jina-colbert-v2ziNew model that expands capabilities of colbert-v1 with multilingual and context length of 8192, 2024 yearz cc-by-nc-4.0gQ@)hfzonnx/model.onnxzonnx/model.onnx_data)modeldim descriptionlicense size_in_GBsources model_fileadditional_filessupported_jina_colbert_modelsc eZdZdZdZdZdZedee fdZ ede e fdZ d deeefd ed edeeefffd ZxZS) JinaColbertiizreturnctS)N)JinaColbertEmbeddingWorkerclss x/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastembed/late_interaction/jina_colbert.py_get_worker_classzJinaColbert._get_worker_classs))ctS)zLists the supported models. Returns: list[DenseModelDescription]: A list of DenseModelDescription objects containing the model information. )rrs r_list_supported_modelsz"JinaColbert._list_supported_modelss -,r onnx_inputis_dockwargsc <t|||}|sd|ddd|S)Nattention_mask)super_preprocess_onnx_input)selfr"r#r$ __class__s rr)z"JinaColbert._preprocess_onnx_input(s1W3JG ./J' ( +r)T)__name__ __module__ __qualname__QUERY_MARKER_TOKEN_IDDOCUMENT_MARKER_TOKEN_IDMIN_QUERY_LENGTH MASK_TOKEN classmethodrrrlistrr!dictstrrboolrr) __classcell__)r+s@rrrs"%J*$'=">**-t,A'B--AEsJ/9=PS c:o rrc$eZdZdedededefdZy)r model_name cache_dirr$rc td||dd|S)Nr&)r:r;threads)r)r*r:r;r$s rinit_embeddingz)JinaColbertEmbeddingWorker.init_embedding4s' !    rN)r,r-r.r6rrr?r>rrrr3s$    P[ rrN)typingrrfastembed.common.typesr"fastembed.late_interaction.colbertrr"fastembed.common.model_descriptionrr rr4__annotations__rrr>rrrEsm-NQ& 78$01  >t$9: '< !7 r