@langchain/classic 1.0.0-alpha.1

package/dist/util/set.cjs

@@ -0,0 +1,35 @@
+
+//#region src/util/set.ts
+/**
+* Source: https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Set#implementing_basic_set_operations
+*/
+/**
+* returns intersection of two sets
+*/
+function intersection(setA, setB) {
+	const _intersection = /* @__PURE__ */ new Set();
+	for (const elem of setB) if (setA.has(elem)) _intersection.add(elem);
+	return _intersection;
+}
+/**
+* returns union of two sets
+*/
+function union(setA, setB) {
+	const _union = new Set(setA);
+	for (const elem of setB) _union.add(elem);
+	return _union;
+}
+/**
+* returns difference of two sets
+*/
+function difference(setA, setB) {
+	const _difference = new Set(setA);
+	for (const elem of setB) _difference.delete(elem);
+	return _difference;
+}
+
+//#endregion
+exports.difference = difference;
+exports.intersection = intersection;
+exports.union = union;
+//# sourceMappingURL=set.cjs.map

package/dist/util/set.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"set.cjs","names":["setA: Set<T>","setB: Set<T>"],"sources":["../../src/util/set.ts"],"sourcesContent":["/**\n * Source: https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Set#implementing_basic_set_operations\n */\n\n/**\n * returns intersection of two sets\n */\nexport function intersection<T>(setA: Set<T>, setB: Set<T>) {\n  const _intersection = new Set<T>();\n  for (const elem of setB) {\n    if (setA.has(elem)) {\n      _intersection.add(elem);\n    }\n  }\n  return _intersection;\n}\n\n/**\n * returns union of two sets\n */\nexport function union<T>(setA: Set<T>, setB: Set<T>) {\n  const _union = new Set(setA);\n  for (const elem of setB) {\n    _union.add(elem);\n  }\n  return _union;\n}\n\n/**\n * returns difference of two sets\n */\nexport function difference<T>(setA: Set<T>, setB: Set<T>) {\n  const _difference = new Set(setA);\n  for (const elem of setB) {\n    _difference.delete(elem);\n  }\n  return _difference;\n}\n"],"mappings":";;;;;;;;AAOA,SAAgB,aAAgBA,MAAcC,MAAc;CAC1D,MAAM,gCAAgB,IAAI;AAC1B,MAAK,MAAM,QAAQ,KACjB,KAAI,KAAK,IAAI,KAAK,EAChB,cAAc,IAAI,KAAK;AAG3B,QAAO;AACR;;;;AAKD,SAAgB,MAASD,MAAcC,MAAc;CACnD,MAAM,SAAS,IAAI,IAAI;AACvB,MAAK,MAAM,QAAQ,MACjB,OAAO,IAAI,KAAK;AAElB,QAAO;AACR;;;;AAKD,SAAgB,WAAcD,MAAcC,MAAc;CACxD,MAAM,cAAc,IAAI,IAAI;AAC5B,MAAK,MAAM,QAAQ,MACjB,YAAY,OAAO,KAAK;AAE1B,QAAO;AACR"}

package/dist/util/set.js

@@ -0,0 +1,32 @@
+//#region src/util/set.ts
+/**
+* Source: https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Set#implementing_basic_set_operations
+*/
+/**
+* returns intersection of two sets
+*/
+function intersection(setA, setB) {
+	const _intersection = /* @__PURE__ */ new Set();
+	for (const elem of setB) if (setA.has(elem)) _intersection.add(elem);
+	return _intersection;
+}
+/**
+* returns union of two sets
+*/
+function union(setA, setB) {
+	const _union = new Set(setA);
+	for (const elem of setB) _union.add(elem);
+	return _union;
+}
+/**
+* returns difference of two sets
+*/
+function difference(setA, setB) {
+	const _difference = new Set(setA);
+	for (const elem of setB) _difference.delete(elem);
+	return _difference;
+}
+
+//#endregion
+export { difference, intersection, union };
+//# sourceMappingURL=set.js.map

package/dist/util/set.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"set.js","names":["setA: Set<T>","setB: Set<T>"],"sources":["../../src/util/set.ts"],"sourcesContent":["/**\n * Source: https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Set#implementing_basic_set_operations\n */\n\n/**\n * returns intersection of two sets\n */\nexport function intersection<T>(setA: Set<T>, setB: Set<T>) {\n  const _intersection = new Set<T>();\n  for (const elem of setB) {\n    if (setA.has(elem)) {\n      _intersection.add(elem);\n    }\n  }\n  return _intersection;\n}\n\n/**\n * returns union of two sets\n */\nexport function union<T>(setA: Set<T>, setB: Set<T>) {\n  const _union = new Set(setA);\n  for (const elem of setB) {\n    _union.add(elem);\n  }\n  return _union;\n}\n\n/**\n * returns difference of two sets\n */\nexport function difference<T>(setA: Set<T>, setB: Set<T>) {\n  const _difference = new Set(setA);\n  for (const elem of setB) {\n    _difference.delete(elem);\n  }\n  return _difference;\n}\n"],"mappings":";;;;;;;AAOA,SAAgB,aAAgBA,MAAcC,MAAc;CAC1D,MAAM,gCAAgB,IAAI;AAC1B,MAAK,MAAM,QAAQ,KACjB,KAAI,KAAK,IAAI,KAAK,EAChB,cAAc,IAAI,KAAK;AAG3B,QAAO;AACR;;;;AAKD,SAAgB,MAASD,MAAcC,MAAc;CACnD,MAAM,SAAS,IAAI,IAAI;AACvB,MAAK,MAAM,QAAQ,MACjB,OAAO,IAAI,KAAK;AAElB,QAAO;AACR;;;;AAKD,SAAgB,WAAcD,MAAcC,MAAc;CACxD,MAAM,cAAc,IAAI,IAAI;AAC5B,MAAK,MAAM,QAAQ,MACjB,YAAY,OAAO,KAAK;AAE1B,QAAO;AACR"}

package/dist/util/sql_utils.cjs

@@ -0,0 +1,178 @@
+const require_sql_db_prompt = require('../chains/sql_db/sql_db_prompt.cjs');
+
+//#region src/util/sql_utils.ts
+const verifyListTablesExistInDatabase = (tablesFromDatabase, listTables, errorPrefixMsg) => {
+	const onlyTableNames = tablesFromDatabase.map((table) => table.tableName);
+	if (listTables.length > 0) {
+		for (const tableName of listTables) if (!onlyTableNames.includes(tableName)) throw new Error(`${errorPrefixMsg} the table ${tableName} was not found in the database`);
+	}
+};
+const verifyIncludeTablesExistInDatabase = (tablesFromDatabase, includeTables) => {
+	verifyListTablesExistInDatabase(tablesFromDatabase, includeTables, "Include tables not found in database:");
+};
+const verifyIgnoreTablesExistInDatabase = (tablesFromDatabase, ignoreTables) => {
+	verifyListTablesExistInDatabase(tablesFromDatabase, ignoreTables, "Ignore tables not found in database:");
+};
+const formatToSqlTable = (rawResultsTableAndColumn) => {
+	const sqlTable = [];
+	for (const oneResult of rawResultsTableAndColumn) {
+		const sqlColumn = {
+			columnName: oneResult.column_name,
+			dataType: oneResult.data_type,
+			isNullable: oneResult.is_nullable === "YES"
+		};
+		const currentTable = sqlTable.find((oneTable) => oneTable.tableName === oneResult.table_name);
+		if (currentTable) currentTable.columns.push(sqlColumn);
+		else {
+			const newTable = {
+				tableName: oneResult.table_name,
+				columns: [sqlColumn]
+			};
+			sqlTable.push(newTable);
+		}
+	}
+	return sqlTable;
+};
+const getTableAndColumnsName = async (appDataSource) => {
+	let sql;
+	if (appDataSource.options.type === "postgres") {
+		const schema = appDataSource.options?.schema ?? "public";
+		sql = `SELECT 
+            t.table_name, 
+            c.* 
+          FROM 
+            information_schema.tables t 
+              JOIN information_schema.columns c 
+                ON t.table_name = c.table_name 
+          WHERE 
+            t.table_schema = '${schema}' 
+              AND c.table_schema = '${schema}' 
+          ORDER BY 
+            t.table_name,
+            c.ordinal_position;`;
+		const rep = await appDataSource.query(sql);
+		return formatToSqlTable(rep);
+	}
+	if (appDataSource.options.type === "sqlite" || appDataSource.options.type === "better-sqlite3" || appDataSource.options.type === "sqljs") {
+		sql = "SELECT \n   m.name AS table_name,\n   p.name AS column_name,\n   p.type AS data_type,\n   CASE \n      WHEN p.\"notnull\" = 0 THEN 'YES' \n      ELSE 'NO' \n   END AS is_nullable \nFROM \n   sqlite_master m \nJOIN \n   pragma_table_info(m.name) p \nWHERE \n   m.type = 'table' AND \n   m.name NOT LIKE 'sqlite_%';\n";
+		const rep = await appDataSource.query(sql);
+		return formatToSqlTable(rep);
+	}
+	if (appDataSource.options.type === "mysql" || appDataSource.options.type === "aurora-mysql") {
+		sql = `SELECT TABLE_NAME AS table_name, COLUMN_NAME AS column_name, DATA_TYPE AS data_type, IS_NULLABLE AS is_nullable FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = '${appDataSource.options.database}';`;
+		const rep = await appDataSource.query(sql);
+		return formatToSqlTable(rep);
+	}
+	if (appDataSource.options.type === "mssql") {
+		const schema = appDataSource.options?.schema;
+		const sql$1 = `SELECT
+    TABLE_NAME AS table_name,
+    COLUMN_NAME AS column_name,
+    DATA_TYPE AS data_type,
+    IS_NULLABLE AS is_nullable
+    FROM INFORMATION_SCHEMA.COLUMNS
+    ${schema && `WHERE TABLE_SCHEMA = '${schema}'`} 
+ORDER BY TABLE_NAME, ORDINAL_POSITION;`;
+		const rep = await appDataSource.query(sql$1);
+		return formatToSqlTable(rep);
+	}
+	if (appDataSource.options.type === "sap") {
+		const schema = appDataSource.options?.schema ?? "public";
+		sql = `SELECT
+        TABLE_NAME,
+        COLUMN_NAME,
+        DATA_TYPE_NAME AS data_type,
+        CASE WHEN IS_NULLABLE='TRUE' THEN 'YES' ELSE 'NO' END AS is_nullable
+      FROM TABLE_COLUMNS
+      WHERE SCHEMA_NAME='${schema}'`;
+		const rep = await appDataSource.query(sql);
+		const repLowerCase = [];
+		rep.forEach((_rep) => repLowerCase.push({
+			table_name: _rep.TABLE_NAME,
+			column_name: _rep.COLUMN_NAME,
+			data_type: _rep.DATA_TYPE,
+			is_nullable: _rep.IS_NULLABLE
+		}));
+		return formatToSqlTable(repLowerCase);
+	}
+	if (appDataSource.options.type === "oracle") {
+		const schemaName = appDataSource.options.schema;
+		const sql$1 = `  
+      SELECT
+          TABLE_NAME AS "table_name",
+          COLUMN_NAME AS "column_name",
+          DATA_TYPE AS "data_type",
+          NULLABLE AS "is_nullable"
+      FROM ALL_TAB_COLS
+      WHERE
+          OWNER = UPPER('${schemaName}')`;
+		const rep = await appDataSource.query(sql$1);
+		return formatToSqlTable(rep);
+	}
+	throw new Error("Database type not implemented yet");
+};
+const formatSqlResponseToSimpleTableString = (rawResult) => {
+	if (!rawResult || !Array.isArray(rawResult) || rawResult.length === 0) return "";
+	let globalString = "";
+	for (const oneRow of rawResult) globalString += `${Object.values(oneRow).reduce((completeString, columnValue) => `${completeString} ${columnValue}`, "")}\n`;
+	return globalString;
+};
+const generateTableInfoFromTables = async (tables, appDataSource, nbSampleRow, customDescription) => {
+	if (!tables) return "";
+	let globalString = "";
+	for (const currentTable of tables) {
+		const tableCustomDescription = customDescription && Object.keys(customDescription).includes(currentTable.tableName) ? `${customDescription[currentTable.tableName]}\n` : "";
+		let schema = null;
+		if (appDataSource.options.type === "postgres") schema = appDataSource.options?.schema ?? "public";
+		else if (appDataSource.options.type === "mssql") schema = appDataSource.options?.schema;
+		else if (appDataSource.options.type === "sap") schema = appDataSource.options?.schema ?? appDataSource.options?.username ?? "public";
+		else if (appDataSource.options.type === "oracle") schema = appDataSource.options.schema;
+		let sqlCreateTableQuery = schema ? `CREATE TABLE "${schema}"."${currentTable.tableName}" (\n` : `CREATE TABLE ${currentTable.tableName} (\n`;
+		for (const [key, currentColumn] of currentTable.columns.entries()) {
+			if (key > 0) sqlCreateTableQuery += ", ";
+			sqlCreateTableQuery += `${currentColumn.columnName} ${currentColumn.dataType} ${currentColumn.isNullable ? "" : "NOT NULL"}`;
+		}
+		sqlCreateTableQuery += ") \n";
+		let sqlSelectInfoQuery;
+		if (appDataSource.options.type === "mysql") sqlSelectInfoQuery = `SELECT * FROM \`${currentTable.tableName}\` LIMIT ${nbSampleRow};\n`;
+		else if (appDataSource.options.type === "postgres") {
+			const schema$1 = appDataSource.options?.schema ?? "public";
+			sqlSelectInfoQuery = `SELECT * FROM "${schema$1}"."${currentTable.tableName}" LIMIT ${nbSampleRow};\n`;
+		} else if (appDataSource.options.type === "mssql") {
+			const schema$1 = appDataSource.options?.schema;
+			sqlSelectInfoQuery = schema$1 ? `SELECT TOP ${nbSampleRow} * FROM ${schema$1}.[${currentTable.tableName}];\n` : `SELECT TOP ${nbSampleRow} * FROM [${currentTable.tableName}];\n`;
+		} else if (appDataSource.options.type === "sap") {
+			const schema$1 = appDataSource.options?.schema ?? appDataSource.options?.username ?? "public";
+			sqlSelectInfoQuery = `SELECT * FROM "${schema$1}"."${currentTable.tableName}" LIMIT ${nbSampleRow};\n`;
+		} else if (appDataSource.options.type === "oracle") sqlSelectInfoQuery = `SELECT * FROM "${schema}"."${currentTable.tableName}" WHERE ROWNUM <= '${nbSampleRow}'`;
+		else sqlSelectInfoQuery = `SELECT * FROM "${currentTable.tableName}" LIMIT ${nbSampleRow};\n`;
+		const columnNamesConcatString = `${currentTable.columns.reduce((completeString, column) => `${completeString} ${column.columnName}`, "")}\n`;
+		let sample = "";
+		try {
+			const infoObjectResult = nbSampleRow ? await appDataSource.query(sqlSelectInfoQuery) : null;
+			sample = formatSqlResponseToSimpleTableString(infoObjectResult);
+		} catch (error) {
+			console.log(error);
+		}
+		globalString = globalString.concat(tableCustomDescription + sqlCreateTableQuery + sqlSelectInfoQuery + columnNamesConcatString + sample);
+	}
+	return globalString;
+};
+const getPromptTemplateFromDataSource = (appDataSource) => {
+	if (appDataSource.options.type === "postgres") return require_sql_db_prompt.SQL_POSTGRES_PROMPT;
+	if (appDataSource.options.type === "sqlite") return require_sql_db_prompt.SQL_SQLITE_PROMPT;
+	if (appDataSource.options.type === "mysql") return require_sql_db_prompt.SQL_MYSQL_PROMPT;
+	if (appDataSource.options.type === "mssql") return require_sql_db_prompt.SQL_MSSQL_PROMPT;
+	if (appDataSource.options.type === "sap") return require_sql_db_prompt.SQL_SAP_HANA_PROMPT;
+	if (appDataSource.options.type === "oracle") return require_sql_db_prompt.SQL_ORACLE_PROMPT;
+	return require_sql_db_prompt.DEFAULT_SQL_DATABASE_PROMPT;
+};
+
+//#endregion
+exports.generateTableInfoFromTables = generateTableInfoFromTables;
+exports.getPromptTemplateFromDataSource = getPromptTemplateFromDataSource;
+exports.getTableAndColumnsName = getTableAndColumnsName;
+exports.verifyIgnoreTablesExistInDatabase = verifyIgnoreTablesExistInDatabase;
+exports.verifyIncludeTablesExistInDatabase = verifyIncludeTablesExistInDatabase;
+exports.verifyListTablesExistInDatabase = verifyListTablesExistInDatabase;
+//# sourceMappingURL=sql_utils.cjs.map

package/dist/util/sql_utils.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql_utils.cjs","names":["tablesFromDatabase: Array<SqlTable>","listTables: Array<string>","errorPrefixMsg: string","onlyTableNames: Array<string>","table: SqlTable","includeTables: Array<string>","ignoreTables: Array<string>","rawResultsTableAndColumn: Array<RawResultTableAndColumn>","sqlTable: Array<SqlTable>","appDataSource: DataSource","sql","rep: Array<{ [key: string]: string }>","repLowerCase: Array<RawResultTableAndColumn>","rawResult: unknown","tables: Array<SqlTable> | undefined","nbSampleRow: number","customDescription?: Record<string, string>","schema","SQL_POSTGRES_PROMPT","SQL_SQLITE_PROMPT","SQL_MYSQL_PROMPT","SQL_MSSQL_PROMPT","SQL_SAP_HANA_PROMPT","SQL_ORACLE_PROMPT","DEFAULT_SQL_DATABASE_PROMPT"],"sources":["../../src/util/sql_utils.ts"],"sourcesContent":["import type { DataSource, DataSourceOptions } from \"typeorm\";\nimport type { PromptTemplate } from \"@langchain/core/prompts\";\nimport {\n  DEFAULT_SQL_DATABASE_PROMPT,\n  SQL_SAP_HANA_PROMPT,\n  SQL_MSSQL_PROMPT,\n  SQL_MYSQL_PROMPT,\n  SQL_POSTGRES_PROMPT,\n  SQL_SQLITE_PROMPT,\n  SQL_ORACLE_PROMPT,\n} from \"../chains/sql_db/sql_db_prompt.js\";\n\ninterface RawResultTableAndColumn {\n  table_name: string;\n  column_name: string;\n  data_type: string | undefined;\n  is_nullable: string;\n}\n\nexport interface SqlDatabaseParams {\n  includesTables?: Array<string>;\n  ignoreTables?: Array<string>;\n  sampleRowsInTableInfo?: number;\n  customDescription?: Record<string, string>;\n}\n\nexport interface SqlDatabaseOptionsParams extends SqlDatabaseParams {\n  appDataSourceOptions: DataSourceOptions;\n}\n\nexport interface SqlDatabaseDataSourceParams extends SqlDatabaseParams {\n  appDataSource: DataSource;\n}\n\nexport type SerializedSqlDatabase = SqlDatabaseOptionsParams & {\n  _type: string;\n};\n\nexport interface SqlTable {\n  tableName: string;\n  columns: SqlColumn[];\n}\n\nexport interface SqlColumn {\n  columnName: string;\n  dataType?: string;\n  isNullable?: boolean;\n}\n\nexport const verifyListTablesExistInDatabase = (\n  tablesFromDatabase: Array<SqlTable>,\n  listTables: Array<string>,\n  errorPrefixMsg: string\n): void => {\n  const onlyTableNames: Array<string> = tablesFromDatabase.map(\n    (table: SqlTable) => table.tableName\n  );\n  if (listTables.length > 0) {\n    for (const tableName of listTables) {\n      if (!onlyTableNames.includes(tableName)) {\n        throw new Error(\n          `${errorPrefixMsg} the table ${tableName} was not found in the database`\n        );\n      }\n    }\n  }\n};\n\nexport const verifyIncludeTablesExistInDatabase = (\n  tablesFromDatabase: Array<SqlTable>,\n  includeTables: Array<string>\n): void => {\n  verifyListTablesExistInDatabase(\n    tablesFromDatabase,\n    includeTables,\n    \"Include tables not found in database:\"\n  );\n};\n\nexport const verifyIgnoreTablesExistInDatabase = (\n  tablesFromDatabase: Array<SqlTable>,\n  ignoreTables: Array<string>\n): void => {\n  verifyListTablesExistInDatabase(\n    tablesFromDatabase,\n    ignoreTables,\n    \"Ignore tables not found in database:\"\n  );\n};\n\nconst formatToSqlTable = (\n  rawResultsTableAndColumn: Array<RawResultTableAndColumn>\n): Array<SqlTable> => {\n  const sqlTable: Array<SqlTable> = [];\n  for (const oneResult of rawResultsTableAndColumn) {\n    const sqlColumn = {\n      columnName: oneResult.column_name,\n      dataType: oneResult.data_type,\n      isNullable: oneResult.is_nullable === \"YES\",\n    };\n    const currentTable = sqlTable.find(\n      (oneTable) => oneTable.tableName === oneResult.table_name\n    );\n    if (currentTable) {\n      currentTable.columns.push(sqlColumn);\n    } else {\n      const newTable = {\n        tableName: oneResult.table_name,\n        columns: [sqlColumn],\n      };\n      sqlTable.push(newTable);\n    }\n  }\n\n  return sqlTable;\n};\n\nexport const getTableAndColumnsName = async (\n  appDataSource: DataSource\n): Promise<Array<SqlTable>> => {\n  let sql;\n  if (appDataSource.options.type === \"postgres\") {\n    const schema = appDataSource.options?.schema ?? \"public\";\n    sql = `SELECT \n            t.table_name, \n            c.* \n          FROM \n            information_schema.tables t \n              JOIN information_schema.columns c \n                ON t.table_name = c.table_name \n          WHERE \n            t.table_schema = '${schema}' \n              AND c.table_schema = '${schema}' \n          ORDER BY \n            t.table_name,\n            c.ordinal_position;`;\n    const rep = await appDataSource.query(sql);\n\n    return formatToSqlTable(rep);\n  }\n\n  if (\n    appDataSource.options.type === \"sqlite\" ||\n    appDataSource.options.type === \"better-sqlite3\" ||\n    appDataSource.options.type === \"sqljs\"\n  ) {\n    sql =\n      \"SELECT \\n\" +\n      \"   m.name AS table_name,\\n\" +\n      \"   p.name AS column_name,\\n\" +\n      \"   p.type AS data_type,\\n\" +\n      \"   CASE \\n\" +\n      \"      WHEN p.\\\"notnull\\\" = 0 THEN 'YES' \\n\" +\n      \"      ELSE 'NO' \\n\" +\n      \"   END AS is_nullable \\n\" +\n      \"FROM \\n\" +\n      \"   sqlite_master m \\n\" +\n      \"JOIN \\n\" +\n      \"   pragma_table_info(m.name) p \\n\" +\n      \"WHERE \\n\" +\n      \"   m.type = 'table' AND \\n\" +\n      \"   m.name NOT LIKE 'sqlite_%';\\n\";\n\n    const rep = await appDataSource.query(sql);\n\n    return formatToSqlTable(rep);\n  }\n\n  if (\n    appDataSource.options.type === \"mysql\" ||\n    appDataSource.options.type === \"aurora-mysql\"\n  ) {\n    sql =\n      \"SELECT \" +\n      \"TABLE_NAME AS table_name, \" +\n      \"COLUMN_NAME AS column_name, \" +\n      \"DATA_TYPE AS data_type, \" +\n      \"IS_NULLABLE AS is_nullable \" +\n      \"FROM INFORMATION_SCHEMA.COLUMNS \" +\n      `WHERE TABLE_SCHEMA = '${appDataSource.options.database}';`;\n\n    const rep = await appDataSource.query(sql);\n\n    return formatToSqlTable(rep);\n  }\n\n  if (appDataSource.options.type === \"mssql\") {\n    const schema = appDataSource.options?.schema;\n    const sql = `SELECT\n    TABLE_NAME AS table_name,\n    COLUMN_NAME AS column_name,\n    DATA_TYPE AS data_type,\n    IS_NULLABLE AS is_nullable\n    FROM INFORMATION_SCHEMA.COLUMNS\n    ${schema && `WHERE TABLE_SCHEMA = '${schema}'`} \nORDER BY TABLE_NAME, ORDINAL_POSITION;`;\n\n    const rep = await appDataSource.query(sql);\n    return formatToSqlTable(rep);\n  }\n\n  if (appDataSource.options.type === \"sap\") {\n    const schema = appDataSource.options?.schema ?? \"public\";\n    sql = `SELECT\n        TABLE_NAME,\n        COLUMN_NAME,\n        DATA_TYPE_NAME AS data_type,\n        CASE WHEN IS_NULLABLE='TRUE' THEN 'YES' ELSE 'NO' END AS is_nullable\n      FROM TABLE_COLUMNS\n      WHERE SCHEMA_NAME='${schema}'`;\n\n    const rep: Array<{ [key: string]: string }> = await appDataSource.query(\n      sql\n    );\n\n    const repLowerCase: Array<RawResultTableAndColumn> = [];\n    rep.forEach((_rep) =>\n      repLowerCase.push({\n        table_name: _rep.TABLE_NAME,\n        column_name: _rep.COLUMN_NAME,\n        data_type: _rep.DATA_TYPE,\n        is_nullable: _rep.IS_NULLABLE,\n      })\n    );\n\n    return formatToSqlTable(repLowerCase);\n  }\n  if (appDataSource.options.type === \"oracle\") {\n    const schemaName = appDataSource.options.schema;\n    const sql = `  \n      SELECT\n          TABLE_NAME AS \"table_name\",\n          COLUMN_NAME AS \"column_name\",\n          DATA_TYPE AS \"data_type\",\n          NULLABLE AS \"is_nullable\"\n      FROM ALL_TAB_COLS\n      WHERE\n          OWNER = UPPER('${schemaName}')`;\n    const rep = await appDataSource.query(sql);\n    return formatToSqlTable(rep);\n  }\n  throw new Error(\"Database type not implemented yet\");\n};\n\nconst formatSqlResponseToSimpleTableString = (rawResult: unknown): string => {\n  if (!rawResult || !Array.isArray(rawResult) || rawResult.length === 0) {\n    return \"\";\n  }\n\n  let globalString = \"\";\n  for (const oneRow of rawResult) {\n    globalString += `${Object.values(oneRow).reduce(\n      (completeString, columnValue) => `${completeString} ${columnValue}`,\n      \"\"\n    )}\\n`;\n  }\n\n  return globalString;\n};\n\nexport const generateTableInfoFromTables = async (\n  tables: Array<SqlTable> | undefined,\n  appDataSource: DataSource,\n  nbSampleRow: number,\n  customDescription?: Record<string, string>\n): Promise<string> => {\n  if (!tables) {\n    return \"\";\n  }\n\n  let globalString = \"\";\n  for (const currentTable of tables) {\n    // Add the custom info of the table\n    const tableCustomDescription =\n      customDescription &&\n      Object.keys(customDescription).includes(currentTable.tableName)\n        ? `${customDescription[currentTable.tableName]}\\n`\n        : \"\";\n    // Add the creation of the table in SQL\n    let schema = null;\n    if (appDataSource.options.type === \"postgres\") {\n      schema = appDataSource.options?.schema ?? \"public\";\n    } else if (appDataSource.options.type === \"mssql\") {\n      schema = appDataSource.options?.schema;\n    } else if (appDataSource.options.type === \"sap\") {\n      schema =\n        appDataSource.options?.schema ??\n        appDataSource.options?.username ??\n        \"public\";\n    } else if (appDataSource.options.type === \"oracle\") {\n      schema = appDataSource.options.schema;\n    }\n    let sqlCreateTableQuery = schema\n      ? `CREATE TABLE \"${schema}\".\"${currentTable.tableName}\" (\\n`\n      : `CREATE TABLE ${currentTable.tableName} (\\n`;\n    for (const [key, currentColumn] of currentTable.columns.entries()) {\n      if (key > 0) {\n        sqlCreateTableQuery += \", \";\n      }\n      sqlCreateTableQuery += `${currentColumn.columnName} ${\n        currentColumn.dataType\n      } ${currentColumn.isNullable ? \"\" : \"NOT NULL\"}`;\n    }\n    sqlCreateTableQuery += \") \\n\";\n\n    let sqlSelectInfoQuery;\n    if (appDataSource.options.type === \"mysql\") {\n      // We use backticks to quote the table names and thus allow for example spaces in table names\n      sqlSelectInfoQuery = `SELECT * FROM \\`${currentTable.tableName}\\` LIMIT ${nbSampleRow};\\n`;\n    } else if (appDataSource.options.type === \"postgres\") {\n      const schema = appDataSource.options?.schema ?? \"public\";\n      sqlSelectInfoQuery = `SELECT * FROM \"${schema}\".\"${currentTable.tableName}\" LIMIT ${nbSampleRow};\\n`;\n    } else if (appDataSource.options.type === \"mssql\") {\n      const schema = appDataSource.options?.schema;\n      sqlSelectInfoQuery = schema\n        ? `SELECT TOP ${nbSampleRow} * FROM ${schema}.[${currentTable.tableName}];\\n`\n        : `SELECT TOP ${nbSampleRow} * FROM [${currentTable.tableName}];\\n`;\n    } else if (appDataSource.options.type === \"sap\") {\n      const schema =\n        appDataSource.options?.schema ??\n        appDataSource.options?.username ??\n        \"public\";\n      sqlSelectInfoQuery = `SELECT * FROM \"${schema}\".\"${currentTable.tableName}\" LIMIT ${nbSampleRow};\\n`;\n    } else if (appDataSource.options.type === \"oracle\") {\n      sqlSelectInfoQuery = `SELECT * FROM \"${schema}\".\"${currentTable.tableName}\" WHERE ROWNUM <= '${nbSampleRow}'`;\n    } else {\n      sqlSelectInfoQuery = `SELECT * FROM \"${currentTable.tableName}\" LIMIT ${nbSampleRow};\\n`;\n    }\n\n    const columnNamesConcatString = `${currentTable.columns.reduce(\n      (completeString, column) => `${completeString} ${column.columnName}`,\n      \"\"\n    )}\\n`;\n\n    let sample = \"\";\n    try {\n      const infoObjectResult = nbSampleRow\n        ? await appDataSource.query(sqlSelectInfoQuery)\n        : null;\n      sample = formatSqlResponseToSimpleTableString(infoObjectResult);\n    } catch (error) {\n      // If the request fails we catch it and only display a log message\n      console.log(error);\n    }\n\n    globalString = globalString.concat(\n      tableCustomDescription +\n        sqlCreateTableQuery +\n        sqlSelectInfoQuery +\n        columnNamesConcatString +\n        sample\n    );\n  }\n\n  return globalString;\n};\n\nexport const getPromptTemplateFromDataSource = (\n  appDataSource: DataSource\n): PromptTemplate => {\n  if (appDataSource.options.type === \"postgres\") {\n    return SQL_POSTGRES_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"sqlite\") {\n    return SQL_SQLITE_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"mysql\") {\n    return SQL_MYSQL_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"mssql\") {\n    return SQL_MSSQL_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"sap\") {\n    return SQL_SAP_HANA_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"oracle\") {\n    return SQL_ORACLE_PROMPT;\n  }\n\n  return DEFAULT_SQL_DATABASE_PROMPT;\n};\n"],"mappings":";;;AAiDA,MAAa,kCAAkC,CAC7CA,oBACAC,YACAC,mBACS;CACT,MAAMC,iBAAgC,mBAAmB,IACvD,CAACC,UAAoB,MAAM,UAC5B;AACD,KAAI,WAAW,SAAS,GACtB;OAAK,MAAM,aAAa,WACtB,KAAI,CAAC,eAAe,SAAS,UAAU,CACrC,OAAM,IAAI,MACR,GAAG,eAAe,WAAW,EAAE,UAAU,8BAA8B,CAAC;CAG7E;AAEJ;AAED,MAAa,qCAAqC,CAChDJ,oBACAK,kBACS;CACT,gCACE,oBACA,eACA,wCACD;AACF;AAED,MAAa,oCAAoC,CAC/CL,oBACAM,iBACS;CACT,gCACE,oBACA,cACA,uCACD;AACF;AAED,MAAM,mBAAmB,CACvBC,6BACoB;CACpB,MAAMC,WAA4B,CAAE;AACpC,MAAK,MAAM,aAAa,0BAA0B;EAChD,MAAM,YAAY;GAChB,YAAY,UAAU;GACtB,UAAU,UAAU;GACpB,YAAY,UAAU,gBAAgB;EACvC;EACD,MAAM,eAAe,SAAS,KAC5B,CAAC,aAAa,SAAS,cAAc,UAAU,WAChD;AACD,MAAI,cACF,aAAa,QAAQ,KAAK,UAAU;OAC/B;GACL,MAAM,WAAW;IACf,WAAW,UAAU;IACrB,SAAS,CAAC,SAAU;GACrB;GACD,SAAS,KAAK,SAAS;EACxB;CACF;AAED,QAAO;AACR;AAED,MAAa,yBAAyB,OACpCC,kBAC6B;CAC7B,IAAI;AACJ,KAAI,cAAc,QAAQ,SAAS,YAAY;EAC7C,MAAM,SAAS,cAAc,SAAS,UAAU;EAChD,MAAM,CAAC;;;;;;;;8BAQmB,EAAE,OAAO;oCACH,EAAE,OAAO;;;+BAGd,CAAC;EAC5B,MAAM,MAAM,MAAM,cAAc,MAAM,IAAI;AAE1C,SAAO,iBAAiB,IAAI;CAC7B;AAED,KACE,cAAc,QAAQ,SAAS,YAC/B,cAAc,QAAQ,SAAS,oBAC/B,cAAc,QAAQ,SAAS,SAC/B;EACA,MACE;EAgBF,MAAM,MAAM,MAAM,cAAc,MAAM,IAAI;AAE1C,SAAO,iBAAiB,IAAI;CAC7B;AAED,KACE,cAAc,QAAQ,SAAS,WAC/B,cAAc,QAAQ,SAAS,gBAC/B;EACA,MACE,CAMC,sKAAsB,EAAE,cAAc,QAAQ,SAAS,EAAE,CAAC;EAE7D,MAAM,MAAM,MAAM,cAAc,MAAM,IAAI;AAE1C,SAAO,iBAAiB,IAAI;CAC7B;AAED,KAAI,cAAc,QAAQ,SAAS,SAAS;EAC1C,MAAM,SAAS,cAAc,SAAS;EACtC,MAAMC,QAAM,CAAC;;;;;;IAMb,EAAE,UAAU,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC,CAAC;sCACb,CAAC;EAEnC,MAAM,MAAM,MAAM,cAAc,MAAMA,MAAI;AAC1C,SAAO,iBAAiB,IAAI;CAC7B;AAED,KAAI,cAAc,QAAQ,SAAS,OAAO;EACxC,MAAM,SAAS,cAAc,SAAS,UAAU;EAChD,MAAM,CAAC;;;;;;yBAMc,EAAE,OAAO,CAAC,CAAC;EAEhC,MAAMC,MAAwC,MAAM,cAAc,MAChE,IACD;EAED,MAAMC,eAA+C,CAAE;EACvD,IAAI,QAAQ,CAAC,SACX,aAAa,KAAK;GAChB,YAAY,KAAK;GACjB,aAAa,KAAK;GAClB,WAAW,KAAK;GAChB,aAAa,KAAK;EACnB,EAAC,CACH;AAED,SAAO,iBAAiB,aAAa;CACtC;AACD,KAAI,cAAc,QAAQ,SAAS,UAAU;EAC3C,MAAM,aAAa,cAAc,QAAQ;EACzC,MAAMF,QAAM,CAAC;;;;;;;;yBAQQ,EAAE,WAAW,EAAE,CAAC;EACrC,MAAM,MAAM,MAAM,cAAc,MAAMA,MAAI;AAC1C,SAAO,iBAAiB,IAAI;CAC7B;AACD,OAAM,IAAI,MAAM;AACjB;AAED,MAAM,uCAAuC,CAACG,cAA+B;AAC3E,KAAI,CAAC,aAAa,CAAC,MAAM,QAAQ,UAAU,IAAI,UAAU,WAAW,EAClE,QAAO;CAGT,IAAI,eAAe;AACnB,MAAK,MAAM,UAAU,WACnB,gBAAgB,GAAG,OAAO,OAAO,OAAO,CAAC,OACvC,CAAC,gBAAgB,gBAAgB,GAAG,eAAe,CAAC,EAAE,aAAa,EACnE,GACD,CAAC,EAAE,CAAC;AAGP,QAAO;AACR;AAED,MAAa,8BAA8B,OACzCC,QACAL,eACAM,aACAC,sBACoB;AACpB,KAAI,CAAC,OACH,QAAO;CAGT,IAAI,eAAe;AACnB,MAAK,MAAM,gBAAgB,QAAQ;EAEjC,MAAM,yBACJ,qBACA,OAAO,KAAK,kBAAkB,CAAC,SAAS,aAAa,UAAU,GAC3D,GAAG,kBAAkB,aAAa,WAAW,EAAE,CAAC,GAChD;EAEN,IAAI,SAAS;AACb,MAAI,cAAc,QAAQ,SAAS,YACjC,SAAS,cAAc,SAAS,UAAU;WACjC,cAAc,QAAQ,SAAS,SACxC,SAAS,cAAc,SAAS;WACvB,cAAc,QAAQ,SAAS,OACxC,SACE,cAAc,SAAS,UACvB,cAAc,SAAS,YACvB;WACO,cAAc,QAAQ,SAAS,UACxC,SAAS,cAAc,QAAQ;EAEjC,IAAI,sBAAsB,SACtB,CAAC,cAAc,EAAE,OAAO,GAAG,EAAE,aAAa,UAAU,KAAK,CAAC,GAC1D,CAAC,aAAa,EAAE,aAAa,UAAU,IAAI,CAAC;AAChD,OAAK,MAAM,CAAC,KAAK,cAAc,IAAI,aAAa,QAAQ,SAAS,EAAE;AACjE,OAAI,MAAM,GACR,uBAAuB;GAEzB,uBAAuB,GAAG,cAAc,WAAW,CAAC,EAClD,cAAc,SACf,CAAC,EAAE,cAAc,aAAa,KAAK,YAAY;EACjD;EACD,uBAAuB;EAEvB,IAAI;AACJ,MAAI,cAAc,QAAQ,SAAS,SAEjC,qBAAqB,CAAC,gBAAgB,EAAE,aAAa,UAAU,SAAS,EAAE,YAAY,GAAG,CAAC;WACjF,cAAc,QAAQ,SAAS,YAAY;GACpD,MAAMC,WAAS,cAAc,SAAS,UAAU;GAChD,qBAAqB,CAAC,eAAe,EAAEA,SAAO,GAAG,EAAE,aAAa,UAAU,QAAQ,EAAE,YAAY,GAAG,CAAC;EACrG,WAAU,cAAc,QAAQ,SAAS,SAAS;GACjD,MAAMA,WAAS,cAAc,SAAS;GACtC,qBAAqBA,WACjB,CAAC,WAAW,EAAE,YAAY,QAAQ,EAAEA,SAAO,EAAE,EAAE,aAAa,UAAU,IAAI,CAAC,GAC3E,CAAC,WAAW,EAAE,YAAY,SAAS,EAAE,aAAa,UAAU,IAAI,CAAC;EACtE,WAAU,cAAc,QAAQ,SAAS,OAAO;GAC/C,MAAMA,WACJ,cAAc,SAAS,UACvB,cAAc,SAAS,YACvB;GACF,qBAAqB,CAAC,eAAe,EAAEA,SAAO,GAAG,EAAE,aAAa,UAAU,QAAQ,EAAE,YAAY,GAAG,CAAC;EACrG,WAAU,cAAc,QAAQ,SAAS,UACxC,qBAAqB,CAAC,eAAe,EAAE,OAAO,GAAG,EAAE,aAAa,UAAU,mBAAmB,EAAE,YAAY,CAAC,CAAC;OAE7G,qBAAqB,CAAC,eAAe,EAAE,aAAa,UAAU,QAAQ,EAAE,YAAY,GAAG,CAAC;EAG1F,MAAM,0BAA0B,GAAG,aAAa,QAAQ,OACtD,CAAC,gBAAgB,WAAW,GAAG,eAAe,CAAC,EAAE,OAAO,YAAY,EACpE,GACD,CAAC,EAAE,CAAC;EAEL,IAAI,SAAS;AACb,MAAI;GACF,MAAM,mBAAmB,cACrB,MAAM,cAAc,MAAM,mBAAmB,GAC7C;GACJ,SAAS,qCAAqC,iBAAiB;EAChE,SAAQ,OAAO;GAEd,QAAQ,IAAI,MAAM;EACnB;EAED,eAAe,aAAa,OAC1B,yBACE,sBACA,qBACA,0BACA,OACH;CACF;AAED,QAAO;AACR;AAED,MAAa,kCAAkC,CAC7CR,kBACmB;AACnB,KAAI,cAAc,QAAQ,SAAS,WACjC,QAAOS;AAGT,KAAI,cAAc,QAAQ,SAAS,SACjC,QAAOC;AAGT,KAAI,cAAc,QAAQ,SAAS,QACjC,QAAOC;AAGT,KAAI,cAAc,QAAQ,SAAS,QACjC,QAAOC;AAGT,KAAI,cAAc,QAAQ,SAAS,MACjC,QAAOC;AAGT,KAAI,cAAc,QAAQ,SAAS,SACjC,QAAOC;AAGT,QAAOC;AACR"}

package/dist/util/sql_utils.d.cts

@@ -0,0 +1,31 @@
+import { PromptTemplate } from "@langchain/core/prompts";
+import { DataSource, DataSourceOptions } from "typeorm";
+
+//#region src/util/sql_utils.d.ts
+interface SqlDatabaseParams {
+  includesTables?: Array<string>;
+  ignoreTables?: Array<string>;
+  sampleRowsInTableInfo?: number;
+  customDescription?: Record<string, string>;
+}
+interface SqlDatabaseOptionsParams extends SqlDatabaseParams {
+  appDataSourceOptions: DataSourceOptions;
+}
+interface SqlDatabaseDataSourceParams extends SqlDatabaseParams {
+  appDataSource: DataSource;
+}
+type SerializedSqlDatabase = SqlDatabaseOptionsParams & {
+  _type: string;
+};
+interface SqlTable {
+  tableName: string;
+  columns: SqlColumn[];
+}
+interface SqlColumn {
+  columnName: string;
+  dataType?: string;
+  isNullable?: boolean;
+}
+//#endregion
+export { SerializedSqlDatabase, SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams, SqlTable };
+//# sourceMappingURL=sql_utils.d.cts.map

package/dist/util/sql_utils.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql_utils.d.cts","names":["DataSource","DataSourceOptions","PromptTemplate","SqlDatabaseParams","Array","Record","SqlDatabaseOptionsParams","SqlDatabaseDataSourceParams","SerializedSqlDatabase","SqlTable","SqlColumn","verifyListTablesExistInDatabase","verifyIncludeTablesExistInDatabase","verifyIgnoreTablesExistInDatabase","getTableAndColumnsName","Promise","generateTableInfoFromTables","getPromptTemplateFromDataSource"],"sources":["../../src/util/sql_utils.d.ts"],"sourcesContent":["import type { DataSource, DataSourceOptions } from \"typeorm\";\nimport type { PromptTemplate } from \"@langchain/core/prompts\";\nexport interface SqlDatabaseParams {\n    includesTables?: Array<string>;\n    ignoreTables?: Array<string>;\n    sampleRowsInTableInfo?: number;\n    customDescription?: Record<string, string>;\n}\nexport interface SqlDatabaseOptionsParams extends SqlDatabaseParams {\n    appDataSourceOptions: DataSourceOptions;\n}\nexport interface SqlDatabaseDataSourceParams extends SqlDatabaseParams {\n    appDataSource: DataSource;\n}\nexport type SerializedSqlDatabase = SqlDatabaseOptionsParams & {\n    _type: string;\n};\nexport interface SqlTable {\n    tableName: string;\n    columns: SqlColumn[];\n}\nexport interface SqlColumn {\n    columnName: string;\n    dataType?: string;\n    isNullable?: boolean;\n}\nexport declare const verifyListTablesExistInDatabase: (tablesFromDatabase: SqlTable[], listTables: string[], errorPrefixMsg: string) => void;\nexport declare const verifyIncludeTablesExistInDatabase: (tablesFromDatabase: SqlTable[], includeTables: string[]) => void;\nexport declare const verifyIgnoreTablesExistInDatabase: (tablesFromDatabase: SqlTable[], ignoreTables: string[]) => void;\nexport declare const getTableAndColumnsName: (appDataSource: DataSource) => Promise<SqlTable[]>;\nexport declare const generateTableInfoFromTables: (tables: SqlTable[] | undefined, appDataSource: DataSource, nbSampleRow: number, customDescription?: Record<string, string> | undefined) => Promise<string>;\nexport declare const getPromptTemplateFromDataSource: (appDataSource: DataSource) => PromptTemplate<any, any>;\n"],"mappings":";;;;UAEiBG,iBAAAA;mBACIC;EADJD,YAAAA,CAAAA,EAEEC,KAFe,CAAA,MAAA,CAAA;EAAA,qBAAA,CAAA,EAAA,MAAA;EAAA,iBACbA,CAAAA,EAGGC,MAHHD,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA;;AAGGC,UAEPC,wBAAAA,SAAiCH,iBAF1BE,CAAAA;EAAM,oBAAA,EAGJJ,iBAHI;AAE9B;AAAyC,UAGxBM,2BAAAA,SAAoCJ,iBAHZ,CAAA;EAAA,aACfF,EAGPD,UAHOC;;AADyC,KAMvDO,qBAAAA,GAAwBF,wBAN+B,GAAA;EAGlDC,KAAAA,EAAAA,MAAAA;CAA2B;AACzBP,UAKFS,QAAAA,CALET;EAAU,SADwBG,EAAAA,MAAAA;EAAiB,OAAA,EAQzDO,SARyD,EAAA;AAGtE;AAGiBD,UAIAC,SAAAA,CAFJA;EAEIA,UAAAA,EAAS,MAAA"}

package/dist/util/sql_utils.d.ts

@@ -0,0 +1,31 @@
+import { PromptTemplate } from "@langchain/core/prompts";
+import { DataSource, DataSourceOptions } from "typeorm";
+
+//#region src/util/sql_utils.d.ts
+interface SqlDatabaseParams {
+  includesTables?: Array<string>;
+  ignoreTables?: Array<string>;
+  sampleRowsInTableInfo?: number;
+  customDescription?: Record<string, string>;
+}
+interface SqlDatabaseOptionsParams extends SqlDatabaseParams {
+  appDataSourceOptions: DataSourceOptions;
+}
+interface SqlDatabaseDataSourceParams extends SqlDatabaseParams {
+  appDataSource: DataSource;
+}
+type SerializedSqlDatabase = SqlDatabaseOptionsParams & {
+  _type: string;
+};
+interface SqlTable {
+  tableName: string;
+  columns: SqlColumn[];
+}
+interface SqlColumn {
+  columnName: string;
+  dataType?: string;
+  isNullable?: boolean;
+}
+//#endregion
+export { SerializedSqlDatabase, SqlDatabaseDataSourceParams, SqlDatabaseOptionsParams, SqlTable };
+//# sourceMappingURL=sql_utils.d.ts.map

package/dist/util/sql_utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql_utils.d.ts","names":["DataSource","DataSourceOptions","PromptTemplate","SqlDatabaseParams","Array","Record","SqlDatabaseOptionsParams","SqlDatabaseDataSourceParams","SerializedSqlDatabase","SqlTable","SqlColumn","verifyListTablesExistInDatabase","verifyIncludeTablesExistInDatabase","verifyIgnoreTablesExistInDatabase","getTableAndColumnsName","Promise","generateTableInfoFromTables","getPromptTemplateFromDataSource"],"sources":["../../src/util/sql_utils.d.ts"],"sourcesContent":["import type { DataSource, DataSourceOptions } from \"typeorm\";\nimport type { PromptTemplate } from \"@langchain/core/prompts\";\nexport interface SqlDatabaseParams {\n    includesTables?: Array<string>;\n    ignoreTables?: Array<string>;\n    sampleRowsInTableInfo?: number;\n    customDescription?: Record<string, string>;\n}\nexport interface SqlDatabaseOptionsParams extends SqlDatabaseParams {\n    appDataSourceOptions: DataSourceOptions;\n}\nexport interface SqlDatabaseDataSourceParams extends SqlDatabaseParams {\n    appDataSource: DataSource;\n}\nexport type SerializedSqlDatabase = SqlDatabaseOptionsParams & {\n    _type: string;\n};\nexport interface SqlTable {\n    tableName: string;\n    columns: SqlColumn[];\n}\nexport interface SqlColumn {\n    columnName: string;\n    dataType?: string;\n    isNullable?: boolean;\n}\nexport declare const verifyListTablesExistInDatabase: (tablesFromDatabase: SqlTable[], listTables: string[], errorPrefixMsg: string) => void;\nexport declare const verifyIncludeTablesExistInDatabase: (tablesFromDatabase: SqlTable[], includeTables: string[]) => void;\nexport declare const verifyIgnoreTablesExistInDatabase: (tablesFromDatabase: SqlTable[], ignoreTables: string[]) => void;\nexport declare const getTableAndColumnsName: (appDataSource: DataSource) => Promise<SqlTable[]>;\nexport declare const generateTableInfoFromTables: (tables: SqlTable[] | undefined, appDataSource: DataSource, nbSampleRow: number, customDescription?: Record<string, string> | undefined) => Promise<string>;\nexport declare const getPromptTemplateFromDataSource: (appDataSource: DataSource) => PromptTemplate<any, any>;\n"],"mappings":";;;;UAEiBG,iBAAAA;mBACIC;EADJD,YAAAA,CAAAA,EAEEC,KAFe,CAAA,MAAA,CAAA;EAAA,qBAAA,CAAA,EAAA,MAAA;EAAA,iBACbA,CAAAA,EAGGC,MAHHD,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA;;AAGGC,UAEPC,wBAAAA,SAAiCH,iBAF1BE,CAAAA;EAAM,oBAAA,EAGJJ,iBAHI;AAE9B;AAAyC,UAGxBM,2BAAAA,SAAoCJ,iBAHZ,CAAA;EAAA,aACfF,EAGPD,UAHOC;;AADyC,KAMvDO,qBAAAA,GAAwBF,wBAN+B,GAAA;EAGlDC,KAAAA,EAAAA,MAAAA;CAA2B;AACzBP,UAKFS,QAAAA,CALET;EAAU,SADwBG,EAAAA,MAAAA;EAAiB,OAAA,EAQzDO,SARyD,EAAA;AAGtE;AAGiBD,UAIAC,SAAAA,CAFJA;EAEIA,UAAAA,EAAS,MAAA"}

package/dist/util/sql_utils.js

@@ -0,0 +1,173 @@
+import { DEFAULT_SQL_DATABASE_PROMPT, SQL_MSSQL_PROMPT, SQL_MYSQL_PROMPT, SQL_ORACLE_PROMPT, SQL_POSTGRES_PROMPT, SQL_SAP_HANA_PROMPT, SQL_SQLITE_PROMPT } from "../chains/sql_db/sql_db_prompt.js";
+
+//#region src/util/sql_utils.ts
+const verifyListTablesExistInDatabase = (tablesFromDatabase, listTables, errorPrefixMsg) => {
+	const onlyTableNames = tablesFromDatabase.map((table) => table.tableName);
+	if (listTables.length > 0) {
+		for (const tableName of listTables) if (!onlyTableNames.includes(tableName)) throw new Error(`${errorPrefixMsg} the table ${tableName} was not found in the database`);
+	}
+};
+const verifyIncludeTablesExistInDatabase = (tablesFromDatabase, includeTables) => {
+	verifyListTablesExistInDatabase(tablesFromDatabase, includeTables, "Include tables not found in database:");
+};
+const verifyIgnoreTablesExistInDatabase = (tablesFromDatabase, ignoreTables) => {
+	verifyListTablesExistInDatabase(tablesFromDatabase, ignoreTables, "Ignore tables not found in database:");
+};
+const formatToSqlTable = (rawResultsTableAndColumn) => {
+	const sqlTable = [];
+	for (const oneResult of rawResultsTableAndColumn) {
+		const sqlColumn = {
+			columnName: oneResult.column_name,
+			dataType: oneResult.data_type,
+			isNullable: oneResult.is_nullable === "YES"
+		};
+		const currentTable = sqlTable.find((oneTable) => oneTable.tableName === oneResult.table_name);
+		if (currentTable) currentTable.columns.push(sqlColumn);
+		else {
+			const newTable = {
+				tableName: oneResult.table_name,
+				columns: [sqlColumn]
+			};
+			sqlTable.push(newTable);
+		}
+	}
+	return sqlTable;
+};
+const getTableAndColumnsName = async (appDataSource) => {
+	let sql;
+	if (appDataSource.options.type === "postgres") {
+		const schema = appDataSource.options?.schema ?? "public";
+		sql = `SELECT 
+            t.table_name, 
+            c.* 
+          FROM 
+            information_schema.tables t 
+              JOIN information_schema.columns c 
+                ON t.table_name = c.table_name 
+          WHERE 
+            t.table_schema = '${schema}' 
+              AND c.table_schema = '${schema}' 
+          ORDER BY 
+            t.table_name,
+            c.ordinal_position;`;
+		const rep = await appDataSource.query(sql);
+		return formatToSqlTable(rep);
+	}
+	if (appDataSource.options.type === "sqlite" || appDataSource.options.type === "better-sqlite3" || appDataSource.options.type === "sqljs") {
+		sql = "SELECT \n   m.name AS table_name,\n   p.name AS column_name,\n   p.type AS data_type,\n   CASE \n      WHEN p.\"notnull\" = 0 THEN 'YES' \n      ELSE 'NO' \n   END AS is_nullable \nFROM \n   sqlite_master m \nJOIN \n   pragma_table_info(m.name) p \nWHERE \n   m.type = 'table' AND \n   m.name NOT LIKE 'sqlite_%';\n";
+		const rep = await appDataSource.query(sql);
+		return formatToSqlTable(rep);
+	}
+	if (appDataSource.options.type === "mysql" || appDataSource.options.type === "aurora-mysql") {
+		sql = `SELECT TABLE_NAME AS table_name, COLUMN_NAME AS column_name, DATA_TYPE AS data_type, IS_NULLABLE AS is_nullable FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = '${appDataSource.options.database}';`;
+		const rep = await appDataSource.query(sql);
+		return formatToSqlTable(rep);
+	}
+	if (appDataSource.options.type === "mssql") {
+		const schema = appDataSource.options?.schema;
+		const sql$1 = `SELECT
+    TABLE_NAME AS table_name,
+    COLUMN_NAME AS column_name,
+    DATA_TYPE AS data_type,
+    IS_NULLABLE AS is_nullable
+    FROM INFORMATION_SCHEMA.COLUMNS
+    ${schema && `WHERE TABLE_SCHEMA = '${schema}'`} 
+ORDER BY TABLE_NAME, ORDINAL_POSITION;`;
+		const rep = await appDataSource.query(sql$1);
+		return formatToSqlTable(rep);
+	}
+	if (appDataSource.options.type === "sap") {
+		const schema = appDataSource.options?.schema ?? "public";
+		sql = `SELECT
+        TABLE_NAME,
+        COLUMN_NAME,
+        DATA_TYPE_NAME AS data_type,
+        CASE WHEN IS_NULLABLE='TRUE' THEN 'YES' ELSE 'NO' END AS is_nullable
+      FROM TABLE_COLUMNS
+      WHERE SCHEMA_NAME='${schema}'`;
+		const rep = await appDataSource.query(sql);
+		const repLowerCase = [];
+		rep.forEach((_rep) => repLowerCase.push({
+			table_name: _rep.TABLE_NAME,
+			column_name: _rep.COLUMN_NAME,
+			data_type: _rep.DATA_TYPE,
+			is_nullable: _rep.IS_NULLABLE
+		}));
+		return formatToSqlTable(repLowerCase);
+	}
+	if (appDataSource.options.type === "oracle") {
+		const schemaName = appDataSource.options.schema;
+		const sql$1 = `  
+      SELECT
+          TABLE_NAME AS "table_name",
+          COLUMN_NAME AS "column_name",
+          DATA_TYPE AS "data_type",
+          NULLABLE AS "is_nullable"
+      FROM ALL_TAB_COLS
+      WHERE
+          OWNER = UPPER('${schemaName}')`;
+		const rep = await appDataSource.query(sql$1);
+		return formatToSqlTable(rep);
+	}
+	throw new Error("Database type not implemented yet");
+};
+const formatSqlResponseToSimpleTableString = (rawResult) => {
+	if (!rawResult || !Array.isArray(rawResult) || rawResult.length === 0) return "";
+	let globalString = "";
+	for (const oneRow of rawResult) globalString += `${Object.values(oneRow).reduce((completeString, columnValue) => `${completeString} ${columnValue}`, "")}\n`;
+	return globalString;
+};
+const generateTableInfoFromTables = async (tables, appDataSource, nbSampleRow, customDescription) => {
+	if (!tables) return "";
+	let globalString = "";
+	for (const currentTable of tables) {
+		const tableCustomDescription = customDescription && Object.keys(customDescription).includes(currentTable.tableName) ? `${customDescription[currentTable.tableName]}\n` : "";
+		let schema = null;
+		if (appDataSource.options.type === "postgres") schema = appDataSource.options?.schema ?? "public";
+		else if (appDataSource.options.type === "mssql") schema = appDataSource.options?.schema;
+		else if (appDataSource.options.type === "sap") schema = appDataSource.options?.schema ?? appDataSource.options?.username ?? "public";
+		else if (appDataSource.options.type === "oracle") schema = appDataSource.options.schema;
+		let sqlCreateTableQuery = schema ? `CREATE TABLE "${schema}"."${currentTable.tableName}" (\n` : `CREATE TABLE ${currentTable.tableName} (\n`;
+		for (const [key, currentColumn] of currentTable.columns.entries()) {
+			if (key > 0) sqlCreateTableQuery += ", ";
+			sqlCreateTableQuery += `${currentColumn.columnName} ${currentColumn.dataType} ${currentColumn.isNullable ? "" : "NOT NULL"}`;
+		}
+		sqlCreateTableQuery += ") \n";
+		let sqlSelectInfoQuery;
+		if (appDataSource.options.type === "mysql") sqlSelectInfoQuery = `SELECT * FROM \`${currentTable.tableName}\` LIMIT ${nbSampleRow};\n`;
+		else if (appDataSource.options.type === "postgres") {
+			const schema$1 = appDataSource.options?.schema ?? "public";
+			sqlSelectInfoQuery = `SELECT * FROM "${schema$1}"."${currentTable.tableName}" LIMIT ${nbSampleRow};\n`;
+		} else if (appDataSource.options.type === "mssql") {
+			const schema$1 = appDataSource.options?.schema;
+			sqlSelectInfoQuery = schema$1 ? `SELECT TOP ${nbSampleRow} * FROM ${schema$1}.[${currentTable.tableName}];\n` : `SELECT TOP ${nbSampleRow} * FROM [${currentTable.tableName}];\n`;
+		} else if (appDataSource.options.type === "sap") {
+			const schema$1 = appDataSource.options?.schema ?? appDataSource.options?.username ?? "public";
+			sqlSelectInfoQuery = `SELECT * FROM "${schema$1}"."${currentTable.tableName}" LIMIT ${nbSampleRow};\n`;
+		} else if (appDataSource.options.type === "oracle") sqlSelectInfoQuery = `SELECT * FROM "${schema}"."${currentTable.tableName}" WHERE ROWNUM <= '${nbSampleRow}'`;
+		else sqlSelectInfoQuery = `SELECT * FROM "${currentTable.tableName}" LIMIT ${nbSampleRow};\n`;
+		const columnNamesConcatString = `${currentTable.columns.reduce((completeString, column) => `${completeString} ${column.columnName}`, "")}\n`;
+		let sample = "";
+		try {
+			const infoObjectResult = nbSampleRow ? await appDataSource.query(sqlSelectInfoQuery) : null;
+			sample = formatSqlResponseToSimpleTableString(infoObjectResult);
+		} catch (error) {
+			console.log(error);
+		}
+		globalString = globalString.concat(tableCustomDescription + sqlCreateTableQuery + sqlSelectInfoQuery + columnNamesConcatString + sample);
+	}
+	return globalString;
+};
+const getPromptTemplateFromDataSource = (appDataSource) => {
+	if (appDataSource.options.type === "postgres") return SQL_POSTGRES_PROMPT;
+	if (appDataSource.options.type === "sqlite") return SQL_SQLITE_PROMPT;
+	if (appDataSource.options.type === "mysql") return SQL_MYSQL_PROMPT;
+	if (appDataSource.options.type === "mssql") return SQL_MSSQL_PROMPT;
+	if (appDataSource.options.type === "sap") return SQL_SAP_HANA_PROMPT;
+	if (appDataSource.options.type === "oracle") return SQL_ORACLE_PROMPT;
+	return DEFAULT_SQL_DATABASE_PROMPT;
+};
+
+//#endregion
+export { generateTableInfoFromTables, getPromptTemplateFromDataSource, getTableAndColumnsName, verifyIgnoreTablesExistInDatabase, verifyIncludeTablesExistInDatabase, verifyListTablesExistInDatabase };
+//# sourceMappingURL=sql_utils.js.map

package/dist/util/sql_utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sql_utils.js","names":["tablesFromDatabase: Array<SqlTable>","listTables: Array<string>","errorPrefixMsg: string","onlyTableNames: Array<string>","table: SqlTable","includeTables: Array<string>","ignoreTables: Array<string>","rawResultsTableAndColumn: Array<RawResultTableAndColumn>","sqlTable: Array<SqlTable>","appDataSource: DataSource","sql","rep: Array<{ [key: string]: string }>","repLowerCase: Array<RawResultTableAndColumn>","rawResult: unknown","tables: Array<SqlTable> | undefined","nbSampleRow: number","customDescription?: Record<string, string>","schema"],"sources":["../../src/util/sql_utils.ts"],"sourcesContent":["import type { DataSource, DataSourceOptions } from \"typeorm\";\nimport type { PromptTemplate } from \"@langchain/core/prompts\";\nimport {\n  DEFAULT_SQL_DATABASE_PROMPT,\n  SQL_SAP_HANA_PROMPT,\n  SQL_MSSQL_PROMPT,\n  SQL_MYSQL_PROMPT,\n  SQL_POSTGRES_PROMPT,\n  SQL_SQLITE_PROMPT,\n  SQL_ORACLE_PROMPT,\n} from \"../chains/sql_db/sql_db_prompt.js\";\n\ninterface RawResultTableAndColumn {\n  table_name: string;\n  column_name: string;\n  data_type: string | undefined;\n  is_nullable: string;\n}\n\nexport interface SqlDatabaseParams {\n  includesTables?: Array<string>;\n  ignoreTables?: Array<string>;\n  sampleRowsInTableInfo?: number;\n  customDescription?: Record<string, string>;\n}\n\nexport interface SqlDatabaseOptionsParams extends SqlDatabaseParams {\n  appDataSourceOptions: DataSourceOptions;\n}\n\nexport interface SqlDatabaseDataSourceParams extends SqlDatabaseParams {\n  appDataSource: DataSource;\n}\n\nexport type SerializedSqlDatabase = SqlDatabaseOptionsParams & {\n  _type: string;\n};\n\nexport interface SqlTable {\n  tableName: string;\n  columns: SqlColumn[];\n}\n\nexport interface SqlColumn {\n  columnName: string;\n  dataType?: string;\n  isNullable?: boolean;\n}\n\nexport const verifyListTablesExistInDatabase = (\n  tablesFromDatabase: Array<SqlTable>,\n  listTables: Array<string>,\n  errorPrefixMsg: string\n): void => {\n  const onlyTableNames: Array<string> = tablesFromDatabase.map(\n    (table: SqlTable) => table.tableName\n  );\n  if (listTables.length > 0) {\n    for (const tableName of listTables) {\n      if (!onlyTableNames.includes(tableName)) {\n        throw new Error(\n          `${errorPrefixMsg} the table ${tableName} was not found in the database`\n        );\n      }\n    }\n  }\n};\n\nexport const verifyIncludeTablesExistInDatabase = (\n  tablesFromDatabase: Array<SqlTable>,\n  includeTables: Array<string>\n): void => {\n  verifyListTablesExistInDatabase(\n    tablesFromDatabase,\n    includeTables,\n    \"Include tables not found in database:\"\n  );\n};\n\nexport const verifyIgnoreTablesExistInDatabase = (\n  tablesFromDatabase: Array<SqlTable>,\n  ignoreTables: Array<string>\n): void => {\n  verifyListTablesExistInDatabase(\n    tablesFromDatabase,\n    ignoreTables,\n    \"Ignore tables not found in database:\"\n  );\n};\n\nconst formatToSqlTable = (\n  rawResultsTableAndColumn: Array<RawResultTableAndColumn>\n): Array<SqlTable> => {\n  const sqlTable: Array<SqlTable> = [];\n  for (const oneResult of rawResultsTableAndColumn) {\n    const sqlColumn = {\n      columnName: oneResult.column_name,\n      dataType: oneResult.data_type,\n      isNullable: oneResult.is_nullable === \"YES\",\n    };\n    const currentTable = sqlTable.find(\n      (oneTable) => oneTable.tableName === oneResult.table_name\n    );\n    if (currentTable) {\n      currentTable.columns.push(sqlColumn);\n    } else {\n      const newTable = {\n        tableName: oneResult.table_name,\n        columns: [sqlColumn],\n      };\n      sqlTable.push(newTable);\n    }\n  }\n\n  return sqlTable;\n};\n\nexport const getTableAndColumnsName = async (\n  appDataSource: DataSource\n): Promise<Array<SqlTable>> => {\n  let sql;\n  if (appDataSource.options.type === \"postgres\") {\n    const schema = appDataSource.options?.schema ?? \"public\";\n    sql = `SELECT \n            t.table_name, \n            c.* \n          FROM \n            information_schema.tables t \n              JOIN information_schema.columns c \n                ON t.table_name = c.table_name \n          WHERE \n            t.table_schema = '${schema}' \n              AND c.table_schema = '${schema}' \n          ORDER BY \n            t.table_name,\n            c.ordinal_position;`;\n    const rep = await appDataSource.query(sql);\n\n    return formatToSqlTable(rep);\n  }\n\n  if (\n    appDataSource.options.type === \"sqlite\" ||\n    appDataSource.options.type === \"better-sqlite3\" ||\n    appDataSource.options.type === \"sqljs\"\n  ) {\n    sql =\n      \"SELECT \\n\" +\n      \"   m.name AS table_name,\\n\" +\n      \"   p.name AS column_name,\\n\" +\n      \"   p.type AS data_type,\\n\" +\n      \"   CASE \\n\" +\n      \"      WHEN p.\\\"notnull\\\" = 0 THEN 'YES' \\n\" +\n      \"      ELSE 'NO' \\n\" +\n      \"   END AS is_nullable \\n\" +\n      \"FROM \\n\" +\n      \"   sqlite_master m \\n\" +\n      \"JOIN \\n\" +\n      \"   pragma_table_info(m.name) p \\n\" +\n      \"WHERE \\n\" +\n      \"   m.type = 'table' AND \\n\" +\n      \"   m.name NOT LIKE 'sqlite_%';\\n\";\n\n    const rep = await appDataSource.query(sql);\n\n    return formatToSqlTable(rep);\n  }\n\n  if (\n    appDataSource.options.type === \"mysql\" ||\n    appDataSource.options.type === \"aurora-mysql\"\n  ) {\n    sql =\n      \"SELECT \" +\n      \"TABLE_NAME AS table_name, \" +\n      \"COLUMN_NAME AS column_name, \" +\n      \"DATA_TYPE AS data_type, \" +\n      \"IS_NULLABLE AS is_nullable \" +\n      \"FROM INFORMATION_SCHEMA.COLUMNS \" +\n      `WHERE TABLE_SCHEMA = '${appDataSource.options.database}';`;\n\n    const rep = await appDataSource.query(sql);\n\n    return formatToSqlTable(rep);\n  }\n\n  if (appDataSource.options.type === \"mssql\") {\n    const schema = appDataSource.options?.schema;\n    const sql = `SELECT\n    TABLE_NAME AS table_name,\n    COLUMN_NAME AS column_name,\n    DATA_TYPE AS data_type,\n    IS_NULLABLE AS is_nullable\n    FROM INFORMATION_SCHEMA.COLUMNS\n    ${schema && `WHERE TABLE_SCHEMA = '${schema}'`} \nORDER BY TABLE_NAME, ORDINAL_POSITION;`;\n\n    const rep = await appDataSource.query(sql);\n    return formatToSqlTable(rep);\n  }\n\n  if (appDataSource.options.type === \"sap\") {\n    const schema = appDataSource.options?.schema ?? \"public\";\n    sql = `SELECT\n        TABLE_NAME,\n        COLUMN_NAME,\n        DATA_TYPE_NAME AS data_type,\n        CASE WHEN IS_NULLABLE='TRUE' THEN 'YES' ELSE 'NO' END AS is_nullable\n      FROM TABLE_COLUMNS\n      WHERE SCHEMA_NAME='${schema}'`;\n\n    const rep: Array<{ [key: string]: string }> = await appDataSource.query(\n      sql\n    );\n\n    const repLowerCase: Array<RawResultTableAndColumn> = [];\n    rep.forEach((_rep) =>\n      repLowerCase.push({\n        table_name: _rep.TABLE_NAME,\n        column_name: _rep.COLUMN_NAME,\n        data_type: _rep.DATA_TYPE,\n        is_nullable: _rep.IS_NULLABLE,\n      })\n    );\n\n    return formatToSqlTable(repLowerCase);\n  }\n  if (appDataSource.options.type === \"oracle\") {\n    const schemaName = appDataSource.options.schema;\n    const sql = `  \n      SELECT\n          TABLE_NAME AS \"table_name\",\n          COLUMN_NAME AS \"column_name\",\n          DATA_TYPE AS \"data_type\",\n          NULLABLE AS \"is_nullable\"\n      FROM ALL_TAB_COLS\n      WHERE\n          OWNER = UPPER('${schemaName}')`;\n    const rep = await appDataSource.query(sql);\n    return formatToSqlTable(rep);\n  }\n  throw new Error(\"Database type not implemented yet\");\n};\n\nconst formatSqlResponseToSimpleTableString = (rawResult: unknown): string => {\n  if (!rawResult || !Array.isArray(rawResult) || rawResult.length === 0) {\n    return \"\";\n  }\n\n  let globalString = \"\";\n  for (const oneRow of rawResult) {\n    globalString += `${Object.values(oneRow).reduce(\n      (completeString, columnValue) => `${completeString} ${columnValue}`,\n      \"\"\n    )}\\n`;\n  }\n\n  return globalString;\n};\n\nexport const generateTableInfoFromTables = async (\n  tables: Array<SqlTable> | undefined,\n  appDataSource: DataSource,\n  nbSampleRow: number,\n  customDescription?: Record<string, string>\n): Promise<string> => {\n  if (!tables) {\n    return \"\";\n  }\n\n  let globalString = \"\";\n  for (const currentTable of tables) {\n    // Add the custom info of the table\n    const tableCustomDescription =\n      customDescription &&\n      Object.keys(customDescription).includes(currentTable.tableName)\n        ? `${customDescription[currentTable.tableName]}\\n`\n        : \"\";\n    // Add the creation of the table in SQL\n    let schema = null;\n    if (appDataSource.options.type === \"postgres\") {\n      schema = appDataSource.options?.schema ?? \"public\";\n    } else if (appDataSource.options.type === \"mssql\") {\n      schema = appDataSource.options?.schema;\n    } else if (appDataSource.options.type === \"sap\") {\n      schema =\n        appDataSource.options?.schema ??\n        appDataSource.options?.username ??\n        \"public\";\n    } else if (appDataSource.options.type === \"oracle\") {\n      schema = appDataSource.options.schema;\n    }\n    let sqlCreateTableQuery = schema\n      ? `CREATE TABLE \"${schema}\".\"${currentTable.tableName}\" (\\n`\n      : `CREATE TABLE ${currentTable.tableName} (\\n`;\n    for (const [key, currentColumn] of currentTable.columns.entries()) {\n      if (key > 0) {\n        sqlCreateTableQuery += \", \";\n      }\n      sqlCreateTableQuery += `${currentColumn.columnName} ${\n        currentColumn.dataType\n      } ${currentColumn.isNullable ? \"\" : \"NOT NULL\"}`;\n    }\n    sqlCreateTableQuery += \") \\n\";\n\n    let sqlSelectInfoQuery;\n    if (appDataSource.options.type === \"mysql\") {\n      // We use backticks to quote the table names and thus allow for example spaces in table names\n      sqlSelectInfoQuery = `SELECT * FROM \\`${currentTable.tableName}\\` LIMIT ${nbSampleRow};\\n`;\n    } else if (appDataSource.options.type === \"postgres\") {\n      const schema = appDataSource.options?.schema ?? \"public\";\n      sqlSelectInfoQuery = `SELECT * FROM \"${schema}\".\"${currentTable.tableName}\" LIMIT ${nbSampleRow};\\n`;\n    } else if (appDataSource.options.type === \"mssql\") {\n      const schema = appDataSource.options?.schema;\n      sqlSelectInfoQuery = schema\n        ? `SELECT TOP ${nbSampleRow} * FROM ${schema}.[${currentTable.tableName}];\\n`\n        : `SELECT TOP ${nbSampleRow} * FROM [${currentTable.tableName}];\\n`;\n    } else if (appDataSource.options.type === \"sap\") {\n      const schema =\n        appDataSource.options?.schema ??\n        appDataSource.options?.username ??\n        \"public\";\n      sqlSelectInfoQuery = `SELECT * FROM \"${schema}\".\"${currentTable.tableName}\" LIMIT ${nbSampleRow};\\n`;\n    } else if (appDataSource.options.type === \"oracle\") {\n      sqlSelectInfoQuery = `SELECT * FROM \"${schema}\".\"${currentTable.tableName}\" WHERE ROWNUM <= '${nbSampleRow}'`;\n    } else {\n      sqlSelectInfoQuery = `SELECT * FROM \"${currentTable.tableName}\" LIMIT ${nbSampleRow};\\n`;\n    }\n\n    const columnNamesConcatString = `${currentTable.columns.reduce(\n      (completeString, column) => `${completeString} ${column.columnName}`,\n      \"\"\n    )}\\n`;\n\n    let sample = \"\";\n    try {\n      const infoObjectResult = nbSampleRow\n        ? await appDataSource.query(sqlSelectInfoQuery)\n        : null;\n      sample = formatSqlResponseToSimpleTableString(infoObjectResult);\n    } catch (error) {\n      // If the request fails we catch it and only display a log message\n      console.log(error);\n    }\n\n    globalString = globalString.concat(\n      tableCustomDescription +\n        sqlCreateTableQuery +\n        sqlSelectInfoQuery +\n        columnNamesConcatString +\n        sample\n    );\n  }\n\n  return globalString;\n};\n\nexport const getPromptTemplateFromDataSource = (\n  appDataSource: DataSource\n): PromptTemplate => {\n  if (appDataSource.options.type === \"postgres\") {\n    return SQL_POSTGRES_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"sqlite\") {\n    return SQL_SQLITE_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"mysql\") {\n    return SQL_MYSQL_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"mssql\") {\n    return SQL_MSSQL_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"sap\") {\n    return SQL_SAP_HANA_PROMPT;\n  }\n\n  if (appDataSource.options.type === \"oracle\") {\n    return SQL_ORACLE_PROMPT;\n  }\n\n  return DEFAULT_SQL_DATABASE_PROMPT;\n};\n"],"mappings":";;;AAiDA,MAAa,kCAAkC,CAC7CA,oBACAC,YACAC,mBACS;CACT,MAAMC,iBAAgC,mBAAmB,IACvD,CAACC,UAAoB,MAAM,UAC5B;AACD,KAAI,WAAW,SAAS,GACtB;OAAK,MAAM,aAAa,WACtB,KAAI,CAAC,eAAe,SAAS,UAAU,CACrC,OAAM,IAAI,MACR,GAAG,eAAe,WAAW,EAAE,UAAU,8BAA8B,CAAC;CAG7E;AAEJ;AAED,MAAa,qCAAqC,CAChDJ,oBACAK,kBACS;CACT,gCACE,oBACA,eACA,wCACD;AACF;AAED,MAAa,oCAAoC,CAC/CL,oBACAM,iBACS;CACT,gCACE,oBACA,cACA,uCACD;AACF;AAED,MAAM,mBAAmB,CACvBC,6BACoB;CACpB,MAAMC,WAA4B,CAAE;AACpC,MAAK,MAAM,aAAa,0BAA0B;EAChD,MAAM,YAAY;GAChB,YAAY,UAAU;GACtB,UAAU,UAAU;GACpB,YAAY,UAAU,gBAAgB;EACvC;EACD,MAAM,eAAe,SAAS,KAC5B,CAAC,aAAa,SAAS,cAAc,UAAU,WAChD;AACD,MAAI,cACF,aAAa,QAAQ,KAAK,UAAU;OAC/B;GACL,MAAM,WAAW;IACf,WAAW,UAAU;IACrB,SAAS,CAAC,SAAU;GACrB;GACD,SAAS,KAAK,SAAS;EACxB;CACF;AAED,QAAO;AACR;AAED,MAAa,yBAAyB,OACpCC,kBAC6B;CAC7B,IAAI;AACJ,KAAI,cAAc,QAAQ,SAAS,YAAY;EAC7C,MAAM,SAAS,cAAc,SAAS,UAAU;EAChD,MAAM,CAAC;;;;;;;;8BAQmB,EAAE,OAAO;oCACH,EAAE,OAAO;;;+BAGd,CAAC;EAC5B,MAAM,MAAM,MAAM,cAAc,MAAM,IAAI;AAE1C,SAAO,iBAAiB,IAAI;CAC7B;AAED,KACE,cAAc,QAAQ,SAAS,YAC/B,cAAc,QAAQ,SAAS,oBAC/B,cAAc,QAAQ,SAAS,SAC/B;EACA,MACE;EAgBF,MAAM,MAAM,MAAM,cAAc,MAAM,IAAI;AAE1C,SAAO,iBAAiB,IAAI;CAC7B;AAED,KACE,cAAc,QAAQ,SAAS,WAC/B,cAAc,QAAQ,SAAS,gBAC/B;EACA,MACE,CAMC,sKAAsB,EAAE,cAAc,QAAQ,SAAS,EAAE,CAAC;EAE7D,MAAM,MAAM,MAAM,cAAc,MAAM,IAAI;AAE1C,SAAO,iBAAiB,IAAI;CAC7B;AAED,KAAI,cAAc,QAAQ,SAAS,SAAS;EAC1C,MAAM,SAAS,cAAc,SAAS;EACtC,MAAMC,QAAM,CAAC;;;;;;IAMb,EAAE,UAAU,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC,CAAC;sCACb,CAAC;EAEnC,MAAM,MAAM,MAAM,cAAc,MAAMA,MAAI;AAC1C,SAAO,iBAAiB,IAAI;CAC7B;AAED,KAAI,cAAc,QAAQ,SAAS,OAAO;EACxC,MAAM,SAAS,cAAc,SAAS,UAAU;EAChD,MAAM,CAAC;;;;;;yBAMc,EAAE,OAAO,CAAC,CAAC;EAEhC,MAAMC,MAAwC,MAAM,cAAc,MAChE,IACD;EAED,MAAMC,eAA+C,CAAE;EACvD,IAAI,QAAQ,CAAC,SACX,aAAa,KAAK;GAChB,YAAY,KAAK;GACjB,aAAa,KAAK;GAClB,WAAW,KAAK;GAChB,aAAa,KAAK;EACnB,EAAC,CACH;AAED,SAAO,iBAAiB,aAAa;CACtC;AACD,KAAI,cAAc,QAAQ,SAAS,UAAU;EAC3C,MAAM,aAAa,cAAc,QAAQ;EACzC,MAAMF,QAAM,CAAC;;;;;;;;yBAQQ,EAAE,WAAW,EAAE,CAAC;EACrC,MAAM,MAAM,MAAM,cAAc,MAAMA,MAAI;AAC1C,SAAO,iBAAiB,IAAI;CAC7B;AACD,OAAM,IAAI,MAAM;AACjB;AAED,MAAM,uCAAuC,CAACG,cAA+B;AAC3E,KAAI,CAAC,aAAa,CAAC,MAAM,QAAQ,UAAU,IAAI,UAAU,WAAW,EAClE,QAAO;CAGT,IAAI,eAAe;AACnB,MAAK,MAAM,UAAU,WACnB,gBAAgB,GAAG,OAAO,OAAO,OAAO,CAAC,OACvC,CAAC,gBAAgB,gBAAgB,GAAG,eAAe,CAAC,EAAE,aAAa,EACnE,GACD,CAAC,EAAE,CAAC;AAGP,QAAO;AACR;AAED,MAAa,8BAA8B,OACzCC,QACAL,eACAM,aACAC,sBACoB;AACpB,KAAI,CAAC,OACH,QAAO;CAGT,IAAI,eAAe;AACnB,MAAK,MAAM,gBAAgB,QAAQ;EAEjC,MAAM,yBACJ,qBACA,OAAO,KAAK,kBAAkB,CAAC,SAAS,aAAa,UAAU,GAC3D,GAAG,kBAAkB,aAAa,WAAW,EAAE,CAAC,GAChD;EAEN,IAAI,SAAS;AACb,MAAI,cAAc,QAAQ,SAAS,YACjC,SAAS,cAAc,SAAS,UAAU;WACjC,cAAc,QAAQ,SAAS,SACxC,SAAS,cAAc,SAAS;WACvB,cAAc,QAAQ,SAAS,OACxC,SACE,cAAc,SAAS,UACvB,cAAc,SAAS,YACvB;WACO,cAAc,QAAQ,SAAS,UACxC,SAAS,cAAc,QAAQ;EAEjC,IAAI,sBAAsB,SACtB,CAAC,cAAc,EAAE,OAAO,GAAG,EAAE,aAAa,UAAU,KAAK,CAAC,GAC1D,CAAC,aAAa,EAAE,aAAa,UAAU,IAAI,CAAC;AAChD,OAAK,MAAM,CAAC,KAAK,cAAc,IAAI,aAAa,QAAQ,SAAS,EAAE;AACjE,OAAI,MAAM,GACR,uBAAuB;GAEzB,uBAAuB,GAAG,cAAc,WAAW,CAAC,EAClD,cAAc,SACf,CAAC,EAAE,cAAc,aAAa,KAAK,YAAY;EACjD;EACD,uBAAuB;EAEvB,IAAI;AACJ,MAAI,cAAc,QAAQ,SAAS,SAEjC,qBAAqB,CAAC,gBAAgB,EAAE,aAAa,UAAU,SAAS,EAAE,YAAY,GAAG,CAAC;WACjF,cAAc,QAAQ,SAAS,YAAY;GACpD,MAAMC,WAAS,cAAc,SAAS,UAAU;GAChD,qBAAqB,CAAC,eAAe,EAAEA,SAAO,GAAG,EAAE,aAAa,UAAU,QAAQ,EAAE,YAAY,GAAG,CAAC;EACrG,WAAU,cAAc,QAAQ,SAAS,SAAS;GACjD,MAAMA,WAAS,cAAc,SAAS;GACtC,qBAAqBA,WACjB,CAAC,WAAW,EAAE,YAAY,QAAQ,EAAEA,SAAO,EAAE,EAAE,aAAa,UAAU,IAAI,CAAC,GAC3E,CAAC,WAAW,EAAE,YAAY,SAAS,EAAE,aAAa,UAAU,IAAI,CAAC;EACtE,WAAU,cAAc,QAAQ,SAAS,OAAO;GAC/C,MAAMA,WACJ,cAAc,SAAS,UACvB,cAAc,SAAS,YACvB;GACF,qBAAqB,CAAC,eAAe,EAAEA,SAAO,GAAG,EAAE,aAAa,UAAU,QAAQ,EAAE,YAAY,GAAG,CAAC;EACrG,WAAU,cAAc,QAAQ,SAAS,UACxC,qBAAqB,CAAC,eAAe,EAAE,OAAO,GAAG,EAAE,aAAa,UAAU,mBAAmB,EAAE,YAAY,CAAC,CAAC;OAE7G,qBAAqB,CAAC,eAAe,EAAE,aAAa,UAAU,QAAQ,EAAE,YAAY,GAAG,CAAC;EAG1F,MAAM,0BAA0B,GAAG,aAAa,QAAQ,OACtD,CAAC,gBAAgB,WAAW,GAAG,eAAe,CAAC,EAAE,OAAO,YAAY,EACpE,GACD,CAAC,EAAE,CAAC;EAEL,IAAI,SAAS;AACb,MAAI;GACF,MAAM,mBAAmB,cACrB,MAAM,cAAc,MAAM,mBAAmB,GAC7C;GACJ,SAAS,qCAAqC,iBAAiB;EAChE,SAAQ,OAAO;GAEd,QAAQ,IAAI,MAAM;EACnB;EAED,eAAe,aAAa,OAC1B,yBACE,sBACA,qBACA,0BACA,OACH;CACF;AAED,QAAO;AACR;AAED,MAAa,kCAAkC,CAC7CR,kBACmB;AACnB,KAAI,cAAc,QAAQ,SAAS,WACjC,QAAO;AAGT,KAAI,cAAc,QAAQ,SAAS,SACjC,QAAO;AAGT,KAAI,cAAc,QAAQ,SAAS,QACjC,QAAO;AAGT,KAAI,cAAc,QAAQ,SAAS,QACjC,QAAO;AAGT,KAAI,cAAc,QAAQ,SAAS,MACjC,QAAO;AAGT,KAAI,cAAc,QAAQ,SAAS,SACjC,QAAO;AAGT,QAAO;AACR"}

package/dist/util/time.cjs

@@ -0,0 +1,25 @@
+const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+
+//#region src/util/time.ts
+var time_exports = {};
+require_rolldown_runtime.__export(time_exports, { sleep: () => sleep });
+/**
+* Sleep for a given amount of time.
+* @param ms - The number of milliseconds to sleep for. Defaults to 1000.
+* @returns A promise that resolves when the sleep is complete.
+*/
+async function sleep(ms = 1e3) {
+	return new Promise((resolve) => {
+		setTimeout(resolve, ms);
+	});
+}
+
+//#endregion
+exports.sleep = sleep;
+Object.defineProperty(exports, 'time_exports', {
+  enumerable: true,
+  get: function () {
+    return time_exports;
+  }
+});
+//# sourceMappingURL=time.cjs.map

package/dist/util/time.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"time.cjs","names":[],"sources":["../../src/util/time.ts"],"sourcesContent":["/**\n * Sleep for a given amount of time.\n * @param ms - The number of milliseconds to sleep for. Defaults to 1000.\n * @returns A promise that resolves when the sleep is complete.\n */\nexport async function sleep(ms = 1000): Promise<void> {\n  return new Promise<void>((resolve) => {\n    setTimeout(resolve, ms);\n  });\n}\n"],"mappings":";;;;;;;;;;AAKA,eAAsB,MAAM,KAAK,KAAqB;AACpD,QAAO,IAAI,QAAc,CAAC,YAAY;EACpC,WAAW,SAAS,GAAG;CACxB;AACF"}

package/dist/util/time.d.cts

@@ -0,0 +1,10 @@
+//#region src/util/time.d.ts
+/**
+ * Sleep for a given amount of time.
+ * @param ms - The number of milliseconds to sleep for. Defaults to 1000.
+ * @returns A promise that resolves when the sleep is complete.
+ */
+declare function sleep(ms?: number): Promise<void>;
+//#endregion
+export { sleep };
+//# sourceMappingURL=time.d.cts.map