x

langchain-ai · Nov 17, 2023 · f9600fe · f9600fe
1 parent 6d6ffc6
commit f9600fe
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/langserve/pydantic_v1.py b/langserve/pydantic_v1.py
@@ -13,7 +13,12 @@
 
 try:
     # F401: imported but unused
-    from pydantic.v1 import BaseModel, Field, ValidationError, create_model  # noqa: F401
+    from pydantic.v1 import (  # noqa: F401
+        BaseModel,
+        Field,
+        ValidationError,
+        create_model,
+    )
 except ImportError:
     from pydantic import BaseModel, Field, ValidationError, create_model  # noqa: F401
 

diff --git a/langserve/server.py b/langserve/server.py
@@ -37,12 +37,17 @@
 from langsmith.utils import tracing_is_enabled
 from typing_extensions import Annotated
 
-from langserve.pydantic_v1 import BaseModel, Field, ValidationError, create_model
-
 from langserve.callbacks import AsyncEventAggregatorCallback, CallbackEventDict
 from langserve.lzstring import LZString
 from langserve.playground import serve_playground
-from langserve.pydantic_v1 import _PYDANTIC_MAJOR_VERSION, PYDANTIC_VERSION
+from langserve.pydantic_v1 import (
+    _PYDANTIC_MAJOR_VERSION,
+    PYDANTIC_VERSION,
+    BaseModel,
+    Field,
+    ValidationError,
+    create_model,
+)
 from langserve.schema import (
     BatchResponseMetadata,
     CustomUserType,