NgXUddlZddlmZddlmZmZmZmZddlm Z ddl Z ddl m Z ddl mZmZddlmZmZmZmZdZeed <d ed efd Zd ed efdZGddee 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_URLurlreturnc^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)r url_parseds _/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/jina.pyis_localrs2#JL((jo&&& 5 file_pathct|d5}tj|dcdddS#1swxYwYdS)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)r image_files r get_bytes_strr%s i  C*  1 12299'BBCCCCCCCCCCCCCCCCCCs9AAAcNeZdZUdZeed<dZeed<dZe e ed<e dZ e d ed ed efd Zded eeefdZdeed eeefdZded eefdZdeed eeefdZdS)JinaEmbeddingszJina embedding models.sessionzjina-embeddings-v2-base-en model_nameN jina_api_key)protected_namespacesbefore)modevaluesrcj tt|dd}nG#t$r:} tt|dd}n#t$r|wxYwYd}~nd}~wwxYwtj}|jd|ddd ||d <|S) z/Validate that auth token exists in environment.r* JINA_API_KEYjina_auth_tokenJINA_AUTH_TOKENNzBearer identityzapplication/json) AuthorizationzAccept-Encodingz Content-typer()r r ValueErrorrequestsSessionheadersupdateget_secret_value)clsr/r* original_excr(s rvalidate_environmentz#JinaEmbeddings.validate_environment3s #0$V^^LLLL # # # #4(1BDUVV    # # #"" #  #"$$!L<+H+H+J+J!L!L#- 2     $y s&! A%A  A  AA  A%inputc|jt||jd}d|vrt |d|d}t |d}d|DS)N)r?model)jsondatadetailc|dS)Nindexr+)es rz'JinaEmbeddings._embed..Xs QwZr)keycg|] }|d S) embeddingr+).0results r z)JinaEmbeddings._embed..[sDDD{#DDDr)r(postrr)rB RuntimeErrorsorted)selfr?resp embeddingssorted_embeddingss r_embedzJinaEmbeddings._embedMs|  II!  $&&    tH~.. .&\ #:3G3GHHHED2CDDDDrtextsc,||S)zCall out to Jina's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. rV)rRrWs rembed_documentszJinaEmbeddings.embed_documents]s{{5!!!rtextc:||gdS)zCall out to Jina's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. rrY)rRr[s r embed_queryzJinaEmbeddings.embed_queryfs{{D6""1%%ruriscg}|D]M}t|r%|dt|i6|d|iN||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%rV)rRr^r?uris r embed_imageszJinaEmbeddings.embed_imagesoss + +C}} + g}S'9'9:;;;; eS\****{{5!!!r)__name__ __module__ __qualname____doc__r__annotations__r)strr*rrr model_configr classmethodrr>rfloatrVrZr]rcr+rrr'r'*sU  LLL2J222(,L(9%,,,:2666L_(###$3[$#0ECEDe$5EEEE "T#Y"4U 3D""""&&U &&&& "c "tDK/@ " " " " " "rr')r os.pathrtypingrrrr urllib.parserr7langchain_core.embeddingsr langchain_core.utilsr r pydanticr r rrrrirhboolrr%r'r+rrrtsI ,,,,,,,,,,,,!!!!!!000000LLLLLLLLFFFFFFFFFFFF7 c777 # $     CS CS C C C CR"R"R"R"R"Y R"R"R"R"R"r