o Zh@slddlZddlZddlZddlmZmZddlZddlmZddl m Z m Z e e ZGddde eZdS)N)AnyList) Embeddings) BaseModel ConfigDictcseZdZUdZ dZeed< dZeed< dZeed<e ddd Z d e ffd d Z d ede efddZde ede e efddZd ede efddZdededede fddZZS)OVHCloudEmbeddingsz+ OVHcloud AI Endpoints Embeddings. access_token model_nameZkeplerregionZforbid)extraZprotected_namespaceskwargsc sLtjdi||jdkrtd|jdkrtd|jdkr$tddS)Nrz1Access token is required for OVHCloud embeddings.z/Model name is required for OVHCloud embeddings.z+Region is required for OVHCloud embeddings.r )super__init__r ValueErrorr r )selfr __class__r ^/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/ovhcloud.pyrs   zOVHCloudEmbeddings.__init__textreturncCs|d|dS)zGenerate embeddings from OVHCLOUD AIE. Args: text (str): The text to embed. Returns: List[float]: Embeddings for the text. z text/plainZtext2vec)_send_request_to_ai_endpointsrrr r r_generate_embedding&sz&OVHCloudEmbeddings._generate_embeddingtextscCs|dt|dS)zEmbed a list of documents. Args: texts (List[str]): The list of texts to embed. Returns: List[List[float]]: List of embeddings, one for each input text. zapplication/jsonZbatch_text2vec)rjsondumps)rrr r rembed_documents0s z"OVHCloudEmbeddings.embed_documentscCs ||S)zEmbed a single query text. Args: text (str): The text to embed. Returns: List[float]: Embeddings for the text. )rrr r r embed_query>s zOVHCloudEmbeddings.embed_query contentTypepayloadroutecCs|d|jd}t} |jd|jd|jd|||d}|jdkr]|jd krG t|j d d }t d ||d krEt |q q |jd krQ td tdj|j|jd|S)a*Send a HTTPS request to OVHcloud AI Endpoints Args: contentType (str): The content type of the request, application/json or text/plain. payload (str): The payload of the request. route (str): The route of the request, batch_text2vec or text2vec. zBearer )z content-type AuthorizationTzhttps://z .endpoints.z.ai.cloud.ovh.net/api/)headersdataizRateLimit-Resetrz(Rate limit exceeded. Waiting %d seconds.iz"Unauthorized, retry with new tokenz6Request failed with status code: {status_code}, {text}) status_coder)r requestssessionpostr r r'intr$getloggerinfotimesleeprformatrr)rr r!r"r$r)responseZ reset_timer r rrGs@      z0OVHCloudEmbeddings._send_request_to_ai_endpoints)__name__ __module__ __qualname____doc__r str__annotations__r r rZ model_configrrrfloatrrrr __classcell__r r rrr s,        r)rloggingr/typingrrr(Zlangchain_core.embeddingsrZpydanticrr getLoggerr3r-rr r r rs