@deepagents/text2sql 0.22.0 → 0.24.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/dist/index.js +24 -19
- package/dist/index.js.map +3 -3
- package/dist/lib/adapters/bigquery/bigquery-fk.d.ts +25 -0
- package/dist/lib/adapters/bigquery/bigquery-fk.d.ts.map +1 -0
- package/dist/lib/adapters/bigquery/constraint.bigquery.grounding.d.ts +3 -1
- package/dist/lib/adapters/bigquery/constraint.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/index.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/index.js +356 -201
- package/dist/lib/adapters/bigquery/index.js.map +4 -4
- package/dist/lib/adapters/bigquery/indexes.bigquery.grounding.d.ts +6 -7
- package/dist/lib/adapters/bigquery/indexes.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/info.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/row-count.bigquery.grounding.d.ts +5 -7
- package/dist/lib/adapters/bigquery/row-count.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/table.bigquery.grounding.d.ts +2 -0
- package/dist/lib/adapters/bigquery/table.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/bigquery/view.bigquery.grounding.d.ts +2 -2
- package/dist/lib/adapters/bigquery/view.bigquery.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/groundings/column-values.grounding.d.ts +4 -0
- package/dist/lib/adapters/groundings/column-values.grounding.d.ts.map +1 -1
- package/dist/lib/adapters/groundings/context.d.ts +2 -0
- package/dist/lib/adapters/groundings/context.d.ts.map +1 -1
- package/dist/lib/adapters/groundings/index.js +10 -4
- package/dist/lib/adapters/groundings/index.js.map +2 -2
- package/dist/lib/adapters/mysql/index.js +10 -4
- package/dist/lib/adapters/mysql/index.js.map +2 -2
- package/dist/lib/adapters/postgres/index.js +10 -4
- package/dist/lib/adapters/postgres/index.js.map +2 -2
- package/dist/lib/adapters/spreadsheet/index.js +10 -4
- package/dist/lib/adapters/spreadsheet/index.js.map +2 -2
- package/dist/lib/adapters/sqlite/index.js +10 -4
- package/dist/lib/adapters/sqlite/index.js.map +2 -2
- package/dist/lib/adapters/sqlserver/index.js +10 -4
- package/dist/lib/adapters/sqlserver/index.js.map +2 -2
- package/dist/lib/agents/developer.agent.d.ts +1 -1
- package/dist/lib/sql.d.ts +5 -3
- package/dist/lib/sql.d.ts.map +1 -1
- package/dist/lib/synthesis/index.js +1 -1
- package/dist/lib/synthesis/index.js.map +1 -1
- package/package.json +8 -6
|
@@ -98,7 +98,8 @@ function createGroundingContext() {
|
|
|
98
98
|
tables: [],
|
|
99
99
|
views: [],
|
|
100
100
|
relationships: [],
|
|
101
|
-
info: void 0
|
|
101
|
+
info: void 0,
|
|
102
|
+
cache: /* @__PURE__ */ new Map()
|
|
102
103
|
};
|
|
103
104
|
}
|
|
104
105
|
|
|
@@ -430,9 +431,11 @@ var InfoGrounding = class extends AbstractGrounding {
|
|
|
430
431
|
// packages/text2sql/src/lib/adapters/groundings/column-values.grounding.ts
|
|
431
432
|
var ColumnValuesGrounding = class extends AbstractGrounding {
|
|
432
433
|
lowCardinalityLimit;
|
|
434
|
+
maxValueLength;
|
|
433
435
|
constructor(config = {}) {
|
|
434
436
|
super("columnValues");
|
|
435
437
|
this.lowCardinalityLimit = config.lowCardinalityLimit ?? 20;
|
|
438
|
+
this.maxValueLength = config.maxValueLength ?? 100;
|
|
436
439
|
}
|
|
437
440
|
/**
|
|
438
441
|
* Get values for native ENUM type columns.
|
|
@@ -537,24 +540,27 @@ var ColumnValuesGrounding = class extends AbstractGrounding {
|
|
|
537
540
|
}
|
|
538
541
|
}
|
|
539
542
|
}
|
|
543
|
+
exceedsMaxValueLength(values) {
|
|
544
|
+
return values.some((v) => v.length > this.maxValueLength);
|
|
545
|
+
}
|
|
540
546
|
/**
|
|
541
547
|
* Resolve column values from all sources in priority order.
|
|
542
548
|
*/
|
|
543
549
|
async resolveColumnValues(tableName, column2, constraints2) {
|
|
544
550
|
const enumValues = await this.collectEnumValues(tableName, column2);
|
|
545
|
-
if (enumValues?.length) {
|
|
551
|
+
if (enumValues?.length && !this.exceedsMaxValueLength(enumValues)) {
|
|
546
552
|
return { kind: "Enum", values: enumValues };
|
|
547
553
|
}
|
|
548
554
|
if (constraints2) {
|
|
549
555
|
for (const constraint2 of constraints2) {
|
|
550
556
|
const checkValues = this.parseCheckConstraint(constraint2, column2.name);
|
|
551
|
-
if (checkValues?.length) {
|
|
557
|
+
if (checkValues?.length && !this.exceedsMaxValueLength(checkValues)) {
|
|
552
558
|
return { kind: "Enum", values: checkValues };
|
|
553
559
|
}
|
|
554
560
|
}
|
|
555
561
|
}
|
|
556
562
|
const lowCardValues = await this.collectLowCardinality(tableName, column2);
|
|
557
|
-
if (lowCardValues?.length) {
|
|
563
|
+
if (lowCardValues?.length && !this.exceedsMaxValueLength(lowCardValues)) {
|
|
558
564
|
return { kind: "LowCardinality", values: lowCardValues };
|
|
559
565
|
}
|
|
560
566
|
return void 0;
|