@deepagents/text2sql 7.0.2 → 7.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +3 -2
- package/dist/bin/sql.js +57 -32
- package/dist/index.js +793 -498
- package/dist/index.js.map +4 -4
- package/dist/lib/adapters/adapter.d.ts +23 -1
- package/dist/lib/adapters/adapter.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/bigquery-fk.d.ts +0 -11
- package/dist/lib/adapters/bigquery/bigquery-fk.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/bigquery-rows.d.ts +22 -0
- package/dist/lib/adapters/bigquery/bigquery-rows.d.ts.map +1 -0
- package/dist/lib/adapters/bigquery/bigquery.d.ts +1 -1
- package/dist/lib/adapters/bigquery/bigquery.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/constraint.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/index.d.ts +6 -13
- package/dist/lib/adapters/bigquery/index.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/index.js +244 -51
- package/dist/lib/adapters/bigquery/index.js.map +4 -4
- package/dist/lib/adapters/bigquery/indexes.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/row-count.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/table.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/view.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/clickhouse/clickhouse-rows.d.ts +7 -0
- package/dist/lib/adapters/clickhouse/clickhouse-rows.d.ts.map +1 -0
- package/dist/lib/adapters/clickhouse/clickhouse.d.ts +1 -1
- package/dist/lib/adapters/clickhouse/clickhouse.d.ts.map +1 -1
- package/dist/lib/adapters/clickhouse/constraint.clickhouse.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/clickhouse/index.js +133 -44
- package/dist/lib/adapters/clickhouse/index.js.map +4 -4
- package/dist/lib/adapters/clickhouse/indexes.clickhouse.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/clickhouse/info.clickhouse.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/clickhouse/row-count.clickhouse.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/clickhouse/table.clickhouse.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/clickhouse/view.clickhouse.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/duckdb/duckdb.d.ts +1 -1
- package/dist/lib/adapters/duckdb/duckdb.d.ts.map +1 -1
- package/dist/lib/adapters/duckdb/duckdb.groundings.d.ts.map +1 -1
- package/dist/lib/adapters/duckdb/index.d.ts +8 -10
- package/dist/lib/adapters/duckdb/index.d.ts.map +1 -1
- package/dist/lib/adapters/duckdb/index.js +1549 -1412
- package/dist/lib/adapters/duckdb/index.js.map +4 -4
- package/dist/lib/adapters/groundings/index.js.map +1 -1
- package/dist/lib/adapters/groundings/rows.d.ts +32 -0
- package/dist/lib/adapters/groundings/rows.d.ts.map +1 -0
- package/dist/lib/adapters/mysql/column-stats.mysql.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/mysql/column-values.mysql.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/mysql/constraint.mysql.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/mysql/index.d.ts +2 -4
- package/dist/lib/adapters/mysql/index.d.ts.map +1 -1
- package/dist/lib/adapters/mysql/index.js +230 -76
- package/dist/lib/adapters/mysql/index.js.map +4 -4
- package/dist/lib/adapters/mysql/indexes.mysql.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/mysql/info.mysql.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/mysql/mysql-rows.d.ts +11 -0
- package/dist/lib/adapters/mysql/mysql-rows.d.ts.map +1 -0
- package/dist/lib/adapters/mysql/mysql.d.ts +1 -1
- package/dist/lib/adapters/mysql/mysql.d.ts.map +1 -1
- package/dist/lib/adapters/mysql/row-count.mysql.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/mysql/table.mysql.grounding.d.ts +3 -2
- package/dist/lib/adapters/mysql/table.mysql.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/mysql/view.mysql.grounding.d.ts +2 -2
- package/dist/lib/adapters/mysql/view.mysql.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/column-stats.postgres.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/column-values.postgres.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/constraint.postgres.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/index.d.ts +2 -4
- package/dist/lib/adapters/postgres/index.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/index.js +266 -487
- package/dist/lib/adapters/postgres/index.js.map +4 -4
- package/dist/lib/adapters/postgres/indexes.postgres.grounding.d.ts +1 -5
- package/dist/lib/adapters/postgres/indexes.postgres.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/info.postgres.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/postgres-rows.d.ts +7 -0
- package/dist/lib/adapters/postgres/postgres-rows.d.ts.map +1 -0
- package/dist/lib/adapters/postgres/postgres.d.ts +6 -1
- package/dist/lib/adapters/postgres/postgres.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/row-count.postgres.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/table.postgres.grounding.d.ts +5 -4
- package/dist/lib/adapters/postgres/table.postgres.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/postgres/view.postgres.grounding.d.ts +3 -3
- package/dist/lib/adapters/postgres/view.postgres.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/posthog/definitions.posthog.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/posthog/index.d.ts +4 -5
- package/dist/lib/adapters/posthog/index.d.ts.map +1 -1
- package/dist/lib/adapters/posthog/index.js +349 -287
- package/dist/lib/adapters/posthog/index.js.map +4 -4
- package/dist/lib/adapters/posthog/posthog.d.ts +2 -2
- package/dist/lib/adapters/posthog/posthog.d.ts.map +1 -1
- package/dist/lib/adapters/posthog/posthog.sql-policy.d.ts.map +1 -1
- package/dist/lib/adapters/posthog/schema.posthog.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/posthog/transport.d.ts.map +1 -1
- package/dist/lib/adapters/posthog/types.d.ts +16 -6
- package/dist/lib/adapters/posthog/types.d.ts.map +1 -1
- package/dist/lib/adapters/spreadsheet/index.js +113 -34
- package/dist/lib/adapters/spreadsheet/index.js.map +4 -4
- package/dist/lib/adapters/sqlite/column-stats.sqlite.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlite/column-values.sqlite.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlite/constraint.sqlite.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlite/index.js +136 -49
- package/dist/lib/adapters/sqlite/index.js.map +4 -4
- package/dist/lib/adapters/sqlite/indexes.sqlite.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlite/info.sqlite.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlite/row-count.sqlite.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlite/sqlite-rows.d.ts +41 -0
- package/dist/lib/adapters/sqlite/sqlite-rows.d.ts.map +1 -0
- package/dist/lib/adapters/sqlite/sqlite.d.ts +1 -1
- package/dist/lib/adapters/sqlite/sqlite.d.ts.map +1 -1
- package/dist/lib/adapters/sqlite/table.sqlite.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlite/view.sqlite.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/column-stats.sqlserver.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/column-values.sqlserver.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/constraint.sqlserver.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/index.d.ts +2 -4
- package/dist/lib/adapters/sqlserver/index.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/index.js +223 -455
- package/dist/lib/adapters/sqlserver/index.js.map +4 -4
- package/dist/lib/adapters/sqlserver/indexes.sqlserver.grounding.d.ts +1 -5
- package/dist/lib/adapters/sqlserver/indexes.sqlserver.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/info.sqlserver.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/row-count.sqlserver.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/sqlserver-rows.d.ts +7 -0
- package/dist/lib/adapters/sqlserver/sqlserver-rows.d.ts.map +1 -0
- package/dist/lib/adapters/sqlserver/sqlserver.d.ts +6 -1
- package/dist/lib/adapters/sqlserver/sqlserver.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/table.sqlserver.grounding.d.ts +4 -3
- package/dist/lib/adapters/sqlserver/table.sqlserver.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/sqlserver/view.sqlserver.grounding.d.ts +3 -3
- package/dist/lib/adapters/sqlserver/view.sqlserver.grounding.d.ts.map +1 -1
- package/dist/lib/agents/sql-transform-plugins.d.ts.map +1 -1
- package/dist/lib/checkpoint.d.ts +34 -21
- package/dist/lib/checkpoint.d.ts.map +1 -1
- package/dist/lib/file-cache.d.ts +6 -1
- package/dist/lib/file-cache.d.ts.map +1 -1
- package/dist/lib/fs/columns.d.ts +14 -0
- package/dist/lib/fs/columns.d.ts.map +1 -0
- package/dist/lib/fs/mssql/mssql-fs.d.ts +4 -4
- package/dist/lib/fs/mssql/mssql-fs.d.ts.map +1 -1
- package/dist/lib/fs/postgres/postgres-fs.d.ts +4 -4
- package/dist/lib/fs/postgres/postgres-fs.d.ts.map +1 -1
- package/dist/lib/fs/sqlite/sqlite-fs.d.ts +4 -4
- package/dist/lib/fs/sqlite/sqlite-fs.d.ts.map +1 -1
- package/dist/lib/index-cache.d.ts +4 -3
- package/dist/lib/index-cache.d.ts.map +1 -1
- package/dist/lib/sql.d.ts.map +1 -1
- package/dist/lib/synthesis/decorators/deduplicated-producer.d.ts +1 -2
- package/dist/lib/synthesis/decorators/deduplicated-producer.d.ts.map +1 -1
- package/dist/lib/synthesis/decorators/filtered-producer.d.ts +0 -2
- package/dist/lib/synthesis/decorators/filtered-producer.d.ts.map +1 -1
- package/dist/lib/synthesis/decorators/validated-producer.d.ts +1 -3
- package/dist/lib/synthesis/decorators/validated-producer.d.ts.map +1 -1
- package/dist/lib/synthesis/extractors/base-contextual-extractor.d.ts.map +1 -1
- package/dist/lib/synthesis/extractors/message-extractor.d.ts.map +1 -1
- package/dist/lib/synthesis/extractors/tool-input.d.ts +6 -0
- package/dist/lib/synthesis/extractors/tool-input.d.ts.map +1 -0
- package/dist/lib/synthesis/index.js +124 -114
- package/dist/lib/synthesis/index.js.map +4 -4
- package/dist/lib/synthesis/synthesizers/breadth-evolver.d.ts +0 -2
- package/dist/lib/synthesis/synthesizers/breadth-evolver.d.ts.map +1 -1
- package/dist/lib/synthesis/synthesizers/depth-evolver.d.ts +0 -3
- package/dist/lib/synthesis/synthesizers/depth-evolver.d.ts.map +1 -1
- package/dist/lib/synthesis/synthesizers/schema-synthesizer.d.ts +0 -2
- package/dist/lib/synthesis/synthesizers/schema-synthesizer.d.ts.map +1 -1
- package/package.json +4 -4
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/posthog/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,6BAA6B,GACvC,OAAO,GAAG,QAAQ,GAAG,SAAS,GAAG,OAAO,CAAC;AAE3C,MAAM,MAAM,kBAAkB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAEzD,MAAM,MAAM,gBAAgB,GACxB;IACE,IAAI,EAAE,YAAY,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,kBAAkB,CAAC;CAC7B,GACD;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,QAAQ,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAC3D;IAAE,IAAI,EAAE,qBAAqB,CAAA;CAAE,CAAC;AAEpC,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,gBAAgB,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC5B,KAAK,CAAC,EAAE,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/lib/adapters/posthog/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,6BAA6B,GACvC,OAAO,GAAG,QAAQ,GAAG,SAAS,GAAG,OAAO,CAAC;AAE3C,MAAM,MAAM,kBAAkB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAEzD,MAAM,MAAM,gBAAgB,GACxB;IACE,IAAI,EAAE,YAAY,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,kBAAkB,CAAC;CAC7B,GACD;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,QAAQ,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAC3D;IAAE,IAAI,EAAE,qBAAqB,CAAA;CAAE,CAAC;AAEpC,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,gBAAgB,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;;GAGG;AACH,MAAM,WAAW,aAAa;IAC5B,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,GAAG,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,GAAG,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACrB;AAED,MAAM,WAAW,uBAAuB;IACtC,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,MAAM,EAAE,aAAa,EAAE,CAAC;IACxB,QAAQ,EAAE,aAAa,EAAE,CAAC;IAC1B,OAAO,EAAE,aAAa,EAAE,CAAC;IACzB,WAAW,EAAE,MAAM,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;IACrB,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC;CACnB;AAED,MAAM,MAAM,wBAAwB,GAChC,SAAS,GACT,OAAO,GACP,SAAS,GACT,QAAQ,GACR,UAAU,GACV,MAAM,GACN,SAAS,GACT,OAAO,GACP,MAAM,GACN,YAAY,GACZ,eAAe,GACf,iBAAiB,GACjB,YAAY,GACZ,MAAM,GACN,mBAAmB,GACnB,SAAS,CAAC;AAEd,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE,wBAAwB,CAAC;IAC/B,YAAY,EAAE,OAAO,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,KAAK,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC;IAC/B,EAAE,CAAC,EAAE,MAAM,CAAC;CACb;AAED,MAAM,MAAM,wBAAwB,GAChC,SAAS,GACT,QAAQ,GACR,gBAAgB,GAChB,MAAM,GACN,cAAc,GACd,mBAAmB,GACnB,cAAc,GACd,UAAU,CAAC;AAEf,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,wBAAwB,CAAC;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC;IAC7C,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,KAAK,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;CAC3B;AAED,MAAM,WAAW,iBAAiB;IAChC,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC5B;AAED,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC;IAC7C,KAAK,EAAE,iBAAiB,EAAE,CAAC;CAC5B;AAED,MAAM,WAAW,sBAAsB;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IACvB,QAAQ,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;CAC3B;AAED,MAAM,WAAW,yBAAyB;IACxC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,YAAY,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,QAAQ,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;CAC3B;AAED;;;GAGG;AACH,MAAM,WAAW,gBAAgB;IAC/B,KAAK,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACtD,gEAAgE;IAChE,oBAAoB,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAC3C,sEAAsE;IACtE,uBAAuB,CAAC,OAAO,EAAE;QAC/B,IAAI,EAAE,6BAA6B,CAAC;QACpC,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;CACxB;AAED,MAAM,WAAW,6BAA6B;IAC5C,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC;IAC3B,cAAc,EAAE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAC/C,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB"}
|
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
// packages/text2sql/src/lib/adapters/adapter.ts
|
|
2
2
|
import { format as formatSql } from "sql-formatter";
|
|
3
|
+
import { z } from "zod";
|
|
3
4
|
|
|
4
5
|
// packages/text2sql/src/lib/agents/exceptions.ts
|
|
5
6
|
var sqlValidationMarker = /* @__PURE__ */ Symbol("SQLValidationError");
|
|
@@ -432,6 +433,40 @@ var Adapter = class {
|
|
|
432
433
|
}
|
|
433
434
|
return decoded;
|
|
434
435
|
}
|
|
436
|
+
/**
|
|
437
|
+
* Run an introspection query and check every row against `row`, the shape
|
|
438
|
+
* the driver returns for that query. A row of any other shape throws.
|
|
439
|
+
*/
|
|
440
|
+
async runQuery(sql, row) {
|
|
441
|
+
const rows = z.array(row).safeParse(await this.queryRows(sql));
|
|
442
|
+
if (!rows.success) {
|
|
443
|
+
throw new Error(
|
|
444
|
+
`${this.constructor.name} introspection query returned rows of an unexpected shape:
|
|
445
|
+
${z.prettifyError(rows.error)}
|
|
446
|
+
SQL: ${sql.trim()}`,
|
|
447
|
+
{ cause: rows.error }
|
|
448
|
+
);
|
|
449
|
+
}
|
|
450
|
+
return rows.data;
|
|
451
|
+
}
|
|
452
|
+
/**
|
|
453
|
+
* The rows in a driver's execute() result: the result itself when it is an
|
|
454
|
+
* array, otherwise the first array found at one of `paths` (dot-separated,
|
|
455
|
+
* so `recordsets.0` reads `result.recordsets[0]`).
|
|
456
|
+
*/
|
|
457
|
+
rowsFrom(result, paths) {
|
|
458
|
+
if (Array.isArray(result)) return result;
|
|
459
|
+
for (const path2 of paths) {
|
|
460
|
+
const rows = path2.split(".").reduce(
|
|
461
|
+
(value, key) => typeof value === "object" && value !== null ? Reflect.get(value, key) : void 0,
|
|
462
|
+
result
|
|
463
|
+
);
|
|
464
|
+
if (Array.isArray(rows)) return rows;
|
|
465
|
+
}
|
|
466
|
+
throw new Error(
|
|
467
|
+
`${this.constructor.name} execute() must return an array of rows or an object holding one at ${paths.map((path2) => `result.${path2}`).join(", ")}.`
|
|
468
|
+
);
|
|
469
|
+
}
|
|
435
470
|
/**
|
|
436
471
|
* Convert unknown database value to number.
|
|
437
472
|
* Handles number, bigint, and string types.
|
|
@@ -928,6 +963,12 @@ var TableGrounding = class extends AbstractGrounding {
|
|
|
928
963
|
};
|
|
929
964
|
|
|
930
965
|
// packages/text2sql/src/lib/adapters/sqlite/column-stats.sqlite.grounding.ts
|
|
966
|
+
import { z as z2 } from "zod";
|
|
967
|
+
var statsRow = z2.object({
|
|
968
|
+
min_value: z2.unknown(),
|
|
969
|
+
max_value: z2.unknown(),
|
|
970
|
+
null_fraction: z2.number().nullable()
|
|
971
|
+
});
|
|
931
972
|
var SqliteColumnStatsGrounding = class extends ColumnStatsGrounding {
|
|
932
973
|
#adapter;
|
|
933
974
|
constructor(adapter, config = {}) {
|
|
@@ -947,7 +988,7 @@ var SqliteColumnStatsGrounding = class extends ColumnStatsGrounding {
|
|
|
947
988
|
AVG(CASE WHEN ${columnIdentifier} IS NULL THEN 1.0 ELSE 0.0 END) AS null_fraction
|
|
948
989
|
FROM ${tableIdentifier}
|
|
949
990
|
`;
|
|
950
|
-
const rows = await this.#adapter.runQuery(sql);
|
|
991
|
+
const rows = await this.#adapter.runQuery(sql, statsRow);
|
|
951
992
|
if (!rows.length) {
|
|
952
993
|
return void 0;
|
|
953
994
|
}
|
|
@@ -995,6 +1036,18 @@ var SqliteColumnStatsGrounding = class extends ColumnStatsGrounding {
|
|
|
995
1036
|
}
|
|
996
1037
|
};
|
|
997
1038
|
|
|
1039
|
+
// packages/text2sql/src/lib/adapters/groundings/rows.ts
|
|
1040
|
+
import { z as z3 } from "zod";
|
|
1041
|
+
var nameRow = z3.object({ name: z3.string() });
|
|
1042
|
+
var versionRow = z3.object({ version: z3.string() });
|
|
1043
|
+
var databaseRow = z3.object({ db: z3.string() });
|
|
1044
|
+
var viewDefinitionRow = z3.object({
|
|
1045
|
+
definition: z3.string().nullable()
|
|
1046
|
+
});
|
|
1047
|
+
var distinctValueRow = z3.object({ value: z3.unknown() });
|
|
1048
|
+
var textValueRow = z3.object({ value: z3.string() });
|
|
1049
|
+
var numericValue = z3.union([z3.number(), z3.bigint(), z3.string()]);
|
|
1050
|
+
|
|
998
1051
|
// packages/text2sql/src/lib/adapters/sqlite/column-values.sqlite.grounding.ts
|
|
999
1052
|
var SqliteColumnValuesGrounding = class extends ColumnValuesGrounding {
|
|
1000
1053
|
#adapter;
|
|
@@ -1012,7 +1065,7 @@ var SqliteColumnValuesGrounding = class extends ColumnValuesGrounding {
|
|
|
1012
1065
|
WHERE ${columnIdentifier} IS NOT NULL
|
|
1013
1066
|
LIMIT ${limit}
|
|
1014
1067
|
`;
|
|
1015
|
-
const rows = await this.#adapter.runQuery(sql);
|
|
1068
|
+
const rows = await this.#adapter.runQuery(sql, distinctValueRow);
|
|
1016
1069
|
if (!rows.length || rows.length > this.lowCardinalityLimit) {
|
|
1017
1070
|
return void 0;
|
|
1018
1071
|
}
|
|
@@ -1049,6 +1102,38 @@ var SqliteColumnValuesGrounding = class extends ColumnValuesGrounding {
|
|
|
1049
1102
|
}
|
|
1050
1103
|
};
|
|
1051
1104
|
|
|
1105
|
+
// packages/text2sql/src/lib/adapters/sqlite/sqlite-rows.ts
|
|
1106
|
+
import { z as z4 } from "zod";
|
|
1107
|
+
var sqliteInteger = z4.union([z4.number(), z4.bigint()]).transform(Number);
|
|
1108
|
+
var sqliteMasterSqlRow = z4.object({ sql: z4.string().nullable() });
|
|
1109
|
+
var tableInfoRow = z4.object({
|
|
1110
|
+
cid: sqliteInteger,
|
|
1111
|
+
name: z4.string(),
|
|
1112
|
+
type: z4.string(),
|
|
1113
|
+
notnull: sqliteInteger,
|
|
1114
|
+
dflt_value: z4.string().nullable(),
|
|
1115
|
+
pk: sqliteInteger
|
|
1116
|
+
});
|
|
1117
|
+
var foreignKeyListRow = z4.object({
|
|
1118
|
+
id: sqliteInteger,
|
|
1119
|
+
seq: sqliteInteger,
|
|
1120
|
+
table: z4.string(),
|
|
1121
|
+
from: z4.string(),
|
|
1122
|
+
to: z4.string().nullable()
|
|
1123
|
+
});
|
|
1124
|
+
var indexListRow = z4.object({
|
|
1125
|
+
seq: sqliteInteger,
|
|
1126
|
+
name: z4.string(),
|
|
1127
|
+
unique: sqliteInteger,
|
|
1128
|
+
origin: z4.string(),
|
|
1129
|
+
partial: sqliteInteger
|
|
1130
|
+
});
|
|
1131
|
+
var indexInfoRow = z4.object({
|
|
1132
|
+
seqno: sqliteInteger,
|
|
1133
|
+
cid: sqliteInteger,
|
|
1134
|
+
name: z4.string().nullable()
|
|
1135
|
+
});
|
|
1136
|
+
|
|
1052
1137
|
// packages/text2sql/src/lib/adapters/sqlite/info.sqlite.grounding.ts
|
|
1053
1138
|
var SqliteInfoGrounding = class extends InfoGrounding {
|
|
1054
1139
|
#adapter;
|
|
@@ -1058,7 +1143,8 @@ var SqliteInfoGrounding = class extends InfoGrounding {
|
|
|
1058
1143
|
}
|
|
1059
1144
|
async collectInfo() {
|
|
1060
1145
|
const rows = await this.#adapter.runQuery(
|
|
1061
|
-
"SELECT sqlite_version() AS version"
|
|
1146
|
+
"SELECT sqlite_version() AS version",
|
|
1147
|
+
versionRow
|
|
1062
1148
|
);
|
|
1063
1149
|
return {
|
|
1064
1150
|
dialect: "sqlite",
|
|
@@ -1071,6 +1157,8 @@ var SqliteInfoGrounding = class extends InfoGrounding {
|
|
|
1071
1157
|
};
|
|
1072
1158
|
|
|
1073
1159
|
// packages/text2sql/src/lib/adapters/sqlite/row-count.sqlite.grounding.ts
|
|
1160
|
+
import { z as z5 } from "zod";
|
|
1161
|
+
var countRow = z5.object({ count: sqliteInteger });
|
|
1074
1162
|
var SqliteRowCountGrounding = class extends RowCountGrounding {
|
|
1075
1163
|
#adapter;
|
|
1076
1164
|
constructor(adapter, config = {}) {
|
|
@@ -1079,9 +1167,10 @@ var SqliteRowCountGrounding = class extends RowCountGrounding {
|
|
|
1079
1167
|
}
|
|
1080
1168
|
async getRowCount(tableName) {
|
|
1081
1169
|
const rows = await this.#adapter.runQuery(
|
|
1082
|
-
`SELECT COUNT(*) as count FROM ${this.#adapter.quoteIdentifier(tableName)}
|
|
1170
|
+
`SELECT COUNT(*) as count FROM ${this.#adapter.quoteIdentifier(tableName)}`,
|
|
1171
|
+
countRow
|
|
1083
1172
|
);
|
|
1084
|
-
return
|
|
1173
|
+
return rows[0]?.count;
|
|
1085
1174
|
}
|
|
1086
1175
|
};
|
|
1087
1176
|
|
|
@@ -1504,17 +1593,8 @@ var Sqlite = class extends Adapter {
|
|
|
1504
1593
|
#quoteIdentifier(name) {
|
|
1505
1594
|
return `'${name.replace(/'/g, "''")}'`;
|
|
1506
1595
|
}
|
|
1507
|
-
async
|
|
1508
|
-
|
|
1509
|
-
if (Array.isArray(result)) {
|
|
1510
|
-
return result;
|
|
1511
|
-
}
|
|
1512
|
-
if (result && typeof result === "object" && "rows" in result && Array.isArray(result.rows)) {
|
|
1513
|
-
return result.rows;
|
|
1514
|
-
}
|
|
1515
|
-
throw new Error(
|
|
1516
|
-
"Sqlite adapter execute() must return an array of rows or an object with a rows array when introspecting."
|
|
1517
|
-
);
|
|
1596
|
+
async queryRows(sql) {
|
|
1597
|
+
return this.rowsFrom(await this.#options.execute(sql), ["rows"]);
|
|
1518
1598
|
}
|
|
1519
1599
|
quoteIdentifier(name) {
|
|
1520
1600
|
return `"${name.replace(/"/g, '""')}"`;
|
|
@@ -1537,45 +1617,44 @@ var SqliteTableGrounding = class extends TableGrounding {
|
|
|
1537
1617
|
this.#adapter = adapter;
|
|
1538
1618
|
}
|
|
1539
1619
|
async getAllTableNames() {
|
|
1540
|
-
const rows = await this.#adapter.runQuery(
|
|
1541
|
-
|
|
1542
|
-
|
|
1620
|
+
const rows = await this.#adapter.runQuery(
|
|
1621
|
+
`SELECT name FROM sqlite_master WHERE type='table' ORDER BY name`,
|
|
1622
|
+
nameRow
|
|
1543
1623
|
);
|
|
1624
|
+
return rows.map((row) => row.name).filter((name) => !name.startsWith("sqlite_"));
|
|
1544
1625
|
}
|
|
1545
1626
|
async getTable(tableName) {
|
|
1546
1627
|
const columns = await this.#adapter.runQuery(
|
|
1547
|
-
`PRAGMA table_info(${this.#quoteIdentifier(tableName)})
|
|
1628
|
+
`PRAGMA table_info(${this.#quoteIdentifier(tableName)})`,
|
|
1629
|
+
tableInfoRow
|
|
1548
1630
|
);
|
|
1549
1631
|
return {
|
|
1550
1632
|
name: tableName,
|
|
1551
1633
|
rawName: tableName,
|
|
1552
|
-
columns: columns.map((col) => ({
|
|
1553
|
-
name: col.name ?? "unknown",
|
|
1554
|
-
type: col.type ?? "unknown"
|
|
1555
|
-
}))
|
|
1634
|
+
columns: columns.map((col) => ({ name: col.name, type: col.type }))
|
|
1556
1635
|
};
|
|
1557
1636
|
}
|
|
1558
1637
|
async findOutgoingRelations(tableName) {
|
|
1559
1638
|
const rows = await this.#adapter.runQuery(
|
|
1560
|
-
`PRAGMA foreign_key_list(${this.#quoteIdentifier(tableName)})
|
|
1639
|
+
`PRAGMA foreign_key_list(${this.#quoteIdentifier(tableName)})`,
|
|
1640
|
+
foreignKeyListRow
|
|
1561
1641
|
);
|
|
1562
1642
|
const groups = /* @__PURE__ */ new Map();
|
|
1563
1643
|
for (const row of rows) {
|
|
1564
|
-
if (row.
|
|
1644
|
+
if (row.to === null) {
|
|
1565
1645
|
continue;
|
|
1566
1646
|
}
|
|
1567
|
-
const
|
|
1568
|
-
const existing = groups.get(id);
|
|
1647
|
+
const existing = groups.get(row.id);
|
|
1569
1648
|
if (!existing) {
|
|
1570
|
-
groups.set(id, {
|
|
1649
|
+
groups.set(row.id, {
|
|
1571
1650
|
table: tableName,
|
|
1572
|
-
from: [
|
|
1573
|
-
referenced_table:
|
|
1574
|
-
to: [
|
|
1651
|
+
from: [row.from],
|
|
1652
|
+
referenced_table: row.table,
|
|
1653
|
+
to: [row.to]
|
|
1575
1654
|
});
|
|
1576
1655
|
} else {
|
|
1577
|
-
existing.from.push(
|
|
1578
|
-
existing.to.push(
|
|
1656
|
+
existing.from.push(row.from);
|
|
1657
|
+
existing.to.push(row.to);
|
|
1579
1658
|
}
|
|
1580
1659
|
}
|
|
1581
1660
|
return Array.from(groups.values());
|