NgddlZddlZddlZddlmZmZmZmZddlZ ddl m Z ddl m Z ddlmZddlmZmZmZddlmZe dd d Gd d ee ZdS)N)AnyDictListOptional) deprecated) Embeddings)run_in_executor) BaseModel ConfigDictmodel_validator)Selfz0.2.11z1.0zlangchain_aws.BedrockEmbeddings)sinceremovalalternative_importceZdZUdZ dZeed< dZee ed< dZ ee ed< dZ e ed< dZ ee ed< dZee ed < d Zeed < ed d ZeddefdZde deefdZdeedeefdZdee deeefdZde deefdZde deefdZdee deeefdZdS)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)modereturncx|j|S ddl}|j||j}n|}i}|jr |j|d<|jr |j|d<|jd i||_n=#t $rt dt$r}td||d}~wwxYw|S) zJValidate that AWS credentials to and python package exists in environment.Nr) 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. Bedrock error: )r#) rboto3rSessionrr ImportError Exception ValueError)selfr$session client_paramses b/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/bedrock.pyvalidate_environmentz&BedrockEmbeddings.validate_environmentPs ; "K  LLL,8--T5R-SS --//M @/3/? m,  B040A n-('.LLmLLDKK   >    >:;>>    sA1A=="B7B22B7textc|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 contentTyper8 embeddings embeddingz$Error raised by inference endpoint: N)replaceoslineseprsplitrkeysjsondumpsr invoke_modelloadsgetreadr'r() r)r/provider _model_kwargs input_bodyr8response response_bodyr,s r-_embedding_funcz!BedrockEmbeddings._embedding_funcwsm||BJ,,=&&s++A.)/R & & x  :??#4#444+< <(#'&Jw  '+J{ #z*%% I{// ). 0H!Jx||F';';'@'@'B'BCCM8##$((66q99%((555 I I IGAGGHH H IsA>Er<ctj|}|tj|z }|S)z)Normalize the embedding to a unit vector.)nparraylinalgnormtolist)r)r<embnorm_embs r-_normalize_vectorz#BedrockEmbeddings._normalize_vectors9hz"",,,   r6cg}|D]H}||}|jr||}||I|S)zCompute doc embeddings using a Bedrock model. Args: texts: The list of texts to embed Returns: List of embeddings, one for each text. )rNrrWappend)r)r6resultsr/rLs r-embed_documentsz!BedrockEmbeddings.embed_documentssb % %D++D11H~ <11(;; NN8 $ $ $ $rXch||}|jr||S|S)zCompute query embeddings using a Bedrock model. Args: text: The text to embed. Returns: Embeddings for the text. )rNrrW)r)r/r=s r- embed_queryzBedrockEmbeddings.embed_querys;((.. > 5)))44 4rXc>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 r^)r)r/s r- aembed_queryzBedrockEmbeddings.aembed_querys/%T4+;TBBBBBBBBBrXcfKtjfd|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)r`).0r/r)s r- z6BedrockEmbeddings.aembed_documents..s''R'R'RD(9(9$(?(?'R'R'RrXN)asynciogatherlist)r)r6results` r-aembed_documentsz"BedrockEmbeddings.aembed_documentssH~'R'R'R'RE'R'R'RSSSSSSSF||rX)__name__ __module__ __qualname____doc__rr__annotations__rrstrrrrrrrboolr model_configr r r.rfloatrNrWr\r^r`rirrXr-rrs  "FC!%K#%%%/3hsm2221Hc000L$(L(4.'''1"&L(3-&&&CItE:H2FFFL_'"""$d$$$#"$L#IC#IDK#I#I#I#IJ!DK!DK!!!! T#Y4U 3D(U  Cs CtE{ C C C C DI $tE{:K      rXr)rerCr?typingrrrrnumpyrPlangchain_core._api.deprecationrlangchain_core.embeddingsrlangchain_core.runnables.configr pydanticr r r typing_extensionsr rrrXr-rzs ,,,,,,,,,,,,666666000000;;;;;;;;;;;;;;;;""""""   8 KKKKK :KK  KKKrX