@deepagents/text2sql 0.38.0 → 0.39.0

package/dist/lib/adapters/postgres/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,
@@ -1599,6 +1610,8 @@ var POSTGRES_ERROR_MAP = {
   }
 };
 var LOW_CARDINALITY_LIMIT = 20;
+var noopProgress2 = () => {
+};
 function isPostgresError(error) {
   return typeof error === "object" && error !== null && "code" in error && typeof error.code === "string";
 }
@@ -1747,12 +1760,12 @@ var Postgres = class extends Adapter {
     }
     return Array.from(relationships.values());
   }
-  async #annotateRowCounts(tables2, onProgress) {
+  async #annotateRowCounts(tables2, onProgress = noopProgress2) {
     const total = tables2.length;
     for (let i = 0; i < tables2.length; i++) {
       const table2 = tables2[i];
       const tableIdentifier = this.#formatQualifiedTableName(table2);
-      onProgress?.({
+      onProgress({
         type: "phase:progress",
         phase: "row_counts",
         table: table2.name,
@@ -1835,7 +1848,7 @@ var Postgres = class extends Adapter {
       }
     }
   }
-  async #annotateColumnStats(tables2, onProgress) {
+  async #annotateColumnStats(tables2, onProgress = noopProgress2) {
     if (!tables2.length) {
       return;
     }
@@ -1843,7 +1856,7 @@ var Postgres = class extends Adapter {
     for (let i = 0; i < tables2.length; i++) {
       const table2 = tables2[i];
       const tableIdentifier = this.#formatQualifiedTableName(table2);
-      onProgress?.({
+      onProgress({
         type: "phase:progress",
         phase: "column_stats",
         table: table2.name,
@@ -1884,12 +1897,12 @@ var Postgres = class extends Adapter {
       }
     }
   }
-  async #annotateLowCardinalityColumns(tables2, onProgress) {
+  async #annotateLowCardinalityColumns(tables2, onProgress = noopProgress2) {
     const total = tables2.length;
     for (let i = 0; i < tables2.length; i++) {
       const table2 = tables2[i];
       const tableIdentifier = this.#formatQualifiedTableName(table2);
-      onProgress?.({
+      onProgress({
         type: "phase:progress",
         phase: "low_cardinality",
         table: table2.name,
@@ -2222,7 +2235,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,