github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-03 12:07:36 +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/spark_sql/tool.py
View File

@@ -26,7 +26,7 @@ class BaseSparkSQLTool(BaseModel):
)
class QuerySparkSQLTool(BaseSparkSQLTool, BaseTool):
class QuerySparkSQLTool(BaseSparkSQLTool, BaseTool): # type: ignore[override]
"""Tool for querying a Spark SQL."""
name: str = "query_sql_db"
@@ -45,7 +45,7 @@ class QuerySparkSQLTool(BaseSparkSQLTool, BaseTool):
return self.db.run_no_throw(query)
class InfoSparkSQLTool(BaseSparkSQLTool, BaseTool):
class InfoSparkSQLTool(BaseSparkSQLTool, BaseTool): # type: ignore[override]
"""Tool for getting metadata about a Spark SQL."""
name: str = "schema_sql_db"
@@ -65,7 +65,7 @@ class InfoSparkSQLTool(BaseSparkSQLTool, BaseTool):
return self.db.get_table_info_no_throw(table_names.split(", "))
class ListSparkSQLTool(BaseSparkSQLTool, BaseTool):
class ListSparkSQLTool(BaseSparkSQLTool, BaseTool): # type: ignore[override]
"""Tool for getting tables names."""
name: str = "list_tables_sql_db"
@@ -80,7 +80,7 @@ class ListSparkSQLTool(BaseSparkSQLTool, BaseTool):
return ", ".join(self.db.get_usable_table_names())
class QueryCheckerTool(BaseSparkSQLTool, BaseTool):
class QueryCheckerTool(BaseSparkSQLTool, BaseTool): # type: ignore[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/"""