From e8b185e607c361a84c6a3d097c3546e3a608fa00 Mon Sep 17 00:00:00 2001
From: Thiago Moretto <thiago.moretto@gmail.com>
Date: Wed, 14 Aug 2024 14:05:44 -0300
Subject: [PATCH] Fix NL2SQL pydantic version conflict

---
 src/crewai_tools/tools/nl2sql/nl2sql_tool.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/crewai_tools/tools/nl2sql/nl2sql_tool.py b/src/crewai_tools/tools/nl2sql/nl2sql_tool.py
index 7ddcd090c..22c3a299b 100644
--- a/src/crewai_tools/tools/nl2sql/nl2sql_tool.py
+++ b/src/crewai_tools/tools/nl2sql/nl2sql_tool.py
@@ -1,12 +1,11 @@
 from typing import Any, Union
 
 from ..base_tool import BaseTool
-from pydantic import Field
+from pydantic import BaseModel, Field
 from sqlalchemy import create_engine, text
 from sqlalchemy.orm import sessionmaker
 
-from typing import Optional, Type, Any
-from pydantic.v1 import BaseModel
+from typing import Type, Any
 
 class NL2SQLToolInput(BaseModel):
     sql_query: str = Field(