U h@sddlZddlZddlZddlmZmZmZmZddlZ ddl m Z ddl m Z ddlmZmZddlmZe ddd d Gd d d ee ZdS) N)AnyDictListOptional) deprecated) Embeddings) BaseModelroot_validator)run_in_executorz0.2.11z1.0zlangchain_aws.BedrockEmbeddings)ZsinceZremovalZalternative_importc@s(eZdZUdZeed<dZeeed<dZ eeed<dZ eed<dZ ee ed<dZ eeed <d Zeed <Gd d d Zed dde e dddZeeedddZeeeedddZeeeeedddZeeedddZeeedddZeeeeedd 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. clientN region_namecredentials_profile_namezamazon.titan-embed-text-v1model_id model_kwargs endpoint_urlF normalizec@seZdZdZdS)zBedrockEmbeddings.ConfigZforbidN)__name__ __module__ __qualname__extrarrJ/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/bedrock.pyConfigMsrT)preZskip_on_failure)valuesreturnc Cs|ddk r|Sznddl}|ddk r8|j|dd}n|}i}|drX|d|d<|drl|d|d<|jd ||d<WnLtk rtd Yn2tk r}ztd ||W5d}~XYnX|S) zJValidate that AWS credentials to and python package exists in environment.r Nrr)Z profile_namer rbedrock-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)boto3Sessionr ImportError Exception ValueError)clsrrsessionZ client_paramserrrvalidate_environmentPs0    z&BedrockEmbeddings.validate_environment)textrc Cs|tjd}|jdd}|jp&i}|}|dkrVd|krJd|d<|g|d<n||d<t|}zR|j j ||jd d d }t | d  }|dkr| d dWS| d WSWn0tk r}ztd|W5d}~XYnXdS)z'Call out to Bedrock embedding endpoint. .rZcohereZ input_typeZsearch_documenttextsZ inputTextzapplication/json)bodyZmodelIdaccept contentTyper+ embeddings embeddingz$Error raised by inference endpoint: N)replaceoslineseprsplitrkeysjsondumpsr Z invoke_modelloadsgetreadr!r") selfr'ZproviderZ _model_kwargsZ input_bodyr+responseZ response_bodyr%rrr_embedding_funcws.    z!BedrockEmbeddings._embedding_func)r.rcCs"t|}|tj|}|S)z)Normalize the embedding to a unit vector.)nparrayZlinalgZnormtolist)r:r.ZembZnorm_embrrr_normalize_vectors z#BedrockEmbeddings._normalize_vector)r*rcCs6g}|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. )r<rr@append)r:r*resultsr'r;rrrembed_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. )r<rr@)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)r rD)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)rE).0r'r:rr sz6BedrockEmbeddings.aembed_documents..N)asyncioZgatherlist)r:r*resultrrGraembed_documentss z"BedrockEmbeddings.aembed_documents)rrr__doc__r__annotations__r rstrrrrrrrboolrr r&rfloatr<r@rCrDrErLrrrrr s"    &% r )rIr5r1typingrrrrZnumpyr=Zlangchain_core._api.deprecationrZlangchain_core.embeddingsrZlangchain_core.pydantic_v1rr Zlangchain_core.runnables.configr r rrrrs