7|hddlmZddlZddlZddlmZmZmZmZddl Z ddl Z ddl m Z ddl mZddlmZddlmZddZe d d d Gd deeZy)) annotationsN)AnyDictListOptional) deprecated) Embeddings)pre_init) BaseModelc tjd|||}|jdk(rytd|j#tjj $r}td|d}~wwxYw)a Check if an endpoint is live by sending a GET request to the specified URL. Args: url (str): The URL of the endpoint to check. Returns: bool: True if the endpoint is live (status code 200), False otherwise. Raises: Exception: If the endpoint returns a non-successful status code or if there is an error querying the endpoint. POSTheadersdataTz0Endpoint returned a non-successful status code: zError querying the endpoint: N)requestsrequest status_code Exception exceptionsRequestException)urlrpayloadresponsees b/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/nemo.pyis_endpoint_livers=##FCwO   3 &B''(*     / /=7s;<<=s(AAA3 A..A3z0.0.37z1.0.0zDirectly instantiating a NeMoEmbeddings from langchain-community is deprecated. Please use langchain-nvidia-ai-endpoints NVIDIAEmbeddings interface.)sinceremovalmessageceZdZUdZdZded<dZded<dZded <edd Z dd Z dd Z dd Z ddZ ddZddZy)NeMoEmbeddingszNeMo embedding models.int batch_sizezNV-Embed-QA-003strmodelz#http://localhost:8088/v1/embeddingsapi_endpoint_urlcn|d}|d}ddi}tjd|dd}t||||S)zHValidate that the end point is alive using the values that are provided.r(r' Content-Typeapplication/jsonz Hello Worldqueryinputr' input_type)jsondumpsr)clsvaluesrr'rrs rvalidate_environmentz#NeMoEmbeddings.validate_environment>sT'(w"#56**&%   gw/ cdKddi}|j|j||j|d|4d{}|j|j d{}t j |}|dddcdddd{S7]777 #1d{7swYyxYww) zAsync call out to embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. r*r+r-)r0rNrr embedding)postr(r'raise_for_statustextr0loads)selfsessionr:r/rranswers r_aembedding_funczNeMoEmbeddings._aembedding_funcSs"#56<<  ! !$**JO   2 2  % % '#==?*FZZ'F&>!$[1 2 2 2 +  2 2 2 2sV5B0BB0$BB #B B0BB0BB0B-!B$ "B-)B0ctj||j|d}ddi}tjd|j ||}tj |j}|ddd}|S) zCall out to Cohere's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. r-r*r+r rrrr7)r0r1r'rrr(r;r:)r<r:r/rrr response_jsonr7s r_embedding_funczNeMoEmbeddings._embedding_funcks~**(  "#56## D))7  8==1 !&)!,[9 r5cN|Dcgc]}|j|dc}Scc}w)zEmbed a list of document texts. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. passager/rB)r< documentsr:s rembed_documentszNeMoEmbeddings.embed_documentss*NWWT$$Ti$@WWWs"c(|j|dS)Nr,rErF)r<r:s r embed_queryzNeMoEmbeddings.embed_querys##DW#==r5cKtj4d{}|j||dd{}|cdddd{S7277 #1d{7swYyxYww)zCall out to NeMo's embedding endpoint async for embedding query text. Args: text: The text to embed. Returns: Embedding for the text. NrD)aiohttp ClientSessionr?)r<r:r=r7s r aembed_queryzNeMoEmbeddings.aembed_querysc((*  g"33GT9MMI   M    sSA)AA)AAA A)A A)AA)A&A A&"A)c Kg}tj4d{}tdt||jD]f}||||jz}|D]M}|Dcgc]}|j ||d}}t j|d{}|j|Ohdddd{|S7cc}w727#1d{7swY|SxYww)zCall out to NeMo's embedding endpoint async for embedding search docs. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. NrrD) rLrMrangelenr%r?asynciogatherextend) r<texts embeddingsr=batch text_batchr:tasks batch_resultss raembed_documentszNeMoEmbeddings.aembed_documentss ((* 5 5gq#e*doo> 5"554??+BC & 5D%/ --gtYGE +2..%*@$@M%%m4 5 5 5 5"# 5  %A 5 5 5 5"sbCB=C?C B? :CC C+ C6C7C?CCCC CCN)r3rreturnr)r=rr:r&r/r&r\ List[float])r:r&r/r&r\r])rG List[str]r\List[List[float]])r:r&r\r])rUr^r\r_)__name__ __module__ __qualname____doc__r%__annotations__r'r(r r4r?rBrHrJrNr[r5rr"r".sw!J"E3"AcA (22"%2362 206 X> r5r")rr&rzOptional[dict]rrr\bool) __future__rrRr0typingrrrrrLrlangchain_core._api.deprecationrlangchain_core.embeddingsr langchain_core.utilsr pydanticr rr"rer5rrms[" ,,60)=>    GY GGr5