o ZhX@sUddlZddlmZddlmZmZmZmZddlm Z ddl Z ddl m Z ddl mZmZddlmZmZmZmZdZeed <d ed efd d Zded efddZGdddee 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_URLurlreturncCs 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_parsedrZ/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/jina.pyis_locals  r file_pathcCsBt|d}t|dWdS1swYdS)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 ed ed efd dZded eeefddZdeed eeefddZded eefddZdeed eeefddZdS)JinaEmbeddingszJina embedding models.sessionzjina-embeddings-v2-base-en model_nameN jina_api_keyr)Zprotected_namespacesbefore)modevaluesrc Csz tt|dd}Wn&ty0}zz tt|dd}Wn ty%|wWYd}~nd}~wwt}|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_environment3s0     z#JinaEmbeddings.validate_environmentinputcCsV|jjt||jdd}d|vrt|d|d}t|ddd}dd |DS) N)r1model)jsondatadetailcSs|dS)Nindexr)errrXsz'JinaEmbeddings._embed..)keycSsg|]}|dqS)Z embeddingr).0resultrrr [sz)JinaEmbeddings._embed..)r"postrr#r3 RuntimeErrorsorted)selfr1respZ embeddingsZsorted_embeddingsrrr_embedMs  zJinaEmbeddings._embedtextscCs ||S)zCall out to Jina's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. rB)r@rCrrrembed_documents]s zJinaEmbeddings.embed_documentstextcCs||gdS)zCall out to Jina's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. rrD)r@rFrrr embed_queryfszJinaEmbeddings.embed_queryuriscCsBg}|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 rB)r@rHr1urirrr embed_imagesos  zJinaEmbeddings.embed_images)__name__ __module__ __qualname____doc__r__annotations__r#strr$rr r Z model_configr classmethodrr0rfloatrBrErGrLrrrrr!*s    " r!)rZos.pathrtypingrrrr urllib.parserr+Zlangchain_core.embeddingsrZlangchain_core.utilsr r Zpydanticr r r rrrRrQboolrr r!rrrrs