7|hddlZddlZddlZddlmZmZddlZddlmZddl m Z m Z eje ZGdde eZy)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 t|di||jdk(r td|jdk(r td|j dk(r tdy)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 f/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/ovhcloud.pyrzOVHCloudEmbeddings.__init__sb "6"    "PQ Q ??b NO O ;;" JK K textreturnc(|jd|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,jQQrtextscN|jdtj|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 13C  rc$|j|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 contentTypepayloadroutecX|d|jd}tj} |jd|jd|j d|||}|j dk7r|j dk(rY t|jjd d }tjd ||d kDrtj| |j d k(r td  tdj|j |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 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_coder)r requestssessionpostr rr3intr0getloggerinfotimesleeprformatrr&)rr+r,r-r0r5response reset_times rr z0OVHCloudEmbeddings._send_request_to_ai_endpointsGs7(&t'8'8&9:  ""$||t/{4;;-,UG5  $H##s*''3.=!$X%5%5%9%9:KQ%O!PJKK JJW!A~ :. Q ''3.>$%IJJ9 LSS$,$8$8x}}T ==? "r)__name__ __module__ __qualname____doc__r str__annotations__r rr model_configrrrfloatr"r(r*r __classcell__)rs@rr r s.L#EJ'FCH2FLLLRRU R  T#Y  4U 3D  ..U .,#,#),,#58,# ,#rr )r&loggingr;typingrrr4langchain_core.embeddingsrpydanticrr getLoggerr@r9r rrrrNs<  0*   8 $f#Jf#r