@deepagents/text2sql 7.1.1 → 7.1.3

package/dist/lib/adapters/spreadsheet/index.js

@@ -876,7 +876,7 @@ var TableGrounding = class extends AbstractGrounding {
         );
       }
       if (depth < forwardLimit) {
-        const rels = await this.findOutgoingRelations(name);
+        const rels = await this.findOutgoingRelations(name, ctx);
         for (const rel of rels) {
           if (!this.isRelationshipVisible(rel)) continue;
           this.addRelationship(rel, allRelationships, seenRelationships);
@@ -908,7 +908,7 @@ var TableGrounding = class extends AbstractGrounding {
         );
       }
       if (depth < backwardLimit) {
-        const rels = await this.findIncomingRelations(name);
+        const rels = await this.findIncomingRelations(name, ctx);
         for (const rel of rels) {
           if (!this.isRelationshipVisible(rel)) continue;
           this.addRelationship(rel, allRelationships, seenRelationships);
@@ -1701,6 +1701,7 @@ function rowCount(config = {}) {
 }
 
 // packages/text2sql/src/lib/adapters/spreadsheet/spreadsheet.ts
+import { transaction } from "@zukhruf/sqlite-transaction";
 import { DatabaseSync } from "node:sqlite";
 
 // packages/text2sql/src/lib/adapters/spreadsheet/parser.ts
@@ -1839,8 +1840,9 @@ var Spreadsheet = class extends Sqlite {
   #db;
   constructor(options) {
     const sheets = parseFile(options.file);
+    using pending = new DisposableStack();
     const dbPath = options.database ?? ":memory:";
-    const db = new DatabaseSync(dbPath);
+    const db = pending.use(new DatabaseSync(dbPath));
     for (const sheet of sheets) {
       const createSQL = createTableSQL(sheet);
       db.exec(createSQL);
@@ -1851,6 +1853,7 @@ var Spreadsheet = class extends Sqlite {
       grounding: options.grounding
     });
     this.#db = db;
+    pending.move();
   }
   /**
    * Close the underlying SQLite database.
@@ -1875,23 +1878,19 @@ function loadData(db, sheet) {
   const placeholders = sheet.columns.map(() => "?").join(", ");
   const insertSQL = `INSERT INTO "${escapeIdentifier(sheet.name)}" (${columns}) VALUES (${placeholders})`;
   const stmt = db.prepare(insertSQL);
-  db.exec("BEGIN TRANSACTION");
-  try {
-    for (const row of sheet.rows) {
-      const values = sheet.columns.map((col) => {
-        const rawValue = row[col.originalKey];
-        return convertValue(rawValue, col.type);
-      });
-      stmt.run(...values);
-    }
-    db.exec("COMMIT");
-  } catch (error) {
-    try {
-      db.exec("ROLLBACK");
-    } catch {
-    }
-    throw error;
-  }
+  transaction(
+    db,
+    () => {
+      for (const row of sheet.rows) {
+        const values = sheet.columns.map((col) => {
+          const rawValue = row[col.originalKey];
+          return convertValue(rawValue, col.type);
+        });
+        stmt.run(...values);
+      }
+    },
+    { mode: "deferred" }
+  );
 }
 function convertValue(value, type) {
   if (value == null || value === "") {