U h@srddlmZddlmZmZmZmZddlZddlm Z ddl m Z m Z m Z ddlmZmZGddde e ZdS) ) annotations)DictListOptionalcastN) Embeddings) BaseModel SecretStrroot_validatorconvert_to_secret_strget_from_dict_or_envc@seZdZUdZdZded<dZded<dZded <dZded <dZ ded <Gd d d Z e dddddddZ dddddZ dddddZdddddZdS)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_idc@seZdZdZdS)zClovaEmbeddings.ConfigZforbidN)__name__ __module__ __qualname__extrarrH/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/clova.pyConfig8srT)prer)valuesreturncCs@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 )clsrrrrvalidate_environment;s  z$ClovaEmbeddings.validate_environmentz List[str]zList[List[float]])textsrcCs"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_documentsKs zClovaEmbeddings.embed_documentsz List[float])r%rcCs ||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_queryZs zClovaEmbeddings.embed_querycCsd|i}tt|jtt|jdd}tt|j}tj|jd|j d|||d}|j dkr| }d|krd|dkr|ddSt d |j d |j d S) 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: N)rr rZget_secret_valuerrrequestspostrr status_coder* ValueErrorr%)r$r%payloadr)rresponseZ response_datarrrr#fs0   zClovaEmbeddings._embed_text)rrr__doc__r__annotations__rrrrrr r r&r'r#rrrrr s        r) __future__rtypingrrrrr-Zlangchain_core.embeddingsrZlangchain_core.pydantic_v1rr r Zlangchain_core.utilsr r rrrrrs