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

Class SQLDatabaseSequentialChain

java.lang.Object

com.hw.langchain.chains.base.Chain

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

public class SQLDatabaseSequentialChain
extends Chain

Chain for querying SQL database that is a sequential chain.

 The chain is as follows:
 1. Based on the query, determine which tables to use.
 2. Based on those tables, call the normal SQL database chain.

 This is useful in cases where the number of tables in the database is large.
 

Author:

HamaWhite

Field Summary

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

memory

Constructor Summary

Constructors

Constructor	Description
SQLDatabaseSequentialChain(SQLDatabaseChain sqlChain, LLMChain deciderChain)

Method Summary

Modifier and Type	Method	Description
String	chainType()
static SQLDatabaseSequentialChain	fromLLM(BaseLanguageModel llm, SQLDatabase database)
static SQLDatabaseSequentialChain	fromLLM(BaseLanguageModel llm, SQLDatabase database, BasePromptTemplate queryPrompt, BasePromptTemplate deciderPrompt)	Load the necessary chains.
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

SQLDatabaseSequentialChain

public SQLDatabaseSequentialChain(SQLDatabaseChain sqlChain,
 LLMChain deciderChain)

Method Details

fromLLM

public static SQLDatabaseSequentialChain fromLLM(BaseLanguageModel llm,
 SQLDatabase database,
 BasePromptTemplate queryPrompt,
 BasePromptTemplate deciderPrompt)

Load the necessary chains.

fromLLM

public static SQLDatabaseSequentialChain fromLLM(BaseLanguageModel llm,
 SQLDatabase database)

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