From df825b767cba4828b386830598bf72f5f71a4288 Mon Sep 17 00:00:00 2001 From: "David L. Qiu" Date: Tue, 26 Dec 2023 15:14:50 -0800 Subject: [PATCH] fix pydantic import --- packages/jupyter-ai/jupyter_ai/chat_handlers/base.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py index 9f179de51..bcbba00ba 100644 --- a/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py +++ b/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py @@ -18,9 +18,7 @@ from jupyter_ai.config_manager import ConfigManager, Logger from jupyter_ai.models import AgentChatMessage, ChatMessage, HumanChatMessage from jupyter_ai_magics.providers import BaseProvider - -# necessary to prevent circular import -from pydantic import BaseModel +from langchain.pydantic_v1 import BaseModel if TYPE_CHECKING: from jupyter_ai.handlers import RootChatHandler