langchain_nvidia_ai_endpoints 0.1.2

langchain_nvidia_ai_endpoints.callbacks

Callback Handler that prints to std out.

Classes

callbacks.UsageCallbackHandler()

Callback Handler that tracks OpenAI info.

Functions

callbacks.get_token_cost_for_model(...[, ...])

Get the cost in USD for a given model and number of tokens.

callbacks.get_usage_callback([price_map, ...])

Get the OpenAI callback handler in a context manager.

callbacks.standardize_model_name(model_name)

Standardize the model name to a format that can be used in the OpenAI API.

langchain_nvidia_ai_endpoints.chat_models

Chat Model Components Derived from ChatModel/NVIDIA

Classes

chat_models.ChatNVIDIA

NVIDIA chat model.

langchain_nvidia_ai_endpoints.embeddings

Embeddings Components Derived from NVEModel/Embeddings

Classes

embeddings.NVIDIAEmbeddings

Client to NVIDIA embeddings models.

langchain_nvidia_ai_endpoints.reranking

Classes

reranking.NVIDIARerank

LangChain Document Compressor that uses the NVIDIA NeMo Retriever Reranking API.

reranking.Ranking

Create a new model by parsing and validating input data from keyword arguments.

langchain_nvidia_ai_endpoints.tools

OpenAI chat wrapper.

Classes

tools.ServerToolsMixin()