@deepagents/text2sql 7.1.0 → 7.1.2

package/dist/lib/adapters/postgres/column-values.postgres.grounding.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"column-values.postgres.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/postgres/column-values.postgres.grounding.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EACL,KAAK,MAAM,EACX,qBAAqB,EACrB,KAAK,2BAA2B,EACjC,MAAM,0CAA0C,CAAC;AAkBlD,qBAAa,6BAA8B,SAAQ,qBAAqB;;gBAK1D,OAAO,EAAE,OAAO,EAAE,MAAM,GAAE,2BAAgC;cA4C7C,iBAAiB,CACxC,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;cAmCP,qBAAqB,CAC5C,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;CAwCjC"}
+{"version":3,"file":"column-values.postgres.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/postgres/column-values.postgres.grounding.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EACL,KAAK,MAAM,EACX,qBAAqB,EACrB,KAAK,2BAA2B,EACjC,MAAM,0CAA0C,CAAC;AAkBlD,qBAAa,6BAA8B,SAAQ,qBAAqB;;gBAK1D,OAAO,EAAE,OAAO,EAAE,MAAM,GAAE,2BAAgC;cA0C7C,iBAAiB,CACxC,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;cAkCP,qBAAqB,CAC5C,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;CAwCjC"}

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

@@ -950,7 +950,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);
@@ -982,7 +982,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);
@@ -1213,6 +1213,7 @@ var PostgresColumnStatsGrounding = class extends ColumnStatsGrounding {
 };
 
 // packages/text2sql/src/lib/adapters/postgres/column-values.postgres.grounding.ts
+import pMemoize from "p-memoize";
 import { z as z4 } from "zod";
 var enumValueRow = z4.object({
   type_name: z4.string(),
@@ -1226,20 +1227,19 @@ var columnTypeRow = z4.object({
 var textValueRow2 = z4.object({ value: z4.string().nullable() });
 var PostgresColumnValuesGrounding = class extends ColumnValuesGrounding {
   #adapter;
-  #enumCache = /* @__PURE__ */ new Map();
-  #enumCacheLoaded = false;
+  /** Every ENUM type's values, loaded once on first use; a failed load runs again. */
+  #enumValues;
   constructor(adapter, config = {}) {
     super(config);
     this.#adapter = adapter;
+    this.#enumValues = pMemoize(() => this.#loadEnumValues());
   }
   /**
-   * Load all ENUM types and their values into cache.
-   * This is more efficient than querying per-column.
+   * Load all ENUM types and their values, keyed by schema-qualified and by
+   * bare type name. One query is more efficient than one per column.
    */
-  async #loadEnumCache() {
-    if (this.#enumCacheLoaded) {
-      return;
-    }
+  async #loadEnumValues() {
+    const enumValues = /* @__PURE__ */ new Map();
     const rows = await this.#adapter.runQuery(
       `
       SELECT
@@ -1255,21 +1255,21 @@ var PostgresColumnValuesGrounding = class extends ColumnValuesGrounding {
     );
     for (const row of rows) {
       const key = `${row.type_schema}.${row.type_name}`;
-      const existing = this.#enumCache.get(key) ?? [];
+      const existing = enumValues.get(key) ?? [];
       existing.push(row.enum_value);
-      this.#enumCache.set(key, existing);
+      enumValues.set(key, existing);
       const simpleKey = row.type_name;
-      const simpleExisting = this.#enumCache.get(simpleKey) ?? [];
+      const simpleExisting = enumValues.get(simpleKey) ?? [];
       simpleExisting.push(row.enum_value);
-      this.#enumCache.set(simpleKey, simpleExisting);
+      enumValues.set(simpleKey, simpleExisting);
     }
-    this.#enumCacheLoaded = true;
+    return enumValues;
   }
   async collectEnumValues(tableName, column2) {
     if (column2.type.toLowerCase() !== "user-defined") {
       return void 0;
     }
-    await this.#loadEnumCache();
+    const enumValues = await this.#enumValues();
     const { schema, table: table2 } = this.#adapter.parseTableName(tableName);
     const rows = await this.#adapter.runQuery(
       `
@@ -1286,7 +1286,7 @@ var PostgresColumnValuesGrounding = class extends ColumnValuesGrounding {
     }
     const { udt_name, udt_schema } = rows[0];
     const fullKey = `${udt_schema}.${udt_name}`;
-    const values = this.#enumCache.get(fullKey) ?? this.#enumCache.get(udt_name);
+    const values = enumValues.get(fullKey) ?? enumValues.get(udt_name);
     return values?.length ? values : void 0;
   }
   async collectLowCardinality(tableName, column2) {