@deepagents/text2sql 0.23.0 → 0.25.0

package/dist/lib/adapters/sqlite/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;
@@ -1300,9 +1305,11 @@ var SqliteTableGrounding = 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.
@@ -1347,15 +1354,19 @@ var SqliteViewGrounding = class extends ViewGrounding {
     return rows.map((row) => row.name).filter((name) => typeof name === "string");
   }
   async getView(viewName) {
-    const defRows = await this.#adapter.runQuery(
-      `SELECT sql FROM sqlite_master WHERE type='view' AND name=${this.#quoteIdentifier(viewName)}`
-    );
+    let definition;
+    if (this.includeDefinition) {
+      const defRows = await this.#adapter.runQuery(
+        `SELECT sql FROM sqlite_master WHERE type='view' AND name=${this.#quoteIdentifier(viewName)}`
+      );
+      definition = defRows[0]?.sql ?? void 0;
+    }
     const columns = await this.#adapter.runQuery(
       `PRAGMA table_info(${this.#quoteIdentifier(viewName)})`
     );
     return {
       name: viewName,
-      definition: defRows[0]?.sql ?? void 0,
+      definition,
       columns: columns.map((col) => ({
         name: col.name ?? "unknown",
         type: col.type ?? "unknown"