farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-28 17:01:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update SqlDatabaseChain.ts

Browse Source 
update types
This commit is contained in:
Henry Heng
2023-09-12 17:38:13 +01:00
committed by GitHub
parent 8fcafe83bc
commit 35ddb30489
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts
+8 -8
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,