🪛 fix pydantic V2 schemas

shroominic · Aug 22, 2023 · 74ebbc7 · 74ebbc7
1 parent 855366b
commit 74ebbc7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/codeinterpreterapi/schema/file.py b/codeinterpreterapi/schema/file.py
@@ -1,6 +1,6 @@
 import asyncio
 
-from pydantic import BaseModel
+from langchain.pydantic_v1 import BaseModel
 
 
 class File(BaseModel):
@@ -60,7 +60,7 @@ def get_image(self):
         img = Image.open(img_io)
 
         # Convert image to RGB if it's not
-        if img.mode not in ("RGB", "L"):  # L is for greyscale images
+        if img.mode not in ("RGB", "L"):  # L is for grayscale images
             img = img.convert("RGB")
 
         return img

diff --git a/codeinterpreterapi/schema/input.py b/codeinterpreterapi/schema/input.py
@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+from langchain.pydantic_v1 import BaseModel
 
 
 class CodeInput(BaseModel):