@deepagents/text2sql 0.23.0 → 0.24.0

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;