github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-07 14:03:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

all: test 3.13 ci (#27197)

Browse Source 
Co-authored-by: Bagatur <baskaryan@gmail.com>
Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
This commit is contained in:
Erick Friis
2024-10-25 12:56:58 -07:00
committed by GitHub
parent 06df15c9c0
commit 600b7bdd61
237 changed files with 3668 additions and 4656 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libs/community/langchain_community/tools/sql_database/tool.py
View File

@@ -32,7 +32,7 @@ class _QuerySQLDataBaseToolInput(BaseModel):
query: str = Field(..., description="A detailed and correct SQL query.")
class QuerySQLDataBaseTool(BaseSQLDatabaseTool, BaseTool):
class QuerySQLDataBaseTool(BaseSQLDatabaseTool, BaseTool): # type: ignore[override, override]
"""Tool for querying a SQL database."""
name: str = "sql_db_query"
@@ -62,7 +62,7 @@ class _InfoSQLDatabaseToolInput(BaseModel):
)
class InfoSQLDatabaseTool(BaseSQLDatabaseTool, BaseTool):
class InfoSQLDatabaseTool(BaseSQLDatabaseTool, BaseTool): # type: ignore[override, override]
"""Tool for getting metadata about a SQL database."""
name: str = "sql_db_schema"
@@ -84,7 +84,7 @@ class _ListSQLDataBaseToolInput(BaseModel):
tool_input: str = Field("", description="An empty string")
class ListSQLDatabaseTool(BaseSQLDatabaseTool, BaseTool):
class ListSQLDatabaseTool(BaseSQLDatabaseTool, BaseTool): # type: ignore[override, override]
"""Tool for getting tables names."""
name: str = "sql_db_list_tables"
@@ -104,7 +104,7 @@ class _QuerySQLCheckerToolInput(BaseModel):
query: str = Field(..., description="A detailed and SQL query to be checked.")
class QuerySQLCheckerTool(BaseSQLDatabaseTool, BaseTool):
class QuerySQLCheckerTool(BaseSQLDatabaseTool, BaseTool): # type: ignore[override, override]
"""Use an LLM to check if a query is correct.
Adapted from https://www.patterns.app/blog/2023/01/18/crunchbot-sql-analyst-gpt/"""