@langchain/classic 1.0.16 → 1.0.18-dev-1771365493298

package/dist/util/sql_utils.cjs

@@ -50,22 +50,19 @@ const getTableAndColumnsName = async (appDataSource) => {
           ORDER BY 
             t.table_name,
             c.ordinal_position;`;
-		const rep = await appDataSource.query(sql);
-		return formatToSqlTable(rep);
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	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);
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	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);
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	if (appDataSource.options.type === "mssql") {
 		const schema = appDataSource.options?.schema;
-		const sql$1 = `SELECT
+		const sql = `SELECT
     TABLE_NAME AS table_name,
     COLUMN_NAME AS column_name,
     DATA_TYPE AS data_type,
@@ -73,18 +70,16 @@ const getTableAndColumnsName = async (appDataSource) => {
     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);
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	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}'`;
+      WHERE SCHEMA_NAME='${appDataSource.options?.schema ?? "public"}'`;
 		const rep = await appDataSource.query(sql);
 		const repLowerCase = [];
 		rep.forEach((_rep) => repLowerCase.push({
@@ -96,8 +91,7 @@ ORDER BY TABLE_NAME, ORDINAL_POSITION;`;
 		return formatToSqlTable(repLowerCase);
 	}
 	if (appDataSource.options.type === "oracle") {
-		const schemaName = appDataSource.options.schema;
-		const sql$1 = `  
+		const sql = `  
       SELECT
           TABLE_NAME AS "table_name",
           COLUMN_NAME AS "column_name",
@@ -105,9 +99,8 @@ ORDER BY TABLE_NAME, ORDINAL_POSITION;`;
           NULLABLE AS "is_nullable"
       FROM ALL_TAB_COLS
       WHERE
-          OWNER = UPPER('${schemaName}')`;
-		const rep = await appDataSource.query(sql$1);
-		return formatToSqlTable(rep);
+          OWNER = UPPER('${appDataSource.options.schema}')`;
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	throw new Error("Database type not implemented yet");
 };
@@ -136,22 +129,17 @@ const generateTableInfoFromTables = async (tables, appDataSource, nbSampleRow, c
 		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.type === "postgres" ? appDataSource.options?.schema ?? "public" : "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 if (appDataSource.options.type === "postgres") sqlSelectInfoQuery = `SELECT * FROM "${appDataSource.options.type === "postgres" ? appDataSource.options?.schema ?? "public" : "public"}"."${currentTable.tableName}" LIMIT ${nbSampleRow};\n`;
+		else if (appDataSource.options.type === "mssql") {
+			const schema = appDataSource.options?.schema;
+			sqlSelectInfoQuery = schema ? `SELECT TOP ${nbSampleRow} * FROM ${schema}.[${currentTable.tableName}];\n` : `SELECT TOP ${nbSampleRow} * FROM [${currentTable.tableName}];\n`;
+		} else if (appDataSource.options.type === "sap") sqlSelectInfoQuery = `SELECT * FROM "${appDataSource.options?.schema ?? appDataSource.options?.username ?? "public"}"."${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);
+			sample = formatSqlResponseToSimpleTableString(nbSampleRow ? await appDataSource.query(sqlSelectInfoQuery) : null);
 		} catch (error) {
 			console.log(error);
 		}

package/dist/util/sql_utils.cjs.map

@@ -1 +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 (\n    appDataSource.options.type === \"postgres\" ||\n    appDataSource.options.type === \"aurora-postgres\"\n  ) {\n    const schema =\n      appDataSource.options.type === \"postgres\"\n        ? (appDataSource.options?.schema ?? \"public\")\n        : \"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 }> =\n      await appDataSource.query(sql);\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 === \"aurora-postgres\") {\n      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 =\n        appDataSource.options.type === \"postgres\"\n          ? (appDataSource.options?.schema ?? \"public\")\n          : \"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 (\n    appDataSource.options.type === \"postgres\" ||\n    appDataSource.options.type === \"aurora-postgres\"\n  ) {\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,KACE,cAAc,QAAQ,SAAS,cAC/B,cAAc,QAAQ,SAAS,mBAC/B;EACA,MAAM,SACJ,cAAc,QAAQ,SAAS,aAC1B,cAAc,SAAS,UAAU,WAClC;EACN,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,MACJ,MAAM,cAAc,MAAM,IAAI;EAEhC,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,mBACxC,SAAS;WACA,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,WACJ,cAAc,QAAQ,SAAS,aAC1B,cAAc,SAAS,UAAU,WAClC;GACN,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,KACE,cAAc,QAAQ,SAAS,cAC/B,cAAc,QAAQ,SAAS,kBAE/B,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"}
+{"version":3,"file":"sql_utils.cjs","names":["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 (\n    appDataSource.options.type === \"postgres\" ||\n    appDataSource.options.type === \"aurora-postgres\"\n  ) {\n    const schema =\n      appDataSource.options.type === \"postgres\"\n        ? (appDataSource.options?.schema ?? \"public\")\n        : \"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 }> =\n      await appDataSource.query(sql);\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 === \"aurora-postgres\") {\n      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 =\n        appDataSource.options.type === \"postgres\"\n          ? (appDataSource.options?.schema ?? \"public\")\n          : \"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 (\n    appDataSource.options.type === \"postgres\" ||\n    appDataSource.options.type === \"aurora-postgres\"\n  ) {\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,mCACX,oBACA,YACA,mBACS;CACT,MAAM,iBAAgC,mBAAmB,KACtD,UAAoB,MAAM,UAC5B;AACD,KAAI,WAAW,SAAS,GACtB;OAAK,MAAM,aAAa,WACtB,KAAI,CAAC,eAAe,SAAS,UAAU,CACrC,OAAM,IAAI,MACR,GAAG,eAAe,aAAa,UAAU,gCAC1C;;;AAMT,MAAa,sCACX,oBACA,kBACS;AACT,iCACE,oBACA,eACA,wCACD;;AAGH,MAAa,qCACX,oBACA,iBACS;AACT,iCACE,oBACA,cACA,uCACD;;AAGH,MAAM,oBACJ,6BACoB;CACpB,MAAM,WAA4B,EAAE;AACpC,MAAK,MAAM,aAAa,0BAA0B;EAChD,MAAM,YAAY;GAChB,YAAY,UAAU;GACtB,UAAU,UAAU;GACpB,YAAY,UAAU,gBAAgB;GACvC;EACD,MAAM,eAAe,SAAS,MAC3B,aAAa,SAAS,cAAc,UAAU,WAChD;AACD,MAAI,aACF,cAAa,QAAQ,KAAK,UAAU;OAC/B;GACL,MAAM,WAAW;IACf,WAAW,UAAU;IACrB,SAAS,CAAC,UAAU;IACrB;AACD,YAAS,KAAK,SAAS;;;AAI3B,QAAO;;AAGT,MAAa,yBAAyB,OACpC,kBAC6B;CAC7B,IAAI;AACJ,KACE,cAAc,QAAQ,SAAS,cAC/B,cAAc,QAAQ,SAAS,mBAC/B;EACA,MAAM,SACJ,cAAc,QAAQ,SAAS,aAC1B,cAAc,SAAS,UAAU,WAClC;AACN,QAAM;;;;;;;;gCAQsB,OAAO;sCACD,OAAO;;;;AAMzC,SAAO,iBAFK,MAAM,cAAc,MAAM,IAAI,CAEd;;AAG9B,KACE,cAAc,QAAQ,SAAS,YAC/B,cAAc,QAAQ,SAAS,oBAC/B,cAAc,QAAQ,SAAS,SAC/B;AACA,QACE;AAkBF,SAAO,iBAFK,MAAM,cAAc,MAAM,IAAI,CAEd;;AAG9B,KACE,cAAc,QAAQ,SAAS,WAC/B,cAAc,QAAQ,SAAS,gBAC/B;AACA,QACE,yKAMyB,cAAc,QAAQ,SAAS;AAI1D,SAAO,iBAFK,MAAM,cAAc,MAAM,IAAI,CAEd;;AAG9B,KAAI,cAAc,QAAQ,SAAS,SAAS;EAC1C,MAAM,SAAS,cAAc,SAAS;EACtC,MAAM,MAAM;;;;;;MAMV,UAAU,yBAAyB,OAAO,GAAG;;AAI/C,SAAO,iBADK,MAAM,cAAc,MAAM,IAAI,CACd;;AAG9B,KAAI,cAAc,QAAQ,SAAS,OAAO;AAExC,QAAM;;;;;;2BADS,cAAc,SAAS,UAAU,SAOlB;EAE9B,MAAM,MACJ,MAAM,cAAc,MAAM,IAAI;EAEhC,MAAM,eAA+C,EAAE;AACvD,MAAI,SAAS,SACX,aAAa,KAAK;GAChB,YAAY,KAAK;GACjB,aAAa,KAAK;GAClB,WAAW,KAAK;GAChB,aAAa,KAAK;GACnB,CAAC,CACH;AAED,SAAO,iBAAiB,aAAa;;AAEvC,KAAI,cAAc,QAAQ,SAAS,UAAU;EAE3C,MAAM,MAAM;;;;;;;;2BADO,cAAc,QAAQ,OASP;AAElC,SAAO,iBADK,MAAM,cAAc,MAAM,IAAI,CACd;;AAE9B,OAAM,IAAI,MAAM,oCAAoC;;AAGtD,MAAM,wCAAwC,cAA+B;AAC3E,KAAI,CAAC,aAAa,CAAC,MAAM,QAAQ,UAAU,IAAI,UAAU,WAAW,EAClE,QAAO;CAGT,IAAI,eAAe;AACnB,MAAK,MAAM,UAAU,UACnB,iBAAgB,GAAG,OAAO,OAAO,OAAO,CAAC,QACtC,gBAAgB,gBAAgB,GAAG,eAAe,GAAG,eACtD,GACD,CAAC;AAGJ,QAAO;;AAGT,MAAa,8BAA8B,OACzC,QACA,eACA,aACA,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,MAC7C;EAEN,IAAI,SAAS;AACb,MAAI,cAAc,QAAQ,SAAS,WACjC,UAAS,cAAc,SAAS,UAAU;WACjC,cAAc,QAAQ,SAAS,kBACxC,UAAS;WACA,cAAc,QAAQ,SAAS,QACxC,UAAS,cAAc,SAAS;WACvB,cAAc,QAAQ,SAAS,MACxC,UACE,cAAc,SAAS,UACvB,cAAc,SAAS,YACvB;WACO,cAAc,QAAQ,SAAS,SACxC,UAAS,cAAc,QAAQ;EAEjC,IAAI,sBAAsB,SACtB,iBAAiB,OAAO,KAAK,aAAa,UAAU,SACpD,gBAAgB,aAAa,UAAU;AAC3C,OAAK,MAAM,CAAC,KAAK,kBAAkB,aAAa,QAAQ,SAAS,EAAE;AACjE,OAAI,MAAM,EACR,wBAAuB;AAEzB,0BAAuB,GAAG,cAAc,WAAW,GACjD,cAAc,SACf,GAAG,cAAc,aAAa,KAAK;;AAEtC,yBAAuB;EAEvB,IAAI;AACJ,MAAI,cAAc,QAAQ,SAAS,QAEjC,sBAAqB,mBAAmB,aAAa,UAAU,WAAW,YAAY;WAC7E,cAAc,QAAQ,SAAS,WAKxC,sBAAqB,kBAHnB,cAAc,QAAQ,SAAS,aAC1B,cAAc,SAAS,UAAU,WAClC,SACwC,KAAK,aAAa,UAAU,UAAU,YAAY;WACvF,cAAc,QAAQ,SAAS,SAAS;GACjD,MAAM,SAAS,cAAc,SAAS;AACtC,wBAAqB,SACjB,cAAc,YAAY,UAAU,OAAO,IAAI,aAAa,UAAU,QACtE,cAAc,YAAY,WAAW,aAAa,UAAU;aACvD,cAAc,QAAQ,SAAS,MAKxC,sBAAqB,kBAHnB,cAAc,SAAS,UACvB,cAAc,SAAS,YACvB,SAC4C,KAAK,aAAa,UAAU,UAAU,YAAY;WACvF,cAAc,QAAQ,SAAS,SACxC,sBAAqB,kBAAkB,OAAO,KAAK,aAAa,UAAU,qBAAqB,YAAY;MAE3G,sBAAqB,kBAAkB,aAAa,UAAU,UAAU,YAAY;EAGtF,MAAM,0BAA0B,GAAG,aAAa,QAAQ,QACrD,gBAAgB,WAAW,GAAG,eAAe,GAAG,OAAO,cACxD,GACD,CAAC;EAEF,IAAI,SAAS;AACb,MAAI;AAIF,YAAS,qCAHgB,cACrB,MAAM,cAAc,MAAM,mBAAmB,GAC7C,KAC2D;WACxD,OAAO;AAEd,WAAQ,IAAI,MAAM;;AAGpB,iBAAe,aAAa,OAC1B,yBACE,sBACA,qBACA,0BACA,OACH;;AAGH,QAAO;;AAGT,MAAa,mCACX,kBACmB;AACnB,KACE,cAAc,QAAQ,SAAS,cAC/B,cAAc,QAAQ,SAAS,kBAE/B,QAAOA;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"}

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

@@ -1 +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//# sourceMappingURL=sql_utils.d.ts.map"],"mappings":";;;;UAEiBG,iBAAAA;mBACIC;EADJD,YAAAA,CAAAA,EAEEC,KAFe,CAAA,MAAA,CAAA;EACbA,qBAAAA,CAAAA,EAAAA,MAAAA;EACFA,iBAAAA,CAAAA,EAEKC,MAFLD,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA;;AAEW,UAEbE,wBAAAA,SAAiCH,iBAFpB,CAAA;EAEbG,oBAAAA,EACSL,iBADe;AAGzC;AAGYO,UAHKD,2BAAAA,SAAoCJ,iBAGO,CAAA;EAG3CM,aAAQ,EALNT,UAONU;AAEb;KAPYF,qBAAAA,GAAwBF;;;UAGnBG,QAAAA;;WAEJC;;UAEIA,SAAAA"}
+{"version":3,"file":"sql_utils.d.cts","names":[],"sources":["../../src/util/sql_utils.ts"],"mappings":";;;;UAmBiB,iBAAA;EACf,cAAA,GAAiB,KAAA;EACjB,YAAA,GAAe,KAAA;EACf,qBAAA;EACA,iBAAA,GAAoB,MAAA;AAAA;AAAA,UAGL,wBAAA,SAAiC,iBAAA;EAChD,oBAAA,EAAsB,iBAAA;AAAA;AAAA,UAGP,2BAAA,SAAoC,iBAAA;EACnD,aAAA,EAAe,UAAA;AAAA;AAAA,KAGL,qBAAA,GAAwB,wBAAA;EAClC,KAAA;AAAA;AAAA,UAGe,QAAA;EACf,SAAA;EACA,OAAA,EAAS,SAAA;AAAA;AAAA,UAGM,SAAA;EACf,UAAA;EACA,QAAA;EACA,UAAA;AAAA"}

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

@@ -1 +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//# sourceMappingURL=sql_utils.d.ts.map"],"mappings":";;;;UAEiBG,iBAAAA;mBACIC;EADJD,YAAAA,CAAAA,EAEEC,KAFe,CAAA,MAAA,CAAA;EACbA,qBAAAA,CAAAA,EAAAA,MAAAA;EACFA,iBAAAA,CAAAA,EAEKC,MAFLD,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA;;AAEW,UAEbE,wBAAAA,SAAiCH,iBAFpB,CAAA;EAEbG,oBAAAA,EACSL,iBADe;AAGzC;AAGYO,UAHKD,2BAAAA,SAAoCJ,iBAGO,CAAA;EAG3CM,aAAQ,EALNT,UAONU;AAEb;KAPYF,qBAAAA,GAAwBF;;;UAGnBG,QAAAA;;WAEJC;;UAEIA,SAAAA"}
+{"version":3,"file":"sql_utils.d.ts","names":[],"sources":["../../src/util/sql_utils.ts"],"mappings":";;;;UAmBiB,iBAAA;EACf,cAAA,GAAiB,KAAA;EACjB,YAAA,GAAe,KAAA;EACf,qBAAA;EACA,iBAAA,GAAoB,MAAA;AAAA;AAAA,UAGL,wBAAA,SAAiC,iBAAA;EAChD,oBAAA,EAAsB,iBAAA;AAAA;AAAA,UAGP,2BAAA,SAAoC,iBAAA;EACnD,aAAA,EAAe,UAAA;AAAA;AAAA,KAGL,qBAAA,GAAwB,wBAAA;EAClC,KAAA;AAAA;AAAA,UAGe,QAAA;EACf,SAAA;EACA,OAAA,EAAS,SAAA;AAAA;AAAA,UAGM,SAAA;EACf,UAAA;EACA,QAAA;EACA,UAAA;AAAA"}

package/dist/util/sql_utils.js

@@ -50,22 +50,19 @@ const getTableAndColumnsName = async (appDataSource) => {
           ORDER BY 
             t.table_name,
             c.ordinal_position;`;
-		const rep = await appDataSource.query(sql);
-		return formatToSqlTable(rep);
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	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);
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	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);
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	if (appDataSource.options.type === "mssql") {
 		const schema = appDataSource.options?.schema;
-		const sql$1 = `SELECT
+		const sql = `SELECT
     TABLE_NAME AS table_name,
     COLUMN_NAME AS column_name,
     DATA_TYPE AS data_type,
@@ -73,18 +70,16 @@ const getTableAndColumnsName = async (appDataSource) => {
     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);
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	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}'`;
+      WHERE SCHEMA_NAME='${appDataSource.options?.schema ?? "public"}'`;
 		const rep = await appDataSource.query(sql);
 		const repLowerCase = [];
 		rep.forEach((_rep) => repLowerCase.push({
@@ -96,8 +91,7 @@ ORDER BY TABLE_NAME, ORDINAL_POSITION;`;
 		return formatToSqlTable(repLowerCase);
 	}
 	if (appDataSource.options.type === "oracle") {
-		const schemaName = appDataSource.options.schema;
-		const sql$1 = `  
+		const sql = `  
       SELECT
           TABLE_NAME AS "table_name",
           COLUMN_NAME AS "column_name",
@@ -105,9 +99,8 @@ ORDER BY TABLE_NAME, ORDINAL_POSITION;`;
           NULLABLE AS "is_nullable"
       FROM ALL_TAB_COLS
       WHERE
-          OWNER = UPPER('${schemaName}')`;
-		const rep = await appDataSource.query(sql$1);
-		return formatToSqlTable(rep);
+          OWNER = UPPER('${appDataSource.options.schema}')`;
+		return formatToSqlTable(await appDataSource.query(sql));
 	}
 	throw new Error("Database type not implemented yet");
 };
@@ -136,22 +129,17 @@ const generateTableInfoFromTables = async (tables, appDataSource, nbSampleRow, c
 		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.type === "postgres" ? appDataSource.options?.schema ?? "public" : "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 if (appDataSource.options.type === "postgres") sqlSelectInfoQuery = `SELECT * FROM "${appDataSource.options.type === "postgres" ? appDataSource.options?.schema ?? "public" : "public"}"."${currentTable.tableName}" LIMIT ${nbSampleRow};\n`;
+		else if (appDataSource.options.type === "mssql") {
+			const schema = appDataSource.options?.schema;
+			sqlSelectInfoQuery = schema ? `SELECT TOP ${nbSampleRow} * FROM ${schema}.[${currentTable.tableName}];\n` : `SELECT TOP ${nbSampleRow} * FROM [${currentTable.tableName}];\n`;
+		} else if (appDataSource.options.type === "sap") sqlSelectInfoQuery = `SELECT * FROM "${appDataSource.options?.schema ?? appDataSource.options?.username ?? "public"}"."${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);
+			sample = formatSqlResponseToSimpleTableString(nbSampleRow ? await appDataSource.query(sqlSelectInfoQuery) : null);
 		} catch (error) {
 			console.log(error);
 		}

package/dist/util/sql_utils.js.map

@@ -1 +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 (\n    appDataSource.options.type === \"postgres\" ||\n    appDataSource.options.type === \"aurora-postgres\"\n  ) {\n    const schema =\n      appDataSource.options.type === \"postgres\"\n        ? (appDataSource.options?.schema ?? \"public\")\n        : \"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 }> =\n      await appDataSource.query(sql);\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 === \"aurora-postgres\") {\n      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 =\n        appDataSource.options.type === \"postgres\"\n          ? (appDataSource.options?.schema ?? \"public\")\n          : \"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 (\n    appDataSource.options.type === \"postgres\" ||\n    appDataSource.options.type === \"aurora-postgres\"\n  ) {\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,KACE,cAAc,QAAQ,SAAS,cAC/B,cAAc,QAAQ,SAAS,mBAC/B;EACA,MAAM,SACJ,cAAc,QAAQ,SAAS,aAC1B,cAAc,SAAS,UAAU,WAClC;EACN,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,MACJ,MAAM,cAAc,MAAM,IAAI;EAEhC,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,mBACxC,SAAS;WACA,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,WACJ,cAAc,QAAQ,SAAS,aAC1B,cAAc,SAAS,UAAU,WAClC;GACN,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,KACE,cAAc,QAAQ,SAAS,cAC/B,cAAc,QAAQ,SAAS,kBAE/B,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"}
+{"version":3,"file":"sql_utils.js","names":[],"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 (\n    appDataSource.options.type === \"postgres\" ||\n    appDataSource.options.type === \"aurora-postgres\"\n  ) {\n    const schema =\n      appDataSource.options.type === \"postgres\"\n        ? (appDataSource.options?.schema ?? \"public\")\n        : \"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 }> =\n      await appDataSource.query(sql);\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 === \"aurora-postgres\") {\n      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 =\n        appDataSource.options.type === \"postgres\"\n          ? (appDataSource.options?.schema ?? \"public\")\n          : \"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 (\n    appDataSource.options.type === \"postgres\" ||\n    appDataSource.options.type === \"aurora-postgres\"\n  ) {\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,mCACX,oBACA,YACA,mBACS;CACT,MAAM,iBAAgC,mBAAmB,KACtD,UAAoB,MAAM,UAC5B;AACD,KAAI,WAAW,SAAS,GACtB;OAAK,MAAM,aAAa,WACtB,KAAI,CAAC,eAAe,SAAS,UAAU,CACrC,OAAM,IAAI,MACR,GAAG,eAAe,aAAa,UAAU,gCAC1C;;;AAMT,MAAa,sCACX,oBACA,kBACS;AACT,iCACE,oBACA,eACA,wCACD;;AAGH,MAAa,qCACX,oBACA,iBACS;AACT,iCACE,oBACA,cACA,uCACD;;AAGH,MAAM,oBACJ,6BACoB;CACpB,MAAM,WAA4B,EAAE;AACpC,MAAK,MAAM,aAAa,0BAA0B;EAChD,MAAM,YAAY;GAChB,YAAY,UAAU;GACtB,UAAU,UAAU;GACpB,YAAY,UAAU,gBAAgB;GACvC;EACD,MAAM,eAAe,SAAS,MAC3B,aAAa,SAAS,cAAc,UAAU,WAChD;AACD,MAAI,aACF,cAAa,QAAQ,KAAK,UAAU;OAC/B;GACL,MAAM,WAAW;IACf,WAAW,UAAU;IACrB,SAAS,CAAC,UAAU;IACrB;AACD,YAAS,KAAK,SAAS;;;AAI3B,QAAO;;AAGT,MAAa,yBAAyB,OACpC,kBAC6B;CAC7B,IAAI;AACJ,KACE,cAAc,QAAQ,SAAS,cAC/B,cAAc,QAAQ,SAAS,mBAC/B;EACA,MAAM,SACJ,cAAc,QAAQ,SAAS,aAC1B,cAAc,SAAS,UAAU,WAClC;AACN,QAAM;;;;;;;;gCAQsB,OAAO;sCACD,OAAO;;;;AAMzC,SAAO,iBAFK,MAAM,cAAc,MAAM,IAAI,CAEd;;AAG9B,KACE,cAAc,QAAQ,SAAS,YAC/B,cAAc,QAAQ,SAAS,oBAC/B,cAAc,QAAQ,SAAS,SAC/B;AACA,QACE;AAkBF,SAAO,iBAFK,MAAM,cAAc,MAAM,IAAI,CAEd;;AAG9B,KACE,cAAc,QAAQ,SAAS,WAC/B,cAAc,QAAQ,SAAS,gBAC/B;AACA,QACE,yKAMyB,cAAc,QAAQ,SAAS;AAI1D,SAAO,iBAFK,MAAM,cAAc,MAAM,IAAI,CAEd;;AAG9B,KAAI,cAAc,QAAQ,SAAS,SAAS;EAC1C,MAAM,SAAS,cAAc,SAAS;EACtC,MAAM,MAAM;;;;;;MAMV,UAAU,yBAAyB,OAAO,GAAG;;AAI/C,SAAO,iBADK,MAAM,cAAc,MAAM,IAAI,CACd;;AAG9B,KAAI,cAAc,QAAQ,SAAS,OAAO;AAExC,QAAM;;;;;;2BADS,cAAc,SAAS,UAAU,SAOlB;EAE9B,MAAM,MACJ,MAAM,cAAc,MAAM,IAAI;EAEhC,MAAM,eAA+C,EAAE;AACvD,MAAI,SAAS,SACX,aAAa,KAAK;GAChB,YAAY,KAAK;GACjB,aAAa,KAAK;GAClB,WAAW,KAAK;GAChB,aAAa,KAAK;GACnB,CAAC,CACH;AAED,SAAO,iBAAiB,aAAa;;AAEvC,KAAI,cAAc,QAAQ,SAAS,UAAU;EAE3C,MAAM,MAAM;;;;;;;;2BADO,cAAc,QAAQ,OASP;AAElC,SAAO,iBADK,MAAM,cAAc,MAAM,IAAI,CACd;;AAE9B,OAAM,IAAI,MAAM,oCAAoC;;AAGtD,MAAM,wCAAwC,cAA+B;AAC3E,KAAI,CAAC,aAAa,CAAC,MAAM,QAAQ,UAAU,IAAI,UAAU,WAAW,EAClE,QAAO;CAGT,IAAI,eAAe;AACnB,MAAK,MAAM,UAAU,UACnB,iBAAgB,GAAG,OAAO,OAAO,OAAO,CAAC,QACtC,gBAAgB,gBAAgB,GAAG,eAAe,GAAG,eACtD,GACD,CAAC;AAGJ,QAAO;;AAGT,MAAa,8BAA8B,OACzC,QACA,eACA,aACA,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,MAC7C;EAEN,IAAI,SAAS;AACb,MAAI,cAAc,QAAQ,SAAS,WACjC,UAAS,cAAc,SAAS,UAAU;WACjC,cAAc,QAAQ,SAAS,kBACxC,UAAS;WACA,cAAc,QAAQ,SAAS,QACxC,UAAS,cAAc,SAAS;WACvB,cAAc,QAAQ,SAAS,MACxC,UACE,cAAc,SAAS,UACvB,cAAc,SAAS,YACvB;WACO,cAAc,QAAQ,SAAS,SACxC,UAAS,cAAc,QAAQ;EAEjC,IAAI,sBAAsB,SACtB,iBAAiB,OAAO,KAAK,aAAa,UAAU,SACpD,gBAAgB,aAAa,UAAU;AAC3C,OAAK,MAAM,CAAC,KAAK,kBAAkB,aAAa,QAAQ,SAAS,EAAE;AACjE,OAAI,MAAM,EACR,wBAAuB;AAEzB,0BAAuB,GAAG,cAAc,WAAW,GACjD,cAAc,SACf,GAAG,cAAc,aAAa,KAAK;;AAEtC,yBAAuB;EAEvB,IAAI;AACJ,MAAI,cAAc,QAAQ,SAAS,QAEjC,sBAAqB,mBAAmB,aAAa,UAAU,WAAW,YAAY;WAC7E,cAAc,QAAQ,SAAS,WAKxC,sBAAqB,kBAHnB,cAAc,QAAQ,SAAS,aAC1B,cAAc,SAAS,UAAU,WAClC,SACwC,KAAK,aAAa,UAAU,UAAU,YAAY;WACvF,cAAc,QAAQ,SAAS,SAAS;GACjD,MAAM,SAAS,cAAc,SAAS;AACtC,wBAAqB,SACjB,cAAc,YAAY,UAAU,OAAO,IAAI,aAAa,UAAU,QACtE,cAAc,YAAY,WAAW,aAAa,UAAU;aACvD,cAAc,QAAQ,SAAS,MAKxC,sBAAqB,kBAHnB,cAAc,SAAS,UACvB,cAAc,SAAS,YACvB,SAC4C,KAAK,aAAa,UAAU,UAAU,YAAY;WACvF,cAAc,QAAQ,SAAS,SACxC,sBAAqB,kBAAkB,OAAO,KAAK,aAAa,UAAU,qBAAqB,YAAY;MAE3G,sBAAqB,kBAAkB,aAAa,UAAU,UAAU,YAAY;EAGtF,MAAM,0BAA0B,GAAG,aAAa,QAAQ,QACrD,gBAAgB,WAAW,GAAG,eAAe,GAAG,OAAO,cACxD,GACD,CAAC;EAEF,IAAI,SAAS;AACb,MAAI;AAIF,YAAS,qCAHgB,cACrB,MAAM,cAAc,MAAM,mBAAmB,GAC7C,KAC2D;WACxD,OAAO;AAEd,WAAQ,IAAI,MAAM;;AAGpB,iBAAe,aAAa,OAC1B,yBACE,sBACA,qBACA,0BACA,OACH;;AAGH,QAAO;;AAGT,MAAa,mCACX,kBACmB;AACnB,KACE,cAAc,QAAQ,SAAS,cAC/B,cAAc,QAAQ,SAAS,kBAE/B,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"}

package/dist/util/time.cjs

@@ -1,8 +1,8 @@
-const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_runtime = require('../_virtual/_rolldown/runtime.cjs');
 
 //#region src/util/time.ts
-var time_exports = {};
-require_rolldown_runtime.__export(time_exports, { sleep: () => sleep });
+var time_exports = /* @__PURE__ */ require_runtime.__exportAll({ sleep: () => sleep });
 /**
 * Sleep for a given amount of time.
 * @param ms - The number of milliseconds to sleep for. Defaults to 1000.

package/dist/util/time.cjs.map

@@ -1 +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"}
+{"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,SAAe,YAAY;AACpC,aAAW,SAAS,GAAG;GACvB"}

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

@@ -1 +1 @@
-{"version":3,"file":"time.d.cts","names":["sleep","Promise"],"sources":["../../src/util/time.d.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 declare function sleep(ms?: number): Promise<void>;\n//# sourceMappingURL=time.d.ts.map"],"mappings":";;AAKA;;;;iBAAwBA,KAAAA,eAAoBC"}
+{"version":3,"file":"time.d.cts","names":[],"sources":["../../src/util/time.ts"],"mappings":";;AAKA;;;;iBAAsB,KAAA,CAAM,EAAA,YAAY,OAAA"}

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

@@ -1 +1 @@
-{"version":3,"file":"time.d.ts","names":["sleep","Promise"],"sources":["../../src/util/time.d.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 declare function sleep(ms?: number): Promise<void>;\n//# sourceMappingURL=time.d.ts.map"],"mappings":";;AAKA;;;;iBAAwBA,KAAAA,eAAoBC"}
+{"version":3,"file":"time.d.ts","names":[],"sources":["../../src/util/time.ts"],"mappings":";;AAKA;;;;iBAAsB,KAAA,CAAM,EAAA,YAAY,OAAA"}

package/dist/util/time.js

@@ -1,8 +1,7 @@
-import { __export } from "../_virtual/rolldown_runtime.js";
+import { __exportAll } from "../_virtual/_rolldown/runtime.js";
 
 //#region src/util/time.ts
-var time_exports = {};
-__export(time_exports, { sleep: () => sleep });
+var time_exports = /* @__PURE__ */ __exportAll({ sleep: () => sleep });
 /**
 * Sleep for a given amount of time.
 * @param ms - The number of milliseconds to sleep for. Defaults to 1000.

package/dist/util/time.js.map

@@ -1 +1 @@
-{"version":3,"file":"time.js","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"}
+{"version":3,"file":"time.js","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,SAAe,YAAY;AACpC,aAAW,SAAS,GAAG;GACvB"}

package/dist/vectorstores/memory.cjs

@@ -1,12 +1,12 @@
-const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_runtime = require('../_virtual/_rolldown/runtime.cjs');
 const require_similarities = require('../util/ml-distance/similarities.cjs');
-const __langchain_core_documents = require_rolldown_runtime.__toESM(require("@langchain/core/documents"));
-const __langchain_core_utils_math = require_rolldown_runtime.__toESM(require("@langchain/core/utils/math"));
-const __langchain_core_vectorstores = require_rolldown_runtime.__toESM(require("@langchain/core/vectorstores"));
+let _langchain_core_documents = require("@langchain/core/documents");
+let _langchain_core_utils_math = require("@langchain/core/utils/math");
+let _langchain_core_vectorstores = require("@langchain/core/vectorstores");
 
 //#region src/vectorstores/memory.ts
-var memory_exports = {};
-require_rolldown_runtime.__export(memory_exports, { MemoryVectorStore: () => MemoryVectorStore });
+var memory_exports = /* @__PURE__ */ require_runtime.__exportAll({ MemoryVectorStore: () => MemoryVectorStore });
 /**
 * In-memory, ephemeral vector store.
 *
@@ -117,13 +117,13 @@ require_rolldown_runtime.__export(memory_exports, { MemoryVectorStore: () => Mem
 *
 * <br />
 */
-var MemoryVectorStore = class MemoryVectorStore extends __langchain_core_vectorstores.VectorStore {
+var MemoryVectorStore = class MemoryVectorStore extends _langchain_core_vectorstores.VectorStore {
 	memoryVectors = [];
 	similarity;
 	_vectorstoreType() {
 		return "memory";
 	}
-	constructor(embeddings, { similarity,...rest } = {}) {
+	constructor(embeddings, { similarity, ...rest } = {}) {
 		super(embeddings, rest);
 		this.similarity = similarity ?? require_similarities.cosine;
 	}
@@ -158,15 +158,13 @@ var MemoryVectorStore = class MemoryVectorStore extends __langchain_core_vectors
 	async _queryVectors(query, k, filter) {
 		const filterFunction = (memoryVector) => {
 			if (!filter) return true;
-			const doc = new __langchain_core_documents.Document({
+			return filter(new _langchain_core_documents.Document({
 				metadata: memoryVector.metadata,
 				pageContent: memoryVector.content,
 				id: memoryVector.id
-			});
-			return filter(doc);
+			}));
 		};
-		const filteredMemoryVectors = this.memoryVectors.filter(filterFunction);
-		return filteredMemoryVectors.map((vector, index) => ({
+		return this.memoryVectors.filter(filterFunction).map((vector, index) => ({
 			similarity: this.similarity(query, vector.embedding),
 			index,
 			metadata: vector.metadata,
@@ -186,20 +184,16 @@ var MemoryVectorStore = class MemoryVectorStore extends __langchain_core_vectors
 	* @returns Promise that resolves with an array of tuples, each containing a `Document` and its similarity score.
 	*/
 	async similaritySearchVectorWithScore(query, k, filter) {
-		const searches = await this._queryVectors(query, k, filter);
-		const result = searches.map((search) => [new __langchain_core_documents.Document({
+		return (await this._queryVectors(query, k, filter)).map((search) => [new _langchain_core_documents.Document({
 			metadata: search.metadata,
 			pageContent: search.content,
 			id: search.id
 		}), search.similarity]);
-		return result;
 	}
 	async maxMarginalRelevanceSearch(query, options) {
 		const queryEmbedding = await this.embeddings.embedQuery(query);
 		const searches = await this._queryVectors(queryEmbedding, options.fetchK ?? 20, options.filter);
-		const embeddingList = searches.map((searchResp) => searchResp.embedding);
-		const mmrIndexes = (0, __langchain_core_utils_math.maximalMarginalRelevance)(queryEmbedding, embeddingList, options.lambda, options.k);
-		return mmrIndexes.map((idx) => new __langchain_core_documents.Document({
+		return (0, _langchain_core_utils_math.maximalMarginalRelevance)(queryEmbedding, searches.map((searchResp) => searchResp.embedding), options.lambda, options.k).map((idx) => new _langchain_core_documents.Document({
 			metadata: searches[idx].metadata,
 			pageContent: searches[idx].content,
 			id: searches[idx].id
@@ -219,7 +213,7 @@ var MemoryVectorStore = class MemoryVectorStore extends __langchain_core_vectors
 		const docs = [];
 		for (let i = 0; i < texts.length; i += 1) {
 			const metadata = Array.isArray(metadatas) ? metadatas[i] : metadatas;
-			const newDoc = new __langchain_core_documents.Document({
+			const newDoc = new _langchain_core_documents.Document({
 				pageContent: texts[i],
 				metadata
 			});
@@ -249,8 +243,7 @@ var MemoryVectorStore = class MemoryVectorStore extends __langchain_core_vectors
 	* @returns Promise that resolves with a new `MemoryVectorStore` instance.
 	*/
 	static async fromExistingIndex(embeddings, dbConfig) {
-		const instance = new this(embeddings, dbConfig);
-		return instance;
+		return new this(embeddings, dbConfig);
 	}
 };