@deepagents/text2sql 7.1.1 → 7.1.2

package/dist/lib/adapters/bigquery/constraint.bigquery.grounding.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"constraint.bigquery.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/bigquery/constraint.bigquery.grounding.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAS,eAAe,EAAE,MAAM,eAAe,CAAC;AAC5D,OAAO,EACL,mBAAmB,EACnB,KAAK,yBAAyB,EAC/B,MAAM,uCAAuC,CAAC;AAC/C,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAGjE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAgB9C,MAAM,WAAW,iCAAkC,SAAQ,yBAAyB;CAAG;AAEvF,qBAAa,2BAA4B,SAAQ,mBAAmB;;gBAKhE,OAAO,EAAE,QAAQ,EACjB,MAAM,GAAE,iCAAsC;IAMjC,OAAO,CAAC,GAAG,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;cA4LnC,cAAc,CACrC,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,eAAe,EAAE,CAAC;CAG9B"}
+{"version":3,"file":"constraint.bigquery.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/bigquery/constraint.bigquery.grounding.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAS,eAAe,EAAE,MAAM,eAAe,CAAC;AAC5D,OAAO,EACL,mBAAmB,EACnB,KAAK,yBAAyB,EAC/B,MAAM,uCAAuC,CAAC;AAC/C,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAGjE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAgB9C,MAAM,WAAW,iCAAkC,SAAQ,yBAAyB;CAAG;AAEvF,qBAAa,2BAA4B,SAAQ,mBAAmB;;gBAIhE,OAAO,EAAE,QAAQ,EACjB,MAAM,GAAE,iCAAsC;IAMjC,OAAO,CAAC,GAAG,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;cAgMnC,cAAc,CACrC,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,eAAe,EAAE,CAAC;CAG9B"}

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

@@ -760,7 +760,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);
@@ -792,7 +792,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);
@@ -1400,13 +1400,11 @@ var tableKeyColumnRow = keyColumnUsageRow.extend({
 });
 var BigQueryConstraintGrounding = class extends ConstraintGrounding {
   #adapter;
-  #cache;
   constructor(adapter, config = {}) {
     super(config);
     this.#adapter = adapter;
   }
   async execute(ctx) {
-    this.#cache = ctx.cache;
     const byDataset = /* @__PURE__ */ new Map();
     for (const table2 of ctx.tables) {
       const { schema: dataset } = this.#adapter.parseTableName(table2.name);
@@ -1429,7 +1427,7 @@ var BigQueryConstraintGrounding = class extends ConstraintGrounding {
         });
       }
       try {
-        await this.#batchConstraints(dataset, tables2);
+        await this.#batchConstraints(dataset, tables2, ctx.cache);
       } catch (error) {
         console.warn(
           "Error collecting constraints for dataset",
@@ -1439,7 +1437,7 @@ var BigQueryConstraintGrounding = class extends ConstraintGrounding {
       }
     }
   }
-  async #batchConstraints(dataset, tables2) {
+  async #batchConstraints(dataset, tables2, cache) {
     const tableNames = tables2.map(
       (t) => this.#adapter.parseTableName(t.name).table
     );
@@ -1449,7 +1447,7 @@ var BigQueryConstraintGrounding = class extends ConstraintGrounding {
       constraintsByTable.set(name, []);
     }
     await this.#batchColumnMetadata(dataset, inList, constraintsByTable);
-    await this.#batchKeyConstraints(dataset, inList, constraintsByTable);
+    await this.#batchKeyConstraints(dataset, inList, constraintsByTable, cache);
     for (const table2 of tables2) {
       const rawName = this.#adapter.parseTableName(table2.name).table;
       table2.constraints = constraintsByTable.get(rawName) ?? [];
@@ -1486,7 +1484,7 @@ var BigQueryConstraintGrounding = class extends ConstraintGrounding {
       }
     }
   }
-  async #batchKeyConstraints(dataset, inList, constraintsByTable) {
+  async #batchKeyConstraints(dataset, inList, constraintsByTable, cache) {
     const rows = await this.#adapter.runQuery(
       `
       SELECT
@@ -1546,7 +1544,7 @@ var BigQueryConstraintGrounding = class extends ConstraintGrounding {
             dataset,
             constraintName,
             childColumns,
-            this.#cache
+            cache
           );
           if (resolution) {
             constraints2.push({
@@ -1844,13 +1842,11 @@ var childKeyColumnRow = keyColumnUsageRow.extend({
 });
 var BigQueryTableGrounding = class extends TableGrounding {
   #adapter;
-  #cache;
   constructor(adapter, config = {}) {
     super(config);
     this.#adapter = adapter;
   }
   async execute(ctx) {
-    this.#cache = ctx.cache;
     await super.execute(ctx);
     ctx.tables = ctx.tables.filter((t) => t.columns.length > 0);
     const tableNames = new Set(ctx.tables.map((t) => t.name));
@@ -1916,7 +1912,7 @@ var BigQueryTableGrounding = class extends TableGrounding {
       columns
     };
   }
-  async findOutgoingRelations(tableName) {
+  async findOutgoingRelations(tableName, ctx) {
     const { schema: dataset, table: table2 } = this.#adapter.parseTableName(tableName);
     const rows = await this.#adapter.runQuery(
       `
@@ -1953,7 +1949,7 @@ var BigQueryTableGrounding = class extends TableGrounding {
         dataset,
         constraintName,
         columns,
-        this.#cache
+        ctx.cache
       );
       if (resolution) {
         rels.push({
@@ -1966,7 +1962,7 @@ var BigQueryTableGrounding = class extends TableGrounding {
     }
     return rels;
   }
-  async findIncomingRelations(tableName) {
+  async findIncomingRelations(tableName, ctx) {
     const { schema: referencedDataset, table: referencedTable } = this.#adapter.parseTableName(tableName);
     const rels = [];
     for (const constraintDataset of this.#adapter.datasets) {
@@ -1985,14 +1981,15 @@ var BigQueryTableGrounding = class extends TableGrounding {
           constraintDataset,
           row.constraint_name,
           referencedDataset,
-          referencedTable
+          referencedTable,
+          ctx.cache
         );
         if (rel) rels.push(rel);
       }
     }
     return rels;
   }
-  async #resolveIncomingRelationship(constraintDataset, constraintName, expectedReferencedDataset, expectedReferencedTable) {
+  async #resolveIncomingRelationship(constraintDataset, constraintName, expectedReferencedDataset, expectedReferencedTable, cache) {
     const keyRows = await this.#adapter.runQuery(
       `
       SELECT
@@ -2024,7 +2021,7 @@ var BigQueryTableGrounding = class extends TableGrounding {
       constraintDataset,
       constraintName,
       childColumns,
-      this.#cache
+      cache
     );
     if (!resolution || resolution.referencedDataset !== expectedReferencedDataset || resolution.referencedTable !== expectedReferencedTable) {
       return void 0;