NgyddlmZddlmZmZmZmZmZddlZddl m Z ddl m Z ddl mZmZddlmZmZmZmZe dd d Gd d ee ZdS)) annotations)AnyDictListOptionalcastN) deprecated) Embeddingsconvert_to_secret_strget_from_dict_or_env) BaseModel ConfigDict SecretStrmodel_validatorz0.3.4z1.0.0z$langchain_community.ClovaXEmbeddings)sinceremovalalternative_importceZdZUdZdZded< dZded< dZded < dZded < dZ ded < e d Z e de ddZddZddZddZdS) ClovaEmbeddingsa Clova's embedding service. To use this service, you should have the following environment variables set with your API tokens and application ID, or pass them as named parameters to the constructor: - ``CLOVA_EMB_API_KEY``: API key for accessing Clova's embedding service. - ``CLOVA_EMB_APIGW_API_KEY``: API gateway key for enhanced security. - ``CLOVA_EMB_APP_ID``: Application ID for identifying your application. Example: .. code-block:: python from langchain_community.embeddings import ClovaEmbeddings embeddings = ClovaEmbeddings( clova_emb_api_key='your_clova_emb_api_key', clova_emb_apigw_api_key='your_clova_emb_apigw_api_key', app_id='your_app_id' ) query_text = "This is a test query." query_result = embeddings.embed_query(query_text) document_text = "This is a test document." document_result = embeddings.embed_documents([document_text]) zChttps://clovastudio.apigw.ntruss.com/testapp/v1/api-tools/embeddingstr endpoint_urlzclir-emb-dolphinmodelNzOptional[SecretStr]clova_emb_api_keyclova_emb_apigw_api_keyapp_idforbid)extrabefore)modevaluesrreturnrctt|dd|d<tt|dd|d<tt|dd|d<|S)z'Validate api key exists in environment.rCLOVA_EMB_API_KEYrCLOVA_EMB_APIGW_API_KEYrCLOVA_EMB_APP_IDr )clsr!s `/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/clova.pyvalidate_environmentz$ClovaEmbeddings.validate_environmentBs'< )<>Q R R' ' "#-B 13L  - - () 1 3E F F  x texts List[str]List[List[float]]cdg}|D]*}|||+|S)z Embed a list of texts and return their embeddings. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. )append _embed_text)selfr+ embeddingstexts r(embed_documentszClovaEmbeddings.embed_documentsSsC  6 6D   d..t44 5 5 5 5r*r3 List[float]c,||S)z Embed a single query text and return its embedding. Args: text: The text to embed. Returns: Embeddings for the text. )r0)r1r3s r( embed_queryzClovaEmbeddings.embed_querybs%%%r*cd|i}tt|jtt|jdd}tt|j}t j|jd|j d|||}|j dkr0| }d|vrd|dvr|ddStd |j d |j ) zT Internal method to call the embedding API and handle the response. r3zapplication/json)zX-NCP-CLOVASTUDIO-API-KEYzX-NCP-APIGW-API-KEYz Content-Type/)headersjsonresult embeddingzAPI request failed with status z: )rrrget_secret_valuerrrequestspostrr status_coder; ValueErrorr3)r1r3payloadr:rresponse response_datas r(r0zClovaEmbeddings._embed_textns*4.*.41**  #'47$$  .  i-->>@@= 8 84: 8 8 8 8     3 & &$MMOOM=(([M(rUs""""""22222222222222666666000000LLLLLLLLFFFFFFFFFFFF   = } } } } } i} }  } } } r*