From ff3b862c9f247674942f5592f28f09bd8bd87711 Mon Sep 17 00:00:00 2001 From: Santhosh Kumar H M Date: Tue, 10 Oct 2023 19:18:24 +0530 Subject: [PATCH] Update SqlDatabaseChain.ts --- .../nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts b/packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts index 6fc42b456..ac33fa0ea 100644 --- a/packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts +++ b/packages/components/nodes/chains/SqlDatabaseChain/SqlDatabaseChain.ts @@ -1,5 +1,5 @@ import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' -import { SqlDatabaseChain, SqlDatabaseChainInput } from 'langchain/chains/sql_db' +import { SqlDatabaseChain, SqlDatabaseChainInput, DEFAULT_SQL_DATABASE_PROMPT } from 'langchain/chains/sql_db' import { getBaseClasses, getInputVariables } from '../../../src/utils' import { DataSource } from 'typeorm' import { SqlDatabase } from 'langchain/sql_db' @@ -7,7 +7,6 @@ import { BaseLanguageModel } from 'langchain/base_language' import { PromptTemplate, PromptTemplateInput } from 'langchain/prompts' import { ConsoleCallbackHandler, CustomChainHandler, additionalCallbacks } from '../../../src/handler' import { DataSourceOptions } from 'typeorm/data-source' -import { DEFAULT_SQL_DATABASE_PROMPT } from 'langchain/chains/sql_db' type DatabaseType = 'sqlite' | 'postgres' | 'mssql' | 'mysql'