U h@sUddlZddlmZddlmZmZmZmZddlm Z ddl Z ddl m Z ddl mZmZmZddlmZmZdZeed <eed d d Zeed ddZGdddee ZdS)N)exists)AnyDictListOptional)urlparse) Embeddings) BaseModel SecretStrroot_validator)convert_to_secret_strget_from_dict_or_envz!https://api.jina.ai/v1/embeddings JINA_API_URL)urlreturncCs t|}|jdkrt|jSdS)zCheck if a URL is a local file. Args: url (str): The URL to check. Returns: bool: True if the URL is a local file, False otherwise. )fileF)rschemerpath)rZ url_parsedrG/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/jina.pyis_locals   r) file_pathrc Cs6t|d"}t|dW5QRSQRXdS)zGet the bytes string of a file. Args: file_path (str): The path to the file. Returns: str: The bytes string of the file. rbzutf-8N)openbase64 b64encodereaddecode)rZ image_filerrr get_bytes_strs rc@seZdZUdZeed<dZeed<dZe e ed<e dde e d d d Z eeeed d dZeeeeedddZeeedddZeeeeedddZdS)JinaEmbeddingszJina embedding models.sessionzjina-embeddings-v2-base-en model_nameN jina_api_keyT)pre)valuesrc Csztt|dd}WnPtk rd}z2ztt|dd}Wntk rR|YnXW5d}~XYnXt}|jd|ddd ||d <|S) z/Validate that auth token exists in environment.r#Z JINA_API_KEYZjina_auth_tokenZJINA_AUTH_TOKENNzBearer identityzapplication/json) AuthorizationzAccept-Encodingz Content-typer!)r r ValueErrorrequestsSessionheadersupdateZget_secret_value)clsr%r#Z original_excr!rrrvalidate_environment1s(   z#JinaEmbeddings.validate_environment)inputrcCsV|jjt||jdd}d|kr0t|d|d}t|ddd}dd |DS) N)r/model)jsondataZdetailcSs|dS)Nindexr)errrUz'JinaEmbeddings._embed..)keycSsg|] }|dqS)Z embeddingr).0resultrrr Xsz)JinaEmbeddings._embed..)r!postrr"r1 RuntimeErrorsorted)selfr/respZ embeddingsZsorted_embeddingsrrr_embedJs  zJinaEmbeddings._embed)textsrcCs ||S)zCall out to Jina's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. r@)r>rArrrembed_documentsZszJinaEmbeddings.embed_documents)textrcCs||gdS)zCall out to Jina's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. rrB)r>rDrrr embed_querycszJinaEmbeddings.embed_query)urisrcCsBg}|D].}t|r(|dt|iq|d|iq||S)zCall out to Jina's image embedding endpoint. Args: uris: The list of uris to embed. Returns: List of embeddings, one for each text. bytesr)rappendrr@)r>rFr/urirrr embed_imagesls zJinaEmbeddings.embed_images)__name__ __module__ __qualname____doc__r__annotations__r"strr#rr r rr.rfloatr@rCrErJrrrrr *s    r )ros.pathrtypingrrrr urllib.parserr)Zlangchain_core.embeddingsrZlangchain_core.pydantic_v1r r r Zlangchain_core.utilsr r rrPrOboolrrr rrrrs