U h@sRddlZddlmZmZmZmZddlmZddl m Z m Z Gddde eZ dS)N)AnyDictListOptional) Embeddings) BaseModelroot_validatorc@seZdZUdZdZeed<dZee ed<GdddZ e dd e e d d d Z eeeeed ddZeeedddZeeeeed ddZeeedddZdS)SpacyEmbeddingsaEmbeddings by spaCy models. Attributes: model_name (str): Name of a spaCy model. nlp (Any): The spaCy model loaded into memory. Methods: embed_documents(texts: List[str]) -> List[List[float]]: Generates embeddings for a list of documents. embed_query(text: str) -> List[float]: Generates an embedding for a single piece of text. en_core_web_sm model_nameNnlpc@seZdZdZdS)zSpacyEmbeddings.ConfigZforbidN)__name__ __module__ __qualname__extrarrS/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/spacy_embeddings.pyConfigsrT)pre)valuesreturnc Cs|ddkrd|d<|d}tjddkr8tdzddl}|||d<Wn*tk r|td|d |d YnX|S) aE Validates that the spaCy package and the model are installed. Args: values (Dict): The values provided to the class constructor. Returns: The validated values. Raises: ValueError: If the spaCy package or the model are not installed. r Nr spacyzDSpaCy package not found. Please install it with `pip install spacy`.rr z SpaCy model 'z>' not found. Please install it with `python -m spacy download z%`or provide a valid spaCy model name.)get importlibutil find_spec ValueErrorrloadOSError)clsrr rrrrvalidate_environments  z$SpacyEmbeddings.validate_environment)textsrcsfdd|DS)z Generates embeddings for a list of documents. Args: texts (List[str]): The documents to generate embeddings for. Returns: A list of embeddings, one for each document. csg|]}|jqSrr Zvectortolist).0textselfrr Osz3SpacyEmbeddings.embed_documents..rr'r!rr&rembed_documentsEs zSpacyEmbeddings.embed_documents)r%rcCs||jS)z Generates an embedding for a single piece of text. Args: text (str): The text to generate an embedding for. Returns: The embedding for the text. r"r'r%rrr embed_queryQs zSpacyEmbeddings.embed_querycs tddS)aA Asynchronously generates embeddings for a list of documents. This method is not implemented and raises a NotImplementedError. Args: texts (List[str]): The documents to generate embeddings for. Raises: NotImplementedError: This method is not implemented. 3Asynchronous embedding generation is not supported.NNotImplementedErrorr)rrraembed_documents]s z SpacyEmbeddings.aembed_documentscs tddS)a< Asynchronously generates an embedding for a single piece of text. This method is not implemented and raises a NotImplementedError. Args: text (str): The text to generate an embedding for. Raises: NotImplementedError: This method is not implemented. r-Nr.r+rrr aembed_queryjs zSpacyEmbeddings.aembed_query)r rr__doc__r str__annotations__r rrrrrr rfloatr*r,r0r1rrrrr s  (   r ) importlib.utilrtypingrrrrZlangchain_core.embeddingsrZlangchain_core.pydantic_v1rrr rrrrs