@deepagents/text2sql 7.1.1 → 7.1.3

package/dist/lib/adapters/bigquery/table.bigquery.grounding.d.ts

@@ -11,7 +11,7 @@ export declare class BigQueryTableGrounding extends TableGrounding {
     protected applyFilter(): Promise<string[]>;
     protected getAllTableNames(): Promise<string[]>;
     protected getTable(tableName: string): Promise<Table>;
-    protected findOutgoingRelations(tableName: string): Promise<Relationship[]>;
-    protected findIncomingRelations(tableName: string): Promise<Relationship[]>;
+    protected findOutgoingRelations(tableName: string, ctx: GroundingContext): Promise<Relationship[]>;
+    protected findIncomingRelations(tableName: string, ctx: GroundingContext): Promise<Relationship[]>;
 }
 //# sourceMappingURL=table.bigquery.grounding.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"table.bigquery.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/bigquery/table.bigquery.grounding.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EACL,cAAc,EACd,KAAK,oBAAoB,EAC1B,MAAM,kCAAkC,CAAC;AAQ1C,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAc9C,MAAM,WAAW,4BAA6B,SAAQ,oBAAoB;CAAG;AAE7E,qBAAa,sBAAuB,SAAQ,cAAc;;gBAI5C,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAE,4BAAiC;IAKzD,OAAO,CAAC,GAAG,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;cAWnC,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;cAKhC,gBAAgB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;cAuBrC,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;cA0C3C,qBAAqB,CAC5C,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,YAAY,EAAE,CAAC;cAwDD,qBAAqB,CAC5C,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,YAAY,EAAE,CAAC;CAiG3B"}
+{"version":3,"file":"table.bigquery.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/bigquery/table.bigquery.grounding.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EACL,cAAc,EACd,KAAK,oBAAoB,EAC1B,MAAM,kCAAkC,CAAC;AAQ1C,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAc9C,MAAM,WAAW,4BAA6B,SAAQ,oBAAoB;CAAG;AAE7E,qBAAa,sBAAuB,SAAQ,cAAc;;gBAG5C,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAE,4BAAiC;IAKzD,OAAO,CAAC,GAAG,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;cAUnC,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;cAKhC,gBAAgB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;cAuBrC,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;cA0C3C,qBAAqB,CAC5C,SAAS,EAAE,MAAM,EACjB,GAAG,EAAE,gBAAgB,GACpB,OAAO,CAAC,YAAY,EAAE,CAAC;cAwDD,qBAAqB,CAC5C,SAAS,EAAE,MAAM,EACjB,GAAG,EAAE,gBAAgB,GACpB,OAAO,CAAC,YAAY,EAAE,CAAC;CAmG3B"}

package/dist/lib/adapters/clickhouse/clickhouse.sql-policy.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"clickhouse.sql-policy.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/clickhouse/clickhouse.sql-policy.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,kBAAkB,EACnB,MAAM,kBAAkB,CAAC;AAS1B,qBAAa,2BAA4B,YAAW,iBAAiB;;gBAKvD,KAAK,EAAE,eAAe;IAI5B,OAAO,CACX,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,gBAAgB,GACxB,OAAO,CAAC,kBAAkB,GAAG,IAAI,CAAC;CA8HtC"}
+{"version":3,"file":"clickhouse.sql-policy.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/clickhouse/clickhouse.sql-policy.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,kBAAkB,EACnB,MAAM,kBAAkB,CAAC;AAS1B,qBAAa,2BAA4B,YAAW,iBAAiB;;gBAQvD,KAAK,EAAE,eAAe;IAK5B,OAAO,CACX,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,gBAAgB,GACxB,OAAO,CAAC,kBAAkB,GAAG,IAAI,CAAC;CA2HtC"}

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

@@ -519,6 +519,9 @@ SQL: ${sql.trim()}`,
   }
 };
 
+// packages/text2sql/src/lib/adapters/clickhouse/clickhouse.sql-policy.ts
+import pMemoize from "p-memoize";
+
 // packages/text2sql/src/lib/sql-scope-error.ts
 function buildScopeParseErrorPayload(sql, parserDialect, error) {
   const parserError = error instanceof Error ? error.message : String(error ?? "Unknown error");
@@ -536,14 +539,18 @@ function buildScopeParseErrorPayload(sql, parserDialect, error) {
 var READ_ONLY_MESSAGE = "only SELECT queries allowed";
 var ClickHouseSqlPolicyAnalyzer = class {
   #query;
-  #readonlyCheck;
+  /**
+   * The connection's current database, read by the probe that also checks
+   * readonly = 1. Kept once the probe passes; a failed probe runs again.
+   */
   #currentDatabase;
   constructor(query) {
     this.#query = query;
+    this.#currentDatabase = pMemoize(() => this.#verifyReadonly());
   }
   async analyze(sql, context) {
     try {
-      await (this.#readonlyCheck ??= this.#verifyReadonly());
+      const currentDatabase = await this.#currentDatabase();
       const ast = await this.#explain(`EXPLAIN AST ${sql}`);
       const astAnalysis = analyzeAst(ast);
       if (!astAnalysis.isSelect) {
@@ -564,13 +571,10 @@ var ClickHouseSqlPolicyAnalyzer = class {
         analyzedTree,
         "analyzed"
       );
-      if (!this.#currentDatabase) {
-        throw new Error("ClickHouse current database was not established.");
-      }
       const relations = [
         ...new Set(
           [...syntacticRelations, ...analyzedRelations].map(
-            (relation) => qualifyRelation(relation, this.#currentDatabase)
+            (relation) => qualifyRelation(relation, currentDatabase)
           )
         )
       ];
@@ -607,7 +611,7 @@ var ClickHouseSqlPolicyAnalyzer = class {
         `ClickHouse connection must have effective readonly = 1; received ${String(value)}.`
       );
     }
-    this.#currentDatabase = row.database;
+    return row.database;
   }
   async #explain(sql) {
     const rows = readRows(await this.#query(sql));
@@ -1331,7 +1335,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);
@@ -1363,7 +1367,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);