a bg@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@s&eZdZUdZdZeed<dZee ed<dZ ee ed<dZ e ed<dZ ee ed<dZee ed <d Zeed <ed d dZeddedddZe eedddZeeeedddZee eeedddZe eedddZe eeddd Zee eeedd!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)mode)returnc Cs|jdur|Szdddl}|jdur2|j|jd}n|}i}|jrN|j|d<|jr^|j|d<|jd i||_WnLtytdYn4ty}ztd||WYd}~n d}~00|S) 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_paramserrt/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/bedrock.pyvalidate_environmentPs4     z&BedrockEmbeddings.validate_environment)textrc Cs|tjd}|jdd}|jp&i}i|}|dkrXd|vrLd|d<|g|d<n||d<t|}zR|j j ||jd d d }t | d  }|dkr| d dWS| d WSWn2ty}ztd|WYd}~n d}~00dS)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_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. )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# z6BedrockEmbeddings.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@rArBrKrrrr#r s"     &% r )rGr2r.typingrrrrnumpyr:Zlangchain_core._api.deprecationrZlangchain_core.embeddingsrZlangchain_core.runnables.configrZpydanticr r r Ztyping_extensionsr r rrrr#s