@deepagents/text2sql 0.32.0 → 0.34.0

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

@@ -1,18 +1,17 @@
 import { type Adapter } from '../adapter.ts';
 import { type ColumnStatsGroundingConfig } from '../groundings/column-stats.grounding.ts';
+import { type ColumnValuesGroundingConfig } from '../groundings/column-values.grounding.ts';
 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 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';
 import type { ViewGroundingConfig } from '../groundings/view.grounding.ts';
 import { SqlServerColumnStatsGrounding } from './column-stats.sqlserver.grounding.ts';
+import { SqlServerColumnValuesGrounding } from './column-values.sqlserver.grounding.ts';
 import { SqlServerConstraintGrounding } from './constraint.sqlserver.grounding.ts';
 import { SqlServerIndexesGrounding } from './indexes.sqlserver.grounding.ts';
 import { SqlServerInfoGrounding } from './info.sqlserver.grounding.ts';
-import { SqlServerColumnValuesGrounding } from './column-values.sqlserver.grounding.ts';
 import { SqlServerRowCountGrounding } from './row-count.sqlserver.grounding.ts';
 import { SqlServer } from './sqlserver.ts';
 import { SqlServerTableGrounding } from './table.sqlserver.grounding.ts';
@@ -26,7 +25,6 @@ export declare function columnValues(config?: ColumnValuesGroundingConfig): (ada
 export declare function indexes(config?: IndexesGroundingConfig): (adapter: Adapter) => SqlServerIndexesGrounding;
 export declare function rowCount(config?: RowCountGroundingConfig): (adapter: Adapter) => SqlServerRowCountGrounding;
 export declare function constraints(config?: ConstraintGroundingConfig): (adapter: Adapter) => SqlServerConstraintGrounding;
-export declare function report(config?: ReportGroundingConfig): (adapter: Adapter) => ReportGrounding;
 declare const _default: {
     tables: typeof tables;
     info: typeof info;
@@ -36,7 +34,6 @@ declare const _default: {
     indexes: typeof indexes;
     rowCount: typeof rowCount;
     constraints: typeof constraints;
-    report: typeof report;
     SqlServer: typeof SqlServer;
 };
 export default _default;

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/sqlserver/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,6BAA6B,EAAE,MAAM,uCAAuC,CAAC;AACtF,OAAO,EAAE,4BAA4B,EAAE,MAAM,qCAAqC,CAAC;AACnF,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,8BAA8B,EAAE,MAAM,wCAAwC,CAAC;AACxF,OAAO,EAAE,0BAA0B,EAAE,MAAM,oCAAoC,CAAC;AAChF,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,uBAAuB,EAAE,MAAM,gCAAgC,CAAC;AACzE,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAEvE,cAAc,gBAAgB,CAAC;AAE/B,wBAAgB,MAAM,CAAC,MAAM,GAAE,oBAAyB,IAC9C,SAAS,OAAO,6BACzB;AAED,wBAAgB,IAAI,CAAC,MAAM,GAAE,mBAAwB,IAC3C,SAAS,OAAO,4BACzB;AAED,wBAAgB,KAAK,CAAC,MAAM,GAAE,mBAAwB,IAC5C,SAAS,OAAO,4BAGzB;AAED,wBAAgB,WAAW,CAAC,MAAM,GAAE,0BAA+B,IACzD,SAAS,OAAO,mCAGzB;AAED,wBAAgB,YAAY,CAAC,MAAM,GAAE,2BAAgC,IAC3D,SAAS,OAAO,oCAGzB;AAED,wBAAgB,OAAO,CAAC,MAAM,GAAE,sBAA2B,IACjD,SAAS,OAAO,+BAGzB;AAED,wBAAgB,QAAQ,CAAC,MAAM,GAAE,uBAA4B,IACnD,SAAS,OAAO,gCAGzB;AAED,wBAAgB,WAAW,CAAC,MAAM,GAAE,yBAA8B,IACxD,SAAS,OAAO,kCAGzB;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/sqlserver/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,2BAA2B,EAAE,MAAM,0CAA0C,CAAC;AAC5F,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,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,6BAA6B,EAAE,MAAM,uCAAuC,CAAC;AACtF,OAAO,EAAE,8BAA8B,EAAE,MAAM,wCAAwC,CAAC;AACxF,OAAO,EAAE,4BAA4B,EAAE,MAAM,qCAAqC,CAAC;AACnF,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,oCAAoC,CAAC;AAChF,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,uBAAuB,EAAE,MAAM,gCAAgC,CAAC;AACzE,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAEvE,cAAc,gBAAgB,CAAC;AAE/B,wBAAgB,MAAM,CAAC,MAAM,GAAE,oBAAyB,IAC9C,SAAS,OAAO,6BACzB;AAED,wBAAgB,IAAI,CAAC,MAAM,GAAE,mBAAwB,IAC3C,SAAS,OAAO,4BACzB;AAED,wBAAgB,KAAK,CAAC,MAAM,GAAE,mBAAwB,IAC5C,SAAS,OAAO,4BAGzB;AAED,wBAAgB,WAAW,CAAC,MAAM,GAAE,0BAA+B,IACzD,SAAS,OAAO,mCAGzB;AAED,wBAAgB,YAAY,CAAC,MAAM,GAAE,2BAAgC,IAC3D,SAAS,OAAO,oCAGzB;AAED,wBAAgB,OAAO,CAAC,MAAM,GAAE,sBAA2B,IACjD,SAAS,OAAO,+BAGzB;AAED,wBAAgB,QAAQ,CAAC,MAAM,GAAE,uBAA4B,IACnD,SAAS,OAAO,gCAGzB;AAED,wBAAgB,WAAW,CAAC,MAAM,GAAE,yBAA8B,IACxD,SAAS,OAAO,kCAGzB;;;;;;;;;;;;AAED,wBAUE"}

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

@@ -346,9 +346,6 @@ var Adapter = class {
         this.#relationshipToFragment(rel, sourceTable, targetTable)
       );
     }
-    if (ctx.report) {
-      fragments.push({ name: "businessContext", data: ctx.report });
-    }
     return fragments;
   }
   /**
@@ -658,64 +655,6 @@ var ColumnStatsGrounding = class extends AbstractGrounding {
   }
 };
 
-// packages/text2sql/src/lib/adapters/groundings/constraint.grounding.ts
-var ConstraintGrounding = class extends AbstractGrounding {
-  constructor(config = {}) {
-    super("constraint");
-  }
-  /**
-   * Execute the grounding process.
-   * Annotates tables in ctx.tables with their constraints.
-   */
-  async execute(ctx) {
-    for (const table2 of ctx.tables) {
-      try {
-        table2.constraints = await this.getConstraints(table2.name);
-      } catch (error) {
-        console.warn("Error collecting constraints for", table2.name, error);
-      }
-    }
-  }
-};
-
-// packages/text2sql/src/lib/adapters/groundings/indexes.grounding.ts
-var IndexesGrounding = class extends AbstractGrounding {
-  constructor(config = {}) {
-    super("index");
-  }
-  /**
-   * Execute the grounding process.
-   * Annotates tables in ctx.tables with their indexes and marks indexed columns.
-   */
-  async execute(ctx) {
-    for (const table2 of ctx.tables) {
-      table2.indexes = await this.getIndexes(table2.name);
-      for (const index2 of table2.indexes ?? []) {
-        for (const colName of index2.columns) {
-          const column2 = table2.columns.find((c) => c.name === colName);
-          if (column2) {
-            column2.isIndexed = true;
-          }
-        }
-      }
-    }
-  }
-};
-
-// packages/text2sql/src/lib/adapters/groundings/info.grounding.ts
-var InfoGrounding = class extends AbstractGrounding {
-  constructor(config = {}) {
-    super("dialectInfo");
-  }
-  /**
-   * Execute the grounding process.
-   * Writes database info to ctx.info.
-   */
-  async execute(ctx) {
-    ctx.info = await this.collectInfo();
-  }
-};
-
 // packages/text2sql/src/lib/adapters/groundings/column-values.grounding.ts
 var ColumnValuesGrounding = class extends AbstractGrounding {
   lowCardinalityLimit;
@@ -855,103 +794,61 @@ var ColumnValuesGrounding = class extends AbstractGrounding {
   }
 };
 
-// packages/text2sql/src/lib/adapters/groundings/report.grounding.ts
-import { groq } from "@ai-sdk/groq";
-import { tool } from "ai";
-import dedent from "dedent";
-import z from "zod";
-import "@deepagents/agent";
-import {
-  ContextEngine,
-  InMemoryContextStore,
-  agent,
-  fragment,
-  persona,
-  user
-} from "@deepagents/context";
-var ReportGrounding = class extends AbstractGrounding {
-  #adapter;
-  #model;
-  #cache;
-  #forceRefresh;
-  constructor(adapter, config = {}) {
-    super("business_context");
-    this.#adapter = adapter;
-    this.#model = config.model ?? groq("openai/gpt-oss-20b");
-    this.#cache = config.cache;
-    this.#forceRefresh = config.forceRefresh ?? false;
+// packages/text2sql/src/lib/adapters/groundings/constraint.grounding.ts
+var ConstraintGrounding = class extends AbstractGrounding {
+  constructor(config = {}) {
+    super("constraint");
   }
+  /**
+   * Execute the grounding process.
+   * Annotates tables in ctx.tables with their constraints.
+   */
   async execute(ctx) {
-    if (!this.#forceRefresh && this.#cache) {
-      const cached = await this.#cache.get();
-      if (cached) {
-        ctx.report = cached;
-        return;
+    for (const table2 of ctx.tables) {
+      try {
+        table2.constraints = await this.getConstraints(table2.name);
+      } catch (error) {
+        console.warn("Error collecting constraints for", table2.name, error);
       }
     }
-    const report2 = await this.#generateReport();
-    ctx.report = report2;
-    if (this.#cache) {
-      await this.#cache.set(report2);
-    }
   }
-  async #generateReport() {
-    const context = new ContextEngine({
-      store: new InMemoryContextStore(),
-      chatId: `report-gen-${crypto.randomUUID()}`,
-      userId: "system"
-    });
-    context.set(
-      persona({
-        name: "db-report-agent",
-        role: "Database analyst",
-        objective: "Analyze the database and write a contextual report about what it represents"
-      }),
-      fragment(
-        "instructions",
-        dedent`
-          Write a business context that helps another agent answer questions accurately.
-
-          For EACH table, do queries ONE AT A TIME:
-          1. SELECT COUNT(*) to get row count
-          2. SELECT * LIMIT 3 to see sample data
-
-          Then write a report with:
-          - What business this database is for
-          - For each table: purpose, row count, and example of what the data looks like
-
-          Include concrete examples like "Track prices are $0.99",
-          "Customer names like 'Luís Gonçalves'", etc.
+};
 
-          Keep it 400-600 words, conversational style.
-        `
-      ),
-      user(
-        "Please analyze the database and write a contextual report about what this database represents."
-      )
-    );
-    const adapter = this.#adapter;
-    const reportAgent = agent({
-      name: "db-report-agent",
-      model: this.#model,
-      context,
-      tools: {
-        query_database: tool({
-          description: "Execute a SELECT query to explore the database and gather insights.",
-          inputSchema: z.object({
-            sql: z.string().describe("The SELECT query to execute"),
-            purpose: z.string().describe(
-              "What insight you are trying to gather with this query"
-            )
-          }),
-          execute: ({ sql }) => {
-            return adapter.execute(sql);
+// packages/text2sql/src/lib/adapters/groundings/indexes.grounding.ts
+var IndexesGrounding = class extends AbstractGrounding {
+  constructor(config = {}) {
+    super("index");
+  }
+  /**
+   * Execute the grounding process.
+   * Annotates tables in ctx.tables with their indexes and marks indexed columns.
+   */
+  async execute(ctx) {
+    for (const table2 of ctx.tables) {
+      table2.indexes = await this.getIndexes(table2.name);
+      for (const index2 of table2.indexes ?? []) {
+        for (const colName of index2.columns) {
+          const column2 = table2.columns.find((c) => c.name === colName);
+          if (column2) {
+            column2.isIndexed = true;
           }
-        })
+        }
       }
-    });
-    const result = await reportAgent.generate({});
-    return result.text;
+    }
+  }
+};
+
+// packages/text2sql/src/lib/adapters/groundings/info.grounding.ts
+var InfoGrounding = class extends AbstractGrounding {
+  constructor(config = {}) {
+    super("dialectInfo");
+  }
+  /**
+   * Execute the grounding process.
+   * Writes database info to ctx.info.
+   */
+  async execute(ctx) {
+    ctx.info = await this.collectInfo();
   }
 };
 
@@ -1217,6 +1114,59 @@ var SqlServerColumnStatsGrounding = class extends ColumnStatsGrounding {
   }
 };
 
+// packages/text2sql/src/lib/adapters/sqlserver/column-values.sqlserver.grounding.ts
+var SqlServerColumnValuesGrounding = class extends ColumnValuesGrounding {
+  #adapter;
+  constructor(adapter, config = {}) {
+    super(config);
+    this.#adapter = adapter;
+  }
+  #isHighCardinality(column2) {
+    const nDistinct = column2.stats?.nDistinct;
+    if (nDistinct != null) {
+      if (nDistinct > 0) return nDistinct > this.lowCardinalityLimit;
+      return true;
+    }
+    const type = column2.type.toLowerCase();
+    if (/nvarchar\s*\(\s*max\s*\)|varchar\s*\(\s*max\s*\)|text|ntext|xml|image|varbinary\s*\(\s*max\s*\)|uniqueidentifier/.test(
+      type
+    )) {
+      return true;
+    }
+    const varcharMatch = type.match(/n?varchar\s*\(\s*(\d+)\s*\)/);
+    if (varcharMatch && parseInt(varcharMatch[1]) > 255) {
+      return true;
+    }
+    return false;
+  }
+  async collectLowCardinality(tableName, column2) {
+    if (this.#isHighCardinality(column2)) {
+      return void 0;
+    }
+    const { schema, table: table2 } = this.#adapter.parseTableName(tableName);
+    const tableIdentifier = `[${this.#adapter.escape(schema)}].[${this.#adapter.escape(table2)}]`;
+    const columnIdentifier = `[${this.#adapter.escape(column2.name)}]`;
+    const limit = this.lowCardinalityLimit + 1;
+    const sql = `
+      SELECT DISTINCT TOP ${limit} CAST(${columnIdentifier} AS NVARCHAR(MAX)) AS value
+      FROM ${tableIdentifier}
+      WHERE ${columnIdentifier} IS NOT NULL
+    `;
+    const rows = await this.#adapter.runQuery(sql);
+    if (!rows.length || rows.length > this.lowCardinalityLimit) {
+      return void 0;
+    }
+    const values = [];
+    for (const row of rows) {
+      if (row.value == null) {
+        return void 0;
+      }
+      values.push(row.value);
+    }
+    return values.length ? values : void 0;
+  }
+};
+
 // packages/text2sql/src/lib/adapters/sqlserver/constraint.sqlserver.grounding.ts
 var SqlServerConstraintGrounding = class extends ConstraintGrounding {
   #adapter;
@@ -1448,59 +1398,6 @@ var SqlServerInfoGrounding = class extends InfoGrounding {
   }
 };
 
-// packages/text2sql/src/lib/adapters/sqlserver/column-values.sqlserver.grounding.ts
-var SqlServerColumnValuesGrounding = class extends ColumnValuesGrounding {
-  #adapter;
-  constructor(adapter, config = {}) {
-    super(config);
-    this.#adapter = adapter;
-  }
-  #isHighCardinality(column2) {
-    const nDistinct = column2.stats?.nDistinct;
-    if (nDistinct != null) {
-      if (nDistinct > 0) return nDistinct > this.lowCardinalityLimit;
-      return true;
-    }
-    const type = column2.type.toLowerCase();
-    if (/nvarchar\s*\(\s*max\s*\)|varchar\s*\(\s*max\s*\)|text|ntext|xml|image|varbinary\s*\(\s*max\s*\)|uniqueidentifier/.test(
-      type
-    )) {
-      return true;
-    }
-    const varcharMatch = type.match(/n?varchar\s*\(\s*(\d+)\s*\)/);
-    if (varcharMatch && parseInt(varcharMatch[1]) > 255) {
-      return true;
-    }
-    return false;
-  }
-  async collectLowCardinality(tableName, column2) {
-    if (this.#isHighCardinality(column2)) {
-      return void 0;
-    }
-    const { schema, table: table2 } = this.#adapter.parseTableName(tableName);
-    const tableIdentifier = `[${this.#adapter.escape(schema)}].[${this.#adapter.escape(table2)}]`;
-    const columnIdentifier = `[${this.#adapter.escape(column2.name)}]`;
-    const limit = this.lowCardinalityLimit + 1;
-    const sql = `
-      SELECT DISTINCT TOP ${limit} CAST(${columnIdentifier} AS NVARCHAR(MAX)) AS value
-      FROM ${tableIdentifier}
-      WHERE ${columnIdentifier} IS NOT NULL
-    `;
-    const rows = await this.#adapter.runQuery(sql);
-    if (!rows.length || rows.length > this.lowCardinalityLimit) {
-      return void 0;
-    }
-    const values = [];
-    for (const row of rows) {
-      if (row.value == null) {
-        return void 0;
-      }
-      values.push(row.value);
-    }
-    return values.length ? values : void 0;
-  }
-};
-
 // packages/text2sql/src/lib/adapters/sqlserver/row-count.sqlserver.grounding.ts
 var SqlServerRowCountGrounding = class extends RowCountGrounding {
   #adapter;
@@ -2260,9 +2157,6 @@ function constraints(config = {}) {
     return new SqlServerConstraintGrounding(adapter, config);
   };
 }
-function report(config = {}) {
-  return (adapter) => new ReportGrounding(adapter, config);
-}
 var sqlserver_default = {
   tables,
   info,
@@ -2272,7 +2166,6 @@ var sqlserver_default = {
   indexes,
   rowCount,
   constraints,
-  report,
   SqlServer
 };
 export {
@@ -2284,7 +2177,6 @@ export {
   formatSqlServerError,
   indexes,
   info,
-  report,
   rowCount,
   tables,
   views