7|hXUddlZddlmZddlmZmZmZmZddlm Z ddl Z ddl m Z ddl mZmZddlmZmZmZmZdZeed <d ed efd Zd ed efdZGddee Zy)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|jSy)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 b/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/jina.pyis_localrs-#JL(joo&&  file_pathct|d5}tj|jj dcdddS#1swYyxYw)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%sI i C*  1299'BCCCs 2A  AceZdZUdZeed<dZeed<dZe e ed<e dZ e d ed ed efd Zded eeefdZdeed eeefdZded eefdZdeed eeefdZy)JinaEmbeddingszJina embedding models.sessionzjina-embeddings-v2-base-en model_nameN jina_api_key)protected_namespacesbefore)modevaluesrc@ tt|dd}tj}|j j d|jddd ||d <|S#t$r1} tt|dd}n#t$r|wxYwYd}~d}~wwxYw) 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^^LL""$#*<+H+H+J*K!L#- 2  $y ! # #4(1BDUV   #"" # #s)A## B-BB BBBinputc|jjt||jdj }d|vrt |d|d}t |d}|Dcgc]}|d c}Scc}w)N)r?model)jsondatadetailc |dS)Nindexr+)es rz'JinaEmbeddings._embed..Xs QwZr)key embedding)r(postrr)rB RuntimeErrorsorted)selfr?resp embeddingssorted_embeddingsresults r_embedzJinaEmbeddings._embedMs||  I! $&   tH~. .&\ #:3GH3DD{#DDDs& A5textsc$|j|S)zCall out to Jina's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. rS)rNrTs rembed_documentszJinaEmbeddings.embed_documents]s{{5!!rtextc,|j|gdS)zCall out to Jina's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. rrV)rNrXs r embed_queryzJinaEmbeddings.embed_queryfs{{D6"1%%ruriscg}|D]=}t|r|jdt|i+|jd|i?|j|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%rS)rNr[r?uris r embed_imageszJinaEmbeddings.embed_imagesosV +C} g}S'9:; eS\*  + {{5!!r)__name__ __module__ __qualname____doc__r__annotations__r)strr*rrr model_configr classmethodrr>rfloatrSrWrZr`r+rrr'r'*s L2J2(,L(9%,26L(#$3$0ECEDe$5E "T#Y"4U 3D"&&U & "c "tDK/@ "rr')r os.pathrtypingrrrr urllib.parserr7langchain_core.embeddingsr langchain_core.utilsr r pydanticr r rrrrfreboolrr%r'r+rrrqsf ,,!0LFF7 c7 # $  CS CS CR"Y R"r