jddlmZmZmZmZddlmZddlZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZGddeZdS))AnyIterableSequenceget_args)deepcopyN) BaseModel)grpc)AsyncQdrantBase) ModelEmbedder)models) common_types) GrpcToRest)INFERENCE_OBJECT_TYPES)ModelSchemaParser) FastEmbedMiscc eZdZUdZeed<dedeffd Zede e e e e jfffdZede e e e e jfffdZede e e e e jfffd Zede e e e e jfffd Zede e e e efffd Zed e de e e jffd Zd e de fdZedejeezeeezejzejzejze jze jze j zdzde jdzfdZ!de j"de j"fdZ#de$e j"de$e j"fdZ% dde&e'e&zdede dzde'e&fdZ( dde'e e e&fe&zde dzde dzde'e&fdZ)xZ*S)AsyncQdrantFastembedMixin_FASTEMBED_INSTALLEDparser is_local_modectj|j_t |||_t dS)N)rr)r is_installed __class__rr _model_embeddersuper__init__)selfrrrs RC:\Projects\DSCons\.venv\Lib\site-packages\qdrant_client\async_qdrant_fastembed.pyrz"AsyncQdrantFastembedMixin.__init__sG.;.H.J.J+,F-XXX returnc,KtjS)zLists the supported dense text models. Returns: dict[str, tuple[int, models.Distance]]: A dict of model names, their dimensions and distance metrics. )rlist_text_modelsclss rr#z*AsyncQdrantFastembedMixin.list_text_models$s-///r c,KtjS)zLists the supported image dense models. Returns: dict[str, tuple[int, models.Distance]]: A dict of model names, their dimensions and distance metrics. )rlist_image_modelsr$s rr'z+AsyncQdrantFastembedMixin.list_image_models-s.000r c,KtjS)zLists the supported late interaction text models. Returns: dict[str, tuple[int, models.Distance]]: A dict of model names, their dimensions and distance metrics. )r!list_late_interaction_text_modelsr$s rr)z;AsyncQdrantFastembedMixin.list_late_interaction_text_models6s>@@@r c,KtjS)zLists the supported late interaction multimodal models. Returns: dict[str, tuple[int, models.Distance]]: A dict of model names, their dimensions and distance metrics. )r'list_late_interaction_multimodal_modelsr$s rr+zAAsyncQdrantFastembedMixin.list_late_interaction_multimodal_models?sDFFFr c,KtjS)zLists the supported sparse text models. Returns: dict[str, dict[str, Any]]: A dict of model names and their descriptions. )rlist_sparse_modelsr$s rr-z,AsyncQdrantFastembedMixin.list_sparse_modelsJs/111r model_namecdtjtjtjtjtjfD]}||x}r|cS|tjvrtdtd|)NzFSparse embeddings do not return fixed embedding size and distance typezUnsupported embedding model: ) rimport_fastembedr#r'r)r+getr- ValueError)r%r. descriptionsparamss r_get_model_paramsz+AsyncQdrantFastembedMixin._get_model_paramsSs&(((  * , ,  + - -  ; = =  A C C    L &))*555v    9;; ; ;X EEEFFFr c8||\}}|S)aoGet the size of the embeddings produced by the specified model. Args: model_name: the name of the model to get the embedding size for. Returns: int: the size of the embeddings produced by the model. Raises: ValueError: If sparse model name is passed or model is not found in the supported models. )r.)r5)rr.embeddings_size_s rget_embedding_sizez,AsyncQdrantFastembedMixin.get_embedding_sizeds% $555LL!r queryNct|ttjr|St|tjrt j|St|tjr't j| St|trt j|St|ttj rEt|tj rtj|n|}t j|St|ttrt j|S|dSt!dt#|)acResolves query interface into a models.Query object Args: query: models.QueryInterface - query as a model or a plain structure like list[float] Returns: Optional[models.Query]: query as it was, models.Query(nearest=query) or None Raises: ValueError: if query is not of supported type )nearestNzUnsupported query type: ) isinstancertypesQuery SparseVectorr NearestQuerynpndarraytolistlistPointIdr rconvert_point_idrr2type)r%r:s r_resolve_queryz(AsyncQdrantFastembedMixin._resolve_queryrsV2 eXek22 3 3 L eU/ 0 0 6&u555 5 eRZ ( ( ?&u||~~>>> > eT " " 6&u555 5 eXem44 5 5 66@ 6U6U` +E222[` &u555 5 eX&<== > > 6&u555 5 =4ADKKAABBBr cbt|}||j|_|S)zResolve QueryRequest query field Args: query: models.QueryRequest - query request to resolve Returns: models.QueryRequest: A deepcopy of the query request with resolved query field )rrIr:)rr:s r_resolve_query_requestz0AsyncQdrantFastembedMixin._resolve_query_requests+))%+66  r requestsc fd|DS)aResolve query field for each query request in a batch Args: requests: Sequence[models.QueryRequest] - query requests to resolve Returns: Sequence[models.QueryRequest]: A list of deep copied query requests with resolved query fields c:g|]}|S)rK).0r:rs r zJAsyncQdrantFastembedMixin._resolve_query_batch_request..s'IIIu++E22IIIr rO)rrLs` r_resolve_query_batch_requestz6AsyncQdrantFastembedMixin._resolve_query_batch_requests JIIIIIIIr F raw_modelsis_query batch_sizec#^K|j|||p|jEd{VdS)N)rSrTrU)r embed_modelsDEFAULT_BATCH_SIZE)rrSrTrUs r _embed_modelsz'AsyncQdrantFastembedMixin._embed_modelssa '44!!rFrEfloatr@r? NumpyArrayDocumentImageInferenceObjectrI QueryRequestrKrrRrrrYr] __classcell__)rs@rrrs0 0tCsFO7K1L,L'M000[01S%V_8L2M-M(N111[1AS%V_H\B]=]8^AAA[AG c5fo-.. /GGG[G2c4S>.A)B222[2G3G5fo9M3NGGG[G S S    )C} u+ tE{    +     /  ,  !  )C  )C)C)C[)CV F,? FDW     J !45 J &% & J J J J !%    33    $J  )      "&#   T#y.1I=>  $J  *  )          r r)typingrrrrcopyrnumpyrBpydanticr qdrant_clientr qdrant_client.async_client_baser "qdrant_client.embed.model_embedderr qdrant_client.httpr qdrant_client.conversionsr r>$qdrant_client.conversions.conversionrqdrant_client.embed.commonr!qdrant_client.embed.schema_parserrqdrant_client.fastembed_commonrrrOr rr}s;544444444444;;;;;;<<<<<<%%%%%%;;;;;;;;;;;;======??????888888s s s s s s s s s s r