Remove unnecessary import and fix name conflict warning.

Author: drazvan

nemoguardrails/llm/providers/_langchain_nvidia_ai_endpoints_patch.py

@@ -22,7 +22,7 @@
 from langchain_core.messages import BaseMessage
 from langchain_core.outputs import ChatResult
 from langchain_core.pydantic_v1 import Field
-from langchain_nvidia_ai_endpoints import ChatNVIDIA
+from langchain_nvidia_ai_endpoints import ChatNVIDIA as ChatNVIDIAOriginal
 
 log = logging.getLogger(__name__)
 
@@ -49,7 +49,9 @@ def wrapper(
     return wrapper
 
 
-class ChatNVIDIA(ChatNVIDIA):
+# NOTE: this needs to have the same name as the original class,
+#   otherwise, there's a check inside `langchain-nvidia-ai-endpoints` that will fail.
+class ChatNVIDIA(ChatNVIDIAOriginal):
     streaming: bool = Field(
         default=False, description="Whether to use streaming or not"
     )

nemoguardrails/llm/providers/providers.py

@@ -243,8 +243,6 @@ def get_llm_provider(model_config: Model) -> Type[BaseLanguageModel]:
             )
     elif model_config.engine == "nvidia_ai_endpoints" or model_config.engine == "nim":
         try:
-            from langchain_nvidia_ai_endpoints import ChatNVIDIA
-
             from ._langchain_nvidia_ai_endpoints_patch import ChatNVIDIA
 
             # Check the version