@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.
Files changed (40) hide show
  1. package/dist/index.js +24 -19
  2. package/dist/index.js.map +3 -3
  3. package/dist/lib/adapters/bigquery/bigquery-fk.d.ts +25 -0
  4. package/dist/lib/adapters/bigquery/bigquery-fk.d.ts.map +1 -0
  5. package/dist/lib/adapters/bigquery/constraint.bigquery.grounding.d.ts +3 -1
  6. package/dist/lib/adapters/bigquery/constraint.bigquery.grounding.d.ts.map +1 -1
  7. package/dist/lib/adapters/bigquery/index.d.ts.map +1 -1
  8. package/dist/lib/adapters/bigquery/index.js +356 -201
  9. package/dist/lib/adapters/bigquery/index.js.map +4 -4
  10. package/dist/lib/adapters/bigquery/indexes.bigquery.grounding.d.ts +6 -7
  11. package/dist/lib/adapters/bigquery/indexes.bigquery.grounding.d.ts.map +1 -1
  12. package/dist/lib/adapters/bigquery/info.bigquery.grounding.d.ts.map +1 -1
  13. package/dist/lib/adapters/bigquery/row-count.bigquery.grounding.d.ts +5 -7
  14. package/dist/lib/adapters/bigquery/row-count.bigquery.grounding.d.ts.map +1 -1
  15. package/dist/lib/adapters/bigquery/table.bigquery.grounding.d.ts +2 -0
  16. package/dist/lib/adapters/bigquery/table.bigquery.grounding.d.ts.map +1 -1
  17. package/dist/lib/adapters/bigquery/view.bigquery.grounding.d.ts +2 -2
  18. package/dist/lib/adapters/bigquery/view.bigquery.grounding.d.ts.map +1 -1
  19. package/dist/lib/adapters/groundings/column-values.grounding.d.ts +4 -0
  20. package/dist/lib/adapters/groundings/column-values.grounding.d.ts.map +1 -1
  21. package/dist/lib/adapters/groundings/context.d.ts +2 -0
  22. package/dist/lib/adapters/groundings/context.d.ts.map +1 -1
  23. package/dist/lib/adapters/groundings/index.js +10 -4
  24. package/dist/lib/adapters/groundings/index.js.map +2 -2
  25. package/dist/lib/adapters/mysql/index.js +10 -4
  26. package/dist/lib/adapters/mysql/index.js.map +2 -2
  27. package/dist/lib/adapters/postgres/index.js +10 -4
  28. package/dist/lib/adapters/postgres/index.js.map +2 -2
  29. package/dist/lib/adapters/spreadsheet/index.js +10 -4
  30. package/dist/lib/adapters/spreadsheet/index.js.map +2 -2
  31. package/dist/lib/adapters/sqlite/index.js +10 -4
  32. package/dist/lib/adapters/sqlite/index.js.map +2 -2
  33. package/dist/lib/adapters/sqlserver/index.js +10 -4
  34. package/dist/lib/adapters/sqlserver/index.js.map +2 -2
  35. package/dist/lib/agents/developer.agent.d.ts +1 -1
  36. package/dist/lib/sql.d.ts +5 -3
  37. package/dist/lib/sql.d.ts.map +1 -1
  38. package/dist/lib/synthesis/index.js +1 -1
  39. package/dist/lib/synthesis/index.js.map +1 -1
  40. 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;