From 3904f2cd40d0bdf6620bb8b1bca38d7a9c68f938 Mon Sep 17 00:00:00 2001 From: nobbbbby <40331529+nobbbbby@users.noreply.github.com> Date: Tue, 2 Jul 2024 01:51:23 +0800 Subject: [PATCH] core: fix NameError (#23658) **Description:** In the chat_models module of the language model, the import statement for BaseModel has been moved from the conditionally imported section to the main import area, fixing `NameError `. **Issue:** fix `NameError ` --- libs/core/langchain_core/language_models/chat_models.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/libs/core/langchain_core/language_models/chat_models.py b/libs/core/langchain_core/language_models/chat_models.py index 329e43a4bc7..51fff5dc3e4 100644 --- a/libs/core/langchain_core/language_models/chat_models.py +++ b/libs/core/langchain_core/language_models/chat_models.py @@ -54,7 +54,7 @@ from langchain_core.outputs import ( RunInfo, ) from langchain_core.prompt_values import ChatPromptValue, PromptValue, StringPromptValue -from langchain_core.pydantic_v1 import Field, root_validator +from langchain_core.pydantic_v1 import BaseModel, Field, root_validator from langchain_core.runnables import RunnableMap, RunnablePassthrough from langchain_core.runnables.config import ensure_config, run_in_executor from langchain_core.tracers._streaming import _StreamingCallbackHandler @@ -62,7 +62,6 @@ from langchain_core.utils.function_calling import convert_to_openai_tool if TYPE_CHECKING: from langchain_core.output_parsers.base import OutputParserLike - from langchain_core.pydantic_v1 import BaseModel from langchain_core.runnables import Runnable, RunnableConfig from langchain_core.tools import BaseTool