7|hddlZddlZddlZddlmZmZmZmZddlZ ddl m Z ddl m Z ddlmZddlmZmZmZddlmZe dd d Gd d ee Zy)N)AnyDictListOptional) deprecated) Embeddings)run_in_executor) BaseModel ConfigDictmodel_validator)Selfz0.2.11z1.0zlangchain_aws.BedrockEmbeddings)sinceremovalalternative_importcveZdZUdZ dZeed< dZee ed< dZ ee ed< dZ e ed< dZ ee ed< dZee ed < d Zeed < ed d ZeddefdZde deefdZdeedeefdZdee deeefdZde deefdZde deefdZdee deeefdZy)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. Nclient region_namecredentials_profile_namezamazon.titan-embed-text-v1model_id model_kwargs endpoint_urlF normalizeforbid)extraprotected_namespacesafter)modereturnc|j|S ddl}|j|j|j}n|j}i}|jr|j|d<|j r|j |d<|jdi||_|S#t $r t dt$r}td||d}~wwxYw) zJValidate that AWS credentials to and python package exists in environment.Nr) profile_namerrzRCould 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. Bedrock error: )zbedrock-runtime) rboto3rSessionrr ImportError Exception ValueError)selfr#session client_paramses e/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/bedrock.pyvalidate_environmentz&BedrockEmbeddings.validate_environmentPs ;; "K  ,,8--T5R5R-S --/M/3/?/? m,  040A0A n-('..LmLDK  >  ::;>   sB BC:C  Ctextcj|jtjd}|jj dd}|j xsi}i|}|dk(rd|j vrd|d<|g|d<n||d<tj|} |jj||jd d }tj|jd j}|dk(r|jd dS|jd S#t$r}td|d}~wwxYw)z'Call out to Bedrock embedding endpoint. .rcohere input_typesearch_documenttexts inputTextzapplication/json)bodymodelIdaccept contentTyper7 embeddings embeddingz$Error raised by inference endpoint: N)replaceoslineseprsplitrkeysjsondumpsr invoke_modelloadsgetreadr&r') r(r.provider _model_kwargs input_bodyr7response response_bodyr+s r,_embedding_funcz!BedrockEmbeddings._embedding_funcws;||BJJ,==&&s+A.))/R & & x :??#44+< <(#'&Jw '+J{ #zz*% I{{// ). 0H!JJx||F';'@'@'BCM8#$((6q99%((55 ICA3GH H IsA3DD D2D--D2r;ctj|}|tjj|z }|j S)z)Normalize the embedding to a unit vector.)nparraylinalgnormtolist)r(r;embnorm_embs r,_normalize_vectorz#BedrockEmbeddings._normalize_vectors4hhz",,  r5cg}|D]A}|j|}|jr|j|}|j|C|S)zCompute doc embeddings using a Bedrock model. Args: texts: The list of texts to embed Returns: List of embeddings, one for each text. )rMrrVappend)r(r5resultsr.rKs r,embed_documentsz!BedrockEmbeddings.embed_documentssR %D++D1H~~11(; NN8 $  %rWcb|j|}|jr|j|S|S)zCompute query embeddings using a Bedrock model. Args: text: The text to embed. Returns: Embeddings for the text. )rMrrV)r(r.r<s r, embed_queryzBedrockEmbeddings.embed_querys2((. >>)))4 4rWcLKtd|j|d{S7w)zAsynchronous compute query embeddings using a Bedrock model. Args: text: The text to embed. Returns: Embeddings for the text. N)r r])r(r.s r, aembed_queryzBedrockEmbeddings.aembed_querys#%T4+;+;TBBBBs $"$cKtj|Dcgc]}|j|c}d{}t|Scc}w7w)zAsynchronous compute doc embeddings using a Bedrock model. Args: texts: The list of texts to embed Returns: List of embeddings, one for each text. N)asynciogatherr_list)r(r5r.results r,aembed_documentsz"BedrockEmbeddings.aembed_documentssB~~E'RD(9(9$(?'RSSF|(SSsA AA A A )__name__ __module__ __qualname____doc__rr__annotations__rrstrrrrrrrboolr model_configr r r-rfloatrMrVr[r]r_rerrWr,rrsY "FC!%K#%/3hsm21Hc0L$(L(4.'1"&L(3-&CItEH2FL'"$d$#$L#IC#IDK#IJ!DK!DK! T#Y4U 3D(U  Cs CtE{ C DI $tE{:K rWr)rarBr>typingrrrrnumpyrOlangchain_core._api.deprecationrlangchain_core.embeddingsrlangchain_core.runnables.configr pydanticr r r typing_extensionsr rrrWr,rvsW ,,60;;;"  8 K :K  KrW