sOi(UddlmZmZmZddlmZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZedd d d d ed dddedddd deddedddd deddedddd d edd!ed"dd#d d$ed"ded%dd&d d$ed%ded'dd(d d$ed'ded)dd*d+d,ed)ded-d.d/d+d0ed1dg Zeeed2<Gd3d4eZGd5d6e Zy7)8)AnyIterableType) NumpyArray)OnnxOutputContext) normalize)OnnxTextEmbeddingOnnxTextEmbeddingWorker)PooledEmbedding)DenseModelDescription ModelSourcez&sentence-transformers/all-MiniLM-L6-v2izText embeddings, Unimodal (text), English, 256 input tokens truncation, Prefixes for queries/documents: not necessary, 2021 year.z apache-2.0g ףp= ?z]https://storage.googleapis.com/qdrant-fastembed/sentence-transformers-all-MiniLM-L6-v2.tar.gzzqdrant/all-MiniLM-L6-v2-onnxT)urlhf_deprecated_tar_structz model.onnx)modeldim descriptionlicense size_in_GBsources model_filez!jinaai/jina-embeddings-v2-base-enizText embeddings, Unimodal (text), English, 8192 input tokens truncation, Prefixes for queries/documents: not necessary, 2023 year.gp= ף?z!xenova/jina-embeddings-v2-base-en)rzonnx/model.onnxz"jinaai/jina-embeddings-v2-small-enigQ?z"xenova/jina-embeddings-v2-small-enz!jinaai/jina-embeddings-v2-base-dezText embeddings, Unimodal (text), Multilingual (German, English), 8192 input tokens truncation, Prefixes for queries/documents: not necessary, 2024 year.g{Gz?zonnx/model_fp16.onnxz#jinaai/jina-embeddings-v2-base-codezText embeddings, Unimodal (text), Multilingual (English, 30 programming languages), 8192 input tokens truncation, Prefixes for queries/documents: not necessary, 2024 year.g{Gz?z!jinaai/jina-embeddings-v2-base-zhzText embeddings, Unimodal (text), supports mixed Chinese-English input text, 8192 input tokens truncation, Prefixes for queries/documents: not necessary, 2024 year.z!jinaai/jina-embeddings-v2-base-eszText embeddings, Unimodal (text), supports mixed Spanish-English input text, 8192 input tokens truncation, Prefixes for queries/documents: not necessary, 2024 year.zthenlper/gte-basezGeneral text embeddings, Unimodal (text), supports English only input text, 512 input tokens truncation, Prefixes for queries/documents: not necessary, 2024 year.mitg)\(?zthenlper/gte-largeizText embeddings, Unimodal (text), English, 512 input tokens truncation, Prefixes for queries/documents: not necessary, 2023 year.g333333?zqdrant/gte-large-onnx"supported_pooled_normalized_modelsc^eZdZedeefdZedeefdZ de de de e fdZy)PooledNormalizedEmbeddingreturnctS)N)PooledNormalizedEmbeddingWorkerclss q/volume1/homes/robertsu/coba/app/.venv/lib/python3.12/site-packages/fastembed/text/pooled_normalized_embedding.py_get_worker_classz+PooledNormalizedEmbedding._get_worker_classs..ctS)zLists the supported models. Returns: list[DenseModelDescription]: A list of DenseModelDescription objects containing the model information. )rrs r!_list_supported_modelsz0PooledNormalizedEmbedding._list_supported_modelss 21r#outputkwargsc |j td|j}|j}t|j ||S)Nz"?//2t,A'B22C'C36C * Cr#rc$eZdZdedededefdZy)r model_name cache_dirr'rc td||dd|S)N)r8r9threadsr6)r)r-r8r9r's r!init_embeddingz.PooledNormalizedEmbeddingWorker.init_embeddings) ) !    r#N)r1r2r3strrr r=r6r#r!rrs-          r#rN)typingrrrfastembed.common.typesrfastembed.common.onnx_modelrfastembed.common.utilsrfastembed.text.onnx_embeddingr r fastembed.text.pooled_embeddingr "fastembed.common.model_descriptionr r rr5__annotations__rrr6r#r!rGs&&.9,T;Q6  Ho-#'   1  HBC$ 2  HCD$ 1  HBC) 3  fDE$ 1  fBC$ 1  fBC$ !  e23$ "  H67 KqC"D)>$?qhCC2  &=  r#