From cdaf2d41c7d862bf8c8f3dc25483b1afd2dcf7cf Mon Sep 17 00:00:00 2001
From: "Brandon Hancock (bhancock_ai)"
 <109994880+bhancockio@users.noreply.github.com>
Date: Fri, 6 Sep 2024 14:22:01 -0400
Subject: [PATCH] move away from pydantic v1 (#1284)

---
 src/crewai/tools/tool_calling.py                    | 2 +-
 src/crewai/tools/tool_output_parser.py              | 2 +-
 src/crewai/utilities/crew_pydantic_output_parser.py | 3 +--
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/crewai/tools/tool_calling.py b/src/crewai/tools/tool_calling.py
index 2a0fd425b..16c5e0bfc 100644
--- a/src/crewai/tools/tool_calling.py
+++ b/src/crewai/tools/tool_calling.py
@@ -1,8 +1,8 @@
 from typing import Any, Dict, Optional
 
+from pydantic import BaseModel, Field
 from pydantic import BaseModel as PydanticBaseModel
 from pydantic import Field as PydanticField
-from pydantic.v1 import BaseModel, Field
 
 
 class ToolCalling(BaseModel):
diff --git a/src/crewai/tools/tool_output_parser.py b/src/crewai/tools/tool_output_parser.py
index d6c3465f1..116dbe7d6 100644
--- a/src/crewai/tools/tool_output_parser.py
+++ b/src/crewai/tools/tool_output_parser.py
@@ -5,7 +5,7 @@ import regex
 from langchain.output_parsers import PydanticOutputParser
 from langchain_core.exceptions import OutputParserException
 from langchain_core.outputs import Generation
-from langchain_core.pydantic_v1 import ValidationError
+from pydantic import ValidationError
 
 
 class ToolOutputParser(PydanticOutputParser):
diff --git a/src/crewai/utilities/crew_pydantic_output_parser.py b/src/crewai/utilities/crew_pydantic_output_parser.py
index f4e9cdd18..37979a524 100644
--- a/src/crewai/utilities/crew_pydantic_output_parser.py
+++ b/src/crewai/utilities/crew_pydantic_output_parser.py
@@ -5,8 +5,7 @@ import regex
 from langchain.output_parsers import PydanticOutputParser
 from langchain_core.exceptions import OutputParserException
 from langchain_core.outputs import Generation
-from langchain_core.pydantic_v1 import ValidationError
-from pydantic import BaseModel
+from pydantic import BaseModel, ValidationError
 
 
 class CrewPydanticOutputParser(PydanticOutputParser):