diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/chat_models.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/chat_models.py index 63ee21c5..06ddca92 100644 --- a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/chat_models.py +++ b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/chat_models.py @@ -313,7 +313,8 @@ def __init__(self, **kwargs: Any): default_hosted_model_name=_DEFAULT_MODEL_NAME, **({"api_key": api_key} if api_key else {}), # only pass if set infer_path="{base_url}/chat/completions", - cls=self.__class__.__name__, + # instead of self.__class__.__name__ to assist in subclassing ChatNVIDIA + cls="ChatNVIDIA", ) # todo: only store the model in one place # the model may be updated to a newer name during initialization