a bgX@sUddlZddlmZddlmZmZmZmZddlm Z ddl Z ddl m Z ddl mZmZddlmZmZmZmZdZeed <eed d d Zeed ddZGdddee ZdS)N)exists)AnyDictListOptional)urlparse) Embeddings)convert_to_secret_strget_from_dict_or_env) BaseModel ConfigDict SecretStrmodel_validatorz!https://api.jina.ai/v1/embeddings JINA_API_URL)urlreturncCs t|}|jdvrt|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_parsedrq/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/jina.pyis_locals   r) file_pathrcCsBt|d$}t|dWdS1s40YdS)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 r c@seZdZUdZeed<dZeed<dZe e ed<e ddZ e d d eeed d d ZeeeedddZeeeeedddZeeedddZeeeeedddZdS)JinaEmbeddingszJina embedding models.sessionzjina-embeddings-v2-base-en model_nameN jina_api_keyr)Zprotected_namespacesbefore)mode)valuesrc Csztt|dd}WnPtyd}z8ztt|dd}WntyN|Yn0WYd}~n d}~00t}|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_environment3s(     z#JinaEmbeddings.validate_environment)inputrcCsV|jjt||jdd}d|vr0t|d|d}t|ddd}dd |DS) N)r1model)jsondatadetailcSs|dS)Nindexr)errrXz'JinaEmbeddings._embed..)keycSsg|] }|dqS)Z embeddingr).0resultrrr [r9z)JinaEmbeddings._embed..)r"postrr#r3 RuntimeErrorsorted)selfr1respZ embeddingsZsorted_embeddingsrrr_embedMs  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. rC)rArDrrrembed_documents]szJinaEmbeddings.embed_documents)textrcCs||gdS)zCall out to Jina's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. rrE)rArGrrr embed_queryfszJinaEmbeddings.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)rappendr rC)rArIr1urirrr embed_imagesos zJinaEmbeddings.embed_images)__name__ __module__ __qualname____doc__r__annotations__r#strr$rr r Z model_configr classmethodrr0rfloatrCrFrHrMrrrrr!*s     r!)ros.pathrtypingrrrr urllib.parserr+Zlangchain_core.embeddingsrZlangchain_core.utilsr r Zpydanticr r r rrrSrRboolrr r!rrrrs