nit

langchain-ai · Dec 6, 2023 · 100b7c9 · 100b7c9
1 parent 867ca6d
commit 100b7c9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 9 deletions.
diff --git a/libs/langchain/langchain/llms/watsonxllm.py b/libs/langchain/langchain/llms/watsonxllm.py
@@ -5,7 +5,7 @@
 from langchain.callbacks.manager import CallbackManagerForLLMRun
 from langchain.llms.base import BaseLLM
 from langchain.pydantic_v1 import Extra, SecretStr, root_validator
-from langchain.schema import LLMResult
+from langchain_core.outputs import LLMResult
 from langchain.schema.output import Generation, GenerationChunk
 from langchain.utils import convert_to_secret_str, get_from_dict_or_env
 

diff --git a/libs/langchain/tests/integration_tests/chat_models/test_volcengine_maas.py b/libs/langchain/tests/integration_tests/chat_models/test_volcengine_maas.py
@@ -2,13 +2,8 @@
 
 from langchain.callbacks.manager import CallbackManager
 from langchain.chat_models.volcengine_maas import VolcEngineMaasChat
-from langchain.schema import (
-    AIMessage,
-    BaseMessage,
-    ChatGeneration,
-    HumanMessage,
-    LLMResult,
-)
+from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
+from langchain_core.outputs import ChatGeneration, LLMResult
 from tests.unit_tests.callbacks.fake_callback_handler import FakeCallbackHandler
 
 

diff --git a/libs/langchain/tests/integration_tests/llms/test_volcengine_maas.py b/libs/langchain/tests/integration_tests/llms/test_volcengine_maas.py
@@ -3,7 +3,7 @@
 from typing import Generator
 
 from langchain.llms.volcengine_maas import VolcEngineMaasLLM
-from langchain.schema import LLMResult
+from langchain_core.outputs import LLMResult
 
 
 def test_default_call() -> None: