o Zh@sddlmZddlZddlZddlmZmZmZmZddl Z ddl Z ddl m Z ddl mZddlmZddlmZdddZe ddddGdddeeZdS)) annotationsN)AnyDictListOptional) deprecated) Embeddings)pre_init) BaseModelurlstrheadersOptional[dict]payloadrreturnboolc Cs^ztjd|||d}|jdkrWdStd|jtjjy.}ztd|d}~ww)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. POSTr dataTz0Endpoint returned a non-successful status code: zError querying the endpoint: N)requestsrequest status_code Exception exceptionsRequestException)r r rresponseerZ/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/nemo.pyis_endpoint_lives r z0.0.37z1.0.0zDirectly instantiating a NeMoEmbeddings from langchain-community is deprecated. Please use langchain-nvidia-ai-endpoints NVIDIAEmbeddings interface.)ZsinceZremovalmessagec@seZdZUdZdZded<dZded<dZded <ed%d dZ d&ddZ d'ddZ d(ddZ d)ddZ d)dd Zd*d"d#Zd$S)+NeMoEmbeddingszNeMo embedding models.int batch_sizezNV-Embed-QA-003r modelz#http://localhost:8088/v1/embeddingsapi_endpoint_urlvaluesrrcCs:|d}|d}ddi}td|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&r rrrrvalidate_environment>s z#NeMoEmbeddings.validate_environmentsessionrtextr. List[float]c sddi}|j|j||j|d|d4IdH#}||IdH}t|}|dddWdIdHS1IdHs@wYdS) zAsync call out to embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. r)r*r,)r/r Nrr embedding)postr'r&raise_for_statusr4r/loads)selfr3r4r.r rZanswerrrr_aembedding_funcSs   0zNeMoEmbeddings._aembedding_funccCsPt||j|d}ddi}tjd|j||d}t|j}|ddd}|S) zCall out to Cohere's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. r,r)r*rrrrr6)r/r0r&rrr'r9r4)r:r4r.rr rZ response_jsonr6rrr_embedding_funcks   zNeMoEmbeddings._embedding_func documents List[str]List[List[float]]csfdd|DS)zEmbed a list of document texts. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. csg|] }j|ddqS)passager.r<.0r4r:rr sz2NeMoEmbeddings.embed_documents..r)r:r=rrErembed_documentss zNeMoEmbeddings.embed_documentscCs|j|ddS)Nr+rArB)r:r4rrr embed_queryszNeMoEmbeddings.embed_queryc sZt4IdH}|||dIdH}|WdIdHS1IdHs&wYdS)zCall out to NeMo's embedding endpoint async for embedding query text. Args: text: The text to embed. Returns: Embedding for the text. Nr@)aiohttp ClientSessionr;)r:r4r3r6rrr aembed_querys  0zNeMoEmbeddings.aembed_querytextsc sg}t4IdH=tdt|jD]'}|||j}|D]}fdd|D}tj|IdH}||q"qWdIdH|S1IdHsNwY|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. Nrcsg|] }|dqS)r@)r;rCr:r3rrrFs z3NeMoEmbeddings.aembed_documents..)rIrJrangelenr%asynciogatherextend)r:rLZ embeddingsbatchZ text_batchr4tasksZ batch_resultsrrMraembed_documentss"   zNeMoEmbeddings.aembed_documentsN)r(rrr)r3rr4r r.r rr5)r4r r.r rr5)r=r>rr?)r4r rr5)rLr>rr?)__name__ __module__ __qualname____doc__r%__annotations__r&r'r r2r;r<rGrHrKrUrrrrr".s         r")r r r rrrrr) __future__rrPr/typingrrrrrIrZlangchain_core._api.deprecationrZlangchain_core.embeddingsrZlangchain_core.utilsr Zpydanticr r r"rrrrs"