U h@srddlZddlZddlmZmZmZmZmZddlZ ddl m Z ddl m Z ddlmZdZGddde e ZdS) N)AnyDictListLiteralOptional) Embeddings) BaseModel)pre_initz0.2.0c@seZdZUdZdZeed<dZeed<e eed<e eed<dZ e d ed <d Z eed <d Z e eed<eed<GdddZeeedddZeeeeedddZeeedddZd S)FastEmbedEmbeddingsaQdrant FastEmbedding models. FastEmbed is a lightweight, fast, Python library built for embedding generation. See more documentation at: * https://github.com/qdrant/fastembed/ * https://qdrant.github.io/fastembed/ To use this class, you must install the `fastembed` Python package. `pip install fastembed` Example: from langchain_community.embeddings import FastEmbedEmbeddings fastembed = FastEmbedEmbeddings() zBAAI/bge-small-en-v1.5 model_namei max_length cache_dirthreadsdefault)rpassagedoc_embed_type batch_sizeNparallel_modelc@seZdZdZdS)zFastEmbedEmbeddings.ConfigZallowN)__name__ __module__ __qualname__extrarrL/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/fastembed.pyConfigFsr)valuesreturncCs|d}|d}|d}|d}ztd}Wntk rRtdYnXtjdtkrltd|j||||d|d <|S) z+Validate that FastEmbed has been installed.r r r r fastembedz\Could not import 'fastembed' Python package. Please install it with `pip install fastembed`.zAFastEmbedEmbeddings requires `pip install -U "fastembed>=0.2.0"`.)r r r rr) get importlib import_moduleModuleNotFoundError ImportErrormetadataversion MIN_VERSIONZ TextEmbedding)clsrr r r rrrrrvalidate_environmentIs*      z(FastEmbedEmbeddings.validate_environment)textsrcCsF|jdkr"|jj||j|jd}n|jj||j|jd}dd|DS)zGenerate embeddings for documents using FastEmbed. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. rrrcSsg|] }|qSr)tolist).0errr ysz7FastEmbedEmbeddings.embed_documents..)rrZ passage_embedrrembed)selfr*Z embeddingsrrrembed_documentsgs z#FastEmbedEmbeddings.embed_documents)textrcCs"t|jj||j|jd}|S)zGenerate query embeddings using FastEmbed. Args: text: The text to embed. Returns: Embeddings for the text. r+)nextrZ query_embedrrr,)r1r3Zquery_embeddingsrrr embed_query{s zFastEmbedEmbeddings.embed_query)rrr__doc__r str__annotations__r intrrrrrrrr rr)rfloatr2r5rrrrr s      r )r!importlib.metadatatypingrrrrrZnumpynpZlangchain_core.embeddingsrZlangchain_core.pydantic_v1rZlangchain_core.utilsr r'r rrrrs