@deepagents/text2sql 0.23.0 → 0.25.0

package/dist/lib/adapters/mysql/view.mysql.grounding.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"view.mysql.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/mysql/view.mysql.grounding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EACL,KAAK,IAAI,EACT,aAAa,EACb,KAAK,mBAAmB,EACzB,MAAM,iCAAiC,CAAC;AAYzC,MAAM,WAAW,wBAAyB,SAAQ,mBAAmB;IACnE,oEAAoE;IACpE,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAED;;;;GAIG;AACH,qBAAa,kBAAmB,SAAQ,aAAa;;gBAIvC,OAAO,EAAE,OAAO,EAAE,MAAM,GAAE,wBAA6B;cAM1C,eAAe,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;cAWpC,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAsDlE"}
+{"version":3,"file":"view.mysql.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/mysql/view.mysql.grounding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EACL,KAAK,IAAI,EACT,aAAa,EACb,KAAK,mBAAmB,EACzB,MAAM,iCAAiC,CAAC;AAYzC,MAAM,WAAW,wBAAyB,SAAQ,mBAAmB;IACnE,oEAAoE;IACpE,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAED;;;;GAIG;AACH,qBAAa,kBAAmB,SAAQ,aAAa;;gBAIvC,OAAO,EAAE,OAAO,EAAE,MAAM,GAAE,wBAA6B;cAM1C,eAAe,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;cAWpC,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAwDlE"}

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

@@ -431,9 +431,11 @@ var InfoGrounding = class extends AbstractGrounding {
 // packages/text2sql/src/lib/adapters/groundings/column-values.grounding.ts
 var ColumnValuesGrounding = class extends AbstractGrounding {
   lowCardinalityLimit;
+  maxValueLength;
   constructor(config = {}) {
     super("columnValues");
     this.lowCardinalityLimit = config.lowCardinalityLimit ?? 20;
+    this.maxValueLength = config.maxValueLength ?? 100;
   }
   /**
    * Get values for native ENUM type columns.
@@ -538,24 +540,27 @@ var ColumnValuesGrounding = class extends AbstractGrounding {
       }
     }
   }
+  exceedsMaxValueLength(values) {
+    return values.some((v) => v.length > this.maxValueLength);
+  }
   /**
    * Resolve column values from all sources in priority order.
    */
   async resolveColumnValues(tableName, column2, constraints2) {
     const enumValues = await this.collectEnumValues(tableName, column2);
-    if (enumValues?.length) {
+    if (enumValues?.length && !this.exceedsMaxValueLength(enumValues)) {
       return { kind: "Enum", values: enumValues };
     }
     if (constraints2) {
       for (const constraint2 of constraints2) {
         const checkValues = this.parseCheckConstraint(constraint2, column2.name);
-        if (checkValues?.length) {
+        if (checkValues?.length && !this.exceedsMaxValueLength(checkValues)) {
           return { kind: "Enum", values: checkValues };
         }
       }
     }
     const lowCardValues = await this.collectLowCardinality(tableName, column2);
-    if (lowCardValues?.length) {
+    if (lowCardValues?.length && !this.exceedsMaxValueLength(lowCardValues)) {
       return { kind: "LowCardinality", values: lowCardValues };
     }
     return void 0;
@@ -1863,9 +1868,11 @@ var PostgresTableGrounding = 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.
@@ -1919,12 +1926,16 @@ var PostgresViewGrounding = class extends ViewGrounding {
   }
   async getView(viewName) {
     const { schema, table: view2 } = this.#adapter.parseTableName(viewName);
-    const defRows = await this.#adapter.runQuery(`
-      SELECT definition
-      FROM pg_views
-      WHERE schemaname = '${this.#adapter.escapeString(schema)}'
-        AND viewname = '${this.#adapter.escapeString(view2)}'
-    `);
+    let definition;
+    if (this.includeDefinition) {
+      const defRows = await this.#adapter.runQuery(`
+        SELECT definition
+        FROM pg_views
+        WHERE schemaname = '${this.#adapter.escapeString(schema)}'
+          AND viewname = '${this.#adapter.escapeString(view2)}'
+      `);
+      definition = defRows[0]?.definition ?? void 0;
+    }
     const columns = await this.#adapter.runQuery(`
       SELECT column_name, data_type
       FROM information_schema.columns
@@ -1936,7 +1947,7 @@ var PostgresViewGrounding = class extends ViewGrounding {
       name: viewName,
       schema,
       rawName: view2,
-      definition: defRows[0]?.definition ?? void 0,
+      definition,
       columns: columns.map((col) => ({
         name: col.column_name ?? "unknown",
         type: col.data_type ?? "unknown"