Package com.hw.langchain.tools.spark.sql.tool

Class QueryCheckerTool

java.lang.Object

com.hw.langchain.tools.base.BaseTool

com.hw.langchain.tools.spark.sql.tool.BaseSparkSqlTool

com.hw.langchain.tools.spark.sql.tool.QueryCheckerTool

public class QueryCheckerTool
extends BaseSparkSqlTool

Use an LLM to check if a query is correct.

Author:

HamaWhite

Field Summary

Fields inherited from class com.hw.langchain.tools.spark.sql.tool.BaseSparkSqlTool

db

Fields inherited from class com.hw.langchain.tools.base.BaseTool

description, name, returnDirect

Constructor Summary

Constructors

Constructor	Description
QueryCheckerTool(SparkSql db, BaseLanguageModel llm)

Method Summary

Modifier and Type	Method	Description
Object	innerRun(String query, Map<String,Object> kwargs)	Use the LLM to check the query.

Methods inherited from class com.hw.langchain.tools.base.BaseTool

args, canEqual, equals, getDescription, getName, hashCode, isReturnDirect, isSingleInput, run, toArgsAndKwargs, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Details

QueryCheckerTool

public QueryCheckerTool(SparkSql db,
 BaseLanguageModel llm)

Method Details

innerRun

public Object innerRun(String query,
 Map<String,Object> kwargs)

Use the LLM to check the query.

Specified by:

innerRun in class BaseTool