@deepagents/text2sql 7.3.0 → 7.4.0

package/dist/lib/adapters/clickhouse/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";
 
@@ -401,12 +402,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()
@@ -520,7 +516,7 @@ SQL: ${sql.trim()}`,
 };
 
 // packages/text2sql/src/lib/adapters/clickhouse/clickhouse.sql-policy.ts
-import pMemoize from "p-memoize";
+import pMemoize2 from "p-memoize";
 
 // packages/text2sql/src/lib/sql-scope-error.ts
 function buildScopeParseErrorPayload(sql, parserDialect, error) {
@@ -546,7 +542,7 @@ var ClickHouseSqlPolicyAnalyzer = class {
   #currentDatabase;
   constructor(query) {
     this.#query = query;
-    this.#currentDatabase = pMemoize(() => this.#verifyReadonly());
+    this.#currentDatabase = pMemoize2(() => this.#verifyReadonly());
   }
   async analyze(sql, context) {
     try {