From 35ddb30489d086ef73f047c3145b4f20596202d7 Mon Sep 17 00:00:00 2001
From: Henry Heng <henryheng@flowiseai.com>
Date: Tue, 12 Sep 2023 17:38:13 +0100
Subject: [PATCH] Update SqlDatabaseChain.ts

update types
---
 .../chains/SqlDatabaseChain/SqlDatabaseChain.ts  | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts b/packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts
index a6e6ab24e..f5fd0cccb 100644
--- a/packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts
+++ b/packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts
@@ -139,14 +139,14 @@ class SqlDatabaseChain_Chains implements INode {
     async init(nodeData: INodeData): Promise<any> {
         const databaseType = nodeData.inputs?.database as DatabaseType
         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 splittedIncludesTables = includesTables == '' ? undefined : includesTables?.split(',')
         const ignoreTables = nodeData.inputs?.ignoreTables
         const splittedIgnoreTables = ignoreTables == '' ? undefined : ignoreTables?.split(',')
-        const sampleRowsInTableInfo = nodeData.inputs?.sampleRowsInTableInfo
-        const topK = nodeData.inputs?.topK
-        const customPrompt = nodeData.inputs?.customPrompt
+        const sampleRowsInTableInfo = nodeData.inputs?.sampleRowsInTableInfo as number
+        const topK = nodeData.inputs?.topK as number
+        const customPrompt = nodeData.inputs?.customPrompt as string
 
         const chain = await getSQLDBChain(
             databaseType,
@@ -164,14 +164,14 @@ class SqlDatabaseChain_Chains implements INode {
     async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<string> {
         const databaseType = nodeData.inputs?.database as DatabaseType
         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 splittedIncludesTables = includesTables == '' ? undefined : includesTables?.split(',')
         const ignoreTables = nodeData.inputs?.ignoreTables
         const splittedIgnoreTables = ignoreTables == '' ? undefined : ignoreTables?.split(',')
-        const sampleRowsInTableInfo = nodeData.inputs?.sampleRowsInTableInfo
-        const topK = nodeData.inputs?.topK
-        const customPrompt = nodeData.inputs?.customPrompt
+        const sampleRowsInTableInfo = nodeData.inputs?.sampleRowsInTableInfo as number
+        const topK = nodeData.inputs?.topK as number
+        const customPrompt = nodeData.inputs?.customPrompt as string
 
         const chain = await getSQLDBChain(
             databaseType,