a bg@slddlZddlZddlZddlmZmZddlZddlmZddl m Z m Z e e ZGddde eZdS)N)AnyList) Embeddings) BaseModel ConfigDictcseZdZUdZdZeed<dZeed<dZeed<e ddd Z e d fd d Z ee ed ddZe ee e edddZee ed ddZeeee dddZZS)OVHCloudEmbeddingsz+ OVHcloud AI Endpoints Embeddings. access_token model_nameZkeplerregionZforbid)extraZprotected_namespaces)kwargsc sLtjfi||jdkr$td|jdkr6td|jdkrHtddS)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__r u/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/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_embedding)textsrcCs|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>szOVHCloudEmbeddings.embed_query) contentTypepayloadroutercCs|d|jd}t}|jd|jd|jd|||d}|jdkr|jdkrt|j d d }t d ||d krt |qnq|jd krtd 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 Authorizationzhttps://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_coderN)r requestssessionpostr r r'intr$getloggerinfotimesleeprformatrr)rr r!r"r$r)responseZ reset_timer r rrGs6       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