@deepagents/text2sql 0.38.0 → 0.39.0

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

@@ -148,6 +148,17 @@ function relationship(input) {
 }
 
 // packages/text2sql/src/lib/adapters/groundings/context.ts
+var noopProgress = () => {
+};
+function timestampProgress(onProgress) {
+  if (!onProgress) return noopProgress;
+  return (progress) => {
+    onProgress({
+      ...progress,
+      timestampMs: progress.timestampMs ?? Date.now()
+    });
+  };
+}
 function createGroundingContext(options = {}) {
   return {
     tables: [],
@@ -155,7 +166,7 @@ function createGroundingContext(options = {}) {
     relationships: [],
     info: void 0,
     cache: /* @__PURE__ */ new Map(),
-    onProgress: options.onProgress
+    onProgress: timestampProgress(options.onProgress)
   };
 }
 
@@ -302,7 +313,7 @@ var Adapter = class {
     for (const fn of this.grounding) {
       const grounding = fn(this);
       if (grounding.phase) {
-        ctx.onProgress?.({
+        ctx.onProgress({
           type: "phase:start",
           phase: grounding.phase,
           message: `Starting ${grounding.name} introspection...`
@@ -312,7 +323,7 @@ var Adapter = class {
         await grounding.execute(ctx);
       } finally {
         if (grounding.phase) {
-          ctx.onProgress?.({
+          ctx.onProgress({
             type: "phase:end",
             phase: grounding.phase,
             message: `Finished ${grounding.name} introspection.`
@@ -694,7 +705,7 @@ var ColumnStatsGrounding = class extends AbstractGrounding {
     const total = allContainers.length;
     for (let i = 0; i < allContainers.length; i++) {
       const container = allContainers[i];
-      ctx.onProgress?.({
+      ctx.onProgress({
         type: "phase:progress",
         phase: "column_stats",
         table: container.name,
@@ -812,7 +823,7 @@ var ColumnValuesGrounding = class extends AbstractGrounding {
     const total = allContainers.length;
     for (let i = 0; i < allContainers.length; i++) {
       const container = allContainers[i];
-      ctx.onProgress?.({
+      ctx.onProgress({
         type: "phase:progress",
         phase: "column_values",
         table: container.name,
@@ -883,7 +894,7 @@ var ConstraintGrounding = class extends AbstractGrounding {
     const total = ctx.tables.length;
     for (let i = 0; i < ctx.tables.length; i++) {
       const table2 = ctx.tables[i];
-      ctx.onProgress?.({
+      ctx.onProgress({
         type: "phase:progress",
         phase: "constraints",
         table: table2.name,
@@ -913,7 +924,7 @@ var IndexesGrounding = class extends AbstractGrounding {
     const total = ctx.tables.length;
     for (let i = 0; i < ctx.tables.length; i++) {
       const table2 = ctx.tables[i];
-      ctx.onProgress?.({
+      ctx.onProgress({
         type: "phase:progress",
         phase: "indexes",
         table: table2.name,
@@ -961,7 +972,7 @@ var RowCountGrounding = class extends AbstractGrounding {
     const total = ctx.tables.length;
     for (let i = 0; i < ctx.tables.length; i++) {
       const table2 = ctx.tables[i];
-      ctx.onProgress?.({
+      ctx.onProgress({
         type: "phase:progress",
         phase: "row_counts",
         table: table2.name,
@@ -1012,7 +1023,7 @@ var TableGrounding = class extends AbstractGrounding {
     if (!forward && !backward) {
       const tables3 = await Promise.all(
         seedTables.map(async (name, index2) => {
-          ctx.onProgress?.({
+          ctx.onProgress({
             type: "phase:progress",
             phase: "tables",
             table: name,
@@ -1047,7 +1058,7 @@ var TableGrounding = class extends AbstractGrounding {
       forwardVisited.add(name);
       if (!tables2[name]) {
         loadedCount++;
-        ctx.onProgress?.({
+        ctx.onProgress({
           type: "phase:progress",
           phase: "tables",
           table: name,
@@ -1079,7 +1090,7 @@ var TableGrounding = class extends AbstractGrounding {
       backwardVisited.add(name);
       if (!tables2[name]) {
         loadedCount++;
-        ctx.onProgress?.({
+        ctx.onProgress({
           type: "phase:progress",
           phase: "tables",
           table: name,
@@ -1659,7 +1670,7 @@ var ViewGrounding = class extends AbstractGrounding {
     const viewNames = await this.applyFilter();
     const views2 = await Promise.all(
       viewNames.map(async (name, index2) => {
-        ctx.onProgress?.({
+        ctx.onProgress({
           type: "phase:progress",
           phase: "views",
           table: name,