Package com.hw.langchain.chains.sql.database.base

Class SQLDatabaseChain

java.lang.Object

com.hw.langchain.chains.base.Chain

com.hw.langchain.chains.sql.database.base.SQLDatabaseChain

public class SQLDatabaseChain
extends Chain

Chain for interacting with SQL Database.

Author:

HamaWhite

Field Summary

Fields inherited from class com.hw.langchain.chains.base.Chain

memory

Constructor Summary

Constructors

Constructor	Description
SQLDatabaseChain(LLMChain llmChain, SQLDatabase database)

Method Summary

Modifier and Type	Method	Description
String	chainType()
static SQLDatabaseChain	fromLLM(BaseLanguageModel llm, SQLDatabase database)
static SQLDatabaseChain	fromLLM(BaseLanguageModel llm, SQLDatabase database, BasePromptTemplate prompt)
protected Map<String,String>	innerCall(Map<String,Object> inputs)	Runs the logic of this chain and returns the output.
List<String>	inputKeys()	Return the singular input key.
List<String>	outputKeys()	Return the singular output key.

Methods inherited from class com.hw.langchain.chains.base.Chain

asyncCall, asyncCall, asyncInnerCall, asyncRun, asyncRun, call, call, prepInputs, run, run

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

SQLDatabaseChain

public SQLDatabaseChain(LLMChain llmChain,
 SQLDatabase database)

Method Details

fromLLM

public static SQLDatabaseChain fromLLM(BaseLanguageModel llm,
 SQLDatabase database)

fromLLM

public static SQLDatabaseChain fromLLM(BaseLanguageModel llm,
 SQLDatabase database,
 BasePromptTemplate prompt)

chainType

public String chainType()

Specified by:

chainType in class Chain

inputKeys

public List<String> inputKeys()

Return the singular input key.

Specified by:

inputKeys in class Chain

Returns:

the list of input keys

outputKeys

public List<String> outputKeys()

Return the singular output key.

Specified by:

outputKeys in class Chain

Returns:

the list of output keys

innerCall

protected Map<String,String> innerCall(Map<String,Object> inputs)

Description copied from class: Chain

Runs the logic of this chain and returns the output.

Specified by:

innerCall in class Chain

Parameters:

inputs - the inputs to be processed by the chain

Returns:

a map containing the output generated by the chain