Fix NL2SQL pydantic version conflict

This commit is contained in:
Thiago Moretto
2024-08-14 14:05:44 -03:00
parent c32023df29
commit e8b185e607

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(