andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

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

Browse Source 
Fix NL2SQL pydantic version conflict
This commit is contained in:
Thiago Moretto
2024-08-19 13:49:06 -03:00
committed by GitHub
parent c32023df29 e8b185e607
commit 859e6162e8
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(