Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update SqlDatabaseChain.ts 

update types
This commit is contained in:
Henry Heng 2023-09-12 17:38:13 +01:00 committed by GitHub
parent 8fcafe83bc
commit 35ddb30489
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts
View File

@ -139,14 +139,14 @@ class SqlDatabaseChain_Chains implements INode {
async init(nodeData: INodeData): Promise<any> { async init(nodeData: INodeData): Promise<any> {
const databaseType = nodeData.inputs?.database as DatabaseType const databaseType = nodeData.inputs?.database as DatabaseType
const model = nodeData.inputs?.model as BaseLanguageModel const model = nodeData.inputs?.model as BaseLanguageModel
const url = nodeData.inputs?.url const url = nodeData.inputs?.url as string
const includesTables = nodeData.inputs?.includesTables const includesTables = nodeData.inputs?.includesTables
const splittedIncludesTables = includesTables == '' ? undefined : includesTables?.split(',') const splittedIncludesTables = includesTables == '' ? undefined : includesTables?.split(',')
const ignoreTables = nodeData.inputs?.ignoreTables const ignoreTables = nodeData.inputs?.ignoreTables
const splittedIgnoreTables = ignoreTables == '' ? undefined : ignoreTables?.split(',') const splittedIgnoreTables = ignoreTables == '' ? undefined : ignoreTables?.split(',')
const sampleRowsInTableInfo = nodeData.inputs?.sampleRowsInTableInfo const sampleRowsInTableInfo = nodeData.inputs?.sampleRowsInTableInfo as number
const topK = nodeData.inputs?.topK const topK = nodeData.inputs?.topK as number
const customPrompt = nodeData.inputs?.customPrompt const customPrompt = nodeData.inputs?.customPrompt as string
const chain = await getSQLDBChain( const chain = await getSQLDBChain(
databaseType, databaseType,
@ -164,14 +164,14 @@ class SqlDatabaseChain_Chains implements INode {
async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<string> { async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<string> {
const databaseType = nodeData.inputs?.database as DatabaseType const databaseType = nodeData.inputs?.database as DatabaseType
const model = nodeData.inputs?.model as BaseLanguageModel const model = nodeData.inputs?.model as BaseLanguageModel
const url = nodeData.inputs?.url const url = nodeData.inputs?.url as string
const includesTables = nodeData.inputs?.includesTables const includesTables = nodeData.inputs?.includesTables
const splittedIncludesTables = includesTables == '' ? undefined : includesTables?.split(',') const splittedIncludesTables = includesTables == '' ? undefined : includesTables?.split(',')
const ignoreTables = nodeData.inputs?.ignoreTables const ignoreTables = nodeData.inputs?.ignoreTables
const splittedIgnoreTables = ignoreTables == '' ? undefined : ignoreTables?.split(',') const splittedIgnoreTables = ignoreTables == '' ? undefined : ignoreTables?.split(',')
const sampleRowsInTableInfo = nodeData.inputs?.sampleRowsInTableInfo const sampleRowsInTableInfo = nodeData.inputs?.sampleRowsInTableInfo as number
const topK = nodeData.inputs?.topK const topK = nodeData.inputs?.topK as number
const customPrompt = nodeData.inputs?.customPrompt const customPrompt = nodeData.inputs?.customPrompt as string
const chain = await getSQLDBChain( const chain = await getSQLDBChain(
databaseType, databaseType,