🪛 fix pydantic V2 schemas

Author: shroominic

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

codeinterpreterapi/schema/input.py

@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+from langchain.pydantic_v1 import BaseModel
 
 
 class CodeInput(BaseModel):