NgyddlmZddlZddlZddlmZmZmZmZddl Z ddl Z ddl m Z ddl mZddlmZddlmZddZe dddGddeeZdS)) annotationsN)AnyDictListOptional) deprecated) Embeddings)pre_init) BaseModelurlstrheadersOptional[dict]payloadrreturnboolc tjd|||}|jdkrdStd|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. POSTrdataTz0Endpoint returned a non-successful status code: zError querying the endpoint: N)requestsrequest status_code Exception exceptionsRequestException)r rrresponsees _/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/nemo.pyis_endpoint_liver!s=#FCwOOO  3 & &4*'**    /===;;;<<<=s#>>A)A$$A)z0.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 ddZ d dZ d!dZ d"dZ d"dZd#dZdS)$NeMoEmbeddingszNeMo embedding models.int batch_sizezNV-Embed-QA-003r modelz#http://localhost:8088/v1/embeddingsapi_endpoint_urlvaluesrrc|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!)clsr,r r*rrs r validate_environmentz#NeMoEmbeddings.validate_environment>s\'(w"#56*#e7 K K   gw/// sessionrtextr3 List[float]c`Kddi}||j||j|d|4d{V }||d{V}t j|}|dddcdddd{VS#1d{VswxYwYdS) zAsync call out to embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. r.r/r1)r4rNrr embedding)postr+r*raise_for_statusr:r4loads)selfr9r:r3rranswers r _aembedding_funczNeMoEmbeddings._aembedding_funcOsX"#56<<  !$*JOO    2 2 2 2 2 2 2 2  % % ' ' '#==??******FZ''F&>!$[1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2sAB B'*B'ctj||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. r1r.r/rrrrr=)r4r5r*rrr+r@r:)rAr:r3rrr response_jsonr=s r _embedding_funczNeMoEmbeddings._embedding_funcgs*TZz J J  "#56# D)7    8=11 !&)!,[9 r8 documents List[str]List[List[float]]c fd|DS)zEmbed a list of document texts. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. c>g|]}|dS)passager3rF).0r:rAs r z2NeMoEmbeddings.embed_documents..s,WWWT$$Ti$@@WWWr8)rArGs` r embed_documentszNeMoEmbeddings.embed_documents~s XWWWYWWWWr8c0||dS)Nr0rMrN)rAr:s r embed_queryzNeMoEmbeddings.embed_querys##DW#===r8cKtj4d{V}|||dd{V}|cdddd{VS#1d{VswxYwYdS)zCall out to NeMo's embedding endpoint async for embedding query text. Args: text: The text to embed. Returns: Embedding for the text. NrL)aiohttp ClientSessionrC)rAr:r9r=s r aembed_queryzNeMoEmbeddings.aembed_querys(**       g"33GT9MMMMMMMMI                              sA AAtextscxKg}tj4d{Vtdt|jD]Q}|||jz}|D]:}fd|D}t j|d{V}||;R dddd{Vn#1d{VswxYwY|S)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. Nrc>g|]}|dS)rL)rC)rOr:rAr9s r rPz3NeMoEmbeddings.aembed_documents..s; --gtYGGr8)rVrWrangelenr)asynciogatherextend) rArY embeddingsbatch text_batchr:tasks batch_resultsr9s ` @r aembed_documentszNeMoEmbeddings.aembed_documentss (** 5 5 5 5 5 5 5gq#e**do>> 5 5"554?+B#BC & 5 5D$.E +2.%*@$@$@$@$@$@$@M%%m4444 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5"sA6B)) B36B3N)r,rrr)r9rr:r r3r rr;)r:r r3r rr;)rGrHrrI)r:r rr;)rYrHrrI)__name__ __module__ __qualname____doc__r)__annotations__r*r+r r7rCrFrRrTrXrfrQr8r r&r&.s! J"E""""AAAAA X 22220. X X X X>>>>    r8r&)r r rrrrrr) __future__rr^r4typingrrrrrVrlangchain_core._api.deprecationrlangchain_core.embeddingsr langchain_core.utilsr pydanticr r!r&rQr8r rrs&"""""" ,,,,,,,,,,,,666666000000))))))====>     Y r8