@deepagents/text2sql 7.3.0 → 7.4.0

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

@@ -1,4 +1,5 @@
 // packages/text2sql/src/lib/adapters/adapter.ts
+import pMemoize from "p-memoize";
 import { format as formatSql } from "sql-formatter";
 import { z } from "zod";
 
@@ -411,12 +412,7 @@ var Adapter = class {
   #decodeShellEscapes(sql) {
     return sql.replace(/\\n/g, "\n").replace(/\\t/g, "	").replace(/\\([().*$])/g, "$1").replace(/\\(?=$)/gm, "");
   }
-  #cachedAllowedEntities = null;
-  async #resolveScope() {
-    if (this.#cachedAllowedEntities) return this.#cachedAllowedEntities;
-    this.#cachedAllowedEntities = await this.resolveAllowedEntities();
-    return this.#cachedAllowedEntities;
-  }
+  #resolveScope = pMemoize(() => this.resolveAllowedEntities());
   async #analyzePolicy(sql) {
     return this.#policyAnalyzer.analyze(sql, {
       resolveAllowedEntities: () => this.#resolveScope()
@@ -1213,7 +1209,7 @@ var PostgresColumnStatsGrounding = class extends ColumnStatsGrounding {
 };
 
 // packages/text2sql/src/lib/adapters/postgres/column-values.postgres.grounding.ts
-import pMemoize from "p-memoize";
+import pMemoize2 from "p-memoize";
 import { z as z4 } from "zod";
 var enumValueRow = z4.object({
   type_name: z4.string(),
@@ -1232,7 +1228,7 @@ var PostgresColumnValuesGrounding = class extends ColumnValuesGrounding {
   constructor(adapter, config = {}) {
     super(config);
     this.#adapter = adapter;
-    this.#enumValues = pMemoize(() => this.#loadEnumValues());
+    this.#enumValues = pMemoize2(() => this.#loadEnumValues());
   }
   /**
    * Load all ENUM types and their values, keyed by schema-qualified and by