昺h| hddlmZmZmZmZmZddlZddlm Z ddl m Z m Z m Z Gdde e ZdS))AnyDictListLiteralOptionalN) Embeddings) BaseModelExtraroot_validatorc$eZdZUdZdZeed< dZeed< e eed< e eed< dZ e d ed < e ed <Gd d Z ededefdZdeedeeefdZdedeefdZdS)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_modelc eZdZdZejZdS)FastEmbedEmbeddings.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__r forbidextrap/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/embeddings/fastembed.pyConfigr5s55 rr!valuesreturnc& ddlm}|d}|d}|d}|d}||||||d<n"#t$r}td |d }~wwxYw|S) z+Validate that FastEmbed has been installed.r) FlagEmbeddingrrrr)rrrrrz\Could not import 'fastembed' Python package. Please install it with `pip install fastembed`.N)fastembed.embeddingr%get ImportError)clsr"r%rrrries r validate_environmentz(FastEmbedEmbeddings.validate_environment:s  9 9 9 9 9 9L11JL11J ;//Ijj++G,}%%#    F8      B    sA,A// B9B  Btextsc|jdkr|j|}n|j|}d|DS)zGenerate embeddings for documents using FastEmbed. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. rc6g|]}|Sr)tolist).0es r z7FastEmbedEmbeddings.embed_documents.._s ///q ///r)rr passage_embedembed)selfr, embeddingss r embed_documentsz#FastEmbedEmbeddings.embed_documentsQsR  ) + +22599JJ**511J//J////rtextcxt|j|}|S)zGenerate query embeddings using FastEmbed. Args: text: The text to embed. Returns: Embeddings for the text. )nextr query_embedr/)r5r8query_embeddingss r embed_queryzFastEmbedEmbeddings.embed_queryas4(,DK,C,CD,I,I'J'J&&(((rN)rrrrrstr__annotations__rintrrrrr!r rr+rfloatr7r=rrr r r sX  /J... J}c]5>NG01=== KKK ^$4,0T#Y04U 3D0000 ) )U ) ) ) ) ) )rr )typingrrrrrnumpynplangchain_core.embeddingsrlangchain_core.pydantic_v1r r r r rrr rGs55555555555555000000GGGGGGGGGGc)c)c)c)c))Zc)c)c)c)c)r