o Zhy@sddlmZddlmZmZmZmZmZddlZddl m Z ddl m Z ddl mZmZddlmZmZmZmZe dd d d Gd d 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)ZsinceZremovalZalternative_importc@seZdZUdZdZded< dZded< dZded < dZded < dZ ded < e d d Z e dde d!ddZd"ddZd#ddZd#dd ZdS)$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_idZforbid)extrabefore)modevaluesrreturnrcCs@tt|dd|d<tt|dd|d<tt|dd|d<|S)z'Validate api key exists in environment.rZCLOVA_EMB_API_KEYrZCLOVA_EMB_APIGW_API_KEYrZCLOVA_EMB_APP_IDr )clsrr[/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/clova.pyvalidate_environmentBs  z$ClovaEmbeddings.validate_environmenttexts List[str]List[List[float]]cCs"g}|D] }|||q|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!Z embeddingstextrrrembed_documentsSs zClovaEmbeddings.embed_documentsr' List[float]cCs ||S)z Embed a single query text and return its embedding. Args: text: The text to embed. Returns: Embeddings for the text. )r%)r&r'rrr embed_querybs zClovaEmbeddings.embed_querycCsd|i}tt|jtt|jdd}tt|j}tj|jd|j d|||d}|j dkrI| }d|vrId|dvrI|ddSt d |j d |j ) zT Internal method to call the embedding API and handle the response. r'zapplication/json)zX-NCP-CLOVASTUDIO-API-KEYzX-NCP-APIGW-API-KEYz Content-Type/)headersjsonresultZ embeddingzAPI request failed with status z: )rrrZget_secret_valuerrrequestspostrr status_coder- ValueErrorr')r&r'payloadr,rresponseZ response_datarrrr%ns0   zClovaEmbeddings._embed_text)rrrr)r!r"rr#)r'rrr))__name__ __module__ __qualname____doc__r__annotations__rrrrrZ model_configr classmethodr r(r*r%rrrrr s,         r) __future__rtypingrrrrrr0Zlangchain_core._api.deprecationrZlangchain_core.embeddingsr Zlangchain_core.utilsr r Zpydanticr rrrrrrrrs