昺h/ UddlmZmZmZmZddlZddlmZddlm Z m Z m Z ddl m Z mZdZeed<Gdd e eZdS) )AnyDictListOptionalN) Embeddings) BaseModel SecretStrroot_validator)convert_to_secret_strget_from_dict_or_envz!https://api.jina.ai/v1/embeddings JINA_API_URLceZdZUdZeed<dZeed<dZe e ed<e de de fd Z d eedeeefd Zd eedeeefd Zd edeefdZdS)JinaEmbeddingszJina embedding models.sessionzjina-embeddings-v2-base-en model_nameN jina_api_keyvaluesreturncj 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)clsrr original_excrs k/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/embeddings/jina.pyvalidate_environmentz#JinaEmbeddings.validate_environments #0$V^^LLLL # # # #4(1BDUVV    # # #"" #  #"$$!L<+H+H+J+J!L!L#- 2     $y s&! A%A  A  AA  A%textsc|jt||jd}d|vrt |d|d}t |d}d|DS)N)inputmodel)jsondatadetailc|dS)Nindex)es r#z'JinaEmbeddings._embed..6s QwZ)keycg|] }|d S) embeddingr.).0results r# z)JinaEmbeddings._embed..9sDDD{#DDDr1)rpostr rr) RuntimeErrorsorted)selfr%resp embeddingssorted_embeddingss r#_embedzJinaEmbeddings._embed+s|  II!  $&&    tH~.. .&\ #:3G3GHHHED2CDDDDr1c,||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;r%s r#embed_documentszJinaEmbeddings.embed_documents;s{{5!!!r1textc:||gdS)zCall out to Jina's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. rrA)r;rCs r# embed_queryzJinaEmbeddings.embed_queryDs{{D6""1%%r1)__name__ __module__ __qualname____doc__r__annotations__rstrrrr r rr$rfloatr?rBrEr.r1r#rr s   LLL2J222(,L(9%,,,^$40EDIE$tE{*;EEEE "T#Y"4U 3D""""&&U &&&&&&r1r)typingrrrrrlangchain_core.embeddingsrlangchain_core.pydantic_v1rr r langchain_core.utilsr r r rKrJrr.r1r#rQs,,,,,,,,,,,,,000000KKKKKKKKKKLLLLLLLL7 c777@&@&@&@&@&Y @&@&@&@&@&r1