-
Notifications
You must be signed in to change notification settings - Fork 34
Description
when attempting the below import:
from databricks_langchain import ChatDatabricks
I get the following error:
ModuleNotFoundError: No module named 'langchain_core.pydantic_v1'
File , line 8
6 import typing
7 import pandas as pd
----> 8 from databricks_langchain import ChatDatabricks
9 from langchain.prompts import PromptTemplate
10 from langchain.schema.runnable import RunnablePassthrough
File , line 8
6 import typing
7 import pandas as pd
----> 8 from databricks_langchain import ChatDatabricks
9 from langchain.prompts import PromptTemplate
10 from langchain.schema.runnable import RunnablePassthrough
File /databricks/python_shell/lib/dbruntime/autoreload/discoverability/hook.py:71, in AutoreloadDiscoverabilityHook._patched_import(self, name, *args, **kwargs)
65 if not self._should_hint and (
66 (module := sys.modules.get(absolute_name)) is not None and
67 (fname := get_allowed_file_name_or_none(module)) is not None and
68 (mtime := os.stat(fname).st_mtime) > self.last_mtime_by_modname.get(
69 absolute_name, float("inf")) and not self._should_hint):
70 self._should_hint = True
---> 71 module = self._original_builtins_import(name, *args, **kwargs)
72 if (fname := fname or get_allowed_file_name_or_none(module)) is not None:
73 mtime = mtime or os.stat(fname).st_mtime
File /local_disk0/.ephemeral_nfs/envs/pythonEnv-7e8d57b3-8a9e-4666-b4b5-80efc317a1b0/lib/python3.12/site-packages/databricks_langchain/init.py:18
16 from unitycatalog.ai.core.base import set_uc_function_client
17 from unitycatalog.ai.core.databricks import DatabricksFunctionClient
---> 18 from unitycatalog.ai.langchain.toolkit import UCFunctionToolkit, UnityCatalogTool
20 from databricks_langchain.chat_models import ChatDatabricks
21 from databricks_langchain.embeddings import DatabricksEmbeddings
File /local_disk0/.ephemeral_nfs/envs/pythonEnv-7e8d57b3-8a9e-4666-b4b5-80efc317a1b0/lib/python3.12/site-packages/unitycatalog/ai/langchain/toolkit.py:13
6 from langchain_core._api.deprecation import LangChainDeprecationWarning
8 warnings.filterwarnings(
9 "ignore",
10 message=r".As of langchain-core 0.3.0, LangChain uses pydantic v2 internally.",
11 category=LangChainDeprecationWarning,
12 )
---> 13 from langchain_core.pydantic_v1 import BaseModel, Field, root_validator
14 from langchain_core.tools import StructuredTool
16 from unitycatalog.ai.core.base import BaseFunctionClient