andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 23:32:39 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Fix NL2SQL pydantic version conflict

Browse Source
This commit is contained in:
Thiago Moretto
2024-08-14 14:05:44 -03:00
parent c32023df29
commit e8b185e607
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/crewai_tools/tools/nl2sql/nl2sql_tool.py
View File

@@ -1,12 +1,11 @@
from typing import Any, Union from typing import Any, Union
from ..base_tool import BaseTool from ..base_tool import BaseTool
from pydantic import Field from pydantic import BaseModel, Field
from sqlalchemy import create_engine, text from sqlalchemy import create_engine, text
from sqlalchemy.orm import sessionmaker from sqlalchemy.orm import sessionmaker
from typing import Optional, Type, Any from typing import Type, Any
from pydantic.v1 import BaseModel
class NL2SQLToolInput(BaseModel): class NL2SQLToolInput(BaseModel):
sql_query: str = Field( sql_query: str = Field(