From ecd826ee54b814022ede66e08f3798943689f74d Mon Sep 17 00:00:00 2001 From: Andre Violante Date: Fri, 30 Aug 2024 12:11:05 -0400 Subject: [PATCH] update docstring --- libs/ai-endpoints/langchain_nvidia_ai_endpoints/embeddings.py | 2 +- libs/ai-endpoints/langchain_nvidia_ai_endpoints/reranking.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/embeddings.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/embeddings.py index 3f1b1de9..428bdb84 100644 --- a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/embeddings.py +++ b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/embeddings.py @@ -91,7 +91,7 @@ def __init__(self, **kwargs: Any): Base URL: - Connect to a self-hosted model with NVIDIA NIM using the `base_url` arg to link to the local host at localhost:8000: - `embedder = NVIDIAEmbeddings(base_url="http://localhost:8080/v1")` + embedder = NVIDIAEmbeddings(base_url="http://localhost:8080/v1") """ super().__init__(**kwargs) self._client = _NVIDIAClient( diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/reranking.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/reranking.py index 00507da7..d1caa69f 100644 --- a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/reranking.py +++ b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/reranking.py @@ -83,7 +83,7 @@ def __init__(self, **kwargs: Any): Base URL: - Connect to a self-hosted model with NVIDIA NIM using the `base_url` arg to link to the local host at localhost:8000: - `ranker = NVIDIARerank(base_url="http://localhost:8000/v1")` + ranker = NVIDIARerank(base_url="http://localhost:8000/v1") Example: >>> from langchain_nvidia_ai_endpoints import NVIDIARerank