update langchain's BaseLLM and BaseChatModel imports

Pouyanpi · Pouyanpi · commit 7485d0656333 · 2025-10-27T17:12:51.000+01:00
Moved all imports of BaseLLM and BaseChatModel to
langchain_core.language_models for consistency and compatibility with
latest langchain-core structure. This removes usage of deprecated
langchain_core.language_models.llms and langchain.chat_models.base
imports across actions, helpers, providers, and tests.
diff --git a/nemoguardrails/actions/llm/generation.py b/nemoguardrails/actions/llm/generation.py
@@ -28,8 +28,7 @@
 
 from jinja2 import meta
 from jinja2.sandbox import SandboxedEnvironment
-from langchain_core.language_models import BaseChatModel
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseChatModel, BaseLLM
 
 from nemoguardrails.actions.actions import ActionResult, action
 from nemoguardrails.actions.llm.utils import (
diff --git a/nemoguardrails/actions/v2_x/generation.py b/nemoguardrails/actions/v2_x/generation.py
@@ -21,8 +21,7 @@
 from ast import literal_eval
 from typing import Any, Dict, List, Optional, Tuple, Union, cast
 
-from langchain_core.language_models import BaseChatModel
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseChatModel, BaseLLM
 from rich.text import Text
 
 from nemoguardrails.actions.actions import action
diff --git a/nemoguardrails/library/content_safety/actions.py b/nemoguardrails/library/content_safety/actions.py
@@ -16,7 +16,7 @@
 import logging
 from typing import Dict, Optional
 
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 
 from nemoguardrails.actions.actions import action
 from nemoguardrails.actions.llm.utils import llm_call
diff --git a/nemoguardrails/library/factchecking/align_score/actions.py b/nemoguardrails/library/factchecking/align_score/actions.py
@@ -16,7 +16,7 @@
 import logging
 from typing import Optional
 
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 
 from nemoguardrails import RailsConfig
 from nemoguardrails.actions import action
diff --git a/nemoguardrails/library/hallucination/actions.py b/nemoguardrails/library/hallucination/actions.py
@@ -16,7 +16,7 @@
 import logging
 from typing import Optional
 
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 from langchain_core.prompts import PromptTemplate
 
 from nemoguardrails import RailsConfig
diff --git a/nemoguardrails/library/llama_guard/actions.py b/nemoguardrails/library/llama_guard/actions.py
@@ -16,7 +16,7 @@
 import logging
 from typing import List, Optional, Tuple
 
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 
 from nemoguardrails.actions import action
 from nemoguardrails.actions.llm.utils import llm_call
diff --git a/nemoguardrails/library/patronusai/actions.py b/nemoguardrails/library/patronusai/actions.py
@@ -19,7 +19,7 @@
 from typing import List, Literal, Optional, Tuple, Union
 
 import aiohttp
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 
 from nemoguardrails.actions import action
 from nemoguardrails.actions.llm.utils import llm_call
diff --git a/nemoguardrails/library/self_check/facts/actions.py b/nemoguardrails/library/self_check/facts/actions.py
@@ -16,7 +16,7 @@
 import logging
 from typing import Optional
 
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 
 from nemoguardrails import RailsConfig
 from nemoguardrails.actions import action
diff --git a/nemoguardrails/library/self_check/input_check/actions.py b/nemoguardrails/library/self_check/input_check/actions.py
@@ -16,7 +16,7 @@
 import logging
 from typing import Optional
 
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 
 from nemoguardrails import RailsConfig
 from nemoguardrails.actions.actions import ActionResult, action
diff --git a/nemoguardrails/library/self_check/output_check/actions.py b/nemoguardrails/library/self_check/output_check/actions.py
@@ -16,7 +16,7 @@
 import logging
 from typing import Optional
 
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 
 from nemoguardrails import RailsConfig
 from nemoguardrails.actions import action
diff --git a/nemoguardrails/library/topic_safety/actions.py b/nemoguardrails/library/topic_safety/actions.py
@@ -16,7 +16,7 @@
 import logging
 from typing import Dict, List, Optional
 
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 
 from nemoguardrails.actions.actions import action
 from nemoguardrails.actions.llm.utils import llm_call
diff --git a/nemoguardrails/llm/helpers.py b/nemoguardrails/llm/helpers.py
@@ -19,7 +19,7 @@
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
 )
-from langchain_core.language_models.llms import LLM, BaseLLM
+from langchain_core.language_models import LLM, BaseLLM
 
 
 def get_llm_instance_wrapper(
diff --git a/nemoguardrails/llm/models/initializer.py b/nemoguardrails/llm/models/initializer.py
@@ -17,8 +17,7 @@
 
 from typing import Any, Dict, Literal, Optional, Union
 
-from langchain_core.language_models import BaseChatModel
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseChatModel, BaseLLM
 
 from .langchain_initializer import ModelInitializationError, init_langchain_model
 
diff --git a/nemoguardrails/llm/models/langchain_initializer.py b/nemoguardrails/llm/models/langchain_initializer.py
@@ -24,8 +24,7 @@
 
 # from langchain_core._api.beta_decorator import LangChainBetaWarning
 # from langchain_core._api.deprecation import LangChainDeprecationWarning
-from langchain_core.language_models import BaseChatModel
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseChatModel, BaseLLM
 
 from nemoguardrails.llm.providers.providers import (
     _get_chat_completion_provider,
diff --git a/nemoguardrails/llm/providers/providers.py b/nemoguardrails/llm/providers/providers.py
@@ -27,10 +27,9 @@
 import warnings
 from typing import Dict, List, Set, Type
 
-from langchain.chat_models.base import BaseChatModel
 from langchain_community import llms
 from langchain_community.chat_models import _module_lookup
-from langchain_core.language_models import BaseLLM
+from langchain_core.language_models import BaseChatModel, BaseLLM
 
 from .trtllm.llm import TRTLLM
 
diff --git a/nemoguardrails/llm/providers/trtllm/llm.py b/nemoguardrails/llm/providers/trtllm/llm.py
@@ -22,7 +22,7 @@
 from typing import Any, Dict, List, Optional
 
 from langchain_core.callbacks.manager import CallbackManagerForLLMRun
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 from pydantic.v1 import Field, root_validator
 
 from nemoguardrails.llm.providers.trtllm.client import TritonClient
diff --git a/nemoguardrails/rails/llm/llmrails.py b/nemoguardrails/rails/llm/llmrails.py
@@ -37,8 +37,7 @@
     cast,
 )
 
-from langchain_core.language_models import BaseChatModel
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseChatModel, BaseLLM
 from typing_extensions import Self
 
 from nemoguardrails.actions.llm.generation import LLMGenerationActions
diff --git a/tests/llm_providers/test_langchain_integration.py b/tests/llm_providers/test_langchain_integration.py
@@ -18,8 +18,7 @@
 from unittest.mock import MagicMock, patch
 
 import pytest
-from langchain.chat_models.base import BaseChatModel
-from langchain_core.language_models import BaseLLM
+from langchain_core.language_models import BaseChatModel, BaseLLM
 
 from nemoguardrails.llm.models.langchain_initializer import init_langchain_model
 from nemoguardrails.llm.providers.providers import (
diff --git a/tests/llm_providers/test_langchain_special_cases.py b/tests/llm_providers/test_langchain_special_cases.py
@@ -24,8 +24,7 @@
 from unittest.mock import patch
 
 import pytest
-from langchain.chat_models.base import BaseChatModel
-from langchain_core.language_models import BaseLLM
+from langchain_core.language_models import BaseChatModel, BaseLLM
 
 from nemoguardrails.llm.models.langchain_initializer import (
     _PROVIDER_INITIALIZERS,
diff --git a/tests/llm_providers/test_providers.py b/tests/llm_providers/test_providers.py
@@ -18,8 +18,7 @@
 from unittest.mock import MagicMock, patch
 
 import pytest
-from langchain.chat_models.base import BaseChatModel
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseChatModel, BaseLLM
 
 from nemoguardrails.llm.providers.providers import (
     _acall,
diff --git a/tests/runnable_rails/test_metadata.py b/tests/runnable_rails/test_metadata.py
@@ -19,11 +19,11 @@
 from unittest.mock import Mock
 
 import pytest
-from langchain.chat_models.base import BaseChatModel
 from langchain_core.callbacks.manager import (
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
 )
+from langchain_core.language_models import BaseChatModel
 from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
 from langchain_core.outputs import ChatGeneration, ChatResult
 from langchain_core.prompts import ChatPromptTemplate
diff --git a/tests/test_configs/with_custom_chat_model/custom_chat_model.py b/tests/test_configs/with_custom_chat_model/custom_chat_model.py
@@ -15,11 +15,11 @@
 
 from typing import List, Optional
 
-from langchain.chat_models.base import BaseChatModel
 from langchain_core.callbacks.manager import (
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
 )
+from langchain_core.language_models import BaseChatModel
 
 
 class CustomChatModel(BaseChatModel):
diff --git a/tests/test_configs/with_custom_llm/custom_llm.py b/tests/test_configs/with_custom_llm/custom_llm.py
@@ -19,7 +19,7 @@
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
 )
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 from langchain_core.outputs import Generation, LLMResult
 
 
diff --git a/tests/test_configs/with_custom_llm_prompt_action_v2_x/actions.py b/tests/test_configs/with_custom_llm_prompt_action_v2_x/actions.py
@@ -16,7 +16,7 @@
 from ast import literal_eval
 from typing import Optional
 
-from langchain_core.language_models.llms import BaseLLM
+from langchain_core.language_models import BaseLLM
 
 from nemoguardrails.actions import action
 from nemoguardrails.actions.llm.utils import llm_call
diff --git a/tests/test_llmrails.py b/tests/test_llmrails.py
@@ -790,7 +790,7 @@ async def test_main_llm_from_config_registered_as_action_param(
     is initialized from the config, it gets properly registered as an action parameter.
     This prevents the regression where actions expecting an 'llm' parameter would receive None.
     """
-    from langchain_core.language_models.llms import BaseLLM
+    from langchain_core.language_models import BaseLLM
 
     from nemoguardrails.actions import action
 
diff --git a/tests/test_streaming.py b/tests/test_streaming.py
@@ -572,8 +572,7 @@ async def test_streaming_error_handling():
 @pytest.fixture
 def custom_streaming_providers():
     """Fixture that registers both custom chat and LLM providers for testing."""
-    from langchain.chat_models.base import BaseChatModel
-    from langchain_core.language_models.llms import BaseLLM
+    from langchain_core.language_models import BaseChatModel, BaseLLM
 
     from nemoguardrails.llm.providers import (
         register_chat_provider,
diff --git a/tests/utils.py b/tests/utils.py
@@ -24,7 +24,7 @@
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
 )
-from langchain_core.language_models.llms import LLM
+from langchain_core.language_models import LLM
 
 from nemoguardrails import LLMRails, RailsConfig
 from nemoguardrails.colang import parse_colang_file
