U ™ÀÂhJ ã@s`ddlZddlZddlmZmZddlZddlmZddlm Z e  e ¡Z Gdd„de eƒZ dS)éN)ÚAnyÚList)Ú Embeddings)Ú BaseModelcsžeZdZUdZdZeed<dZeed<dZeed<Gdd„dƒZ e d œ‡fd d „ Z ee e d œd d„Ze ee e e dœdd„Zee e d œdd„Z‡ZS)ÚOVHCloudEmbeddingsz+ OVHcloud AI Endpoints Embeddings. ÚÚ access_tokenÚ model_nameZkeplerÚregionc@seZdZdZdS)zOVHCloudEmbeddings.ConfigZforbidN)Ú__name__Ú __module__Ú __qualname__Úextra©rrúK/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/ovhcloud.pyÚConfigsr)Úkwargsc sHtƒjf|Ž|jdkr tdƒ‚|jdkr2tdƒ‚|jdkrDtdƒ‚dS)Nrz1Access token is required for OVHCloud embeddings.z/Model name is required for OVHCloud embeddings.z+Region is required for OVHCloud embeddings.)ÚsuperÚ__init__rÚ ValueErrorr r )Úselfr©Ú __class__rrrs   zOVHCloudEmbeddings.__init__)ÚtextÚreturncCs¶dd|j›dœ}t ¡}|jd|j›d|j›d||d}|jdkr®|jd kr†t|j  d d ¡ƒ}t   d |¡|d krt   |¡qnq|jd kr˜tdƒ‚tdj|j|jdƒ‚| ¡S)z«Generate embeddings from OVHCLOUD AIE. Args: text (str): The text to embed. Returns: List[float]: Embeddings for the text. z text/plainzBearer )z content-typeÚ Authorizationzhttps://z .endpoints.z.ai.cloud.ovh.net/api/text2vec)ÚheadersÚdataéÈi­zRateLimit-Resetrz(Rate limit exceeded. Waiting %d seconds.i‘z"Unauthorized, retry with new tokenz6Request failed with status code: {status_code}, {text})Ú status_coderN)rÚrequestsÚsessionÚpostr r rÚintrÚgetÚloggerÚinfoÚtimeÚsleeprÚformatrÚjson)rrrr!ÚresponseZ reset_timerrrÚ_generate_embedding&s4 þý     ÿÿz&OVHCloudEmbeddings._generate_embedding)Útextsrcs‡fdd„|DƒS)zÜCreate a retry decorator for PremAIEmbeddings. Args: texts (List[str]): The list of texts to embed. Returns: List[List[float]]: List of embeddings, one for each input text. csg|]}ˆ |¡‘qSr©r,)Ú.0r©rrrÚ Wsz6OVHCloudEmbeddings.embed_documents..r)rr-rr0rÚembed_documentsOsz"OVHCloudEmbeddings.embed_documentscCs | |¡S)zŸEmbed a single query text. Args: text (str): The text to embed. Returns: List[float]: Embeddings for the text. r.)rrrrrÚ embed_queryYszOVHCloudEmbeddings.embed_query)r r r Ú__doc__rÚstrÚ__annotations__r r rrrrÚfloatr,r2r3Ú __classcell__rrrrr s     ) r)Úloggingr'Útypingrrr Zlangchain_core.embeddingsrZlangchain_core.pydantic_v1rÚ getLoggerr r%rrrrrÚs