@deepagents/text2sql 0.24.0 → 0.25.0

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

@@ -1347,9 +1347,11 @@ var BigQueryTableGrounding = class extends TableGrounding {
 // packages/text2sql/src/lib/adapters/groundings/view.grounding.ts
 var ViewGrounding = class extends AbstractGrounding {
   #filter;
+  includeDefinition;
   constructor(config = {}) {
     super("view");
     this.#filter = config.filter;
+    this.includeDefinition = config.includeDefinition ?? true;
   }
   /**
    * Execute the grounding process.
@@ -1411,13 +1413,17 @@ var BigQueryViewGrounding = class extends ViewGrounding {
   }
   async getView(viewName) {
     const { schema: dataset, table: table2 } = this.#adapter.parseTableName(viewName);
-    const defRows = await this.#adapter.runQuery(`
-      SELECT ddl
-      FROM ${this.#adapter.infoSchemaView(dataset, "TABLES")}
-      WHERE table_name = '${this.#adapter.escapeString(table2)}'
-        AND table_type IN ('VIEW', 'MATERIALIZED VIEW')
-      LIMIT 1
-    `);
+    let definition;
+    if (this.includeDefinition) {
+      const defRows = await this.#adapter.runQuery(`
+        SELECT ddl
+        FROM ${this.#adapter.infoSchemaView(dataset, "TABLES")}
+        WHERE table_name = '${this.#adapter.escapeString(table2)}'
+          AND table_type IN ('VIEW', 'MATERIALIZED VIEW')
+        LIMIT 1
+      `);
+      definition = defRows[0]?.ddl ?? void 0;
+    }
     const columns = await this.#adapter.runQuery(`
       SELECT column_name, data_type
       FROM ${this.#adapter.infoSchemaView(dataset, "COLUMNS")}
@@ -1428,7 +1434,7 @@ var BigQueryViewGrounding = class extends ViewGrounding {
       name: `${dataset}.${table2}`,
       schema: dataset,
       rawName: table2,
-      definition: defRows[0]?.ddl ?? void 0,
+      definition,
       columns: columns.map((c) => ({
         name: c.column_name ?? "unknown",
         type: c.data_type ?? "unknown"