U h@sfddlmZmZmZmZmZmZddlZddlm Z ddl m Z m Z ddl mZGddde e ZdS))AnyDictListMappingOptionalTupleN) Embeddings) BaseModelroot_validatorget_from_dict_or_envc@seZdZUdZdZeed<dZeed<dZeed<dZ e ed <d Z e eed <Gd d d Z eddeedddZeeeefdddZd eeeefeeee dddZeeeee dddZeee dddZd S)!MosaicMLInstructorEmbeddingsaaMosaicML embedding service. To use, you should have the environment variable ``MOSAICML_API_TOKEN`` set with your API token, or pass it as a named parameter to the constructor. Example: .. code-block:: python from langchain_community.llms import MosaicMLInstructorEmbeddings endpoint_url = ( "https://models.hosted-on.mosaicml.hosting/instructor-large/v1/predict" ) mosaic_llm = MosaicMLInstructorEmbeddings( endpoint_url=endpoint_url, mosaicml_api_token="my-api-key" ) zBhttps://models.hosted-on.mosaicml.hosting/instructor-xl/v1/predict endpoint_urlz&Represent the document for retrieval: embed_instructionztjjk rh}ztd|d|j W5d}~XYnX| S)Ninputszapplication/json) Authorizationz Content-Type)headersjsonz$Error raised by inference endpoint: irT)r#z>Error raised by inference API: rate limit exceeded. Response: dataoutputoutputsz#No key data or output in response: zUnexpected response type: zError raised by inference API: z . Response: )rrequestspostr exceptionsRequestException ValueError status_codetimesleepr_embedtextr' isinstancedictlistJSONDecodeError) r r"r#payloadr&responseer1Zparsed_responseZ output_keyskeyZ output_item embeddingsrrrr3=sH      z#MosaicMLInstructorEmbeddings._embed)textsrcs fdd|D}|}|S)zEmbed documents using a MosaicML deployed instructor embedding model. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. csg|]}j|fqSr)r).0r4rrr sz@MosaicMLInstructorEmbeddings.embed_documents..)r3)r r>Zinstruction_pairsr=rrrembed_documentsys  z,MosaicMLInstructorEmbeddings.embed_documents)r4rcCs|j|f}||gd}|S)zEmbed a query using a MosaicML deployed instructor embedding model. Args: text: The text to embed. Returns: Embeddings for the text. r)rr3)r r4Zinstruction_pairZ embeddingrrr embed_querys z(MosaicMLInstructorEmbeddings.embed_query)F)rrr__doc__rstr__annotations__rrrfloatrrrr rrpropertyrrr!rrboolr3rArBrrrrr s*      < r )typingrrrrrrr+Zlangchain_core.embeddingsrZlangchain_core.pydantic_v1r r Zlangchain_core.utilsr r rrrrs