Depend on pydantic directly instead of langchain (#48)

jupyterlab · Jun 10, 2024 · ed3d92e · ed3d92e
1 parent f3ba6e0
commit ed3d92e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/packages/jupyterlab-ws-chat/jupyterlab_ws_chat/handlers.py b/packages/jupyterlab-ws-chat/jupyterlab_ws_chat/handlers.py
@@ -8,7 +8,7 @@
 from typing import Dict, List
 
 from jupyter_server.base.handlers import APIHandler as BaseAPIHandler, JupyterHandler
-from langchain.pydantic_v1 import ValidationError
+from pydantic.v1 import ValidationError
 from tornado import web, websocket
 
 from .config_manager import WriteConflictError

diff --git a/packages/jupyterlab-ws-chat/jupyterlab_ws_chat/models.py b/packages/jupyterlab-ws-chat/jupyterlab_ws_chat/models.py
@@ -3,7 +3,7 @@
 
 from typing import Any, Dict, List, Literal, Optional, Union
 
-from langchain.pydantic_v1 import BaseModel, validator
+from pydantic.v1 import BaseModel, validator
 
 DEFAULT_CHUNK_SIZE = 2000
 DEFAULT_CHUNK_OVERLAP = 100

diff --git a/packages/jupyterlab-ws-chat/pyproject.toml b/packages/jupyterlab-ws-chat/pyproject.toml
@@ -30,7 +30,7 @@ dependencies = [
     "jupyterlab~=4.0",
     "traitlets>=5.0",
     "deepmerge>=1.0",
-    "langchain"
+    "pydantic>=2"
 ]
 dynamic = ["version", "description", "authors", "urls", "keywords"]