昺h'ddlZddlZddlZddlmZmZmZmZddlm Z ddl m Z m Z m Z ddlmZGdde e ZdS)N)AnyDictListOptional) Embeddings) BaseModelExtraroot_validator)run_in_executorceZdZUdZ eed< dZeeed< dZ eeed< dZ eed< dZ ee ed< dZ eeed < Gd d Zed e d e fdZded eefdZdeed eeefdZded eefdZded eefdZdeed eeefdZdS)BedrockEmbeddingsaBedrock embedding models. To authenticate, the AWS client uses the following methods to automatically load credentials: https://boto3.amazonaws.com/v1/documentation/api/latest/guide/credentials.html If a specific credential profile should be used, you must pass the name of the profile from the ~/.aws/credentials file that is to be used. Make sure the credentials / roles used have the required policies to access the Bedrock service. clientN region_namecredentials_profile_namezamazon.titan-embed-text-v1model_id model_kwargs endpoint_urlc eZdZdZejZdS)BedrockEmbeddings.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__r forbidextran/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/embeddings/bedrock.pyConfigrCs55 rrvaluesreturnc||d|S ddl}|d||d}n|}i}|dr |d|d<|dr |d|d<|jd i||d<n:#t$rt d t $r}t d |d}~wwxYw|S) zJValidate that AWS credentials to and python package exists in environment.rNrr) profile_namerrbedrock-runtimezRCould not import boto3 python package. Please install it with `pip install boto3`.zCould not load credentials to authenticate with AWS client. Please check that credentials in the specified profile name are valid.)r$)boto3Sessionr ImportErrorModuleNotFoundError Exception ValueError)clsr r%session client_paramses rvalidate_environmentz&BedrockEmbeddings.validate_environmentHs# (  'M  LLL01=--V    *    sA5B"B9$B44B9textc|tjd}|jdd}|jpi}i|}|dkr"d|vrd|d<|g|d<n||d<tj|} |j ||jd d }tj | d  }|dkr| d dS| d S#t$r}td|d}~wwxYw)z'Call out to Bedrock embedding endpoint. .rcohere input_typesearch_documenttexts inputTextzapplication/json)bodymodelIdaccept contentTyper9 embeddings embeddingz$Error raised by inference endpoint: N)replaceoslineseprsplitrkeysjsondumpsr invoke_modelloadsgetreadr)r*) selfr0provider _model_kwargs input_bodyr9response response_bodyr.s r_embedding_funcz!BedrockEmbeddings._embedding_funcosm||BJ,,=&&s++A.)/R & & x  :??#4#444+< <(#'&Jw  '+J{ #z*%% I{// ). 0H!Jx||F';';'@'@'B'BCCM8##$((66q99%((555 I I IGAGGHH H IsA>Er7chg}|D],}||}||-|S)zCompute doc embeddings using a Bedrock model. Args: texts: The list of texts to embed Returns: List of embeddings, one for each text. )rPappend)rJr7resultsr0rNs rembed_documentsz!BedrockEmbeddings.embed_documentssE % %D++D11H NN8 $ $ $ $rc,||S)zCompute query embeddings using a Bedrock model. Args: text: The text to embed. Returns: Embeddings for the text. )rPrJr0s r embed_queryzBedrockEmbeddings.embed_querys##D)))rc>Ktd|j|d{VS)zAsynchronous compute query embeddings using a Bedrock model. Args: text: The text to embed. Returns: Embeddings for the text. N)r rWrVs r aembed_queryzBedrockEmbeddings.aembed_querys/%T4+;TBBBBBBBBBrcfKtjfd|Dd{V}t|S)zAsynchronous compute doc embeddings using a Bedrock model. Args: texts: The list of texts to embed Returns: List of embeddings, one for each text. c:g|]}|Sr)rY).0r0rJs r z6BedrockEmbeddings.aembed_documents..s''R'R'RD(9(9$(?(?'R'R'RrN)asynciogatherlist)rJr7results` raembed_documentsz"BedrockEmbeddings.aembed_documentssH~'R'R'R'RE'R'R'RSSSSSSSF||r)rrrrr__annotations__rrstrrrrrrrr r/rfloatrPrTrWrYrbrrrr r s  " KKK!%K#%%%/3hsm2221Hc000L$(L(4.'''1"&L(3-&&&C ^$$$4$$$$L#IC#IDK#I#I#I#IJ T#Y 4U 3D     * *U * * * * Cs CtE{ C C C C DI $tE{:K      rr )r^rDr@typingrrrrlangchain_core.embeddingsrlangchain_core.pydantic_v1rr r langchain_core.runnables.configr r rrrrjs ,,,,,,,,,,,,000000GGGGGGGGGG;;;;;;{{{{{ :{{{{{r