From 6aac2eeab5824660a41b9be21cc005f68e262e55 Mon Sep 17 00:00:00 2001 From: Bagatur Date: Tue, 3 Sep 2024 17:42:22 -0700 Subject: [PATCH] fmt --- libs/partners/fireworks/langchain_fireworks/chat_models.py | 6 +++--- libs/partners/fireworks/scripts/check_pydantic.sh | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/libs/partners/fireworks/langchain_fireworks/chat_models.py b/libs/partners/fireworks/langchain_fireworks/chat_models.py index c5e6d9c42fd..4983f51daab 100644 --- a/libs/partners/fireworks/langchain_fireworks/chat_models.py +++ b/libs/partners/fireworks/langchain_fireworks/chat_models.py @@ -805,7 +805,7 @@ class ChatFireworks(BaseChatModel): from typing import Optional from langchain_fireworks import ChatFireworks - from langchain_core.pydantic_v1 import BaseModel, Field + from pydantic import BaseModel, Field class AnswerWithJustification(BaseModel): @@ -836,7 +836,7 @@ class ChatFireworks(BaseChatModel): .. code-block:: python from langchain_fireworks import ChatFireworks - from langchain_core.pydantic_v1 import BaseModel + from pydantic import BaseModel class AnswerWithJustification(BaseModel): @@ -923,7 +923,7 @@ class ChatFireworks(BaseChatModel): .. code-block:: from langchain_fireworks import ChatFireworks - from langchain_core.pydantic_v1 import BaseModel + from pydantic import BaseModel class AnswerWithJustification(BaseModel): answer: str diff --git a/libs/partners/fireworks/scripts/check_pydantic.sh b/libs/partners/fireworks/scripts/check_pydantic.sh index 06b5bb81ae2..1317f5e5391 100755 --- a/libs/partners/fireworks/scripts/check_pydantic.sh +++ b/libs/partners/fireworks/scripts/check_pydantic.sh @@ -20,8 +20,8 @@ result=$(git -C "$repository_path" grep -E '^import pydantic|^from pydantic') if [ -n "$result" ]; then echo "ERROR: The following lines need to be updated:" echo "$result" - echo "Please replace the code with an import from langchain_core.pydantic_v1." + echo "Please replace the code with an import from pydantic." echo "For example, replace 'from pydantic import BaseModel'" - echo "with 'from langchain_core.pydantic_v1 import BaseModel'" + echo "with 'from pydantic import BaseModel'" exit 1 fi