Merge pull request #93 from crewAIInc/fix-nl2sql-pydantic-conflict

Fix NL2SQL pydantic version conflict
This commit is contained in:
Thiago Moretto
2024-08-19 13:49:06 -03:00
committed by GitHub

View File

@@ -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(