linting fix

langchain-ai · Nov 21, 2024 · 2296d31 · 2296d31
1 parent a254d5e
commit 2296d31
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_common.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_common.py
@@ -103,7 +103,7 @@ class _NVIDIAClient(BaseModel):
     last_inputs: Optional[dict] = Field(
         default={}, description="Last inputs sent over to the server"
     )
-    last_response: Response = Field(
+    last_response: Optional[Response] = Field(
         None, description="Last response sent from the server"
     )
     headers_tmpl: dict = Field(

diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/chat_models.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/chat_models.py
@@ -253,7 +253,7 @@ class ChatNVIDIA(BaseChatModel):
             response = model.invoke("Hello")
     """
 
-    _client: _NVIDIAClient = PrivateAttr(_NVIDIAClient)
+    _client: _NVIDIAClient = PrivateAttr(default=_NVIDIAClient())
     base_url: Optional[str] = Field(
         default=None,
         description="Base url for model listing an invocation",

diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/embeddings.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/embeddings.py
@@ -34,7 +34,7 @@ class NVIDIAEmbeddings(BaseModel, Embeddings):
         validate_assignment=True,
     )
 
-    _client: _NVIDIAClient = PrivateAttr(_NVIDIAClient)
+    _client: _NVIDIAClient = PrivateAttr(default=_NVIDIAClient())
     base_url: Optional[str] = Field(
         default=None,
         description="Base url for model listing an invocation",

diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/llm.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/llm.py
@@ -23,7 +23,7 @@ class NVIDIA(LLM):
         validate_assignment=True,
     )
 
-    _client: _NVIDIAClient = PrivateAttr(_NVIDIAClient)
+    _client: _NVIDIAClient = PrivateAttr(default=_NVIDIAClient())
     _default_model_name: str = "nvidia/mistral-nemo-minitron-8b-base"
     base_url: Optional[str] = Field(
         default=None,

diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/reranking.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/reranking.py
@@ -34,7 +34,7 @@ class NVIDIARerank(BaseDocumentCompressor):
         validate_assignment=True,
     )
 
-    _client: _NVIDIAClient = PrivateAttr(_NVIDIAClient)
+    _client: _NVIDIAClient = PrivateAttr(default=_NVIDIAClient())
 
     base_url: Optional[str] = Field(
         default=None,