a bgy@sddlmZddlZddlZddlmZmZmZmZddl Z ddl Z ddl m Z ddl mZddlmZddlmZdd d d d d dZe ddddGdddeeZdS)) annotationsN)AnyDictListOptional) deprecated) Embeddings)pre_init) BaseModelstrzOptional[dict]rbool)urlheaderspayloadreturnc Csrz6tjd|||d}|jdkr$WdStd|jWn6tjjyl}ztd|WYd}~n d}~00dS)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 rrresponseerq/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/nemo.pyis_endpoint_lives rz0.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 d d d Z dddddddZ ddddddZ dddddZ dddddZ dddddZddd d!d"Zd#S)$NeMoEmbeddingszNeMo embedding models.int batch_sizezNV-Embed-QA-003r modelz#http://localhost:8088/v1/embeddingsapi_endpoint_urlr)valuesrcCs:|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%rrrrrvalidate_environment>s  z#NeMoEmbeddings.validate_environmentrz List[float])sessiontextr-rc sddi}|j|j||j|d|d4IdHF}||IdH}t|}|dddWdIdHS1IdHs~0YdS) zAsync call out to embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. r(r)r+)r.rNrr embedding)postr&r%raise_for_statusr3r.loads)selfr2r3r-rrZanswerrrr_aembedding_funcOs   zNeMoEmbeddings._aembedding_func)r3r-rcCsPt||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)rrrrr4)r.r/r%rrr&r7r3)r8r3r-rrrZ response_jsonr4rrr_embedding_funcgs    zNeMoEmbeddings._embedding_funcz List[str]zList[List[float]]) documentsrcsfdd|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:.0r3r8rr z2NeMoEmbeddings.embed_documents..r)r8r;rrArembed_documents~s zNeMoEmbeddings.embed_documents)r3rcCs|j|ddS)Nr*r=r>)r8r3rrr embed_queryszNeMoEmbeddings.embed_queryc sXt4IdH.}|||dIdH}|WdIdHS1IdHsJ0YdS)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 ClientSessionr9)r8r3r2r4rrr aembed_querys zNeMoEmbeddings.aembed_query)textsrc sg}t4IdHxtdt|jD]N}|||j}|D]2}fdd|D}tj|IdH}||qBq(WdIdHq1IdHs0Y|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<)r9r?r8r2rrrBsz3NeMoEmbeddings.aembed_documents..)rFrGrangelenr$asynciogatherextend)r8rIZ embeddingsbatchZ text_batchr3tasksZ batch_resultsrrJraembed_documentss  8zNeMoEmbeddings.aembed_documentsN)__name__ __module__ __qualname____doc__r$__annotations__r%r&r r1r9r:rDrErHrRrrrrr!.s     r!) __future__rrMr.typingrrrrrFrZlangchain_core._api.deprecationrZlangchain_core.embeddingsrZlangchain_core.utilsr Zpydanticr rr!rrrrs