@deepagents/text2sql 0.3.0 → 0.6.0

package/dist/lib/adapters/postgres/postgres.d.ts

@@ -22,5 +22,6 @@ export declare class Postgres extends Adapter {
     runQuery<Row>(sql: string): Promise<Row[]>;
     quoteIdentifier(name: string): string;
     escape(value: string): string;
+    buildSampleRowsQuery(tableName: string, columns: string[] | undefined, limit: number): string;
 }
 //# sourceMappingURL=postgres.d.ts.map

package/dist/lib/adapters/postgres/postgres.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/postgres/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,OAAO,EACP,KAAK,eAAe,EACpB,KAAK,WAAW,EAKhB,KAAK,gBAAgB,EACtB,MAAM,eAAe,CAAC;AAEvB,MAAM,MAAM,sBAAsB,GAAG;IACnC,OAAO,EAAE,eAAe,CAAC;IACzB,QAAQ,CAAC,EAAE,gBAAgB,CAAC;IAC5B,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;CACpB,CAAC;AAwEF,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO;;;;;EA0B9D;AAED,qBAAa,QAAS,SAAQ,OAAO;;IAEnC,SAAkB,SAAS,EAAE,WAAW,EAAE,CAAC;IAC3C,SAAkB,aAAa,YAAY;IAC3C,SAAkB,aAAa,WAAwC;gBAE3D,OAAO,EAAE,sBAAsB;IAY5B,OAAO,CAAC,GAAG,EAAE,MAAM;IAInB,QAAQ,CAAC,GAAG,EAAE,MAAM;IAcpB,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAIhD,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAIrC,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;CA4cvC"}
+{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/postgres/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,OAAO,EACP,KAAK,eAAe,EACpB,KAAK,WAAW,EAKhB,KAAK,gBAAgB,EACtB,MAAM,eAAe,CAAC;AAEvB,MAAM,MAAM,sBAAsB,GAAG;IACnC,OAAO,EAAE,eAAe,CAAC;IACzB,QAAQ,CAAC,EAAE,gBAAgB,CAAC;IAC5B,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;CACpB,CAAC;AAwEF,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO;;;;;EA0B9D;AAED,qBAAa,QAAS,SAAQ,OAAO;;IAEnC,SAAkB,SAAS,EAAE,WAAW,EAAE,CAAC;IAC3C,SAAkB,aAAa,YAAY;IAC3C,SAAkB,aAAa,WAAwC;gBAE3D,OAAO,EAAE,sBAAsB;IAY5B,OAAO,CAAC,GAAG,EAAE,MAAM;IAInB,QAAQ,CAAC,GAAG,EAAE,MAAM;IAcpB,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAIhD,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAIrC,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;IAI7B,oBAAoB,CAC3B,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EAAE,GAAG,SAAS,EAC7B,KAAK,EAAE,MAAM,GACZ,MAAM;CAmdV"}

package/dist/lib/adapters/sqlite/column-values.sqlite.grounding.d.ts

@@ -0,0 +1,17 @@
+import type { Adapter } from '../adapter.ts';
+import { type Column, ColumnValuesGrounding, type ColumnValuesGroundingConfig } from '../groundings/column-values.grounding.ts';
+/**
+ * SQLite implementation of ColumnValuesGrounding.
+ *
+ * Supports:
+ * - CHECK constraints with IN clauses (inherited from base)
+ * - Low cardinality data scan
+ *
+ * Note: SQLite does not have native ENUM types.
+ */
+export declare class SqliteColumnValuesGrounding extends ColumnValuesGrounding {
+    #private;
+    constructor(adapter: Adapter, config?: ColumnValuesGroundingConfig);
+    protected collectLowCardinality(tableName: string, column: Column): Promise<string[] | undefined>;
+}
+//# sourceMappingURL=column-values.sqlite.grounding.d.ts.map

package/dist/lib/adapters/sqlite/column-values.sqlite.grounding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"column-values.sqlite.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/sqlite/column-values.sqlite.grounding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EACL,KAAK,MAAM,EACX,qBAAqB,EACrB,KAAK,2BAA2B,EACjC,MAAM,0CAA0C,CAAC;AAElD;;;;;;;;GAQG;AACH,qBAAa,2BAA4B,SAAQ,qBAAqB;;gBAGxD,OAAO,EAAE,OAAO,EAAE,MAAM,GAAE,2BAAgC;cAK7C,qBAAqB,CAC5C,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;CAmDjC"}

package/dist/lib/adapters/sqlite/constraint.sqlite.grounding.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"constraint.sqlite.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/sqlite/constraint.sqlite.grounding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAC9D,OAAO,EACL,mBAAmB,EACnB,KAAK,yBAAyB,EAC/B,MAAM,uCAAuC,CAAC;AAuB/C;;;;;;GAMG;AACH,qBAAa,yBAA0B,SAAQ,mBAAmB;;gBAGpD,OAAO,EAAE,OAAO,EAAE,MAAM,GAAE,yBAA8B;cAK3C,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;CAgIvF"}
+{"version":3,"file":"constraint.sqlite.grounding.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/sqlite/constraint.sqlite.grounding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAC9D,OAAO,EACL,mBAAmB,EACnB,KAAK,yBAAyB,EAC/B,MAAM,uCAAuC,CAAC;AAuB/C;;;;;;GAMG;AACH,qBAAa,yBAA0B,SAAQ,mBAAmB;;gBAGpD,OAAO,EAAE,OAAO,EAAE,MAAM,GAAE,yBAA8B;cAK3C,cAAc,CACrC,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,eAAe,EAAE,CAAC;CAoK9B"}

package/dist/lib/adapters/sqlite/index.d.ts

@@ -3,7 +3,7 @@ import { type ColumnStatsGroundingConfig } from '../groundings/column-stats.grou
 import { type ConstraintGroundingConfig } from '../groundings/constraint.grounding.ts';
 import { type IndexesGroundingConfig } from '../groundings/indexes.grounding.ts';
 import { type InfoGroundingConfig } from '../groundings/info.grounding.ts';
-import { type LowCardinalityGroundingConfig } from '../groundings/low-cardinality.grounding.ts';
+import { type ColumnValuesGroundingConfig } from '../groundings/column-values.grounding.ts';
 import { ReportGrounding, type ReportGroundingConfig } from '../groundings/report.grounding.ts';
 import { type RowCountGroundingConfig } from '../groundings/row-count.grounding.ts';
 import { type TableGroundingConfig } from '../groundings/table.grounding.ts';
@@ -12,7 +12,7 @@ import { SqliteColumnStatsGrounding } from './column-stats.sqlite.grounding.ts';
 import { SqliteConstraintGrounding } from './constraint.sqlite.grounding.ts';
 import { SqliteIndexesGrounding } from './indexes.sqlite.grounding.ts';
 import { SqliteInfoGrounding } from './info.sqlite.grounding.ts';
-import { SqliteLowCardinalityGrounding } from './low-cardinality.sqlite.grounding.ts';
+import { SqliteColumnValuesGrounding } from './column-values.sqlite.grounding.ts';
 import { SqliteRowCountGrounding } from './row-count.sqlite.grounding.ts';
 import { Sqlite } from './sqlite.ts';
 import { SqliteTableGrounding } from './table.sqlite.grounding.ts';
@@ -22,7 +22,7 @@ export declare function tables(config?: TableGroundingConfig): (adapter: Adapter
 export declare function info(config?: InfoGroundingConfig): (adapter: Adapter) => SqliteInfoGrounding;
 export declare function views(config?: ViewGroundingConfig): (adapter: Adapter) => SqliteViewGrounding;
 export declare function columnStats(config?: ColumnStatsGroundingConfig): (adapter: Adapter) => SqliteColumnStatsGrounding;
-export declare function lowCardinality(config?: LowCardinalityGroundingConfig): (adapter: Adapter) => SqliteLowCardinalityGrounding;
+export declare function columnValues(config?: ColumnValuesGroundingConfig): (adapter: Adapter) => SqliteColumnValuesGrounding;
 export declare function indexes(config?: IndexesGroundingConfig): (adapter: Adapter) => SqliteIndexesGrounding;
 export declare function rowCount(config?: RowCountGroundingConfig): (adapter: Adapter) => SqliteRowCountGrounding;
 export declare function constraints(config?: ConstraintGroundingConfig): (adapter: Adapter) => SqliteConstraintGrounding;
@@ -32,7 +32,7 @@ declare const _default: {
     info: typeof info;
     views: typeof views;
     columnStats: typeof columnStats;
-    lowCardinality: typeof lowCardinality;
+    columnValues: typeof columnValues;
     indexes: typeof indexes;
     rowCount: typeof rowCount;
     constraints: typeof constraints;

package/dist/lib/adapters/sqlite/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/sqlite/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,KAAK,0BAA0B,EAAE,MAAM,yCAAyC,CAAC;AAC1F,OAAO,EAAE,KAAK,yBAAyB,EAAE,MAAM,uCAAuC,CAAC;AACvF,OAAO,EAAE,KAAK,sBAAsB,EAAE,MAAM,oCAAoC,CAAC;AACjF,OAAO,EAAE,KAAK,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AAC3E,OAAO,EAAE,KAAK,6BAA6B,EAAE,MAAM,4CAA4C,CAAC;AAChG,OAAO,EACL,eAAe,EACf,KAAK,qBAAqB,EAC3B,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,KAAK,uBAAuB,EAAE,MAAM,sCAAsC,CAAC;AACpF,OAAO,EAAE,KAAK,oBAAoB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AAC3E,OAAO,EAAE,0BAA0B,EAAE,MAAM,oCAAoC,CAAC;AAChF,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,EAAE,6BAA6B,EAAE,MAAM,uCAAuC,CAAC;AACtF,OAAO,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC1E,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAEjE,cAAc,aAAa,CAAC;AAE5B,wBAAgB,MAAM,CAAC,MAAM,GAAE,oBAAyB,IAC9C,SAAS,OAAO,0BACzB;AAED,wBAAgB,IAAI,CAAC,MAAM,GAAE,mBAAwB,IAC3C,SAAS,OAAO,yBACzB;AAED,wBAAgB,KAAK,CAAC,MAAM,GAAE,mBAAwB,IAC5C,SAAS,OAAO,yBAGzB;AAED,wBAAgB,WAAW,CAAC,MAAM,GAAE,0BAA+B,IACzD,SAAS,OAAO,gCAGzB;AAED,wBAAgB,cAAc,CAAC,MAAM,GAAE,6BAAkC,IAC/D,SAAS,OAAO,mCAGzB;AAED,wBAAgB,OAAO,CAAC,MAAM,GAAE,sBAA2B,IACjD,SAAS,OAAO,4BAGzB;AAED,wBAAgB,QAAQ,CAAC,MAAM,GAAE,uBAA4B,IACnD,SAAS,OAAO,6BAGzB;AAED,wBAAgB,WAAW,CAAC,MAAM,GAAE,yBAA8B,IACxD,SAAS,OAAO,+BAGzB;AAED,wBAAgB,MAAM,CAAC,MAAM,GAAE,qBAA0B,IAC/C,SAAS,OAAO,qBACzB;;;;;;;;;;;;;AAED,wBAWE"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/sqlite/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,KAAK,0BAA0B,EAAE,MAAM,yCAAyC,CAAC;AAC1F,OAAO,EAAE,KAAK,yBAAyB,EAAE,MAAM,uCAAuC,CAAC;AACvF,OAAO,EAAE,KAAK,sBAAsB,EAAE,MAAM,oCAAoC,CAAC;AACjF,OAAO,EAAE,KAAK,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AAC3E,OAAO,EAAE,KAAK,2BAA2B,EAAE,MAAM,0CAA0C,CAAC;AAC5F,OAAO,EACL,eAAe,EACf,KAAK,qBAAqB,EAC3B,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,KAAK,uBAAuB,EAAE,MAAM,sCAAsC,CAAC;AACpF,OAAO,EAAE,KAAK,oBAAoB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AAC3E,OAAO,EAAE,0BAA0B,EAAE,MAAM,oCAAoC,CAAC;AAChF,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,EAAE,2BAA2B,EAAE,MAAM,qCAAqC,CAAC;AAClF,OAAO,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC1E,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAEjE,cAAc,aAAa,CAAC;AAE5B,wBAAgB,MAAM,CAAC,MAAM,GAAE,oBAAyB,IAC9C,SAAS,OAAO,0BACzB;AAED,wBAAgB,IAAI,CAAC,MAAM,GAAE,mBAAwB,IAC3C,SAAS,OAAO,yBACzB;AAED,wBAAgB,KAAK,CAAC,MAAM,GAAE,mBAAwB,IAC5C,SAAS,OAAO,yBAGzB;AAED,wBAAgB,WAAW,CAAC,MAAM,GAAE,0BAA+B,IACzD,SAAS,OAAO,gCAGzB;AAED,wBAAgB,YAAY,CAAC,MAAM,GAAE,2BAAgC,IAC3D,SAAS,OAAO,iCAGzB;AAED,wBAAgB,OAAO,CAAC,MAAM,GAAE,sBAA2B,IACjD,SAAS,OAAO,4BAGzB;AAED,wBAAgB,QAAQ,CAAC,MAAM,GAAE,uBAA4B,IACnD,SAAS,OAAO,6BAGzB;AAED,wBAAgB,WAAW,CAAC,MAAM,GAAE,yBAA8B,IACxD,SAAS,OAAO,+BAGzB;AAED,wBAAgB,MAAM,CAAC,MAAM,GAAE,qBAA0B,IAC/C,SAAS,OAAO,qBACzB;;;;;;;;;;;;;AAED,wBAWE"}

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

@@ -10,9 +10,8 @@ function createGroundingContext() {
 
 // packages/text2sql/src/lib/adapters/adapter.ts
 var Adapter = class {
-  async introspect() {
+  async introspect(ctx = createGroundingContext()) {
     const lines = [];
-    const ctx = createGroundingContext();
     for (const fn of this.grounding) {
       const grounding = fn(this);
       lines.push({
@@ -82,7 +81,7 @@ ${description}
   }
 };
 
-// packages/text2sql/src/lib/adapters/grounding.ticket.ts
+// packages/text2sql/src/lib/adapters/groundings/abstract.grounding.ts
 var AbstractGrounding = class {
   tag;
   constructor(tag) {
@@ -196,31 +195,108 @@ var InfoGrounding = class extends AbstractGrounding {
   }
 };
 
-// packages/text2sql/src/lib/adapters/groundings/low-cardinality.grounding.ts
-var LowCardinalityGrounding = class extends AbstractGrounding {
+// packages/text2sql/src/lib/adapters/groundings/column-values.grounding.ts
+var ColumnValuesGrounding = class extends AbstractGrounding {
+  lowCardinalityLimit;
   constructor(config = {}) {
-    super("low_cardinality");
+    super("column_values");
+    this.lowCardinalityLimit = config.lowCardinalityLimit ?? 20;
+  }
+  /**
+   * Get values for native ENUM type columns.
+   * Return undefined if column is not an ENUM type.
+   * Default implementation returns undefined (no native ENUM support).
+   */
+  async collectEnumValues(_tableName, _column) {
+    return void 0;
+  }
+  /**
+   * Parse CHECK constraint for enum-like IN clause.
+   * Extracts values from patterns like:
+   * - CHECK (status IN ('active', 'inactive'))
+   * - CHECK ((status)::text = ANY (ARRAY['a'::text, 'b'::text]))
+   * - CHECK (status = 'active' OR status = 'inactive')
+   */
+  parseCheckConstraint(constraint, columnName) {
+    if (constraint.type !== "CHECK" || !constraint.definition) {
+      return void 0;
+    }
+    if (constraint.columns && !constraint.columns.includes(columnName)) {
+      return void 0;
+    }
+    const def = constraint.definition;
+    const escapedCol = this.escapeRegex(columnName);
+    const colPattern = `(?:\\(?\\(?${escapedCol}\\)?(?:::(?:text|varchar|character varying))?\\)?)`;
+    const inMatch = def.match(
+      new RegExp(`${colPattern}\\s+IN\\s*\\(([^)]+)\\)`, "i")
+    );
+    if (inMatch) {
+      return this.extractStringValues(inMatch[1]);
+    }
+    const anyMatch = def.match(
+      new RegExp(
+        `${colPattern}\\s*=\\s*ANY\\s*\\(\\s*(?:ARRAY)?\\s*\\[([^\\]]+)\\]`,
+        "i"
+      )
+    );
+    if (anyMatch) {
+      return this.extractStringValues(anyMatch[1]);
+    }
+    const orPattern = new RegExp(
+      `\\b${this.escapeRegex(columnName)}\\b\\s*=\\s*'([^']*)'`,
+      "gi"
+    );
+    const orMatches = [...def.matchAll(orPattern)];
+    if (orMatches.length >= 2) {
+      return orMatches.map((m) => m[1]);
+    }
+    return void 0;
+  }
+  /**
+   * Extract string values from a comma-separated list.
+   */
+  extractStringValues(input) {
+    const values = [];
+    const matches = input.matchAll(/'([^']*)'/g);
+    for (const match of matches) {
+      values.push(match[1]);
+    }
+    return values.length > 0 ? values : void 0;
+  }
+  /**
+   * Escape special regex characters in a string.
+   */
+  escapeRegex(str) {
+    return str.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+  }
+  /**
+   * Get the table from context by name.
+   */
+  getTable(ctx, name) {
+    return ctx.tables.find((t) => t.name === name);
   }
   /**
    * Execute the grounding process.
-   * Annotates columns in ctx.tables and ctx.views with low cardinality values.
+   * Annotates columns in ctx.tables and ctx.views with values.
    */
   async execute(ctx) {
     const allContainers = [...ctx.tables, ...ctx.views];
     for (const container of allContainers) {
+      const table = this.getTable(ctx, container.name);
       for (const column of container.columns) {
         try {
-          const lowCard = await this.collectLowCardinality(
+          const result = await this.resolveColumnValues(
             container.name,
-            column
+            column,
+            table?.constraints
           );
-          if (lowCard) {
-            column.kind = lowCard.kind;
-            column.values = lowCard.values;
+          if (result) {
+            column.kind = result.kind;
+            column.values = result.values;
           }
         } catch (error) {
           console.warn(
-            "Error collecting low cardinality values for",
+            "Error collecting column values for",
             container.name,
             column.name,
             error
@@ -230,6 +306,28 @@ var LowCardinalityGrounding = class extends AbstractGrounding {
     }
     return () => this.#describe();
   }
+  /**
+   * Resolve column values from all sources in priority order.
+   */
+  async resolveColumnValues(tableName, column, constraints2) {
+    const enumValues = await this.collectEnumValues(tableName, column);
+    if (enumValues?.length) {
+      return { kind: "Enum", values: enumValues };
+    }
+    if (constraints2) {
+      for (const constraint of constraints2) {
+        const checkValues = this.parseCheckConstraint(constraint, column.name);
+        if (checkValues?.length) {
+          return { kind: "Enum", values: checkValues };
+        }
+      }
+    }
+    const lowCardValues = await this.collectLowCardinality(tableName, column);
+    if (lowCardValues?.length) {
+      return { kind: "LowCardinality", values: lowCardValues };
+    }
+    return void 0;
+  }
   #describe() {
     return null;
   }
@@ -478,18 +576,61 @@ var TableGrounding = class extends AbstractGrounding {
     }
     return tables2.map((table) => {
       const rowCountInfo = table.rowCount != null ? ` [rows: ${table.rowCount}${table.sizeHint ? `, size: ${table.sizeHint}` : ""}]` : "";
-      const pkConstraint = table.constraints?.find((c) => c.type === "PRIMARY_KEY");
+      const pkConstraint = table.constraints?.find(
+        (c) => c.type === "PRIMARY_KEY"
+      );
       const pkColumns = new Set(pkConstraint?.columns ?? []);
+      const notNullColumns = new Set(
+        table.constraints?.filter((c) => c.type === "NOT_NULL").flatMap((c) => c.columns ?? []) ?? []
+      );
+      const defaultByColumn = /* @__PURE__ */ new Map();
+      for (const c of table.constraints?.filter(
+        (c2) => c2.type === "DEFAULT"
+      ) ?? []) {
+        for (const col of c.columns ?? []) {
+          if (c.defaultValue != null) {
+            defaultByColumn.set(col, c.defaultValue);
+          }
+        }
+      }
+      const uniqueColumns = new Set(
+        table.constraints?.filter((c) => c.type === "UNIQUE" && c.columns?.length === 1).flatMap((c) => c.columns ?? []) ?? []
+      );
+      const fkByColumn = /* @__PURE__ */ new Map();
+      for (const c of table.constraints?.filter(
+        (c2) => c2.type === "FOREIGN_KEY"
+      ) ?? []) {
+        const cols = c.columns ?? [];
+        const refCols = c.referencedColumns ?? [];
+        for (let i = 0; i < cols.length; i++) {
+          const refCol = refCols[i] ?? refCols[0] ?? cols[i];
+          fkByColumn.set(cols[i], `${c.referencedTable}.${refCol}`);
+        }
+      }
       const columns = table.columns.map((column) => {
         const annotations = [];
         const isPrimaryKey = pkColumns.has(column.name);
         if (isPrimaryKey) {
           annotations.push("PK");
         }
+        if (fkByColumn.has(column.name)) {
+          annotations.push(`FK -> ${fkByColumn.get(column.name)}`);
+        }
+        if (uniqueColumns.has(column.name)) {
+          annotations.push("UNIQUE");
+        }
+        if (notNullColumns.has(column.name)) {
+          annotations.push("NOT NULL");
+        }
+        if (defaultByColumn.has(column.name)) {
+          annotations.push(`DEFAULT: ${defaultByColumn.get(column.name)}`);
+        }
         if (column.isIndexed && !isPrimaryKey) {
           annotations.push("Indexed");
         }
-        if (column.kind === "LowCardinality" && column.values?.length) {
+        if (column.kind === "Enum" && column.values?.length) {
+          annotations.push(`Enum: ${column.values.join(", ")}`);
+        } else if (column.kind === "LowCardinality" && column.values?.length) {
           annotations.push(`LowCardinality: ${column.values.join(", ")}`);
         }
         if (column.stats) {
@@ -524,9 +665,19 @@ ${table.indexes.map((index) => {
         const columnsText = index.columns?.length ? index.columns.join(", ") : "expression";
         return `    - ${index.name}${propsText}: ${columnsText}`;
       }).join("\n")}` : "";
+      const multiColumnUniques = table.constraints?.filter(
+        (c) => c.type === "UNIQUE" && (c.columns?.length ?? 0) > 1
+      ) ?? [];
+      const uniqueConstraints = multiColumnUniques.length ? `
+  Unique Constraints:
+${multiColumnUniques.map((c) => `    - ${c.name}: (${c.columns?.join(", ")})`).join("\n")}` : "";
+      const checkConstraints = table.constraints?.filter((c) => c.type === "CHECK") ?? [];
+      const checks = checkConstraints.length ? `
+  Check Constraints:
+${checkConstraints.map((c) => `    - ${c.name}: ${c.definition}`).join("\n")}` : "";
       return `- Table: ${table.name}${rowCountInfo}
   Columns:
-${columns}${indexes2}`;
+${columns}${indexes2}${uniqueConstraints}${checks}`;
     }).join("\n\n");
   }
   #formatTableLabel = (tableName) => {
@@ -699,33 +850,54 @@ var SqliteConstraintGrounding = class extends ConstraintGrounding {
       `SELECT sql FROM sqlite_master WHERE type='table' AND name=${this.#quoteIdentifier(tableName)}`
     );
     if (ddlRows[0]?.sql) {
-      const checkConstraints = this.#parseCheckConstraints(ddlRows[0].sql, tableName);
+      const columnNames = columns.map((c) => c.name);
+      const checkConstraints = this.#parseCheckConstraints(
+        ddlRows[0].sql,
+        tableName,
+        columnNames
+      );
       constraints2.push(...checkConstraints);
     }
     return constraints2;
   }
-  #parseCheckConstraints(ddl, tableName) {
+  #parseCheckConstraints(ddl, tableName, columnNames) {
     const constraints2 = [];
-    const checkRegex = /(?:CONSTRAINT\s+["'`]?(\w+)["'`]?\s+)?CHECK\s*\(([^)]+)\)/gi;
-    let match;
+    const checkStartRegex = /(?:CONSTRAINT\s+["'`]?(\w+)["'`]?\s+)?CHECK\s*\(/gi;
+    let startMatch;
     let index = 0;
-    while ((match = checkRegex.exec(ddl)) !== null) {
-      const name = match[1] || `${tableName}_check_${index}`;
-      const definition = match[2]?.trim();
-      if (definition) {
-        constraints2.push({
-          name,
-          type: "CHECK",
-          definition
-        });
-        index++;
+    while ((startMatch = checkStartRegex.exec(ddl)) !== null) {
+      const name = startMatch[1] || `${tableName}_check_${index}`;
+      const startPos = startMatch.index + startMatch[0].length;
+      let depth = 1;
+      let endPos = startPos;
+      while (endPos < ddl.length && depth > 0) {
+        if (ddl[endPos] === "(") depth++;
+        else if (ddl[endPos] === ")") depth--;
+        endPos++;
+      }
+      if (depth === 0) {
+        const definition = ddl.slice(startPos, endPos - 1).trim();
+        if (definition) {
+          const constraintColumns = columnNames.filter((col) => {
+            const colRegex = new RegExp(`\\b${col}\\b`, "i");
+            return colRegex.test(definition);
+          });
+          constraints2.push({
+            name,
+            type: "CHECK",
+            definition,
+            columns: constraintColumns.length > 0 ? constraintColumns : void 0
+          });
+          index++;
+        }
       }
     }
     const uniqueRegex = /(?:CONSTRAINT\s+["'`]?(\w+)["'`]?\s+)?UNIQUE\s*\(([^)]+)\)/gi;
     let uniqueIndex = 0;
-    while ((match = uniqueRegex.exec(ddl)) !== null) {
-      const name = match[1] || `${tableName}_unique_${uniqueIndex}`;
-      const columnsStr = match[2]?.trim();
+    let uniqueMatch;
+    while ((uniqueMatch = uniqueRegex.exec(ddl)) !== null) {
+      const name = uniqueMatch[1] || `${tableName}_unique_${uniqueIndex}`;
+      const columnsStr = uniqueMatch[2]?.trim();
       if (columnsStr) {
         const columns = columnsStr.split(",").map((c) => c.trim().replace(/["'`]/g, ""));
         constraints2.push({
@@ -794,9 +966,8 @@ var SqliteInfoGrounding = class extends InfoGrounding {
   }
 };
 
-// packages/text2sql/src/lib/adapters/sqlite/low-cardinality.sqlite.grounding.ts
-var LOW_CARDINALITY_LIMIT = 20;
-var SqliteLowCardinalityGrounding = class extends LowCardinalityGrounding {
+// packages/text2sql/src/lib/adapters/sqlite/column-values.sqlite.grounding.ts
+var SqliteColumnValuesGrounding = class extends ColumnValuesGrounding {
   #adapter;
   constructor(adapter, config = {}) {
     super(config);
@@ -805,7 +976,7 @@ var SqliteLowCardinalityGrounding = class extends LowCardinalityGrounding {
   async collectLowCardinality(tableName, column) {
     const tableIdentifier = this.#adapter.quoteIdentifier(tableName);
     const columnIdentifier = this.#adapter.quoteIdentifier(column.name);
-    const limit = LOW_CARDINALITY_LIMIT + 1;
+    const limit = this.lowCardinalityLimit + 1;
     const sql = `
       SELECT DISTINCT ${columnIdentifier} AS value
       FROM ${tableIdentifier}
@@ -813,7 +984,7 @@ var SqliteLowCardinalityGrounding = class extends LowCardinalityGrounding {
       LIMIT ${limit}
     `;
     const rows = await this.#adapter.runQuery(sql);
-    if (!rows.length || rows.length > LOW_CARDINALITY_LIMIT) {
+    if (!rows.length || rows.length > this.lowCardinalityLimit) {
       return void 0;
     }
     const values = [];
@@ -824,10 +995,7 @@ var SqliteLowCardinalityGrounding = class extends LowCardinalityGrounding {
       }
       values.push(formatted);
     }
-    if (!values.length) {
-      return void 0;
-    }
-    return { kind: "LowCardinality", values };
+    return values.length ? values : void 0;
   }
   #normalizeValue(value) {
     if (value === null || value === void 0) {
@@ -965,6 +1133,10 @@ var Sqlite = class extends Adapter {
   escape(value) {
     return value.replace(/"/g, '""');
   }
+  buildSampleRowsQuery(tableName, columns, limit) {
+    const columnList = columns?.length ? columns.map((c) => this.quoteIdentifier(c)).join(", ") : "*";
+    return `SELECT ${columnList} FROM ${this.quoteIdentifier(tableName)} LIMIT ${limit}`;
+  }
 };
 
 // packages/text2sql/src/lib/adapters/sqlite/table.sqlite.grounding.ts
@@ -1163,9 +1335,9 @@ function columnStats(config = {}) {
     return new SqliteColumnStatsGrounding(adapter, config);
   };
 }
-function lowCardinality(config = {}) {
+function columnValues(config = {}) {
   return (adapter) => {
-    return new SqliteLowCardinalityGrounding(adapter, config);
+    return new SqliteColumnValuesGrounding(adapter, config);
   };
 }
 function indexes(config = {}) {
@@ -1191,7 +1363,7 @@ var sqlite_default = {
   info,
   views,
   columnStats,
-  lowCardinality,
+  columnValues,
   indexes,
   rowCount,
   constraints,
@@ -1201,12 +1373,12 @@ var sqlite_default = {
 export {
   Sqlite,
   columnStats,
+  columnValues,
   constraints,
   sqlite_default as default,
   formatError,
   indexes,
   info,
-  lowCardinality,
   report,
   rowCount,
   tables,