NgddlZddlZddlZddlmZmZddlZddlmZddl m Z m Z ej e ZGdde eZdS)N)AnyList) Embeddings) BaseModel ConfigDictceZdZUdZ dZeed< dZeed< dZeed<e dd Z d e ffd Z d ed e efdZde ed e e efdZd ed e efdZdededed e fdZxZS)OVHCloudEmbeddingsz+ OVHcloud AI Endpoints Embeddings. access_token model_namekeplerregionforbid)extraprotected_namespaceskwargsc tjdi||jdkrtd|jdkrtd|jdkrtddS)Nr z1Access 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__s c/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/ovhcloud.pyrzOVHCloudEmbeddings.__init__s}""6"""   " "PQQ Q ?b NOO O ;"  JKK K  textreturnc0|d|dS)zGenerate embeddings from OVHCLOUD AIE. Args: text (str): The text to embed. Returns: List[float]: Embeddings for the text. z text/plaintext2vec)_send_request_to_ai_endpointsrrs r_generate_embeddingz&OVHCloudEmbeddings._generate_embedding&s11,jQQQrtextscT|dtj|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/jsonbatch_text2vec)r jsondumps)rr#s rembed_documentsz"OVHCloudEmbeddings.embed_documents0s.11  5 1 13C   rc,||S)zEmbed a single query text. Args: text (str): The text to embed. Returns: List[float]: Embeddings for the text. )r"r!s r embed_queryzOVHCloudEmbeddings.embed_query>s''---r contentTypepayloadroutecD|d|jd}tj} |d|jd|jd|||}|jdkr|jd kra t|j d d }t d ||d krtj | |jd kr td td|j|j|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 rr3intr0getloggerinfotimesleeprformatrr&)rr+r,r-r0r5response reset_times rr z0OVHCloudEmbeddings._send_request_to_ai_endpointsGse(:t'8::   "$$ #||5t554;55-255  $H#s**'3..=!$X%5%9%9:KQ%O%O!P!PJKK JJWWW!A~~ :... Q '3..>$%IJJJ9 LSS$,$8x}T ==?? "r)__name__ __module__ __qualname____doc__r str__annotations__r rr model_configrrrfloatr"r(r*r __classcell__)rs@rr r s[.L#EJ'FC:H2FFFLLLLLLLLRRU RRRR  T#Y  4U 3D     ..U ....,#,#),,#58,# ,#,#,#,#,#,#,#,#rr )r&loggingr;typingrrr4langchain_core.embeddingsrpydanticrr getLoggerr@r9r rrrrNs  000000********  8 $ $f#f#f#f#f#Jf#f#f#f#f#r