@deepagents/text2sql 7.0.2 → 7.1.1
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 +63 -75
- package/dist/bin/sql.js +82 -599
- package/dist/eval/src/evals/models.d.ts +12 -0
- package/dist/eval/src/evals/models.d.ts.map +1 -0
- package/dist/eval/src/evals/sql-create-context/sql-create-context.eval.d.ts +2 -0
- package/dist/eval/src/evals/sql-create-context/sql-create-context.eval.d.ts.map +1 -0
- package/dist/index.d.ts +0 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +920 -706
- package/dist/index.js.map +4 -4
- package/dist/lib/adapter-index.d.ts +0 -2
- package/dist/lib/adapter-index.d.ts.map +1 -1
- 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 +117 -38
- 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 -52
- 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 +255 -503
- 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 +5 -4
- 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/question.agent.d.ts.map +1 -1
- package/dist/lib/agents/sql-transform-plugins.d.ts.map +1 -1
- package/dist/lib/agents/suggestions.agents.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 +6 -6
- package/dist/lib/index-cache.d.ts.map +1 -1
- package/dist/lib/sql-command.d.ts +1 -4
- package/dist/lib/sql-command.d.ts.map +1 -1
- package/dist/lib/sql.d.ts +0 -5
- 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/segmented-context-extractor.d.ts.map +1 -1
- package/dist/lib/synthesis/extractors/sql-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/persona-generator.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/dist/scripts/scripts/bigquery-test.d.ts +2 -0
- package/dist/scripts/scripts/bigquery-test.d.ts.map +1 -0
- package/package.json +4 -6
- package/dist/lib/index-lock.d.ts +0 -60
- package/dist/lib/index-lock.d.ts.map +0 -1
- package/dist/lib/prompt.d.ts +0 -2
- package/dist/lib/prompt.d.ts.map +0 -1
|
@@ -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
|
|
|
@@ -1501,20 +1590,8 @@ var Sqlite = class extends Adapter {
|
|
|
1501
1590
|
return JSON.stringify(formatError(sql, error));
|
|
1502
1591
|
}
|
|
1503
1592
|
}
|
|
1504
|
-
|
|
1505
|
-
return
|
|
1506
|
-
}
|
|
1507
|
-
async runQuery(sql) {
|
|
1508
|
-
const result = await this.#options.execute(sql);
|
|
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
|
-
);
|
|
1593
|
+
async queryRows(sql) {
|
|
1594
|
+
return this.rowsFrom(await this.#options.execute(sql), ["rows"]);
|
|
1518
1595
|
}
|
|
1519
1596
|
quoteIdentifier(name) {
|
|
1520
1597
|
return `"${name.replace(/"/g, '""')}"`;
|
|
@@ -1537,45 +1614,44 @@ var SqliteTableGrounding = class extends TableGrounding {
|
|
|
1537
1614
|
this.#adapter = adapter;
|
|
1538
1615
|
}
|
|
1539
1616
|
async getAllTableNames() {
|
|
1540
|
-
const rows = await this.#adapter.runQuery(
|
|
1541
|
-
|
|
1542
|
-
|
|
1617
|
+
const rows = await this.#adapter.runQuery(
|
|
1618
|
+
`SELECT name FROM sqlite_master WHERE type='table' ORDER BY name`,
|
|
1619
|
+
nameRow
|
|
1543
1620
|
);
|
|
1621
|
+
return rows.map((row) => row.name).filter((name) => !name.startsWith("sqlite_"));
|
|
1544
1622
|
}
|
|
1545
1623
|
async getTable(tableName) {
|
|
1546
1624
|
const columns = await this.#adapter.runQuery(
|
|
1547
|
-
`PRAGMA table_info(${this.#quoteIdentifier(tableName)})
|
|
1625
|
+
`PRAGMA table_info(${this.#quoteIdentifier(tableName)})`,
|
|
1626
|
+
tableInfoRow
|
|
1548
1627
|
);
|
|
1549
1628
|
return {
|
|
1550
1629
|
name: tableName,
|
|
1551
1630
|
rawName: tableName,
|
|
1552
|
-
columns: columns.map((col) => ({
|
|
1553
|
-
name: col.name ?? "unknown",
|
|
1554
|
-
type: col.type ?? "unknown"
|
|
1555
|
-
}))
|
|
1631
|
+
columns: columns.map((col) => ({ name: col.name, type: col.type }))
|
|
1556
1632
|
};
|
|
1557
1633
|
}
|
|
1558
1634
|
async findOutgoingRelations(tableName) {
|
|
1559
1635
|
const rows = await this.#adapter.runQuery(
|
|
1560
|
-
`PRAGMA foreign_key_list(${this.#quoteIdentifier(tableName)})
|
|
1636
|
+
`PRAGMA foreign_key_list(${this.#quoteIdentifier(tableName)})`,
|
|
1637
|
+
foreignKeyListRow
|
|
1561
1638
|
);
|
|
1562
1639
|
const groups = /* @__PURE__ */ new Map();
|
|
1563
1640
|
for (const row of rows) {
|
|
1564
|
-
if (row.
|
|
1641
|
+
if (row.to === null) {
|
|
1565
1642
|
continue;
|
|
1566
1643
|
}
|
|
1567
|
-
const
|
|
1568
|
-
const existing = groups.get(id);
|
|
1644
|
+
const existing = groups.get(row.id);
|
|
1569
1645
|
if (!existing) {
|
|
1570
|
-
groups.set(id, {
|
|
1646
|
+
groups.set(row.id, {
|
|
1571
1647
|
table: tableName,
|
|
1572
|
-
from: [
|
|
1573
|
-
referenced_table:
|
|
1574
|
-
to: [
|
|
1648
|
+
from: [row.from],
|
|
1649
|
+
referenced_table: row.table,
|
|
1650
|
+
to: [row.to]
|
|
1575
1651
|
});
|
|
1576
1652
|
} else {
|
|
1577
|
-
existing.from.push(
|
|
1578
|
-
existing.to.push(
|
|
1653
|
+
existing.from.push(row.from);
|
|
1654
|
+
existing.to.push(row.to);
|
|
1579
1655
|
}
|
|
1580
1656
|
}
|
|
1581
1657
|
return Array.from(groups.values());
|
|
@@ -1810,7 +1886,10 @@ function loadData(db, sheet) {
|
|
|
1810
1886
|
}
|
|
1811
1887
|
db.exec("COMMIT");
|
|
1812
1888
|
} catch (error) {
|
|
1813
|
-
|
|
1889
|
+
try {
|
|
1890
|
+
db.exec("ROLLBACK");
|
|
1891
|
+
} catch {
|
|
1892
|
+
}
|
|
1814
1893
|
throw error;
|
|
1815
1894
|
}
|
|
1816
1895
|
}
|