@finos/legend-lego 2.0.201 → 2.0.203

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 (75) hide show
  1. package/lib/index.css +2 -2
  2. package/lib/index.css.map +1 -1
  3. package/lib/legend-ai/LegendAIDocEnrichment.d.ts +34 -1
  4. package/lib/legend-ai/LegendAIDocEnrichment.d.ts.map +1 -1
  5. package/lib/legend-ai/LegendAIDocEnrichment.js +890 -1
  6. package/lib/legend-ai/LegendAIDocEnrichment.js.map +1 -1
  7. package/lib/legend-ai/LegendAIServiceRetrieval.d.ts +36 -0
  8. package/lib/legend-ai/LegendAIServiceRetrieval.d.ts.map +1 -0
  9. package/lib/legend-ai/LegendAIServiceRetrieval.js +179 -0
  10. package/lib/legend-ai/LegendAIServiceRetrieval.js.map +1 -0
  11. package/lib/legend-ai/LegendAITypes.d.ts +62 -2
  12. package/lib/legend-ai/LegendAITypes.d.ts.map +1 -1
  13. package/lib/legend-ai/LegendAITypes.js +56 -30
  14. package/lib/legend-ai/LegendAITypes.js.map +1 -1
  15. package/lib/legend-ai/LegendAI_LegendApplicationPlugin_Extension.d.ts +29 -6
  16. package/lib/legend-ai/LegendAI_LegendApplicationPlugin_Extension.d.ts.map +1 -1
  17. package/lib/legend-ai/LegendAI_LegendApplicationPlugin_Extension.js +9 -0
  18. package/lib/legend-ai/LegendAI_LegendApplicationPlugin_Extension.js.map +1 -1
  19. package/lib/legend-ai/__test-utils__/LegendAITestUtils.d.ts +5 -0
  20. package/lib/legend-ai/__test-utils__/LegendAITestUtils.d.ts.map +1 -1
  21. package/lib/legend-ai/__test-utils__/LegendAITestUtils.js +18 -2
  22. package/lib/legend-ai/__test-utils__/LegendAITestUtils.js.map +1 -1
  23. package/lib/legend-ai/components/LegendAIAnalysisPanel.d.ts.map +1 -1
  24. package/lib/legend-ai/components/LegendAIAnalysisPanel.js +27 -8
  25. package/lib/legend-ai/components/LegendAIAnalysisPanel.js.map +1 -1
  26. package/lib/legend-ai/components/LegendAIAnalysisUtils.d.ts +1 -0
  27. package/lib/legend-ai/components/LegendAIAnalysisUtils.d.ts.map +1 -1
  28. package/lib/legend-ai/components/LegendAIAnalysisUtils.js +144 -22
  29. package/lib/legend-ai/components/LegendAIAnalysisUtils.js.map +1 -1
  30. package/lib/legend-ai/components/LegendAICharts.d.ts +4 -0
  31. package/lib/legend-ai/components/LegendAICharts.d.ts.map +1 -1
  32. package/lib/legend-ai/components/LegendAICharts.js +60 -1
  33. package/lib/legend-ai/components/LegendAICharts.js.map +1 -1
  34. package/lib/legend-ai/components/LegendAIChat.d.ts.map +1 -1
  35. package/lib/legend-ai/components/LegendAIChat.js +46 -17
  36. package/lib/legend-ai/components/LegendAIChat.js.map +1 -1
  37. package/lib/legend-ai/components/LegendAIChatHelpers.d.ts +1 -1
  38. package/lib/legend-ai/components/LegendAIChatHelpers.d.ts.map +1 -1
  39. package/lib/legend-ai/components/LegendAIChatHelpers.js +220 -15
  40. package/lib/legend-ai/components/LegendAIChatHelpers.js.map +1 -1
  41. package/lib/legend-ai/components/LegendAIChatInput.d.ts.map +1 -1
  42. package/lib/legend-ai/components/LegendAIChatInput.js +1 -0
  43. package/lib/legend-ai/components/LegendAIChatInput.js.map +1 -1
  44. package/lib/legend-ai/components/LegendAIScopeSelector.d.ts.map +1 -1
  45. package/lib/legend-ai/components/LegendAIScopeSelector.js +5 -2
  46. package/lib/legend-ai/components/LegendAIScopeSelector.js.map +1 -1
  47. package/lib/legend-ai/index.d.ts +4 -3
  48. package/lib/legend-ai/index.d.ts.map +1 -1
  49. package/lib/legend-ai/index.js +4 -3
  50. package/lib/legend-ai/index.js.map +1 -1
  51. package/lib/legend-ai/stores/LegendAIChatProcessors.d.ts +30 -13
  52. package/lib/legend-ai/stores/LegendAIChatProcessors.d.ts.map +1 -1
  53. package/lib/legend-ai/stores/LegendAIChatProcessors.js +776 -216
  54. package/lib/legend-ai/stores/LegendAIChatProcessors.js.map +1 -1
  55. package/lib/legend-ai/stores/LegendAIChatState.d.ts +2 -2
  56. package/lib/legend-ai/stores/LegendAIChatState.d.ts.map +1 -1
  57. package/lib/legend-ai/stores/LegendAIChatState.js +26 -25
  58. package/lib/legend-ai/stores/LegendAIChatState.js.map +1 -1
  59. package/package.json +3 -3
  60. package/src/legend-ai/LegendAIDocEnrichment.ts +1198 -1
  61. package/src/legend-ai/LegendAIServiceRetrieval.ts +228 -0
  62. package/src/legend-ai/LegendAITypes.ts +138 -32
  63. package/src/legend-ai/LegendAI_LegendApplicationPlugin_Extension.ts +62 -0
  64. package/src/legend-ai/__test-utils__/LegendAITestUtils.ts +28 -0
  65. package/src/legend-ai/components/LegendAIAnalysisPanel.tsx +36 -17
  66. package/src/legend-ai/components/LegendAIAnalysisUtils.ts +178 -22
  67. package/src/legend-ai/components/LegendAICharts.tsx +160 -0
  68. package/src/legend-ai/components/LegendAIChat.tsx +107 -14
  69. package/src/legend-ai/components/LegendAIChatHelpers.ts +334 -20
  70. package/src/legend-ai/components/LegendAIChatInput.tsx +2 -1
  71. package/src/legend-ai/components/LegendAIScopeSelector.tsx +7 -2
  72. package/src/legend-ai/index.ts +22 -1
  73. package/src/legend-ai/stores/LegendAIChatProcessors.ts +1205 -321
  74. package/src/legend-ai/stores/LegendAIChatState.ts +80 -69
  75. package/tsconfig.json +1 -0
@@ -1 +1 @@
1
- {"version":3,"file":"LegendAITypes.js","sourceRoot":"","sources":["../../src/legend-ai/LegendAITypes.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAKL,8BAA8B,EAC9B,kBAAkB,EAClB,0BAA0B,GAC3B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAEpD,OAAO,EACL,0CAA0C,GAE3C,MAAM,iDAAiD,CAAC;AAEzD,MAAM,OAAO,eAAe;IAC1B,IAAI,CAAU;IACd,IAAI,CAAU;IACd,aAAa,CAAU;IACvB,YAAY,CAAU;IACtB,uEAAuE;IACvE,QAAQ,CAAW;IACnB,uEAAuE;IACvE,cAAc,CAAU;CACzB;AAED,MAAM,OAAO,kBAAkB;IAC7B,IAAI,CAAU;IACd,IAAI,CAAU;IACd,QAAQ,CAAW;CACpB;AAgBD,MAAM,CAAN,IAAY,oBAGX;AAHD,WAAY,oBAAoB;IAC9B,2CAAmB,CAAA;IACnB,oDAA4B,CAAA;AAC9B,CAAC,EAHW,oBAAoB,KAApB,oBAAoB,QAG/B;AAED,MAAM,OAAO,gBAAgB;IAC3B,KAAK,CAAU;IACf,WAAW,CAAU;IACrB,OAAO,CAAU;IACjB,OAAO,CAAqB;IAC5B,UAAU,CAAY;IACtB;;;;OAIG;IACH,gBAAgB,CAAwB;IACxC;;;;OAIG;IACH,UAAU,CAAwB;IAClC,0GAA0G;IAC1G,eAAe,CAAU;IACzB;;;;OAIG;IACH,yBAAyB,CAAW;IACpC,mDAAmD;IACnD,qBAAqB,CAAU;IAC/B,iFAAiF;IACjF,SAAS,CAAU;IACnB;;;OAGG;IACH,UAAU,CAAyB;CACpC;AAED,MAAM,OAAO,cAAc;IACzB,OAAO,CAAW;IAClB,aAAa,CAAsB;IACnC,YAAY,CAAsB;IAClC,eAAe,CAAY;IAC3B,eAAe,CAAsB;IACrC,eAAe,CAAsB;IACrC,oBAAoB,CAAsB;IAC1C,SAAS,CAAsB;IAC/B,qBAAqB,CAAU;IAC/B,gBAAgB,CAAU;IAC1B,gGAAgG;IAChG,oBAAoB,CAAU;IAC9B,uDAAuD;IACvD,YAAY,CAAU;IACtB,iEAAiE;IACjE,sBAAsB,CAAU;CACjC;AAED,MAAM,CAAC,MAAM,wBAAwB,GAAmB,MAAM,CAAC,MAAM,CAAC;IACpE,OAAO,EAAE,KAAK;IACd,aAAa,EAAE,SAAS;IACxB,YAAY,EAAE,SAAS;IACvB,eAAe,EAAE,SAAS;IAC1B,eAAe,EAAE,SAAS;IAC1B,oBAAoB,EAAE,SAAS;IAC/B,SAAS,EAAE,SAAS;CACrB,CAAC,CAAC;AAEH,0DAA0D;AAC1D,MAAM,CAAC,MAAM,kBAAkB,GAAG,MAAM,CAAC;AACzC,+CAA+C;AAC/C,MAAM,CAAC,MAAM,kBAAkB,GAAG,WAAW,CAAC;AAC9C,6DAA6D;AAC7D,MAAM,CAAC,MAAM,qBAAqB,GAAG,mBAAmB,CAAC;AAEzD;;;;GAIG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,SAAS,CAAC;AAErD,MAAM,UAAU,WAAW;IACzB,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AAC/C,CAAC;AAED,wEAAwE;AACxE,MAAM,CAAC,MAAM,yCAAyC,GACpD,uBAAuB,CAAC;AAE1B,MAAM,UAAU,kBAAkB,CAChC,OAAkC;IAElC,OAAO,OAAO,CAAC,IAAI,CACjB,CAAC,CAAC,EAAmD,EAAE,CACrD,CAAC,YAAY,0CAA0C,CAC1D,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,gBAAgB;IAC3B,UAAU,CAA8B;IACxC,OAAO,CAAoB;CAC5B;AAED,MAAM,UAAU,wBAAwB,CACtC,OAAiB;IAEjB,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC3B,KAAK,EAAE,GAAG;QACV,UAAU,EAAE,GAAG;QACf,KAAK,EAAE,GAAG;KACX,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,CAAN,IAAY,0BAIX;AAJD,WAAY,0BAA0B;IACpC,+CAAiB,CAAA;IACjB,2CAAa,CAAA;IACb,6CAAe,CAAA;AACjB,CAAC,EAJW,0BAA0B,KAA1B,0BAA0B,QAIrC;AAED,MAAM,OAAO,oBAAoB;IAC/B,EAAE,CAAU;IACZ,KAAK,CAAU;IACf,MAAM,CAA8B;CACrC;AAED,MAAM,CAAN,IAAY,iBAMX;AAND,WAAY,iBAAiB;IAC3B,8CAAyB,CAAA;IACzB,wCAAmB,CAAA;IACnB,4CAAuB,CAAA;IACvB,8CAAyB,CAAA;IACzB,wCAAmB,CAAA;AACrB,CAAC,EANW,iBAAiB,KAAjB,iBAAiB,QAM5B;AAED,MAAM,OAAO,oBAAqB,SAAQ,KAAK;IACpC,IAAI,GAAG,sBAAsB,CAAC;IACvC,SAAS,CAAoB;IAE7B,YAAY,OAAe,EAAE,SAA4B;QACvD,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;CACF;AAED,MAAM,CAAN,IAAY,mBAGX;AAHD,WAAY,mBAAmB;IAC7B,oCAAa,CAAA;IACb,8CAAuB,CAAA;AACzB,CAAC,EAHW,mBAAmB,KAAnB,mBAAmB,QAG9B;AAED,MAAM,OAAO,mBAAmB;IAC9B,EAAE,CAAU;IACZ,IAAI,CAA4B;IAChC,IAAI,CAAU;CACf;AAOD,MAAM,CAAN,IAAY,6BAGX;AAHD,WAAY,6BAA6B;IACvC,wDAAuB,CAAA;IACvB,4DAA2B,CAAA;AAC7B,CAAC,EAHW,6BAA6B,KAA7B,6BAA6B,QAGxC;AAWD,MAAM,OAAO,wBAAwB;IACnC,EAAE,CAAU;IACZ,IAAI,CAAiC;IACrC,aAAa,CAA0B;IACvC,GAAG,CAAiB;IACpB,UAAU,CAAiB;IAC3B,WAAW,CAAiB;IAC5B,QAAQ,CAA2B;IACnC,KAAK,CAAiB;IACtB,SAAS,CAA4B;IACrC,UAAU,CAAiB;IAC3B,QAAQ,CAAiB;IACzB,gBAAgB,CAAiB;IACjC,YAAY,CAAW;IACvB,WAAW,CAAW;IACtB,gBAAgB,CAAY;IAC5B,cAAc,CAAiC;CAChD;AAID,MAAM,OAAO,wBAAwB;IACnC,QAAQ,CAAU;IAClB,GAAG,CAAU;IACb;;;OAGG;IACH,MAAM,CAA0B;IAChC,4DAA4D;IAC5D,QAAQ,CAAU;IAClB,yEAAyE;IACzE,aAAa,CAAU;CACxB;AA8CD,MAAM,OAAO,4BAA4B;IACvC,KAAK,CAAU;IACf,WAAW,CAAU;IACrB,YAAY,CAA6B;CAC1C;AA4BD,MAAM,OAAO,uBAAuB;IAClC,IAAI,CAAU;IACd,WAAW,CAAU;IACrB,WAAW,CAAU;IACrB,gBAAgB,CAA4B;IAC5C,iBAAiB,CAAkC;IACnD,IAAI,CAAqB;IACzB,WAAW,CAAU;IACrB,8EAA8E;IAC9E,wBAAwB,CAAqC;IAC7D,iFAAiF;IACjF,oBAAoB,CAAiC;IACrD,+DAA+D;IAC/D,aAAa,CAAU;CACxB;AAED,MAAM,CAAN,IAAY,sBAIX;AAJD,WAAY,sBAAsB;IAChC,mDAAyB,CAAA;IACzB,+CAAqB,CAAA;IACrB,uDAA6B,CAAA;AAC/B,CAAC,EAJW,sBAAsB,KAAtB,sBAAsB,QAIjC;AAED,MAAM,CAAC,MAAM,wBAAwB,GAAsB,MAAM,CAAC,MAAM,CAAC;IACvE,oHAAoH;IACpH,qHAAqH;IACrH,2FAA2F;IAC3F,6FAA6F;IAC7F,6FAA6F;IAC7F,wEAAwE;IACxE,gEAAgE;IAChE,uGAAuG;IACvG,4EAA4E;IAC5E,iFAAiF;IACjF,gFAAgF;IAChF,iFAAiF;IACjF,iDAAiD;IACjD,4CAA4C;IAC5C,qDAAqD;IACrD,yCAAyC;IACzC,gCAAgC;IAChC,mEAAmE;IACnE,uDAAuD;IACvD,+BAA+B;IAC/B,4DAA4D;IAC5D,yCAAyC;IACzC,qEAAqE;IACrE,mFAAmF;IACnF,0GAA0G;IAC1G,0EAA0E;IAC1E,8FAA8F;CAC/F,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0BAA0B,GAAsB,MAAM,CAAC,MAAM,CAAC;IACzE,6EAA6E;IAC7E,sDAAsD;IACtD,2DAA2D;IAC3D,kEAAkE;IAClE,qDAAqD;IACrD,+BAA+B;IAC/B,kFAAkF;IAClF,yCAAyC;IACzC,sEAAsE;IACtE,+EAA+E;IAC/E,mCAAmC;IACnC,6BAA6B;IAC7B,kGAAkG;IAClG,qFAAqF;IACrF,+CAA+C;IAC/C,0DAA0D;IAC1D,gDAAgD;IAChD,gBAAgB;IAChB,6DAA6D;IAC7D,kCAAkC;IAClC,mHAAmH;IACnH,uFAAuF;IACvF,uFAAuF;IACvF,sEAAsE;IACtE,sEAAsE;CACvE,CAAC,CAAC;AAEH,MAAM,mCAAmC,GAAsB;IAC7D,wEAAwE;IACxE,sFAAsF;IACtF,kDAAkD;IAClD,gEAAgE;CACjE,CAAC;AAEF,MAAM,yBAAyB,GAC7B,sEAAsE,CAAC;AAEzE,MAAM,0BAA0B,GAC9B,qFAAqF,CAAC;AAExF,MAAM,6BAA6B,GAAsB;IACvD,wBAAwB;IACxB,0CAA0C;IAC1C,yBAAyB;CAC1B,CAAC;AAEF,SAAS,mBAAmB,CAC1B,QAAgB,EAChB,QAA2B;IAE3B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,KAAK,EAAE,CAAC;QACV,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,4BAA4B;IACvC,MAAM,CAA0B;IAChC,SAAS,CAAU;IACnB,SAAS,CAAU;IACnB,SAAS,CAAW;CACrB;AAED;;;;GAIG;AACH,MAAM,UAAU,0BAA0B,CACxC,QAAgB,EAChB,WAAoB;IAEpB,MAAM,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IAExC,IAAI,mCAAmC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,QAAQ;YACvC,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IAED,MAAM,SAAS,GAAG,mBAAmB,CAAC,CAAC,EAAE,wBAAwB,CAAC,CAAC;IACnE,MAAM,SAAS,GAAG,mBAAmB,CAAC,CAAC,EAAE,0BAA0B,CAAC,CAAC;IAErE,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;QACrC,MAAM,YAAY,GAChB,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,0BAA0B,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1E,MAAM,qBAAqB,GAAG,6BAA6B,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACrE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CACV,CAAC;QACF,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,QAAQ;YACvC,SAAS;YACT,SAAS;YACT,SAAS,EAAE,WAAW,IAAI,CAAC,YAAY,IAAI,CAAC,qBAAqB;SAClE,CAAC;IACJ,CAAC;IACD,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,UAAU;YACzC,SAAS;YACT,SAAS;YACT,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IAED,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;QACnC,IAAI,SAAS,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAC/B,OAAO;gBACL,MAAM,EAAE,sBAAsB,CAAC,UAAU;gBACzC,SAAS;gBACT,SAAS;gBACT,SAAS,EAAE,KAAK;aACjB,CAAC;QACJ,CAAC;QACD,IACE,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC;YACjC,0BAA0B,CAAC,IAAI,CAAC,CAAC,CAAC,EAClC,CAAC;YACD,OAAO;gBACL,MAAM,EAAE,sBAAsB,CAAC,QAAQ;gBACvC,SAAS;gBACT,SAAS;gBACT,SAAS,EAAE,KAAK;aACjB,CAAC;QACJ,CAAC;QACD,MAAM,SAAS,GACb,SAAS,GAAG,SAAS;YACnB,CAAC,CAAC,sBAAsB,CAAC,QAAQ;YACjC,CAAC,CAAC,sBAAsB,CAAC,UAAU,CAAC;QACxC,OAAO;YACL,MAAM,EAAE,SAAS;YACjB,SAAS;YACT,SAAS;YACT,SAAS,EAAE,IAAI;SAChB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,MAAM,EAAE,WAAW;YACjB,CAAC,CAAC,sBAAsB,CAAC,UAAU;YACnC,CAAC,CAAC,sBAAsB,CAAC,QAAQ;QACnC,SAAS;QACT,SAAS;QACT,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CACpC,QAAgB,EAChB,WAAoB;IAEpB,OAAO,0BAA0B,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,MAAM,CAAC;AAClE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,KAAa,EACb,YAAsC,EACtC,iBAAoC;IAMpC,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAC7D,MAAM,cAAc,GAAG,8BAA8B,CACnD,SAAS,EACT,iBAAiB,CAClB,CAAC,MAAM,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAC3C,OAAO;YACL,UAAU,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YAC7C,gBAAgB,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACzC,MAAM,MAAM,GAAuB,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBACpD,MAAM,QAAQ,GAAG,CAAC,CAAC,WAAW,EAAE,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC;gBAC1D,IAAI,QAAQ,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC;gBACrD,CAAC;gBACD,IAAI,CAAC,CAAC,YAAY,CAAC,UAAU,GAAG,CAAC,EAAE,CAAC;oBAClC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;gBACzB,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YACF,yBAAyB,EAAE,KAAK;SACjC,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;YACL,UAAU,EAAE,EAAE;YACd,gBAAgB,EAAE,EAAE;YACpB,yBAAyB,EAAE,IAAI;SAChC,CAAC;IACJ,CAAC;AACH,CAAC"}
1
+ {"version":3,"file":"LegendAITypes.js","sourceRoot":"","sources":["../../src/legend-ai/LegendAITypes.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAKL,8BAA8B,EAC9B,kBAAkB,EAClB,0BAA0B,GAC3B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAEpD,OAAO,EACL,0CAA0C,GAE3C,MAAM,iDAAiD,CAAC;AAEzD,MAAM,OAAO,eAAe;IAC1B,IAAI,CAAU;IACd,IAAI,CAAU;IACd,aAAa,CAAU;IACvB,YAAY,CAAU;IACtB,uEAAuE;IACvE,QAAQ,CAAW;IACnB,uEAAuE;IACvE,cAAc,CAAU;CACzB;AAED,MAAM,OAAO,kBAAkB;IAC7B,IAAI,CAAU;IACd,IAAI,CAAU;IACd,QAAQ,CAAW;CACpB;AAgBD,MAAM,CAAN,IAAY,oBAGX;AAHD,WAAY,oBAAoB;IAC9B,2CAAmB,CAAA;IACnB,oDAA4B,CAAA;AAC9B,CAAC,EAHW,oBAAoB,KAApB,oBAAoB,QAG/B;AAED,MAAM,OAAO,gBAAgB;IAC3B,KAAK,CAAU;IACf,WAAW,CAAU;IACrB,OAAO,CAAU;IACjB,OAAO,CAAqB;IAC5B,UAAU,CAAY;IACtB;;;;OAIG;IACH,gBAAgB,CAAwB;IACxC;;;;OAIG;IACH,UAAU,CAAwB;IAClC,0GAA0G;IAC1G,eAAe,CAAU;IACzB;;;;OAIG;IACH,yBAAyB,CAAW;IACpC,mDAAmD;IACnD,qBAAqB,CAAU;IAC/B,iFAAiF;IACjF,SAAS,CAAU;IACnB;;;OAGG;IACH,UAAU,CAAyB;CACpC;AAED,MAAM,OAAO,cAAc;IACzB,OAAO,CAAW;IAClB,aAAa,CAAsB;IACnC,YAAY,CAAsB;IAClC,eAAe,CAAY;IAC3B,eAAe,CAAsB;IACrC,eAAe,CAAsB;IACrC,oBAAoB,CAAsB;IAC1C,SAAS,CAAsB;IAC/B,qBAAqB,CAAU;IAC/B,gBAAgB,CAAU;IAC1B,gGAAgG;IAChG,oBAAoB,CAAU;IAC9B,uDAAuD;IACvD,YAAY,CAAU;IACtB,iEAAiE;IACjE,sBAAsB,CAAU;CACjC;AAED,MAAM,CAAC,MAAM,wBAAwB,GAAmB,MAAM,CAAC,MAAM,CAAC;IACpE,OAAO,EAAE,KAAK;IACd,aAAa,EAAE,SAAS;IACxB,YAAY,EAAE,SAAS;IACvB,eAAe,EAAE,SAAS;IAC1B,eAAe,EAAE,SAAS;IAC1B,oBAAoB,EAAE,SAAS;IAC/B,SAAS,EAAE,SAAS;CACrB,CAAC,CAAC;AAEH,0DAA0D;AAC1D,MAAM,CAAC,MAAM,kBAAkB,GAAG,MAAM,CAAC;AACzC,+CAA+C;AAC/C,MAAM,CAAC,MAAM,kBAAkB,GAAG,WAAW,CAAC;AAC9C,6DAA6D;AAC7D,MAAM,CAAC,MAAM,qBAAqB,GAAG,mBAAmB,CAAC;AAEzD;;;;GAIG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,SAAS,CAAC;AAErD,MAAM,UAAU,WAAW;IACzB,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AAC/C,CAAC;AAED,wEAAwE;AACxE,MAAM,CAAC,MAAM,yCAAyC,GACpD,uBAAuB,CAAC;AAE1B,MAAM,UAAU,kBAAkB,CAChC,OAAkC;IAElC,OAAO,OAAO,CAAC,IAAI,CACjB,CAAC,CAAC,EAAmD,EAAE,CACrD,CAAC,YAAY,0CAA0C,CAC1D,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,gBAAgB;IAC3B,UAAU,CAA8B;IACxC,OAAO,CAAoB;CAC5B;AAED,MAAM,UAAU,wBAAwB,CACtC,OAAiB;IAEjB,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC3B,KAAK,EAAE,GAAG;QACV,UAAU,EAAE,GAAG;QACf,KAAK,EAAE,GAAG;KACX,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,CAAN,IAAY,0BAIX;AAJD,WAAY,0BAA0B;IACpC,+CAAiB,CAAA;IACjB,2CAAa,CAAA;IACb,6CAAe,CAAA;AACjB,CAAC,EAJW,0BAA0B,KAA1B,0BAA0B,QAIrC;AAED,MAAM,OAAO,oBAAoB;IAC/B,EAAE,CAAU;IACZ,KAAK,CAAU;IACf,MAAM,CAA8B;CACrC;AAED,MAAM,CAAN,IAAY,iBAMX;AAND,WAAY,iBAAiB;IAC3B,8CAAyB,CAAA;IACzB,wCAAmB,CAAA;IACnB,4CAAuB,CAAA;IACvB,8CAAyB,CAAA;IACzB,wCAAmB,CAAA;AACrB,CAAC,EANW,iBAAiB,KAAjB,iBAAiB,QAM5B;AAED,MAAM,OAAO,oBAAqB,SAAQ,KAAK;IACpC,IAAI,GAAG,sBAAsB,CAAC;IACvC,SAAS,CAAoB;IAE7B,YAAY,OAAe,EAAE,SAA4B;QACvD,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;CACF;AAED,MAAM,OAAO,mCAAoC,SAAQ,oBAAoB;IAClE,IAAI,GAAG,qCAAqC,CAAC;IAEtD,YAAY,IAAY;QACtB,KAAK,CAAC,IAAI,EAAE,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;CACF;AAED,MAAM,CAAN,IAAY,mBAGX;AAHD,WAAY,mBAAmB;IAC7B,oCAAa,CAAA;IACb,8CAAuB,CAAA;AACzB,CAAC,EAHW,mBAAmB,KAAnB,mBAAmB,QAG9B;AAED,MAAM,OAAO,mBAAmB;IAC9B,EAAE,CAAU;IACZ,IAAI,CAA4B;IAChC,IAAI,CAAU;CACf;AAOD,MAAM,CAAN,IAAY,6BAGX;AAHD,WAAY,6BAA6B;IACvC,wDAAuB,CAAA;IACvB,4DAA2B,CAAA;AAC7B,CAAC,EAHW,6BAA6B,KAA7B,6BAA6B,QAGxC;AAWD,MAAM,OAAO,wBAAwB;IACnC,EAAE,CAAU;IACZ,IAAI,CAAiC;IACrC,aAAa,CAA0B;IACvC,GAAG,CAAiB;IACpB,UAAU,CAAiB;IAC3B,WAAW,CAAiB;IAC5B,QAAQ,CAA2B;IACnC,KAAK,CAAiB;IACtB,SAAS,CAA4B;IACrC,UAAU,CAAiB;IAC3B,QAAQ,CAAiB;IACzB,gBAAgB,CAAiB;IACjC,YAAY,CAAW;IACvB,WAAW,CAAW;IACtB,gBAAgB,CAAY;IAC5B,cAAc,CAAiC;IAC/C,wBAAwB,CAAY;CACrC;AAID,MAAM,OAAO,wBAAwB;IACnC,QAAQ,CAAU;IAClB,GAAG,CAAU;IACb;;;OAGG;IACH,MAAM,CAA0B;IAChC,4DAA4D;IAC5D,QAAQ,CAAU;IAClB,yEAAyE;IACzE,aAAa,CAAU;CACxB;AA8CD,MAAM,OAAO,4BAA4B;IACvC,KAAK,CAAU;IACf,WAAW,CAAU;IACrB,YAAY,CAA6B;CAC1C;AA4BD,MAAM,OAAO,uBAAuB;IAClC,IAAI,CAAU;IACd,WAAW,CAAU;IACrB,WAAW,CAAU;IACrB,gBAAgB,CAA4B;IAC5C,iBAAiB,CAAkC;IACnD,IAAI,CAAqB;IACzB,WAAW,CAAU;IACrB,8EAA8E;IAC9E,wBAAwB,CAAqC;IAC7D,iFAAiF;IACjF,oBAAoB,CAAiC;IACrD,+DAA+D;IAC/D,aAAa,CAAU;CACxB;AAkED,MAAM,CAAN,IAAY,sBAIX;AAJD,WAAY,sBAAsB;IAChC,mDAAyB,CAAA;IACzB,+CAAqB,CAAA;IACrB,uDAA6B,CAAA;AAC/B,CAAC,EAJW,sBAAsB,KAAtB,sBAAsB,QAIjC;AAED,MAAM,CAAC,MAAM,wBAAwB,GAAsB,MAAM,CAAC,MAAM,CAAC;IACvE,oHAAoH;IACpH,qHAAqH;IACrH,2FAA2F;IAC3F,6FAA6F;IAC7F,6FAA6F;IAC7F,wEAAwE;IACxE,gEAAgE;IAChE,uGAAuG;IACvG,4EAA4E;IAC5E,iFAAiF;IACjF,gFAAgF;IAChF,iFAAiF;IACjF,iDAAiD;IACjD,4CAA4C;IAC5C,qDAAqD;IACrD,yCAAyC;IACzC,gCAAgC;IAChC,mEAAmE;IACnE,uDAAuD;IACvD,+BAA+B;IAC/B,4DAA4D;IAC5D,yCAAyC;IACzC,qEAAqE;IACrE,mFAAmF;IACnF,0GAA0G;IAC1G,0EAA0E;IAC1E,8FAA8F;CAC/F,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0BAA0B,GAAsB,MAAM,CAAC,MAAM,CAAC;IACzE,6EAA6E;IAC7E,sDAAsD;IACtD,2DAA2D;IAC3D,kEAAkE;IAClE,qDAAqD;IACrD,+BAA+B;IAC/B,kFAAkF;IAClF,yCAAyC;IACzC,sEAAsE;IACtE,+EAA+E;IAC/E,mCAAmC;IACnC,6BAA6B;IAC7B,kGAAkG;IAClG,qFAAqF;IACrF,+CAA+C;IAC/C,0DAA0D;IAC1D,gDAAgD;IAChD,gBAAgB;IAChB,6DAA6D;IAC7D,kCAAkC;IAClC,mHAAmH;IACnH,uFAAuF;IACvF,uFAAuF;IACvF,sEAAsE;IACtE,sEAAsE;CACvE,CAAC,CAAC;AAEH,MAAM,mCAAmC,GAAsB;IAC7D,wEAAwE;IACxE,sFAAsF;IACtF,kDAAkD;IAClD,gEAAgE;CACjE,CAAC;AAEF,MAAM,yBAAyB,GAC7B,sEAAsE,CAAC;AAEzE,MAAM,0BAA0B,GAC9B,qFAAqF,CAAC;AAExF,MAAM,6BAA6B,GAAsB;IACvD,wBAAwB;IACxB,0CAA0C;IAC1C,yBAAyB;CAC1B,CAAC;AAEF,SAAS,mBAAmB,CAC1B,QAAgB,EAChB,QAA2B;IAE3B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,KAAK,EAAE,CAAC;QACV,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,4BAA4B;IACvC,MAAM,CAA0B;IAChC,SAAS,CAAU;IACnB,SAAS,CAAU;IACnB,SAAS,CAAW;CACrB;AAED,SAAS,eAAe,CAAC,QAAgB,EAAE,WAAqB;IAC9D,MAAM,OAAO,GAAG,WAAW;SACxB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;SACjC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACZ,IAAI,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,qBAAqB,EAAE,MAAM,CAAC,GAAG,CAAA,KAAK,CAAC,CACtE,CAAC;IACJ,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO,QAAQ,CAAC,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;AACxE,CAAC;AAED,SAAS,mBAAmB,CAC1B,CAAS,EACT,SAAiB,EACjB,SAAiB;IAEjB,IAAI,SAAS,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;QAC/B,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,UAAU;YACzC,SAAS;YACT,SAAS;YACT,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IACD,IAAI,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,0BAA0B,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5E,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,QAAQ;YACvC,SAAS;YACT,SAAS;YACT,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IACD,IAAI,6BAA6B,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACzD,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,QAAQ;YACvC,SAAS;YACT,SAAS;YACT,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IACD,OAAO;QACL,MAAM,EACJ,SAAS,GAAG,SAAS;YACnB,CAAC,CAAC,sBAAsB,CAAC,QAAQ;YACjC,CAAC,CAAC,sBAAsB,CAAC,UAAU;QACvC,SAAS;QACT,SAAS;QACT,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,0BAA0B,CACxC,QAAgB,EAChB,WAAoB,EACpB,WAAsB;IAEtB,MAAM,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IAExC,IAAI,mCAAmC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,QAAQ;YACvC,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IAED,MAAM,SAAS,GAAG,mBAAmB,CAAC,CAAC,EAAE,wBAAwB,CAAC,CAAC;IACnE,MAAM,aAAa,GACjB,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9E,MAAM,SAAS,GAAG,mBAAmB,CACnC,aAAa,EACb,0BAA0B,CAC3B,CAAC;IAEF,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;QACrC,MAAM,YAAY,GAChB,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,0BAA0B,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1E,MAAM,qBAAqB,GAAG,6BAA6B,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACrE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CACV,CAAC;QACF,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,QAAQ;YACvC,SAAS;YACT,SAAS;YACT,SAAS,EAAE,WAAW,IAAI,CAAC,YAAY,IAAI,CAAC,qBAAqB;SAClE,CAAC;IACJ,CAAC;IACD,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,UAAU;YACzC,SAAS;YACT,SAAS;YACT,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IACD,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;QACnC,OAAO,mBAAmB,CAAC,CAAC,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IACtD,CAAC;IAED,OAAO;QACL,MAAM,EAAE,WAAW;YACjB,CAAC,CAAC,sBAAsB,CAAC,UAAU;YACnC,CAAC,CAAC,sBAAsB,CAAC,QAAQ;QACnC,SAAS;QACT,SAAS;QACT,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CACpC,QAAgB,EAChB,WAAoB,EACpB,WAAsB;IAEtB,OAAO,0BAA0B,CAAC,QAAQ,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC,MAAM,CAAC;AAC/E,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,KAAa,EACb,YAAsC,EACtC,iBAAoC;IAMpC,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAC7D,MAAM,cAAc,GAAG,8BAA8B,CACnD,SAAS,EACT,iBAAiB,CAClB,CAAC,MAAM,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAC3C,OAAO;YACL,UAAU,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YAC7C,gBAAgB,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACzC,MAAM,MAAM,GAAuB,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBACpD,MAAM,QAAQ,GAAG,CAAC,CAAC,WAAW,EAAE,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC;gBAC1D,IAAI,QAAQ,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC;gBACrD,CAAC;gBACD,IAAI,CAAC,CAAC,YAAY,CAAC,UAAU,GAAG,CAAC,EAAE,CAAC;oBAClC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;gBACzB,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YACF,yBAAyB,EAAE,KAAK;SACjC,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;YACL,UAAU,EAAE,EAAE;YACd,gBAAgB,EAAE,EAAE;YACpB,yBAAyB,EAAE,IAAI;SAChC,CAAC;IACJ,CAAC;AACH,CAAC"}
@@ -15,7 +15,7 @@
15
15
  */
16
16
  import { LegendApplicationPlugin, type LegendApplicationPluginManager } from '@finos/legend-application';
17
17
  import type { QueryExplicitExecutionContextInfo, TDSRowDataType } from '@finos/legend-graph';
18
- import type { TDSServiceSchema, LegendAIConfig, LegendAIConversationTurn, LegendAIProductMetadata, LegendAIQuestionIntent } from './LegendAITypes.js';
18
+ import type { TDSServiceSchema, LegendAIConfig, LegendAIConversationTurn, LegendAIProductMetadata, LegendAIQuestionIntent, LegendAIModelContext } from './LegendAITypes.js';
19
19
  export declare class LegendAISqlExtractionResult {
20
20
  sql: string | null;
21
21
  failure: string | null;
@@ -44,6 +44,26 @@ export interface LegendAISemanticSearchResolutionDetails {
44
44
  data_product_coordinates: LegendAIOrchestratorDataProductCoordinates;
45
45
  root_entity: string;
46
46
  related_entities: string[];
47
+ enriched_business_context?: LegendAIEnrichedBusinessContext;
48
+ }
49
+ export interface LegendAIEnrichedBusinessContext {
50
+ naturalLanguageQuery?: string;
51
+ businessContextMatch?: LegendAIBusinessContextMatch;
52
+ }
53
+ export interface LegendAIBusinessContextMatch {
54
+ properties?: LegendAIBusinessContextProperty[];
55
+ additionalNlModelContext?: LegendAIAdditionalNlModelContext[];
56
+ }
57
+ export interface LegendAIBusinessContextProperty {
58
+ propertyName: string;
59
+ matchType?: string[];
60
+ propertyValues?: string[];
61
+ probablePropertyValues?: string[];
62
+ }
63
+ export interface LegendAIAdditionalNlModelContext {
64
+ id: string;
65
+ description: string;
66
+ category: string;
47
67
  }
48
68
  export declare class LegendAIOrchestratorRequest {
49
69
  user_question: string;
@@ -115,11 +135,11 @@ export declare abstract class LegendAI_LegendApplicationPlugin_Extension extends
115
135
  /**
116
136
  * Build the LLM prompt for answering a metadata question about the product.
117
137
  */
118
- abstract buildMetadataPrompt(question: string, metadata: LegendAIProductMetadata, history?: LegendAIConversationTurn[]): string;
138
+ abstract buildMetadataPrompt(question: string, metadata: LegendAIProductMetadata, history?: LegendAIConversationTurn[], services?: TDSServiceSchema[], modelContextEnrichment?: string): string;
119
139
  /**
120
140
  * Build the LLM prompt for generating a SQL query from the user question.
121
141
  */
122
- abstract buildGeneratorPrompt(question: string, services: TDSServiceSchema[], coordinates: string, history?: LegendAIConversationTurn[], metadata?: LegendAIProductMetadata): string;
142
+ abstract buildGeneratorPrompt(question: string, services: TDSServiceSchema[], coordinates: string, history?: LegendAIConversationTurn[], metadata?: LegendAIProductMetadata, modelContextEnrichment?: string): string;
123
143
  /**
124
144
  * Build the LLM prompt for verifying and correcting a generated SQL query.
125
145
  */
@@ -129,12 +149,15 @@ export declare abstract class LegendAI_LegendApplicationPlugin_Extension extends
129
149
  * access points. Unlike service-based prompts this uses `p()` syntax,
130
150
  * has no coordinates or parameters, and focuses on lakehouse execution.
131
151
  */
132
- abstract buildAccessPointGeneratorPrompt(question: string, accessPoints: TDSServiceSchema[], history?: LegendAIConversationTurn[]): string;
152
+ abstract buildAccessPointGeneratorPrompt(question: string, accessPoints: TDSServiceSchema[], history?: LegendAIConversationTurn[], modelContextEnrichment?: string): string;
133
153
  /**
134
154
  * Build the LLM prompt for verifying and correcting a SQL query that
135
155
  * targets data product access points using `p()` syntax.
136
156
  */
137
- abstract buildAccessPointJudgePrompt(sql: string, question: string, accessPoints: TDSServiceSchema[], history?: LegendAIConversationTurn[]): string;
157
+ abstract buildAccessPointJudgePrompt(sql: string, question: string, accessPoints: TDSServiceSchema[], history?: LegendAIConversationTurn[], modelContextEnrichment?: string): string;
158
+ buildAccessPointErrorCorrectionPrompt(failedSql: string, question: string, executionError: string, accessPoints: TDSServiceSchema[], history?: LegendAIConversationTurn[], modelContextEnrichment?: string): string;
159
+ normalizeQuestion(question: string, _config: LegendAIConfig): Promise<string>;
160
+ preWarmSchemaAnalysis(_services: TDSServiceSchema[], _config: LegendAIConfig): void;
138
161
  /**
139
162
  * Send a prompt to the LLM service and return the raw response text.
140
163
  * The plugin manages conversation lifecycle internally — callers
@@ -162,7 +185,7 @@ export declare abstract class LegendAI_LegendApplicationPlugin_Extension extends
162
185
  * Resolve root and related entities for a given user query within a data product.
163
186
  * Uses the dataset search API to determine which entities are most relevant.
164
187
  */
165
- abstract resolveEntitiesForQuery(query: string, dataProductCoordinates: LegendAIOrchestratorDataProductCoordinates, config: LegendAIConfig, pureExecutionContext?: QueryExplicitExecutionContextInfo): Promise<LegendAIResolvedEntities>;
188
+ abstract resolveEntitiesForQuery(query: string, dataProductCoordinates: LegendAIOrchestratorDataProductCoordinates, config: LegendAIConfig, pureExecutionContext?: QueryExplicitExecutionContextInfo, modelContext?: LegendAIModelContext): Promise<LegendAIResolvedEntities>;
166
189
  /**
167
190
  * Execute a Pure/Legend query against the engine and return tabular results.
168
191
  * Converts the Pure expression text to a lambda, then executes it using
@@ -1 +1 @@
1
- {"version":3,"file":"LegendAI_LegendApplicationPlugin_Extension.d.ts","sourceRoot":"","sources":["../../src/legend-ai/LegendAI_LegendApplicationPlugin_Extension.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,uBAAuB,EACvB,KAAK,8BAA8B,EACpC,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EACV,iCAAiC,EACjC,cAAc,EACf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EACV,gBAAgB,EAChB,cAAc,EACd,wBAAwB,EACxB,uBAAuB,EACvB,sBAAsB,EACvB,MAAM,oBAAoB,CAAC;AAE5B,qBAAa,2BAA2B;IACtC,GAAG,EAAG,MAAM,GAAG,IAAI,CAAC;IACpB,OAAO,EAAG,MAAM,GAAG,IAAI,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,oBAAY,oBAAoB;IAC9B,IAAI,SAAS;IACb,IAAI,SAAS;CACd;AAED,qBAAa,mBAAmB;IAC9B,OAAO,EAAG,oBAAoB,CAAC;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,8BAA8B;IACzC,OAAO,EAAG,MAAM,EAAE,CAAC;IACnB,IAAI,EAAG,cAAc,EAAE,CAAC;CACzB;AAED,qBAAa,0CAA0C;IACrD,YAAY,EAAG,MAAM,CAAC;IACtB,QAAQ,EAAG,MAAM,CAAC;IAClB,WAAW,EAAG,MAAM,CAAC;IACrB,OAAO,EAAG,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,uCAAuC;IACtD,wBAAwB,EAAE,0CAA0C,CAAC;IACrE,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED,qBAAa,2BAA2B;IACtC,aAAa,EAAG,MAAM,CAAC;IACvB,kCAAkC,EAAG,uCAAuC,CAAC;CAC9E;AAED,qBAAa,4BAA4B;IACvC,YAAY,EAAG,MAAM,CAAC;CACvB;AAED,qBAAa,wBAAwB;IACnC,UAAU,EAAG,MAAM,CAAC;IACpB,eAAe,EAAG,MAAM,EAAE,CAAC;CAC5B;AAED,qBAAa,uBAAuB;IAClC,WAAW,EAAG,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,EAAG,MAAM,CAAC;IACnB,eAAe,EAAG,MAAM,CAAC;CAC1B;AAED,oBAAY,iBAAiB;IAC3B,GAAG,QAAQ;IACX,IAAI,SAAS;IACb,GAAG,QAAQ;IACX,KAAK,UAAU;IACf,IAAI,SAAS;CACd;AAED,qBAAa,2BAA2B;IACtC,SAAS,EAAG,iBAAiB,CAAC;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAG,MAAM,CAAC;CACpB;AAED,qBAAa,iBAAiB;IAC5B,KAAK,EAAG,MAAM,CAAC;IACf,KAAK,EAAG,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,sBAAsB;IACjC,KAAK,EAAG,MAAM,CAAC;IACf,KAAK,EAAG,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,sBAAsB;IACjC,OAAO,EAAG,MAAM,CAAC;IACjB,mBAAmB,CAAC,EAAE,2BAA2B,CAAC;IAClD,UAAU,EAAG,iBAAiB,EAAE,CAAC;IACjC,SAAS,EAAG,sBAAsB,EAAE,CAAC;IACrC,gBAAgB,EAAG,MAAM,EAAE,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,MAAM,4BAA4B,GAAG,iCAAiC,CAAC;AAE7E,8BAAsB,0CAA2C,SAAQ,uBAAuB;IAC9F;;;OAGG;IACH,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAgD;IAEnF,OAAO,CACL,aAAa,EAAE,8BAA8B,CAAC,uBAAuB,CAAC,GACrE,IAAI;IAIP;;;;;OAKG;IACH,QAAQ,CAAC,sBAAsB,CAC7B,QAAQ,EAAE,MAAM,EAChB,WAAW,EAAE,OAAO,EACpB,MAAM,EAAE,cAAc,EACtB,YAAY,CAAC,EAAE,MAAM,EAAE,GACtB,OAAO,CAAC,sBAAsB,CAAC;IAElC;;OAEG;IACH,QAAQ,CAAC,mBAAmB,CAC1B,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,uBAAuB,EACjC,OAAO,CAAC,EAAE,wBAAwB,EAAE,GACnC,MAAM;IAET;;OAEG;IACH,QAAQ,CAAC,oBAAoB,CAC3B,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,wBAAwB,EAAE,EACpC,QAAQ,CAAC,EAAE,uBAAuB,GACjC,MAAM;IAET;;OAEG;IACH,QAAQ,CAAC,gBAAgB,CACvB,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,wBAAwB,EAAE,GACnC,MAAM;IAET;;;;OAIG;IACH,QAAQ,CAAC,+BAA+B,CACtC,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,gBAAgB,EAAE,EAChC,OAAO,CAAC,EAAE,wBAAwB,EAAE,GACnC,MAAM;IAET;;;OAGG;IACH,QAAQ,CAAC,2BAA2B,CAClC,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,gBAAgB,EAAE,EAChC,OAAO,CAAC,EAAE,wBAAwB,EAAE,GACnC,MAAM;IAET;;;;OAIG;IACH,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;IAEzE;;OAEG;IACH,QAAQ,CAAC,UAAU,CACjB,GAAG,EAAE,MAAM,EACX,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,8BAA8B,CAAC;IAE1C;;OAEG;IACH,QAAQ,CAAC,sBAAsB,CAC7B,UAAU,EAAE,MAAM,GACjB,2BAA2B;IAE9B;;OAEG;IACH,QAAQ,CAAC,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,mBAAmB;IAEpE;;;OAGG;IACH,QAAQ,CAAC,4BAA4B,CACnC,OAAO,EAAE,2BAA2B,EACpC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,4BAA4B,CAAC;IAExC;;;OAGG;IACH,QAAQ,CAAC,uBAAuB,CAC9B,KAAK,EAAE,MAAM,EACb,sBAAsB,EAAE,0CAA0C,EAClE,MAAM,EAAE,cAAc,EACtB,oBAAoB,CAAC,EAAE,iCAAiC,GACvD,OAAO,CAAC,wBAAwB,CAAC;IAEpC;;;;OAIG;IACH,QAAQ,CAAC,gBAAgB,CACvB,SAAS,EAAE,MAAM,EACjB,gBAAgB,EAAE,iCAAiC,EACnD,sBAAsB,EAAE,0CAA0C,EAClE,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,8BAA8B,CAAC;IAE1C;;;OAGG;IACH,QAAQ,CAAC,mBAAmB,CAC1B,GAAG,EAAE,MAAM,EACX,sBAAsB,EAAE,0CAA0C,EAClE,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,8BAA8B,CAAC;IAE1C;;;OAGG;IACH,QAAQ,CAAC,mBAAmB,CAC1B,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,EAAE,EACjB,IAAI,EAAE,cAAc,EAAE,EACtB,QAAQ,EAAE,uBAAuB,EACjC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,sBAAsB,GAAG,SAAS,CAAC;IAE9C;;OAEG;IACH,QAAQ,CAAC,sBAAsB,CAC7B,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,uBAAuB,EACjC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,sBAAsB,GAAG,SAAS,CAAC;IAE9C;;OAEG;IACH,QAAQ,CAAC,uBAAuB,CAC9B,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,uBAAuB,EACjC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAE9B;;OAEG;IACH,QAAQ,CAAC,oBAAoB,CAC3B,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,uBAAuB,EACjC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAE9B;;;;OAIG;IACH,QAAQ,CAAC,kBAAkB,CACzB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,uBAAuB,EAAE,EACrC,MAAM,EAAE,cAAc,EACtB,oBAAoB,CAAC,EAAE,iCAAiC,EACxD,sBAAsB,CAAC,EAAE,0CAA0C,GAClE,OAAO,CAAC,wBAAwB,CAAC;IAEpC;;;;;OAKG;IACH,QAAQ,CAAC,cAAc,CACrB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE;QACV,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,aAAa,EAAE,MAAM,EAAE,CAAC;KACzB,EAAE,EACH,aAAa,EAAE,MAAM,EAAE,EACvB,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;IAEhC;;;;;OAKG;IACH,sBAAsB,CACpB,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAI9B;;;;OAIG;IACH,mBAAmB,CACjB,QAAQ,EAAE,gBAAgB,EAC1B,YAAY,EAAE,MAAM,EACpB,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;IAIhC;;;OAGG;IACH,QAAQ,CAAC,0BAA0B,CACjC,SAAS,EAAE,MAAM,EACjB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,EACnB,gBAAgB,CAAC,EAAE,MAAM,EAAE,GAC1B,MAAM;IAET;;;OAGG;IACH,QAAQ,CAAC,4BAA4B,CACnC,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,GAClB,MAAM;CACV"}
1
+ {"version":3,"file":"LegendAI_LegendApplicationPlugin_Extension.d.ts","sourceRoot":"","sources":["../../src/legend-ai/LegendAI_LegendApplicationPlugin_Extension.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,uBAAuB,EACvB,KAAK,8BAA8B,EACpC,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EACV,iCAAiC,EACjC,cAAc,EACf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EACV,gBAAgB,EAChB,cAAc,EACd,wBAAwB,EACxB,uBAAuB,EACvB,sBAAsB,EACtB,oBAAoB,EACrB,MAAM,oBAAoB,CAAC;AAE5B,qBAAa,2BAA2B;IACtC,GAAG,EAAG,MAAM,GAAG,IAAI,CAAC;IACpB,OAAO,EAAG,MAAM,GAAG,IAAI,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,oBAAY,oBAAoB;IAC9B,IAAI,SAAS;IACb,IAAI,SAAS;CACd;AAED,qBAAa,mBAAmB;IAC9B,OAAO,EAAG,oBAAoB,CAAC;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,8BAA8B;IACzC,OAAO,EAAG,MAAM,EAAE,CAAC;IACnB,IAAI,EAAG,cAAc,EAAE,CAAC;CACzB;AAED,qBAAa,0CAA0C;IACrD,YAAY,EAAG,MAAM,CAAC;IACtB,QAAQ,EAAG,MAAM,CAAC;IAClB,WAAW,EAAG,MAAM,CAAC;IACrB,OAAO,EAAG,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,uCAAuC;IACtD,wBAAwB,EAAE,0CAA0C,CAAC;IACrE,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,yBAAyB,CAAC,EAAE,+BAA+B,CAAC;CAC7D;AAED,MAAM,WAAW,+BAA+B;IAC9C,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,oBAAoB,CAAC,EAAE,4BAA4B,CAAC;CACrD;AAED,MAAM,WAAW,4BAA4B;IAC3C,UAAU,CAAC,EAAE,+BAA+B,EAAE,CAAC;IAC/C,wBAAwB,CAAC,EAAE,gCAAgC,EAAE,CAAC;CAC/D;AAED,MAAM,WAAW,+BAA+B;IAC9C,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,sBAAsB,CAAC,EAAE,MAAM,EAAE,CAAC;CACnC;AAED,MAAM,WAAW,gCAAgC;IAC/C,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,2BAA2B;IACtC,aAAa,EAAG,MAAM,CAAC;IACvB,kCAAkC,EAAG,uCAAuC,CAAC;CAC9E;AAED,qBAAa,4BAA4B;IACvC,YAAY,EAAG,MAAM,CAAC;CACvB;AAED,qBAAa,wBAAwB;IACnC,UAAU,EAAG,MAAM,CAAC;IACpB,eAAe,EAAG,MAAM,EAAE,CAAC;CAC5B;AAED,qBAAa,uBAAuB;IAClC,WAAW,EAAG,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,EAAG,MAAM,CAAC;IACnB,eAAe,EAAG,MAAM,CAAC;CAC1B;AAED,oBAAY,iBAAiB;IAC3B,GAAG,QAAQ;IACX,IAAI,SAAS;IACb,GAAG,QAAQ;IACX,KAAK,UAAU;IACf,IAAI,SAAS;CACd;AAED,qBAAa,2BAA2B;IACtC,SAAS,EAAG,iBAAiB,CAAC;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAG,MAAM,CAAC;CACpB;AAED,qBAAa,iBAAiB;IAC5B,KAAK,EAAG,MAAM,CAAC;IACf,KAAK,EAAG,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,sBAAsB;IACjC,KAAK,EAAG,MAAM,CAAC;IACf,KAAK,EAAG,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,sBAAsB;IACjC,OAAO,EAAG,MAAM,CAAC;IACjB,mBAAmB,CAAC,EAAE,2BAA2B,CAAC;IAClD,UAAU,EAAG,iBAAiB,EAAE,CAAC;IACjC,SAAS,EAAG,sBAAsB,EAAE,CAAC;IACrC,gBAAgB,EAAG,MAAM,EAAE,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,MAAM,4BAA4B,GAAG,iCAAiC,CAAC;AAE7E,8BAAsB,0CAA2C,SAAQ,uBAAuB;IAC9F;;;OAGG;IACH,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAgD;IAEnF,OAAO,CACL,aAAa,EAAE,8BAA8B,CAAC,uBAAuB,CAAC,GACrE,IAAI;IAIP;;;;;OAKG;IACH,QAAQ,CAAC,sBAAsB,CAC7B,QAAQ,EAAE,MAAM,EAChB,WAAW,EAAE,OAAO,EACpB,MAAM,EAAE,cAAc,EACtB,YAAY,CAAC,EAAE,MAAM,EAAE,GACtB,OAAO,CAAC,sBAAsB,CAAC;IAElC;;OAEG;IACH,QAAQ,CAAC,mBAAmB,CAC1B,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,uBAAuB,EACjC,OAAO,CAAC,EAAE,wBAAwB,EAAE,EACpC,QAAQ,CAAC,EAAE,gBAAgB,EAAE,EAC7B,sBAAsB,CAAC,EAAE,MAAM,GAC9B,MAAM;IAET;;OAEG;IACH,QAAQ,CAAC,oBAAoB,CAC3B,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,wBAAwB,EAAE,EACpC,QAAQ,CAAC,EAAE,uBAAuB,EAClC,sBAAsB,CAAC,EAAE,MAAM,GAC9B,MAAM;IAET;;OAEG;IACH,QAAQ,CAAC,gBAAgB,CACvB,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,wBAAwB,EAAE,GACnC,MAAM;IAET;;;;OAIG;IACH,QAAQ,CAAC,+BAA+B,CACtC,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,gBAAgB,EAAE,EAChC,OAAO,CAAC,EAAE,wBAAwB,EAAE,EACpC,sBAAsB,CAAC,EAAE,MAAM,GAC9B,MAAM;IAET;;;OAGG;IACH,QAAQ,CAAC,2BAA2B,CAClC,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,gBAAgB,EAAE,EAChC,OAAO,CAAC,EAAE,wBAAwB,EAAE,EACpC,sBAAsB,CAAC,EAAE,MAAM,GAC9B,MAAM;IAET,qCAAqC,CACnC,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,MAAM,EAChB,cAAc,EAAE,MAAM,EACtB,YAAY,EAAE,gBAAgB,EAAE,EAChC,OAAO,CAAC,EAAE,wBAAwB,EAAE,EACpC,sBAAsB,CAAC,EAAE,MAAM,GAC9B,MAAM;IAUT,iBAAiB,CACf,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,MAAM,CAAC;IAIlB,qBAAqB,CACnB,SAAS,EAAE,gBAAgB,EAAE,EAC7B,OAAO,EAAE,cAAc,GACtB,IAAI;IAIP;;;;OAIG;IACH,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;IAEzE;;OAEG;IACH,QAAQ,CAAC,UAAU,CACjB,GAAG,EAAE,MAAM,EACX,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,8BAA8B,CAAC;IAE1C;;OAEG;IACH,QAAQ,CAAC,sBAAsB,CAC7B,UAAU,EAAE,MAAM,GACjB,2BAA2B;IAE9B;;OAEG;IACH,QAAQ,CAAC,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,mBAAmB;IAEpE;;;OAGG;IACH,QAAQ,CAAC,4BAA4B,CACnC,OAAO,EAAE,2BAA2B,EACpC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,4BAA4B,CAAC;IAExC;;;OAGG;IACH,QAAQ,CAAC,uBAAuB,CAC9B,KAAK,EAAE,MAAM,EACb,sBAAsB,EAAE,0CAA0C,EAClE,MAAM,EAAE,cAAc,EACtB,oBAAoB,CAAC,EAAE,iCAAiC,EACxD,YAAY,CAAC,EAAE,oBAAoB,GAClC,OAAO,CAAC,wBAAwB,CAAC;IAEpC;;;;OAIG;IACH,QAAQ,CAAC,gBAAgB,CACvB,SAAS,EAAE,MAAM,EACjB,gBAAgB,EAAE,iCAAiC,EACnD,sBAAsB,EAAE,0CAA0C,EAClE,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,8BAA8B,CAAC;IAE1C;;;OAGG;IACH,QAAQ,CAAC,mBAAmB,CAC1B,GAAG,EAAE,MAAM,EACX,sBAAsB,EAAE,0CAA0C,EAClE,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,8BAA8B,CAAC;IAE1C;;;OAGG;IACH,QAAQ,CAAC,mBAAmB,CAC1B,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,EAAE,EACjB,IAAI,EAAE,cAAc,EAAE,EACtB,QAAQ,EAAE,uBAAuB,EACjC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,sBAAsB,GAAG,SAAS,CAAC;IAE9C;;OAEG;IACH,QAAQ,CAAC,sBAAsB,CAC7B,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,uBAAuB,EACjC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,sBAAsB,GAAG,SAAS,CAAC;IAE9C;;OAEG;IACH,QAAQ,CAAC,uBAAuB,CAC9B,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,uBAAuB,EACjC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAE9B;;OAEG;IACH,QAAQ,CAAC,oBAAoB,CAC3B,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,uBAAuB,EACjC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAE9B;;;;OAIG;IACH,QAAQ,CAAC,kBAAkB,CACzB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,uBAAuB,EAAE,EACrC,MAAM,EAAE,cAAc,EACtB,oBAAoB,CAAC,EAAE,iCAAiC,EACxD,sBAAsB,CAAC,EAAE,0CAA0C,GAClE,OAAO,CAAC,wBAAwB,CAAC;IAEpC;;;;;OAKG;IACH,QAAQ,CAAC,cAAc,CACrB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE;QACV,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,aAAa,EAAE,MAAM,EAAE,CAAC;KACzB,EAAE,EACH,aAAa,EAAE,MAAM,EAAE,EACvB,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;IAEhC;;;;;OAKG;IACH,sBAAsB,CACpB,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAI9B;;;;OAIG;IACH,mBAAmB,CACjB,QAAQ,EAAE,gBAAgB,EAC1B,YAAY,EAAE,MAAM,EACpB,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,MAAM,EAAE,GAAG,SAAS,CAAC;IAIhC;;;OAGG;IACH,QAAQ,CAAC,0BAA0B,CACjC,SAAS,EAAE,MAAM,EACjB,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,EACnB,gBAAgB,CAAC,EAAE,MAAM,EAAE,GAC1B,MAAM;IAET;;;OAGG;IACH,QAAQ,CAAC,4BAA4B,CACnC,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,GAClB,MAAM;CACV"}
@@ -98,6 +98,15 @@ export class LegendAI_LegendApplicationPlugin_Extension extends LegendApplicatio
98
98
  install(pluginManager) {
99
99
  pluginManager.registerApplicationPlugin(this);
100
100
  }
101
+ buildAccessPointErrorCorrectionPrompt(failedSql, question, executionError, accessPoints, history, modelContextEnrichment) {
102
+ return this.buildAccessPointJudgePrompt(failedSql, `${question}\n\nEXECUTION ERROR — the previous SQL failed. Fix it using the SAME access points:\n${executionError}`, accessPoints, history, modelContextEnrichment);
103
+ }
104
+ normalizeQuestion(question, _config) {
105
+ return Promise.resolve(question);
106
+ }
107
+ preWarmSchemaAnalysis(_services, _config) {
108
+ /* no-op by default — plugins override */
109
+ }
101
110
  /**
102
111
  * Given a user question and available services, use LLM to select the most
103
112
  * relevant service(s) for SQL generation. Returns the filtered subset of
@@ -1 +1 @@
1
- {"version":3,"file":"LegendAI_LegendApplicationPlugin_Extension.js","sourceRoot":"","sources":["../../src/legend-ai/LegendAI_LegendApplicationPlugin_Extension.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,uBAAuB,GAExB,MAAM,2BAA2B,CAAC;AAanC,MAAM,OAAO,2BAA2B;IACtC,GAAG,CAAiB;IACpB,OAAO,CAAiB;IACxB,UAAU,CAAU;CACrB;AAED,MAAM,CAAN,IAAY,oBAGX;AAHD,WAAY,oBAAoB;IAC9B,qCAAa,CAAA;IACb,qCAAa,CAAA;AACf,CAAC,EAHW,oBAAoB,KAApB,oBAAoB,QAG/B;AAED,MAAM,OAAO,mBAAmB;IAC9B,OAAO,CAAwB;IAC/B,MAAM,CAAU;IAChB,YAAY,CAAU;CACvB;AAED,MAAM,OAAO,8BAA8B;IACzC,OAAO,CAAY;IACnB,IAAI,CAAoB;CACzB;AAED,MAAM,OAAO,0CAA0C;IACrD,YAAY,CAAU;IACtB,QAAQ,CAAU;IAClB,WAAW,CAAU;IACrB,OAAO,CAAU;CAClB;AAQD,MAAM,OAAO,2BAA2B;IACtC,aAAa,CAAU;IACvB,kCAAkC,CAA2C;CAC9E;AAED,MAAM,OAAO,4BAA4B;IACvC,YAAY,CAAU;CACvB;AAED,MAAM,OAAO,wBAAwB;IACnC,UAAU,CAAU;IACpB,eAAe,CAAY;CAC5B;AAED,MAAM,OAAO,uBAAuB;IAClC,WAAW,CAAU;IACrB,WAAW,CAAU;IACrB,SAAS,CAAU;IACnB,eAAe,CAAU;CAC1B;AAED,MAAM,CAAN,IAAY,iBAMX;AAND,WAAY,iBAAiB;IAC3B,gCAAW,CAAA;IACX,kCAAa,CAAA;IACb,gCAAW,CAAA;IACX,oCAAe,CAAA;IACf,kCAAa,CAAA;AACf,CAAC,EANW,iBAAiB,KAAjB,iBAAiB,QAM5B;AAED,MAAM,OAAO,2BAA2B;IACtC,SAAS,CAAqB;IAC9B,KAAK,CAAU;IACf,KAAK,CAAU;IACf,KAAK,CAAU;IACf,SAAS,CAAU;CACpB;AAED,MAAM,OAAO,iBAAiB;IAC5B,KAAK,CAAU;IACf,KAAK,CAAU;IACf,MAAM,CAAU;CACjB;AAED,MAAM,OAAO,sBAAsB;IACjC,KAAK,CAAU;IACf,KAAK,CAAU;IACf,KAAK,CAAU;IACf,UAAU,CAAU;CACrB;AAED,MAAM,OAAO,sBAAsB;IACjC,OAAO,CAAU;IACjB,mBAAmB,CAA+B;IAClD,UAAU,CAAuB;IACjC,SAAS,CAA4B;IACrC,gBAAgB,CAAY;CAC7B;AAOD,MAAM,OAAgB,0CAA2C,SAAQ,uBAAuB;IAC9F;;;OAGG;IACc,kBAAkB,CAAgD;IAEnF,OAAO,CACL,aAAsE;QAEtE,aAAa,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC;IAChD,CAAC;IAkND;;;;;OAKG;IACH,sBAAsB,CACpB,SAAiB,EACjB,QAA4B,EAC5B,OAAuB;QAEvB,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED;;;;OAIG;IACH,mBAAmB,CACjB,QAA0B,EAC1B,YAAoB,EACpB,OAAuB;QAEvB,OAAO,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC;CAyBF"}
1
+ {"version":3,"file":"LegendAI_LegendApplicationPlugin_Extension.js","sourceRoot":"","sources":["../../src/legend-ai/LegendAI_LegendApplicationPlugin_Extension.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,uBAAuB,GAExB,MAAM,2BAA2B,CAAC;AAcnC,MAAM,OAAO,2BAA2B;IACtC,GAAG,CAAiB;IACpB,OAAO,CAAiB;IACxB,UAAU,CAAU;CACrB;AAED,MAAM,CAAN,IAAY,oBAGX;AAHD,WAAY,oBAAoB;IAC9B,qCAAa,CAAA;IACb,qCAAa,CAAA;AACf,CAAC,EAHW,oBAAoB,KAApB,oBAAoB,QAG/B;AAED,MAAM,OAAO,mBAAmB;IAC9B,OAAO,CAAwB;IAC/B,MAAM,CAAU;IAChB,YAAY,CAAU;CACvB;AAED,MAAM,OAAO,8BAA8B;IACzC,OAAO,CAAY;IACnB,IAAI,CAAoB;CACzB;AAED,MAAM,OAAO,0CAA0C;IACrD,YAAY,CAAU;IACtB,QAAQ,CAAU;IAClB,WAAW,CAAU;IACrB,OAAO,CAAU;CAClB;AAgCD,MAAM,OAAO,2BAA2B;IACtC,aAAa,CAAU;IACvB,kCAAkC,CAA2C;CAC9E;AAED,MAAM,OAAO,4BAA4B;IACvC,YAAY,CAAU;CACvB;AAED,MAAM,OAAO,wBAAwB;IACnC,UAAU,CAAU;IACpB,eAAe,CAAY;CAC5B;AAED,MAAM,OAAO,uBAAuB;IAClC,WAAW,CAAU;IACrB,WAAW,CAAU;IACrB,SAAS,CAAU;IACnB,eAAe,CAAU;CAC1B;AAED,MAAM,CAAN,IAAY,iBAMX;AAND,WAAY,iBAAiB;IAC3B,gCAAW,CAAA;IACX,kCAAa,CAAA;IACb,gCAAW,CAAA;IACX,oCAAe,CAAA;IACf,kCAAa,CAAA;AACf,CAAC,EANW,iBAAiB,KAAjB,iBAAiB,QAM5B;AAED,MAAM,OAAO,2BAA2B;IACtC,SAAS,CAAqB;IAC9B,KAAK,CAAU;IACf,KAAK,CAAU;IACf,KAAK,CAAU;IACf,SAAS,CAAU;CACpB;AAED,MAAM,OAAO,iBAAiB;IAC5B,KAAK,CAAU;IACf,KAAK,CAAU;IACf,MAAM,CAAU;CACjB;AAED,MAAM,OAAO,sBAAsB;IACjC,KAAK,CAAU;IACf,KAAK,CAAU;IACf,KAAK,CAAU;IACf,UAAU,CAAU;CACrB;AAED,MAAM,OAAO,sBAAsB;IACjC,OAAO,CAAU;IACjB,mBAAmB,CAA+B;IAClD,UAAU,CAAuB;IACjC,SAAS,CAA4B;IACrC,gBAAgB,CAAY;CAC7B;AAOD,MAAM,OAAgB,0CAA2C,SAAQ,uBAAuB;IAC9F;;;OAGG;IACc,kBAAkB,CAAgD;IAEnF,OAAO,CACL,aAAsE;QAEtE,aAAa,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC;IAChD,CAAC;IAyED,qCAAqC,CACnC,SAAiB,EACjB,QAAgB,EAChB,cAAsB,EACtB,YAAgC,EAChC,OAAoC,EACpC,sBAA+B;QAE/B,OAAO,IAAI,CAAC,2BAA2B,CACrC,SAAS,EACT,GAAG,QAAQ,wFAAwF,cAAc,EAAE,EACnH,YAAY,EACZ,OAAO,EACP,sBAAsB,CACvB,CAAC;IACJ,CAAC;IAED,iBAAiB,CACf,QAAgB,EAChB,OAAuB;QAEvB,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED,qBAAqB,CACnB,SAA6B,EAC7B,OAAuB;QAEvB,yCAAyC;IAC3C,CAAC;IAiJD;;;;;OAKG;IACH,sBAAsB,CACpB,SAAiB,EACjB,QAA4B,EAC5B,OAAuB;QAEvB,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED;;;;OAIG;IACH,mBAAmB,CACjB,QAA0B,EAC1B,YAAoB,EACpB,OAAuB;QAEvB,OAAO,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC;CAyBF"}
@@ -22,6 +22,11 @@ export declare const TEST__createMockSetter: () => {
22
22
  };
23
23
  export declare const TEST__makeAssistantMessage: (overrides?: Partial<LegendAIAssistantMessage>) => LegendAIAssistantMessage;
24
24
  export declare const TEST__seedAssistant: (setter: MessageSetter) => void;
25
+ /**
26
+ * Retrieve a message by index and assert it is an assistant message,
27
+ * narrowing the type via the discriminated `role` field without a cast.
28
+ */
29
+ export declare const TEST__getAssistantMessage: (messages: LegendAIMessage[], index: number) => LegendAIAssistantMessage;
25
30
  export declare const TEST_DATA__legendAIConfig: LegendAIConfig;
26
31
  export declare const TEST_DATA__legendAIMetadata: LegendAIProductMetadata;
27
32
  export declare const TEST_DATA__legendAIServices: TDSServiceSchema[];
@@ -1 +1 @@
1
- {"version":3,"file":"LegendAITestUtils.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/__test-utils__/LegendAITestUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,KAAK,EACV,aAAa,EACb,wBAAwB,EACzB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,wBAAwB,EAC7B,KAAK,cAAc,EACnB,KAAK,uBAAuB,EAC5B,KAAK,gBAAgB,EAItB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,0CAA0C,EAOhD,MAAM,kDAAkD,CAAC;AAE1D,eAAO,MAAM,sBAAsB,QAAO;IACxC,MAAM,EAAE,aAAa,CAAC;IACtB,WAAW,EAAE,MAAM,eAAe,EAAE,CAAC;CAWtC,CAAC;AACF,eAAO,MAAM,0BAA0B,GACrC,YAAY,OAAO,CAAC,wBAAwB,CAAC,KAC5C,wBAkBD,CAAC;AACH,eAAO,MAAM,mBAAmB,GAAI,QAAQ,aAAa,KAAG,IAK3D,CAAC;AACF,eAAO,MAAM,yBAAyB,EAAE,cASvC,CAAC;AAEF,eAAO,MAAM,2BAA2B,EAAE,uBAIzC,CAAC;AAEF,eAAO,MAAM,2BAA2B,EAAE,gBAAgB,EAUzD,CAAC;AACF,eAAO,MAAM,8BAA8B,GACzC,YAAY,OAAO,CAAC,0CAA0C,CAAC,KAC9D,0CA4E+C,CAAC;AAEnD,eAAO,MAAM,4BAA4B,GACvC,YAAY,OAAO,CAAC,wBAAwB,CAAC,KAC5C,wBASF,CAAC"}
1
+ {"version":3,"file":"LegendAITestUtils.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/__test-utils__/LegendAITestUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,KAAK,EACV,aAAa,EACb,wBAAwB,EACzB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,wBAAwB,EAC7B,KAAK,cAAc,EACnB,KAAK,uBAAuB,EAC5B,KAAK,gBAAgB,EAItB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,0CAA0C,EAOhD,MAAM,kDAAkD,CAAC;AAE1D,eAAO,MAAM,sBAAsB,QAAO;IACxC,MAAM,EAAE,aAAa,CAAC;IACtB,WAAW,EAAE,MAAM,eAAe,EAAE,CAAC;CAWtC,CAAC;AACF,eAAO,MAAM,0BAA0B,GACrC,YAAY,OAAO,CAAC,wBAAwB,CAAC,KAC5C,wBAmBD,CAAC;AACH,eAAO,MAAM,mBAAmB,GAAI,QAAQ,aAAa,KAAG,IAK3D,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,yBAAyB,GACpC,UAAU,eAAe,EAAE,EAC3B,OAAO,MAAM,KACZ,wBAWF,CAAC;AAEF,eAAO,MAAM,yBAAyB,EAAE,cASvC,CAAC;AAEF,eAAO,MAAM,2BAA2B,EAAE,uBAIzC,CAAC;AAEF,eAAO,MAAM,2BAA2B,EAAE,gBAAgB,EAUzD,CAAC;AACF,eAAO,MAAM,8BAA8B,GACzC,YAAY,OAAO,CAAC,0CAA0C,CAAC,KAC9D,0CAiF+C,CAAC;AAEnD,eAAO,MAAM,4BAA4B,GACvC,YAAY,OAAO,CAAC,wBAAwB,CAAC,KAC5C,wBASF,CAAC"}
@@ -13,6 +13,7 @@
13
13
  * See the License for the specific language governing permissions and
14
14
  * limitations under the License.
15
15
  */
16
+ import { guaranteeNonNullable } from '@finos/legend-shared';
16
17
  import { createMock } from '@finos/legend-shared/test';
17
18
  import { LegendAIMessageRole, LegendAIQuestionIntent, } from '../LegendAITypes.js';
18
19
  import { LegendAIJudgeVerdict, LegendAIOrchestratorResponse, LegendAIResolvedEntities, LegendAISqlExecutionResultData, } from '../LegendAI_LegendApplicationPlugin_Extension.js';
@@ -45,6 +46,7 @@ export const TEST__makeAssistantMessage = (overrides) => ({
45
46
  isExecuting: false,
46
47
  suggestedQueries: [],
47
48
  fallbackAction: null,
49
+ queriedAccessPointGroups: [],
48
50
  ...overrides,
49
51
  });
50
52
  export const TEST__seedAssistant = (setter) => {
@@ -53,6 +55,17 @@ export const TEST__seedAssistant = (setter) => {
53
55
  TEST__makeAssistantMessage(),
54
56
  ]);
55
57
  };
58
+ /**
59
+ * Retrieve a message by index and assert it is an assistant message,
60
+ * narrowing the type via the discriminated `role` field without a cast.
61
+ */
62
+ export const TEST__getAssistantMessage = (messages, index) => {
63
+ const msg = guaranteeNonNullable(messages[index], `Expected message at index ${index}`);
64
+ if (msg.role !== LegendAIMessageRole.ASSISTANT) {
65
+ throw new Error(`Expected assistant message at index ${index}, got role '${msg.role}'`);
66
+ }
67
+ return msg;
68
+ };
56
69
  export const TEST_DATA__legendAIConfig = {
57
70
  enabled: true,
58
71
  llmServiceUrl: 'http://localhost/llm',
@@ -87,8 +100,11 @@ export const TEST__createMockLegendAIPlugin = (overrides) => ({
87
100
  buildMetadataPrompt: (_q, _m, _h) => 'metadata prompt',
88
101
  buildGeneratorPrompt: (_q, _s, _c, _h, _m) => 'generator prompt',
89
102
  buildJudgePrompt: (_sql, _q, _s, _c, _h) => 'judge prompt',
90
- buildAccessPointGeneratorPrompt: (_q, _s, _h) => 'ap generator prompt',
91
- buildAccessPointJudgePrompt: (_sql, _q, _s, _h) => 'ap judge prompt',
103
+ buildAccessPointGeneratorPrompt: (_q, _s, _h, _m) => 'ap generator prompt',
104
+ buildAccessPointJudgePrompt: (_sql, _q, _s, _h, _m) => 'ap judge prompt',
105
+ preWarmSchemaAnalysis: () => {
106
+ /* no-op in test */
107
+ },
92
108
  callLLM: createMock(),
93
109
  executeSql: createMock(),
94
110
  extractSqlFromResponse: (_a) => ({
@@ -1 +1 @@
1
- {"version":3,"file":"LegendAITestUtils.js","sourceRoot":"","sources":["../../../src/legend-ai/__test-utils__/LegendAITestUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAKvD,OAAO,EAOL,mBAAmB,EACnB,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAIL,oBAAoB,EACpB,4BAA4B,EAC5B,wBAAwB,EACxB,8BAA8B,GAC/B,MAAM,kDAAkD,CAAC;AAE1D,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAGpC,EAAE;IACF,IAAI,QAAQ,GAAsB,EAAE,CAAC;IACrC,MAAM,MAAM,GAAkB,CAAC,MAAM,EAAE,EAAE;QACvC,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;YACjC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,MAAM,CAAC;QACpB,CAAC;IACH,CAAC,CAAC;IACF,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;AACjD,CAAC,CAAC;AACF,MAAM,CAAC,MAAM,0BAA0B,GAAG,CACxC,SAA6C,EACnB,EAAE,CAAC,CAAC;IAC9B,EAAE,EAAE,oBAAoB;IACxB,IAAI,EAAE,mBAAmB,CAAC,SAAS;IACnC,aAAa,EAAE,EAAE;IACjB,GAAG,EAAE,IAAI;IACT,UAAU,EAAE,IAAI;IAChB,WAAW,EAAE,IAAI;IACjB,QAAQ,EAAE,IAAI;IACd,KAAK,EAAE,IAAI;IACX,SAAS,EAAE,IAAI;IACf,UAAU,EAAE,IAAI;IAChB,QAAQ,EAAE,IAAI;IACd,gBAAgB,EAAE,IAAI;IACtB,YAAY,EAAE,IAAI;IAClB,WAAW,EAAE,KAAK;IAClB,gBAAgB,EAAE,EAAE;IACpB,cAAc,EAAE,IAAI;IACpB,GAAG,SAAS;CACb,CAAC,CAAC;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,MAAqB,EAAQ,EAAE;IACjE,MAAM,CAAC;QACL,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE;QACrE,0BAA0B,EAAE;KAC7B,CAAC,CAAC;AACL,CAAC,CAAC;AACF,MAAM,CAAC,MAAM,yBAAyB,GAAmB;IACvD,OAAO,EAAE,IAAI;IACb,aAAa,EAAE,sBAAsB;IACrC,YAAY,EAAE,YAAY;IAC1B,eAAe,EAAE,sBAAsB;IACvC,eAAe,EAAE,SAAS;IAC1B,oBAAoB,EAAE,SAAS;IAC/B,SAAS,EAAE,SAAS;IACpB,gBAAgB,EAAE,CAAC;CACpB,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAA4B;IAClE,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,qBAAqB;IAClC,gBAAgB,EAAE,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;CAC1D,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAuB;IAC7D;QACE,KAAK,EAAE,cAAc;QACrB,OAAO,EAAE,SAAS;QAClB,OAAO,EAAE;YACP,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;YACnC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE;SACnC;QACD,UAAU,EAAE,EAAE;KACf;CACF,CAAC;AACF,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAC5C,SAA+D,EACnB,EAAE,CAC9C,CAAC;IACC,OAAO,EAAE,GAAG,EAAE,CAAC,aAAa;IAC5B,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO;IACzB,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC;IACjB,sBAAsB,EAAE,CAAC,EAAU,EAAE,EAAW,EAA0B,EAAE,CAC1E,sBAAsB,CAAC,UAAU;IACnC,mBAAmB,EAAE,CACnB,EAAU,EACV,EAA2B,EAC3B,EAA+B,EAC/B,EAAE,CAAC,iBAAiB;IACtB,oBAAoB,EAAE,CACpB,EAAU,EACV,EAAsB,EACtB,EAAU,EACV,EAA+B,EAC/B,EAA4B,EAC5B,EAAE,CAAC,kBAAkB;IACvB,gBAAgB,EAAE,CAChB,IAAY,EACZ,EAAU,EACV,EAAsB,EACtB,EAAU,EACV,EAA+B,EAC/B,EAAE,CAAC,cAAc;IACnB,+BAA+B,EAAE,CAC/B,EAAU,EACV,EAAsB,EACtB,EAA+B,EAC/B,EAAE,CAAC,qBAAqB;IAC1B,2BAA2B,EAAE,CAC3B,IAAY,EACZ,EAAU,EACV,EAAsB,EACtB,EAA+B,EAC/B,EAAE,CAAC,iBAAiB;IACtB,OAAO,EAAE,UAAU,EAAE;IACrB,UAAU,EAAE,UAAU,EAAE;IACxB,sBAAsB,EAAE,CAAC,EAAU,EAA+B,EAAE,CAAC,CAAC;QACpE,GAAG,EAAE,iBAAiB;QACtB,OAAO,EAAE,IAAI;KACd,CAAC;IACF,kBAAkB,EAAE,CAAC,EAAU,EAAuB,EAAE,CAAC,CAAC;QACxD,OAAO,EAAE,oBAAoB,CAAC,IAAI;KACnC,CAAC;IACF,sBAAsB,EAAE,CACtB,EAAU,EACV,QAA4B,EACC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;IAC3D,uBAAuB,EAAE,GAAsC,EAAE;QAC/D,MAAM,QAAQ,GAAG,IAAI,wBAAwB,EAAE,CAAC;QAChD,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC;QACjC,QAAQ,CAAC,eAAe,GAAG,EAAE,CAAC;QAC9B,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IACD,4BAA4B,EAAE,GAA0C,EAAE;QACxE,MAAM,QAAQ,GAAG,IAAI,4BAA4B,EAAE,CAAC;QACpD,QAAQ,CAAC,YAAY,GAAG,gDAAgD,CAAC;QACzE,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IACD,gBAAgB,EAAE,GAA4C,EAAE;QAC9D,MAAM,IAAI,GAAG,IAAI,8BAA8B,EAAE,CAAC;QAClD,IAAI,CAAC,OAAO,GAAG,CAAC,IAAI,CAAC,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1B,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IACD,kBAAkB,EAAE,GAAsC,EAAE;QAC1D,MAAM,QAAQ,GAAG,IAAI,wBAAwB,EAAE,CAAC;QAChD,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC;QACjC,QAAQ,CAAC,eAAe,GAAG,EAAE,CAAC;QAC9B,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IACD,0BAA0B,EAAE,GAAW,EAAE,CAAC,EAAE;IAC5C,4BAA4B,EAAE,GAAW,EAAE,CAAC,EAAE;IAC9C,GAAG,SAAS;CACb,CAA+C,CAAC;AAEnD,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAC1C,SAA6C,EACnB,EAAE;IAC5B,MAAM,EAAE,MAAM,EAAE,GAAG,sBAAsB,EAAE,CAAC;IAC5C,OAAO;QACL,MAAM,EAAE,yBAAyB;QACjC,MAAM,EAAE,8BAA8B,EAAE;QACxC,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,MAAM;QACnB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC,CAAC"}
1
+ {"version":3,"file":"LegendAITestUtils.js","sourceRoot":"","sources":["../../../src/legend-ai/__test-utils__/LegendAITestUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAKvD,OAAO,EAOL,mBAAmB,EACnB,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAIL,oBAAoB,EACpB,4BAA4B,EAC5B,wBAAwB,EACxB,8BAA8B,GAC/B,MAAM,kDAAkD,CAAC;AAE1D,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAGpC,EAAE;IACF,IAAI,QAAQ,GAAsB,EAAE,CAAC;IACrC,MAAM,MAAM,GAAkB,CAAC,MAAM,EAAE,EAAE;QACvC,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;YACjC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,MAAM,CAAC;QACpB,CAAC;IACH,CAAC,CAAC;IACF,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;AACjD,CAAC,CAAC;AACF,MAAM,CAAC,MAAM,0BAA0B,GAAG,CACxC,SAA6C,EACnB,EAAE,CAAC,CAAC;IAC9B,EAAE,EAAE,oBAAoB;IACxB,IAAI,EAAE,mBAAmB,CAAC,SAAS;IACnC,aAAa,EAAE,EAAE;IACjB,GAAG,EAAE,IAAI;IACT,UAAU,EAAE,IAAI;IAChB,WAAW,EAAE,IAAI;IACjB,QAAQ,EAAE,IAAI;IACd,KAAK,EAAE,IAAI;IACX,SAAS,EAAE,IAAI;IACf,UAAU,EAAE,IAAI;IAChB,QAAQ,EAAE,IAAI;IACd,gBAAgB,EAAE,IAAI;IACtB,YAAY,EAAE,IAAI;IAClB,WAAW,EAAE,KAAK;IAClB,gBAAgB,EAAE,EAAE;IACpB,cAAc,EAAE,IAAI;IACpB,wBAAwB,EAAE,EAAE;IAC5B,GAAG,SAAS;CACb,CAAC,CAAC;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,MAAqB,EAAQ,EAAE;IACjE,MAAM,CAAC;QACL,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE;QACrE,0BAA0B,EAAE;KAC7B,CAAC,CAAC;AACL,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACvC,QAA2B,EAC3B,KAAa,EACa,EAAE;IAC5B,MAAM,GAAG,GAAG,oBAAoB,CAC9B,QAAQ,CAAC,KAAK,CAAC,EACf,6BAA6B,KAAK,EAAE,CACrC,CAAC;IACF,IAAI,GAAG,CAAC,IAAI,KAAK,mBAAmB,CAAC,SAAS,EAAE,CAAC;QAC/C,MAAM,IAAI,KAAK,CACb,uCAAuC,KAAK,eAAe,GAAG,CAAC,IAAI,GAAG,CACvE,CAAC;IACJ,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAmB;IACvD,OAAO,EAAE,IAAI;IACb,aAAa,EAAE,sBAAsB;IACrC,YAAY,EAAE,YAAY;IAC1B,eAAe,EAAE,sBAAsB;IACvC,eAAe,EAAE,SAAS;IAC1B,oBAAoB,EAAE,SAAS;IAC/B,SAAS,EAAE,SAAS;IACpB,gBAAgB,EAAE,CAAC;CACpB,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAA4B;IAClE,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,qBAAqB;IAClC,gBAAgB,EAAE,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;CAC1D,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAuB;IAC7D;QACE,KAAK,EAAE,cAAc;QACrB,OAAO,EAAE,SAAS;QAClB,OAAO,EAAE;YACP,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;YACnC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE;SACnC;QACD,UAAU,EAAE,EAAE;KACf;CACF,CAAC;AACF,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAC5C,SAA+D,EACnB,EAAE,CAC9C,CAAC;IACC,OAAO,EAAE,GAAG,EAAE,CAAC,aAAa;IAC5B,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO;IACzB,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC;IACjB,sBAAsB,EAAE,CAAC,EAAU,EAAE,EAAW,EAA0B,EAAE,CAC1E,sBAAsB,CAAC,UAAU;IACnC,mBAAmB,EAAE,CACnB,EAAU,EACV,EAA2B,EAC3B,EAA+B,EAC/B,EAAE,CAAC,iBAAiB;IACtB,oBAAoB,EAAE,CACpB,EAAU,EACV,EAAsB,EACtB,EAAU,EACV,EAA+B,EAC/B,EAA4B,EAC5B,EAAE,CAAC,kBAAkB;IACvB,gBAAgB,EAAE,CAChB,IAAY,EACZ,EAAU,EACV,EAAsB,EACtB,EAAU,EACV,EAA+B,EAC/B,EAAE,CAAC,cAAc;IACnB,+BAA+B,EAAE,CAC/B,EAAU,EACV,EAAsB,EACtB,EAA+B,EAC/B,EAAW,EACX,EAAE,CAAC,qBAAqB;IAC1B,2BAA2B,EAAE,CAC3B,IAAY,EACZ,EAAU,EACV,EAAsB,EACtB,EAA+B,EAC/B,EAAW,EACX,EAAE,CAAC,iBAAiB;IACtB,qBAAqB,EAAE,GAAG,EAAE;QAC1B,mBAAmB;IACrB,CAAC;IACD,OAAO,EAAE,UAAU,EAAE;IACrB,UAAU,EAAE,UAAU,EAAE;IACxB,sBAAsB,EAAE,CAAC,EAAU,EAA+B,EAAE,CAAC,CAAC;QACpE,GAAG,EAAE,iBAAiB;QACtB,OAAO,EAAE,IAAI;KACd,CAAC;IACF,kBAAkB,EAAE,CAAC,EAAU,EAAuB,EAAE,CAAC,CAAC;QACxD,OAAO,EAAE,oBAAoB,CAAC,IAAI;KACnC,CAAC;IACF,sBAAsB,EAAE,CACtB,EAAU,EACV,QAA4B,EACC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;IAC3D,uBAAuB,EAAE,GAAsC,EAAE;QAC/D,MAAM,QAAQ,GAAG,IAAI,wBAAwB,EAAE,CAAC;QAChD,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC;QACjC,QAAQ,CAAC,eAAe,GAAG,EAAE,CAAC;QAC9B,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IACD,4BAA4B,EAAE,GAA0C,EAAE;QACxE,MAAM,QAAQ,GAAG,IAAI,4BAA4B,EAAE,CAAC;QACpD,QAAQ,CAAC,YAAY,GAAG,gDAAgD,CAAC;QACzE,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IACD,gBAAgB,EAAE,GAA4C,EAAE;QAC9D,MAAM,IAAI,GAAG,IAAI,8BAA8B,EAAE,CAAC;QAClD,IAAI,CAAC,OAAO,GAAG,CAAC,IAAI,CAAC,CAAC;QACtB,IAAI,CAAC,IAAI,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1B,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IACD,kBAAkB,EAAE,GAAsC,EAAE;QAC1D,MAAM,QAAQ,GAAG,IAAI,wBAAwB,EAAE,CAAC;QAChD,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC;QACjC,QAAQ,CAAC,eAAe,GAAG,EAAE,CAAC;QAC9B,OAAO,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IACD,0BAA0B,EAAE,GAAW,EAAE,CAAC,EAAE;IAC5C,4BAA4B,EAAE,GAAW,EAAE,CAAC,EAAE;IAC9C,GAAG,SAAS;CACb,CAA+C,CAAC;AAEnD,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAC1C,SAA6C,EACnB,EAAE;IAC5B,MAAM,EAAE,MAAM,EAAE,GAAG,sBAAsB,EAAE,CAAC;IAC5C,OAAO;QACL,MAAM,EAAE,yBAAyB;QACjC,MAAM,EAAE,8BAA8B,EAAE;QACxC,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,MAAM;QACnB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"LegendAIAnalysisPanel.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIAnalysisPanel.tsx"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAI5D,eAAO,MAAM,qBAAqB,GAAI,OAAO;IAC3C,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE,KAAK,CAAC,aAAa,CAAC;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACzD,KAAG,KAAK,CAAC,SAuDT,CAAC"}
1
+ {"version":3,"file":"LegendAIAnalysisPanel.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIAnalysisPanel.tsx"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAQ5D,eAAO,MAAM,qBAAqB,GAAI,OAAO;IAC3C,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE,KAAK,CAAC,aAAa,CAAC;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACzD,KAAG,KAAK,CAAC,SAsET,CAAC"}
@@ -17,18 +17,37 @@ import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
17
17
  import { useMemo } from 'react';
18
18
  import { LegendAIChartType } from '../LegendAI_LegendApplicationPlugin_Extension.js';
19
19
  import { analyzeGridData } from './LegendAIAnalysisUtils.js';
20
- import { LegendAIBarChart, LegendAIDonutChart } from './LegendAICharts.js';
20
+ import { LegendAIBarChart, LegendAIDonutChart, LegendAILineChart, } from './LegendAICharts.js';
21
21
  export const LegendAIAnalysisPanel = (props) => {
22
22
  const { gridData, summary, SummaryRenderer } = props;
23
- const { metrics, chartType, chartData, numericColumnName } = useMemo(() => analyzeGridData(gridData), [gridData]);
24
- let chartTitle;
25
- if (numericColumnName !== undefined) {
26
- chartTitle =
27
- chartType === LegendAIChartType.PIE
23
+ const { metrics, chartType, chartData, numericColumnName, categoryColumnName, } = useMemo(() => analyzeGridData(gridData), [gridData]);
24
+ const chartTitle = useMemo(() => {
25
+ if (chartType === LegendAIChartType.LINE && numericColumnName) {
26
+ return categoryColumnName
27
+ ? `${numericColumnName} over ${categoryColumnName}`
28
+ : `${numericColumnName} Trend`;
29
+ }
30
+ if (numericColumnName !== undefined) {
31
+ return chartType === LegendAIChartType.PIE
28
32
  ? `${numericColumnName} Distribution`
29
33
  : `Top ${chartData.length} by ${numericColumnName}`;
30
- }
34
+ }
35
+ if (categoryColumnName !== undefined) {
36
+ return `${categoryColumnName} Distribution`;
37
+ }
38
+ return undefined;
39
+ }, [chartType, numericColumnName, categoryColumnName, chartData.length]);
31
40
  const chartTitleProp = chartTitle === undefined ? {} : { title: chartTitle };
32
- return (_jsxs("div", { className: "legend-ai-analysis", children: [metrics.length > 0 && (_jsx("div", { className: "legend-ai-analysis__metrics", children: metrics.map((m) => (_jsxs("div", { className: "legend-ai-analysis__metric-card", children: [_jsx("span", { className: "legend-ai-analysis__metric-value", children: m.value }), _jsx("span", { className: "legend-ai-analysis__metric-label", children: m.label }), m.detail !== undefined && (_jsx("span", { className: "legend-ai-analysis__metric-detail", children: m.detail }))] }, m.label))) })), chartData.length > 0 && (_jsx("div", { className: "legend-ai-analysis__chart-section", children: chartType === LegendAIChartType.PIE ? (_jsx(LegendAIDonutChart, { data: chartData, ...chartTitleProp })) : (_jsx(LegendAIBarChart, { data: chartData, ...chartTitleProp })) })), _jsx("div", { className: "legend-ai-analysis__narrative", children: _jsx(SummaryRenderer, { value: summary }) })] }));
41
+ let chartElement;
42
+ if (chartType === LegendAIChartType.LINE) {
43
+ chartElement = _jsx(LegendAILineChart, { data: chartData, ...chartTitleProp });
44
+ }
45
+ else if (chartType === LegendAIChartType.PIE) {
46
+ chartElement = _jsx(LegendAIDonutChart, { data: chartData, ...chartTitleProp });
47
+ }
48
+ else {
49
+ chartElement = _jsx(LegendAIBarChart, { data: chartData, ...chartTitleProp });
50
+ }
51
+ return (_jsxs("div", { className: "legend-ai-analysis", children: [metrics.length > 0 && (_jsx("div", { className: "legend-ai-analysis__metrics", children: metrics.map((m) => (_jsxs("div", { className: "legend-ai-analysis__metric-card", children: [_jsx("span", { className: "legend-ai-analysis__metric-value", children: m.value }), _jsx("span", { className: "legend-ai-analysis__metric-label", children: m.label }), m.detail !== undefined && (_jsx("span", { className: "legend-ai-analysis__metric-detail", children: m.detail }))] }, m.label))) })), chartData.length > 0 && (_jsx("div", { className: "legend-ai-analysis__chart-section", children: chartElement })), _jsx("div", { className: "legend-ai-analysis__narrative", children: _jsx(SummaryRenderer, { value: summary }) })] }));
33
52
  };
34
53
  //# sourceMappingURL=LegendAIAnalysisPanel.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"LegendAIAnalysisPanel.js","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIAnalysisPanel.tsx"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EAAE,iBAAiB,EAAE,MAAM,kDAAkD,CAAC;AAErF,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAE3E,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,KAIrC,EAAmB,EAAE;IACpB,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC;IAErD,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAClE,GAAG,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,EAC/B,CAAC,QAAQ,CAAC,CACX,CAAC;IAEF,IAAI,UAA8B,CAAC;IACnC,IAAI,iBAAiB,KAAK,SAAS,EAAE,CAAC;QACpC,UAAU;YACR,SAAS,KAAK,iBAAiB,CAAC,GAAG;gBACjC,CAAC,CAAC,GAAG,iBAAiB,eAAe;gBACrC,CAAC,CAAC,OAAO,SAAS,CAAC,MAAM,OAAO,iBAAiB,EAAE,CAAC;IAC1D,CAAC;IAED,MAAM,cAAc,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;IAE7E,OAAO,CACL,eAAK,SAAS,EAAC,oBAAoB,aAChC,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,CACrB,cAAK,SAAS,EAAC,6BAA6B,YACzC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAClB,eAAmB,SAAS,EAAC,iCAAiC,aAC5D,eAAM,SAAS,EAAC,kCAAkC,YAC/C,CAAC,CAAC,KAAK,GACH,EACP,eAAM,SAAS,EAAC,kCAAkC,YAC/C,CAAC,CAAC,KAAK,GACH,EACN,CAAC,CAAC,MAAM,KAAK,SAAS,IAAI,CACzB,eAAM,SAAS,EAAC,mCAAmC,YAChD,CAAC,CAAC,MAAM,GACJ,CACR,KAXO,CAAC,CAAC,KAAK,CAYX,CACP,CAAC,GACE,CACP,EAEA,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,CACvB,cAAK,SAAS,EAAC,mCAAmC,YAC/C,SAAS,KAAK,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CACrC,KAAC,kBAAkB,IAAC,IAAI,EAAE,SAAS,KAAM,cAAc,GAAI,CAC5D,CAAC,CAAC,CAAC,CACF,KAAC,gBAAgB,IAAC,IAAI,EAAE,SAAS,KAAM,cAAc,GAAI,CAC1D,GACG,CACP,EAED,cAAK,SAAS,EAAC,+BAA+B,YAC5C,KAAC,eAAe,IAAC,KAAK,EAAE,OAAO,GAAI,GAC/B,IACF,CACP,CAAC;AACJ,CAAC,CAAC"}
1
+ {"version":3,"file":"LegendAIAnalysisPanel.js","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIAnalysisPanel.tsx"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EAAE,iBAAiB,EAAE,MAAM,kDAAkD,CAAC;AAErF,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAC7D,OAAO,EACL,gBAAgB,EAChB,kBAAkB,EAClB,iBAAiB,GAClB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,KAIrC,EAAmB,EAAE;IACpB,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC;IAErD,MAAM,EACJ,OAAO,EACP,SAAS,EACT,SAAS,EACT,iBAAiB,EACjB,kBAAkB,GACnB,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;IAEzD,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,EAAE;QAC9B,IAAI,SAAS,KAAK,iBAAiB,CAAC,IAAI,IAAI,iBAAiB,EAAE,CAAC;YAC9D,OAAO,kBAAkB;gBACvB,CAAC,CAAC,GAAG,iBAAiB,SAAS,kBAAkB,EAAE;gBACnD,CAAC,CAAC,GAAG,iBAAiB,QAAQ,CAAC;QACnC,CAAC;QACD,IAAI,iBAAiB,KAAK,SAAS,EAAE,CAAC;YACpC,OAAO,SAAS,KAAK,iBAAiB,CAAC,GAAG;gBACxC,CAAC,CAAC,GAAG,iBAAiB,eAAe;gBACrC,CAAC,CAAC,OAAO,SAAS,CAAC,MAAM,OAAO,iBAAiB,EAAE,CAAC;QACxD,CAAC;QACD,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;YACrC,OAAO,GAAG,kBAAkB,eAAe,CAAC;QAC9C,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC,EAAE,CAAC,SAAS,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;IAEzE,MAAM,cAAc,GAAG,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;IAE7E,IAAI,YAA6B,CAAC;IAClC,IAAI,SAAS,KAAK,iBAAiB,CAAC,IAAI,EAAE,CAAC;QACzC,YAAY,GAAG,KAAC,iBAAiB,IAAC,IAAI,EAAE,SAAS,KAAM,cAAc,GAAI,CAAC;IAC5E,CAAC;SAAM,IAAI,SAAS,KAAK,iBAAiB,CAAC,GAAG,EAAE,CAAC;QAC/C,YAAY,GAAG,KAAC,kBAAkB,IAAC,IAAI,EAAE,SAAS,KAAM,cAAc,GAAI,CAAC;IAC7E,CAAC;SAAM,CAAC;QACN,YAAY,GAAG,KAAC,gBAAgB,IAAC,IAAI,EAAE,SAAS,KAAM,cAAc,GAAI,CAAC;IAC3E,CAAC;IAED,OAAO,CACL,eAAK,SAAS,EAAC,oBAAoB,aAChC,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,CACrB,cAAK,SAAS,EAAC,6BAA6B,YACzC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAClB,eAAmB,SAAS,EAAC,iCAAiC,aAC5D,eAAM,SAAS,EAAC,kCAAkC,YAC/C,CAAC,CAAC,KAAK,GACH,EACP,eAAM,SAAS,EAAC,kCAAkC,YAC/C,CAAC,CAAC,KAAK,GACH,EACN,CAAC,CAAC,MAAM,KAAK,SAAS,IAAI,CACzB,eAAM,SAAS,EAAC,mCAAmC,YAChD,CAAC,CAAC,MAAM,GACJ,CACR,KAXO,CAAC,CAAC,KAAK,CAYX,CACP,CAAC,GACE,CACP,EAEA,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,CACvB,cAAK,SAAS,EAAC,mCAAmC,YAAE,YAAY,GAAO,CACxE,EAED,cAAK,SAAS,EAAC,+BAA+B,YAC5C,KAAC,eAAe,IAAC,KAAK,EAAE,OAAO,GAAI,GAC/B,IACF,CACP,CAAC;AACJ,CAAC,CAAC"}
@@ -20,6 +20,7 @@ export interface LegendAIGridAnalysis {
20
20
  chartType: LegendAIChartType;
21
21
  chartData: LegendAIChartDataPoint[];
22
22
  numericColumnName: string | undefined;
23
+ categoryColumnName: string | undefined;
23
24
  }
24
25
  export declare function computeKeyMetrics(gridData: LegendAIGridData): LegendAIKeyMetric[];
25
26
  export declare function inferChartType(gridData: LegendAIGridData): LegendAIChartType;
@@ -1 +1 @@
1
- {"version":3,"file":"LegendAIAnalysisUtils.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIAnalysisUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,iBAAiB,EACtB,KAAK,sBAAsB,EAC3B,iBAAiB,EAClB,MAAM,kDAAkD,CAAC;AAC1D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAqB5D,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,iBAAiB,EAAE,CAAC;IAC7B,SAAS,EAAE,iBAAiB,CAAC;IAC7B,SAAS,EAAE,sBAAsB,EAAE,CAAC;IACpC,iBAAiB,EAAE,MAAM,GAAG,SAAS,CAAC;CACvC;AAiID,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,gBAAgB,GACzB,iBAAiB,EAAE,CAKrB;AAoCD,wBAAgB,cAAc,CAAC,QAAQ,EAAE,gBAAgB,GAAG,iBAAiB,CAK5E;AAgED,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,gBAAgB,GACzB,sBAAsB,EAAE,CAE1B;AAED,wBAAgB,eAAe,CAC7B,QAAQ,EAAE,gBAAgB,GACzB,sBAAsB,EAAE,CAE1B;AAgBD,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,gBAAgB,GACzB,MAAM,GAAG,SAAS,CAEpB;AAED,wBAAgB,eAAe,CAC7B,QAAQ,EAAE,gBAAgB,GACzB,oBAAoB,CAatB"}
1
+ {"version":3,"file":"LegendAIAnalysisUtils.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIAnalysisUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,iBAAiB,EACtB,KAAK,sBAAsB,EAC3B,iBAAiB,EAClB,MAAM,kDAAkD,CAAC;AAC1D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAuB5D,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,iBAAiB,EAAE,CAAC;IAC7B,SAAS,EAAE,iBAAiB,CAAC;IAC7B,SAAS,EAAE,sBAAsB,EAAE,CAAC;IACpC,iBAAiB,EAAE,MAAM,GAAG,SAAS,CAAC;IACtC,kBAAkB,EAAE,MAAM,GAAG,SAAS,CAAC;CACxC;AAoQD,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,gBAAgB,GACzB,iBAAiB,EAAE,CAMrB;AA0CD,wBAAgB,cAAc,CAAC,QAAQ,EAAE,gBAAgB,GAAG,iBAAiB,CAK5E;AAgED,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,gBAAgB,GACzB,sBAAsB,EAAE,CAE1B;AAED,wBAAgB,eAAe,CAC7B,QAAQ,EAAE,gBAAgB,GACzB,sBAAsB,EAAE,CAE1B;AAgBD,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,gBAAgB,GACzB,MAAM,GAAG,SAAS,CAEpB;AAgBD,wBAAgB,eAAe,CAC7B,QAAQ,EAAE,gBAAgB,GACzB,oBAAoB,CActB"}
@@ -19,37 +19,58 @@ const CHART_PALETTE_COUNT = 10;
19
19
  const MAX_CHART_ITEMS = 10;
20
20
  const TOP_N_ITEMS = 5;
21
21
  const MAX_PROFILE_SAMPLE = 1000;
22
- const MAX_KEY_METRICS = 4;
22
+ const MAX_KEY_METRICS = 6;
23
23
  const MAX_METRIC_COLUMNS = 5;
24
24
  const MAX_BAR_CHART_ROWS = 20;
25
25
  const MAX_PIE_CHART_ROWS = 6;
26
+ const CONCENTRATION_THRESHOLD = 0.6;
27
+ const DATE_NAME_PATTERN = /(?:date|time|timestamp)|(?:_dt$)/i;
28
+ function looksLikeDateValue(v) {
29
+ if (!isString(v)) {
30
+ return false;
31
+ }
32
+ // ISO date patterns: 2024-01-15, 2024-01-15T10:30:00, etc.
33
+ return /^\d{4}-\d{2}-\d{2}/.test(v);
34
+ }
35
+ function profileRowValues(rows, field) {
36
+ let nonNullCount = 0;
37
+ let stringCount = 0;
38
+ let dateCount = 0;
39
+ const numericValues = [];
40
+ const unique = new Set();
41
+ for (const r of rows) {
42
+ const v = r[field];
43
+ if (v === null || v === undefined) {
44
+ continue;
45
+ }
46
+ nonNullCount++;
47
+ unique.add(isString(v) ? v : JSON.stringify(v));
48
+ if (isNumber(v)) {
49
+ numericValues.push(v);
50
+ }
51
+ if (isString(v)) {
52
+ stringCount++;
53
+ if (looksLikeDateValue(v)) {
54
+ dateCount++;
55
+ }
56
+ }
57
+ }
58
+ return { nonNullCount, stringCount, dateCount, numericValues, unique };
59
+ }
26
60
  function profileColumns(gridData) {
27
61
  const rows = gridData.rowData.length > MAX_PROFILE_SAMPLE
28
62
  ? gridData.rowData.slice(0, MAX_PROFILE_SAMPLE)
29
63
  : gridData.rowData;
30
64
  return gridData.columnDefs.map((col) => {
31
65
  const field = col.field ?? col.colId ?? '';
32
- let nonNullCount = 0;
33
- let stringCount = 0;
34
- const numericValues = [];
35
- const unique = new Set();
36
- for (const r of rows) {
37
- const v = r[field];
38
- if (v !== null && v !== undefined) {
39
- nonNullCount++;
40
- unique.add(String(v));
41
- if (isNumber(v)) {
42
- numericValues.push(v);
43
- }
44
- if (isString(v)) {
45
- stringCount++;
46
- }
47
- }
48
- }
66
+ const { nonNullCount, stringCount, dateCount, numericValues, unique } = profileRowValues(rows, field);
67
+ const isDateCol = (dateCount > 0 && dateCount >= nonNullCount * 0.8) ||
68
+ (nonNullCount === 0 && DATE_NAME_PATTERN.test(field));
49
69
  return {
50
70
  name: field,
51
71
  isNumeric: numericValues.length === nonNullCount && nonNullCount > 0,
52
72
  isString: nonNullCount > 0 && stringCount === nonNullCount,
73
+ isDate: isDateCol,
53
74
  uniqueCount: unique.size,
54
75
  nonNullCount,
55
76
  numericValues,
@@ -103,13 +124,73 @@ function computeNumericMetrics(col, metrics) {
103
124
  });
104
125
  }
105
126
  }
106
- function computeKeyMetricsFromProfiles(profiles, rowCount) {
127
+ function computeConcentrationMetric(profiles, gridData) {
128
+ const catCol = findBestCategoricalColumn(profiles, gridData.rowData.length);
129
+ if (!catCol || catCol.uniqueCount < 3) {
130
+ return undefined;
131
+ }
132
+ const freqMap = new Map();
133
+ for (const row of gridData.rowData) {
134
+ const val = String(row[catCol.name] ?? '');
135
+ if (val.length > 0) {
136
+ freqMap.set(val, (freqMap.get(val) ?? 0) + 1);
137
+ }
138
+ }
139
+ const sorted = [...freqMap.values()].sort((a, b) => b - a);
140
+ const topCount = Math.min(3, sorted.length);
141
+ let topSum = 0;
142
+ for (let i = 0; i < topCount; i++) {
143
+ const count = sorted[i];
144
+ if (count !== undefined) {
145
+ topSum += count;
146
+ }
147
+ }
148
+ const pct = topSum / gridData.rowData.length;
149
+ if (pct >= CONCENTRATION_THRESHOLD) {
150
+ return {
151
+ label: `Top ${topCount} ${catCol.name}`,
152
+ value: `${Math.round(pct * 100)}%`,
153
+ detail: 'of all records',
154
+ };
155
+ }
156
+ return undefined;
157
+ }
158
+ function computeDateRangeMetric(profiles, gridData) {
159
+ const dateCol = profiles.find((c) => c.isDate);
160
+ if (!dateCol) {
161
+ return undefined;
162
+ }
163
+ let minDate;
164
+ let maxDate;
165
+ for (const row of gridData.rowData) {
166
+ const v = row[dateCol.name];
167
+ if (isString(v) && v.length >= 10) {
168
+ const d = v.slice(0, 10);
169
+ if (!minDate || d < minDate) {
170
+ minDate = d;
171
+ }
172
+ if (!maxDate || d > maxDate) {
173
+ maxDate = d;
174
+ }
175
+ }
176
+ }
177
+ if (minDate && maxDate && minDate !== maxDate) {
178
+ return {
179
+ label: 'Date Range',
180
+ value: `${minDate} – ${maxDate}`,
181
+ };
182
+ }
183
+ return undefined;
184
+ }
185
+ function computeKeyMetricsFromProfiles(profiles, rowCount, gridData) {
107
186
  const metrics = [];
108
187
  metrics.push({
109
188
  label: 'Total Rows',
110
189
  value: rowCount.toLocaleString(),
111
190
  });
112
- const numericCol = profiles.find((c) => c.isNumeric && c.numericValues.length > 0);
191
+ // Primary numeric column
192
+ const numericCols = profiles.filter((c) => c.isNumeric && c.numericValues.length > 0);
193
+ const numericCol = numericCols[0];
113
194
  if (numericCol) {
114
195
  computeNumericMetrics(numericCol, metrics);
115
196
  }
@@ -120,14 +201,46 @@ function computeKeyMetricsFromProfiles(profiles, rowCount) {
120
201
  value: stringCol.uniqueCount.toLocaleString(),
121
202
  });
122
203
  }
204
+ // Second numeric column (if distinct from primary)
205
+ const secondNumeric = numericCols[1];
206
+ if (secondNumeric && metrics.length < MAX_KEY_METRICS) {
207
+ let sum2 = 0;
208
+ for (const n of secondNumeric.numericValues) {
209
+ sum2 += n;
210
+ }
211
+ const avg2 = sum2 / secondNumeric.numericValues.length;
212
+ metrics.push({
213
+ label: `Avg ${secondNumeric.name}`,
214
+ value: formatNumber(avg2),
215
+ });
216
+ }
217
+ // Date range metric
218
+ if (gridData && metrics.length < MAX_KEY_METRICS) {
219
+ const dateMetric = computeDateRangeMetric(profiles, gridData);
220
+ if (dateMetric) {
221
+ metrics.push(dateMetric);
222
+ }
223
+ }
224
+ // Concentration metric (top-3 share)
225
+ if (gridData && metrics.length < MAX_KEY_METRICS) {
226
+ const concentrationMetric = computeConcentrationMetric(profiles, gridData);
227
+ if (concentrationMetric) {
228
+ metrics.push(concentrationMetric);
229
+ }
230
+ }
123
231
  return metrics.slice(0, MAX_KEY_METRICS);
124
232
  }
125
233
  export function computeKeyMetrics(gridData) {
126
- return computeKeyMetricsFromProfiles(profileColumns(gridData), gridData.rowData.length);
234
+ return computeKeyMetricsFromProfiles(profileColumns(gridData), gridData.rowData.length, gridData);
127
235
  }
128
236
  function inferChartTypeFromProfiles(profiles, rowCount) {
129
237
  const numericCols = profiles.filter((c) => c.isNumeric);
130
238
  const stringCols = profiles.filter((c) => c.isString && c.uniqueCount > 1);
239
+ const dateCols = profiles.filter((c) => c.isDate);
240
+ // Date + numeric → LINE chart (time series)
241
+ if (dateCols.length >= 1 && numericCols.length >= 1 && rowCount > 1) {
242
+ return LegendAIChartType.LINE;
243
+ }
131
244
  if (stringCols.length >= 1 &&
132
245
  numericCols.length >= 1 &&
133
246
  rowCount <= MAX_BAR_CHART_ROWS) {
@@ -217,17 +330,26 @@ function findNumericColumnNameFromProfiles(profiles, gridData) {
217
330
  export function findNumericColumnName(gridData) {
218
331
  return findNumericColumnNameFromProfiles(profileColumns(gridData), gridData);
219
332
  }
333
+ function findCategoryColumnNameFromProfiles(profiles, gridData) {
334
+ const catCol = findBestCategoricalColumn(profiles, gridData.rowData.length);
335
+ if (!catCol) {
336
+ return undefined;
337
+ }
338
+ const colDef = gridData.columnDefs.find((c) => (c.field ?? c.colId ?? '') === catCol.name);
339
+ return colDef?.headerName ?? colDef?.field;
340
+ }
220
341
  export function analyzeGridData(gridData) {
221
342
  const profiles = profileColumns(gridData);
222
343
  const rowCount = gridData.rowData.length;
223
344
  const chartType = inferChartTypeFromProfiles(profiles, rowCount);
224
345
  return {
225
- metrics: computeKeyMetricsFromProfiles(profiles, rowCount),
346
+ metrics: computeKeyMetricsFromProfiles(profiles, rowCount, gridData),
226
347
  chartType,
227
348
  chartData: chartType === LegendAIChartType.NONE
228
349
  ? []
229
350
  : computeChartDataFromProfiles(profiles, gridData),
230
351
  numericColumnName: findNumericColumnNameFromProfiles(profiles, gridData),
352
+ categoryColumnName: findCategoryColumnNameFromProfiles(profiles, gridData),
231
353
  };
232
354
  }
233
355
  //# sourceMappingURL=LegendAIAnalysisUtils.js.map