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):