o Zh@sddlZddlZddlZddlmZmZmZmZddlZ ddl m Z ddl m Z ddlmZddlmZmZmZddlmZe dd d d Gd d d ee ZdS)N)AnyDictListOptional) deprecated) Embeddings)run_in_executor) BaseModel ConfigDictmodel_validator)Selfz0.2.11z1.0zlangchain_aws.BedrockEmbeddings)ZsinceZremovalZalternative_importc@sBeZdZUdZ dZeed< dZee ed< dZ ee ed< dZ e ed< dZ ee ed< dZee ed < d Zeed < ed d dZedddefddZde deefddZdeedeefddZdee deeefddZde deefddZde deefdd Zdee deeefd!d"ZdS)#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 normalizeZforbid)extraZprotected_namespacesafter)modereturnc Cs|jdur|Sz3ddl}|jdur|j|jd}n|}i}|jr'|j|d<|jr/|j|d<|jd i||_W|StyEtdtyX}ztd||d}~ww) zJValidate that AWS credentials to and python package exists in environment.Nr)Z 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)selfrsessionZ client_paramserr]/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/bedrock.pyvalidate_environmentPs8     z&BedrockEmbeddings.validate_environmenttextc Cs|tjd}|jdd}|jpi}i|}|dkr,d|vr&d|d<|g|d<n||d<t|}z'|j j ||jd d d }t | d  }|dkrW| d dWS| d WStyo}ztd|d}~ww)z'Call out to Bedrock embedding endpoint. .rZcohereZ input_typeZsearch_documenttextsZ inputTextzapplication/json)bodyZmodelIdacceptZ contentTyper) embeddings embeddingz$Error raised by inference endpoint: N)replaceoslineseprsplitrkeysjsondumpsrZ invoke_modelloadsgetreadrr) r r%providerZ _model_kwargsZ input_bodyr)responseZ response_bodyr"rrr#_embedding_funcws2     z!BedrockEmbeddings._embedding_funcr+cCs"t|}|tj|}|S)z)Normalize the embedding to a unit vector.)nparrayZlinalgZnormtolist)r r+ZembZnorm_embrrr#_normalize_vectors z#BedrockEmbeddings._normalize_vectorr(cCs6g}|D]}||}|jr||}||q|S)zCompute doc embeddings using a Bedrock model. Args: texts: The list of texts to embed Returns: List of embeddings, one for each text. )r9rr=append)r r(resultsr%r8rrr#embed_documentss    z!BedrockEmbeddings.embed_documentscCs||}|jr ||S|S)zCompute query embeddings using a Bedrock model. Args: text: The text to embed. Returns: Embeddings for the text. )r9rr=)r r%r,rrr# embed_querys  zBedrockEmbeddings.embed_querycstd|j|IdHS)zAsynchronous compute query embeddings using a Bedrock model. Args: text: The text to embed. Returns: Embeddings for the text. N)rrA)r r%rrr# aembed_querys zBedrockEmbeddings.aembed_querycs(tjfdd|DIdH}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. csg|]}|qSr)rB).0r%r rr# sz6BedrockEmbeddings.aembed_documents..N)asynciogatherlist)r r(resultrrDr#aembed_documentss z"BedrockEmbeddings.aembed_documents)__name__ __module__ __qualname____doc__rr__annotations__rrstrrrrrrrboolr Z model_configr r r$rfloatr9r=r@rArBrJrrrr#r s4     &%" r )rFr2r.typingrrrrnumpyr:Zlangchain_core._api.deprecationrZlangchain_core.embeddingsrZlangchain_core.runnables.configrZpydanticr r r Ztyping_extensionsr r rrrr#s