@askdb/core 1.0.0-beta.40 → 1.0.0-beta.41
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/ask.d.ts +9 -0
- package/dist/ask.d.ts.map +1 -1
- package/dist/ask.js +1 -0
- package/dist/ask.js.map +1 -1
- package/dist/enrichment/suggest.d.ts +7 -0
- package/dist/enrichment/suggest.d.ts.map +1 -1
- package/dist/enrichment/suggest.js +7 -1
- package/dist/enrichment/suggest.js.map +1 -1
- package/dist/exec/postgres.d.ts +6 -0
- package/dist/exec/postgres.d.ts.map +1 -0
- package/dist/exec/postgres.js +30 -0
- package/dist/exec/postgres.js.map +1 -0
- package/dist/modes/post-execute-log.d.ts +9 -0
- package/dist/modes/post-execute-log.d.ts.map +1 -0
- package/dist/modes/post-execute-log.js +26 -0
- package/dist/modes/post-execute-log.js.map +1 -0
- package/dist/sql/generate.d.ts +9 -0
- package/dist/sql/generate.d.ts.map +1 -1
- package/dist/sql/generate.js +7 -1
- package/dist/sql/generate.js.map +1 -1
- package/package.json +4 -4
- package/dist/ai/provider.d.ts +0 -116
- package/dist/ai/provider.d.ts.map +0 -1
- package/dist/ai/provider.js +0 -255
- package/dist/ai/provider.js.map +0 -1
package/dist/ask.d.ts
CHANGED
|
@@ -13,6 +13,7 @@ export type AskDialectGenerateOptions = {
|
|
|
13
13
|
explain?: boolean;
|
|
14
14
|
omitSensitiveIdentifiersFromNlToSqlPrompt?: boolean;
|
|
15
15
|
generateText?: typeof defaultGenerateText;
|
|
16
|
+
providerOptions?: Record<string, unknown>;
|
|
16
17
|
prebuiltDdl?: string;
|
|
17
18
|
tenantPolicy?: import("./schema/v2/tenant-policy.js").NormalizedTenantPolicy;
|
|
18
19
|
tenantScope?: TenantScope;
|
|
@@ -52,6 +53,14 @@ export type AskDialectInput = BuiltInDialectId | DialectSpec | AskDialect;
|
|
|
52
53
|
/** Generic deps the pipeline forwards into the dialect (test-time mock for `generateText`, etc.). */
|
|
53
54
|
export type AskGenerateDeps = {
|
|
54
55
|
generateText?: typeof defaultGenerateText;
|
|
56
|
+
/**
|
|
57
|
+
* Forwarded verbatim to the underlying `generateText` call's `providerOptions`.
|
|
58
|
+
* Resolve provider-portable reasoning/latency effort (e.g. via `@askdb/ai`'s
|
|
59
|
+
* `resolveProviderOptions`) and pass the result here — core stays BYO-model
|
|
60
|
+
* and does not interpret or validate this bag. Unset preserves current
|
|
61
|
+
* `generateText` call shapes exactly.
|
|
62
|
+
*/
|
|
63
|
+
providerOptions?: Record<string, unknown>;
|
|
55
64
|
};
|
|
56
65
|
export type AskPipelineOptions = {
|
|
57
66
|
question: string;
|
package/dist/ask.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ask.d.ts","sourceRoot":"","sources":["../src/ask.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AAC9D,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAE7D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAsB,KAAK,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACxE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAGtD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EACL,KAAK,gBAAgB,EACrB,KAAK,WAAW,EAGjB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EAEL,KAAK,mBAAmB,EAExB,KAAK,aAAa,EACnB,MAAM,8BAA8B,CAAC;AAGtC,0EAA0E;AAC1E,MAAM,MAAM,yBAAyB,GAAG;IACtC,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,yCAAyC,CAAC,EAAE,OAAO,CAAC;IACpD,YAAY,CAAC,EAAE,OAAO,mBAAmB,CAAC;IAC1C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,OAAO,8BAA8B,EAAE,sBAAsB,CAAC;IAC7E,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B,CAAC;AAEF,kGAAkG;AAClG,MAAM,MAAM,QAAQ,GAAG;IACrB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;CAC5B,CAAC;AAEF,sGAAsG;AACtG,MAAM,MAAM,wBAAwB,GAAG;IACrC,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,eAAe,CAAC,EAAE,OAAO,2BAA2B,EAAE,qBAAqB,CAAC;IAC5E,KAAK,CAAC,EAAE,QAAQ,CAAC;CAClB,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,MAAM,UAAU,GAAG;IACvB,QAAQ,CACN,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,mBAAmB,EAC3B,KAAK,EAAE,kBAAkB,EACzB,OAAO,CAAC,EAAE,yBAAyB,GAClC,OAAO,CAAC,wBAAwB,CAAC,CAAC;CACtC,CAAC;AAEF;;;;;GAKG;AACH,MAAM,MAAM,eAAe,GAAG,gBAAgB,GAAG,WAAW,GAAG,UAAU,CAAC;AAE1E,qGAAqG;AACrG,MAAM,MAAM,eAAe,GAAG;IAC5B,YAAY,CAAC,EAAE,OAAO,mBAAmB,CAAC;CAC3C,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,mBAAmB,CAAC;IAC5B,KAAK,EAAE,kBAAkB,CAAC;IAC1B;;;OAGG;IACH,OAAO,EAAE,eAAe,CAAC;IACzB,+EAA+E;IAC/E,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;;OAGG;IACH,yCAAyC,CAAC,EAAE,OAAO,CAAC;IACpD,IAAI,CAAC,EAAE,eAAe,CAAC;IACvB,wFAAwF;IACxF,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB;;;OAGG;IACH,IAAI,CAAC,EAAE,WAAW,CAAC;IACnB;;;;;;;;;OASG;IACH,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,mDAAmD;IACnD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;;;OAIG;IACH,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC;;;;;;OAMG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B;;;;OAIG;IACH,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B;;;OAGG;IACH,aAAa,CAAC,EAAE,mBAAmB,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,eAAe,CAAC,EAAE,OAAO,2BAA2B,EAAE,qBAAqB,CAAC;IAC5E,YAAY,CAAC,EAAE,OAAO,EAAE,CAAC;IACzB,cAAc,CAAC,EAAE,aAAa,EAAE,CAAC;IACjC,+FAA+F;IAC/F,KAAK,CAAC,EAAE,QAAQ,CAAC;CAClB,CAAC;AAEF,wBAAsB,GAAG,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,iBAAiB,CAAC,
|
|
1
|
+
{"version":3,"file":"ask.d.ts","sourceRoot":"","sources":["../src/ask.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AAC9D,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAE7D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAsB,KAAK,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACxE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAGtD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EACL,KAAK,gBAAgB,EACrB,KAAK,WAAW,EAGjB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EAEL,KAAK,mBAAmB,EAExB,KAAK,aAAa,EACnB,MAAM,8BAA8B,CAAC;AAGtC,0EAA0E;AAC1E,MAAM,MAAM,yBAAyB,GAAG;IACtC,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,yCAAyC,CAAC,EAAE,OAAO,CAAC;IACpD,YAAY,CAAC,EAAE,OAAO,mBAAmB,CAAC;IAC1C,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC1C,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,OAAO,8BAA8B,EAAE,sBAAsB,CAAC;IAC7E,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B,CAAC;AAEF,kGAAkG;AAClG,MAAM,MAAM,QAAQ,GAAG;IACrB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;CAC5B,CAAC;AAEF,sGAAsG;AACtG,MAAM,MAAM,wBAAwB,GAAG;IACrC,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,eAAe,CAAC,EAAE,OAAO,2BAA2B,EAAE,qBAAqB,CAAC;IAC5E,KAAK,CAAC,EAAE,QAAQ,CAAC;CAClB,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,MAAM,UAAU,GAAG;IACvB,QAAQ,CACN,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,mBAAmB,EAC3B,KAAK,EAAE,kBAAkB,EACzB,OAAO,CAAC,EAAE,yBAAyB,GAClC,OAAO,CAAC,wBAAwB,CAAC,CAAC;CACtC,CAAC;AAEF;;;;;GAKG;AACH,MAAM,MAAM,eAAe,GAAG,gBAAgB,GAAG,WAAW,GAAG,UAAU,CAAC;AAE1E,qGAAqG;AACrG,MAAM,MAAM,eAAe,GAAG;IAC5B,YAAY,CAAC,EAAE,OAAO,mBAAmB,CAAC;IAC1C;;;;;;OAMG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC3C,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,mBAAmB,CAAC;IAC5B,KAAK,EAAE,kBAAkB,CAAC;IAC1B;;;OAGG;IACH,OAAO,EAAE,eAAe,CAAC;IACzB,+EAA+E;IAC/E,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;;OAGG;IACH,yCAAyC,CAAC,EAAE,OAAO,CAAC;IACpD,IAAI,CAAC,EAAE,eAAe,CAAC;IACvB,wFAAwF;IACxF,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB;;;OAGG;IACH,IAAI,CAAC,EAAE,WAAW,CAAC;IACnB;;;;;;;;;OASG;IACH,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,mDAAmD;IACnD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB;;;;OAIG;IACH,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC;;;;;;OAMG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B;;;;OAIG;IACH,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B;;;OAGG;IACH,aAAa,CAAC,EAAE,mBAAmB,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,eAAe,CAAC,EAAE,OAAO,2BAA2B,EAAE,qBAAqB,CAAC;IAC5E,YAAY,CAAC,EAAE,OAAO,EAAE,CAAC;IACzB,cAAc,CAAC,EAAE,aAAa,EAAE,CAAC;IACjC,+FAA+F;IAC/F,KAAK,CAAC,EAAE,QAAQ,CAAC;CAClB,CAAC;AAEF,wBAAsB,GAAG,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,iBAAiB,CAAC,CA6DjF"}
|
package/dist/ask.js
CHANGED
|
@@ -32,6 +32,7 @@ export async function ask(options) {
|
|
|
32
32
|
explain: explainRequested,
|
|
33
33
|
omitSensitiveIdentifiersFromNlToSqlPrompt: omitSensitive || undefined,
|
|
34
34
|
generateText: options.deps?.generateText,
|
|
35
|
+
providerOptions: options.deps?.providerOptions,
|
|
35
36
|
prebuiltDdl,
|
|
36
37
|
tenantPolicy,
|
|
37
38
|
tenantScope: options.tenantScope,
|
package/dist/ask.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ask.js","sourceRoot":"","sources":["../src/ask.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAExD,OAAO,EAAE,kBAAkB,EAAoB,MAAM,kBAAkB,CAAC;AAExE,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAGvE,OAAO,EAGL,cAAc,EACd,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,EACL,gBAAgB,GAIjB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;
|
|
1
|
+
{"version":3,"file":"ask.js","sourceRoot":"","sources":["../src/ask.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAExD,OAAO,EAAE,kBAAkB,EAAoB,MAAM,kBAAkB,CAAC;AAExE,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAGvE,OAAO,EAGL,cAAc,EACd,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,EACL,gBAAgB,GAIjB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;AA8IrE,MAAM,CAAC,KAAK,UAAU,GAAG,CAAC,OAA2B;IACnD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAC9B,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,kBAAkB,CAAC;IAChD,MAAM,EAAE,IAAI,CAAC,EAAE,KAAK,EAAE,aAAa,CAAC,YAAY,EAAE,IAAI,EAAE,EAAE,eAAe,CAAC,CAAC;IAE3E,iFAAiF;IACjF,MAAM,YAAY,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1F,IAAI,YAAY,EAAE,CAAC;QACjB,mBAAmB,CAAC,YAAY,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACvD,MAAM,EAAE,IAAI,CACV;YACE,KAAK,EAAE,aAAa,CAAC,oBAAoB;YACzC,SAAS,EAAE,OAAO,CAAC,WAAY,CAAC,MAAM,CAAC,IAAI;YAC3C,WAAW,EAAE,YAAY,CAAC,WAAW;SACtC,EACD,wBAAwB,CACzB,CAAC;IACJ,CAAC;IAED,MAAM,gBAAgB,GAAG,OAAO,CAAC,OAAO,IAAI,KAAK,CAAC;IAClD,MAAM,aAAa,GAAG,OAAO,CAAC,yCAAyC,IAAI,KAAK,CAAC;IACjF,MAAM,WAAW,GAAG,MAAM,gBAAgB,CAAC;QACzC,OAAO;QACP,MAAM;QACN,aAAa;KACd,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,QAAQ,CACtC,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,MAAM,EACd,OAAO,CAAC,KAAK,EACb;QACE,MAAM;QACN,OAAO,EAAE,gBAAgB;QACzB,yCAAyC,EAAE,aAAa,IAAI,SAAS;QACrE,YAAY,EAAE,OAAO,CAAC,IAAI,EAAE,YAAY;QACxC,eAAe,EAAE,OAAO,CAAC,IAAI,EAAE,eAAe;QAC9C,WAAW;QACX,YAAY;QACZ,WAAW,EAAE,OAAO,CAAC,WAAW;KACjC,CACF,CAAC;IACF,IAAI,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC;IACxB,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC;IAClC,MAAM,eAAe,GAAG,SAAS,CAAC,eAAe,CAAC;IAClD,MAAM,MAAM,GAAsB,EAAE,GAAG,EAAE,CAAC;IAC1C,IAAI,OAAO,KAAK,SAAS;QAAE,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC;IACpD,IAAI,eAAe,KAAK,SAAS;QAAE,MAAM,CAAC,eAAe,GAAG,eAAe,CAAC;IAC5E,IAAI,SAAS,CAAC,KAAK,KAAK,SAAS;QAAE,MAAM,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC;IAElE,IAAI,YAAY,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,IAAI,GAAG,OAAO,CAAC,aAAa,IAAI,UAAU,CAAC;QACjD,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,EAAE,YAAY,EAAE,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QAChF,MAAM,CAAC,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC;QAC1B,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM,CAAC,cAAc,GAAG,QAAQ,CAAC,QAAQ,CAAC;QAC5E,IAAI,QAAQ,CAAC,IAAI,KAAK,YAAY,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjE,MAAM,CAAC,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC;QACxC,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,KAA+B;IACnD,OAAO,OAAQ,KAAoB,CAAC,QAAQ,KAAK,UAAU,CAAC;AAC9D,CAAC;AAED;;;;GAIG;AACH,SAAS,cAAc,CAAC,KAAsB;IAC5C,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CACb,uBAAuB,KAAK,6EAA6E,CAC1G,CAAC;QACJ,CAAC;QACD,OAAO,aAAa,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;IAC9C,CAAC;IACD,IAAI,YAAY,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACtC,OAAO,aAAa,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,aAAa,CAAC,IAAiB;IACtC,OAAO;QACL,QAAQ,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,CAC7C,iBAAiB,CAAC,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC;KAC5D,CAAC;AACJ,CAAC;AAED,2EAA2E;AAC3E,MAAM,kCAAkC,GAAG,EAAE,CAAC;AAC9C,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAE9B,KAAK,UAAU,gBAAgB,CAAC,IAI/B;IACC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC;IAChD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACpC,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IAEjC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAChC,MAAM,EAAE,IAAI,CACV,EAAE,KAAK,EAAE,aAAa,CAAC,wBAAwB,EAAE,MAAM,EAAE,eAAe,EAAE,EAC1E,8DAA8D,CAC/D,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,wBAAwB,IAAI,kCAAkC,CAAC;IACzF,MAAM,KAAK,GAAG,OAAO,CAAC,qBAAqB,IAAI,MAAM,CAAC,iBAAiB,CAAC;IACxE,IAAI,KAAK,IAAI,SAAS,EAAE,CAAC;QACvB,MAAM,EAAE,IAAI,CACV;YACE,KAAK,EAAE,aAAa,CAAC,wBAAwB;YAC7C,MAAM,EAAE,iBAAiB;YACzB,WAAW,EAAE,KAAK;YAClB,SAAS;SACV,EACD,mEAAmE,CACpE,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,CAAC,GAAG,OAAO,CAAC,UAAU,IAAI,mBAAmB,CAAC;IACpD,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC;QAC9B,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,CAAC;QACD,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE;KAC9C,CAAC,CAAC;IACH,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,EAAE,IAAI,CACV;YACE,KAAK,EAAE,aAAa,CAAC,wBAAwB;YAC7C,MAAM,EAAE,YAAY;YACpB,CAAC;YACD,SAAS;YACT,WAAW,EAAE,KAAK,KAAK,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK;SAC/D,EACD,+CAA+C,CAChD,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,KAAK,GAAG,sBAAsB,CAAC;QACnC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,OAAO;QACP,kCAAkC,EAAE,aAAa;KAClD,CAAC,CAAC;IACH,MAAM,EAAE,IAAI,CACV;QACE,KAAK,EAAE,aAAa,CAAC,qBAAqB;QAC1C,CAAC;QACD,WAAW,EAAE,OAAO,CAAC,MAAM;QAC3B,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,SAAS;QACT,WAAW,EAAE,KAAK,KAAK,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK;KAC/D,EACD,gCAAgC,CACjC,CAAC;IACF,OAAO,KAAK,CAAC,GAAG,CAAC;AACnB,CAAC;AAED,SAAS,UAAU,CAAC,MAA2B;IAC7C,OAAO,UAAU,IAAI,MAAM,CAAC;AAC9B,CAAC"}
|
|
@@ -7,6 +7,13 @@ export type SuggestEnrichmentDeps = {
|
|
|
7
7
|
maxCandidates?: number;
|
|
8
8
|
/** Sampling temperature. Default 0.4 — small variety, mostly grounded. */
|
|
9
9
|
temperature?: number;
|
|
10
|
+
/**
|
|
11
|
+
* Forwarded verbatim to the underlying `generateText` call's `providerOptions`.
|
|
12
|
+
* Enrichment suggestions are non-critical, so hosts may resolve a lower
|
|
13
|
+
* reasoning/latency effort here than for NL→SQL generation (e.g. via
|
|
14
|
+
* `@askdb/ai`'s `resolveProviderOptions`). Omitted when unset.
|
|
15
|
+
*/
|
|
16
|
+
providerOptions?: Record<string, unknown>;
|
|
10
17
|
};
|
|
11
18
|
/**
|
|
12
19
|
* Ask the model for 1–3 enrichment candidates for the given target. The model
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"suggest.d.ts","sourceRoot":"","sources":["../../src/enrichment/suggest.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AAKzD,OAAO,KAAK,EACV,mBAAmB,EACnB,iBAAiB,EACjB,gBAAgB,EACjB,MAAM,YAAY,CAAC;AAEpB,MAAM,MAAM,qBAAqB,GAAG;IAClC,YAAY,CAAC,EAAE,OAAO,mBAAmB,CAAC;IAC1C,oEAAoE;IACpE,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,0EAA0E;IAC1E,WAAW,CAAC,EAAE,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"suggest.d.ts","sourceRoot":"","sources":["../../src/enrichment/suggest.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AAKzD,OAAO,KAAK,EACV,mBAAmB,EACnB,iBAAiB,EACjB,gBAAgB,EACjB,MAAM,YAAY,CAAC;AAEpB,MAAM,MAAM,qBAAqB,GAAG;IAClC,YAAY,CAAC,EAAE,OAAO,mBAAmB,CAAC;IAC1C,oEAAoE;IACpE,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,0EAA0E;IAC1E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;;OAKG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC3C,CAAC;AAEF;;;;;GAKG;AACH,wBAAsB,iBAAiB,CACrC,MAAM,EAAE,gBAAgB,EACxB,OAAO,EAAE,iBAAiB,EAC1B,KAAK,EAAE,aAAa,EACpB,IAAI,GAAE,qBAA0B,GAC/B,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAkBhC;AAED;;;GAGG;AACH,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,mBAAmB,EAAE,CAU/E"}
|
|
@@ -11,9 +11,15 @@ export async function suggestEnrichment(target, context, model, deps = {}) {
|
|
|
11
11
|
const maxCandidates = deps.maxCandidates ?? 3;
|
|
12
12
|
const result = await generateText({
|
|
13
13
|
model,
|
|
14
|
-
|
|
14
|
+
instructions: ENRICHMENT_SYSTEM_PROMPT,
|
|
15
15
|
prompt: buildEnrichmentUserPrompt(target, context),
|
|
16
16
|
temperature: deps.temperature ?? 0.4,
|
|
17
|
+
// Cast: AskDB's public `providerOptions` type is a plain opaque bag
|
|
18
|
+
// (`Record<string, unknown>`) so callers don't need AI SDK JSON types.
|
|
19
|
+
// We don't interpret or validate it — the AI SDK does that.
|
|
20
|
+
...(deps.providerOptions
|
|
21
|
+
? { providerOptions: deps.providerOptions }
|
|
22
|
+
: {}),
|
|
17
23
|
});
|
|
18
24
|
const text = result.text;
|
|
19
25
|
return parseCandidates(text, maxCandidates);
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"suggest.js","sourceRoot":"","sources":["../../src/enrichment/suggest.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AACzD,OAAO,EACL,wBAAwB,EACxB,yBAAyB,GAC1B,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"suggest.js","sourceRoot":"","sources":["../../src/enrichment/suggest.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AACzD,OAAO,EACL,wBAAwB,EACxB,yBAAyB,GAC1B,MAAM,aAAa,CAAC;AAsBrB;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,MAAwB,EACxB,OAA0B,EAC1B,KAAoB,EACpB,OAA8B,EAAE;IAEhC,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,IAAI,mBAAmB,CAAC;IAC9D,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,CAAC,CAAC;IAE9C,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;QAChC,KAAK;QACL,YAAY,EAAE,wBAAwB;QACtC,MAAM,EAAE,yBAAyB,CAAC,MAAM,EAAE,OAAO,CAAC;QAClD,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,GAAG;QACpC,oEAAoE;QACpE,uEAAuE;QACvE,4DAA4D;QAC5D,GAAG,CAAC,IAAI,CAAC,eAAe;YACtB,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,eAAwE,EAAE;YACpG,CAAC,CAAC,EAAE,CAAC;KACR,CAAC,CAAC;IACH,MAAM,IAAI,GAAI,MAA2B,CAAC,IAAI,CAAC;IAC/C,OAAO,eAAe,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;AAC9C,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,GAAW,EAAE,GAAW;IACtD,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IACvC,MAAM,UAAU,GAA0B,EAAE,CAAC;IAC7C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI;YAAE,SAAS;QACpB,UAAU,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1B,IAAI,UAAU,CAAC,MAAM,IAAI,GAAG;YAAE,MAAM;IACtC,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../../src/exec/postgres.ts"],"names":[],"mappings":"AAGA,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC;CACnB,CAAC;AAEF,wBAAsB,qBAAqB,CAAC,gBAAgB,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAuBzG"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import pg from "pg";
|
|
2
|
+
import { SqlExecutionError } from "../errors.js";
|
|
3
|
+
export async function executeReadOnlySelect(connectionString, sql) {
|
|
4
|
+
const pool = new pg.Pool({ connectionString });
|
|
5
|
+
const client = await pool.connect();
|
|
6
|
+
try {
|
|
7
|
+
await client.query("BEGIN READ ONLY");
|
|
8
|
+
const res = await client.query(sql);
|
|
9
|
+
await client.query("COMMIT");
|
|
10
|
+
const fields = res.fields ?? [];
|
|
11
|
+
const columns = fields.map((f) => f.name);
|
|
12
|
+
const rows = res.rows.map((row) => fields.map((f) => row[f.name]));
|
|
13
|
+
return { columns, rows };
|
|
14
|
+
}
|
|
15
|
+
catch (e) {
|
|
16
|
+
try {
|
|
17
|
+
await client.query("ROLLBACK");
|
|
18
|
+
}
|
|
19
|
+
catch {
|
|
20
|
+
/* ignore */
|
|
21
|
+
}
|
|
22
|
+
const message = e instanceof Error ? e.message : String(e);
|
|
23
|
+
throw new SqlExecutionError(`PostgreSQL execution failed: ${message}`, e);
|
|
24
|
+
}
|
|
25
|
+
finally {
|
|
26
|
+
client.release();
|
|
27
|
+
await pool.end();
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=postgres.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"postgres.js","sourceRoot":"","sources":["../../src/exec/postgres.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAOjD,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,gBAAwB,EAAE,GAAW;IAC/E,MAAM,IAAI,GAAG,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC;IAC/C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;IACpC,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QACtC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,KAAK,CAA2B,GAAG,CAAC,CAAC;QAC9D,MAAM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC7B,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,YAAY;QACd,CAAC;QACD,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAC3D,MAAM,IAAI,iBAAiB,CAAC,gCAAgC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IAC5E,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { AskDbLogger } from "../logging/askdb-logger.js";
|
|
2
|
+
import type { AskDbModeV1 } from "./types.js";
|
|
3
|
+
/**
|
|
4
|
+
* After a successful read-only execute, record how the mode treats optional
|
|
5
|
+
* second-pass model context (summaries). v1 does not call the model again;
|
|
6
|
+
* modes differ only in contract + structured logs (`branch`).
|
|
7
|
+
*/
|
|
8
|
+
export declare function logPostExecuteModeBranch(logger: AskDbLogger | undefined, mode: AskDbModeV1, rowCount: number): void;
|
|
9
|
+
//# sourceMappingURL=post-execute-log.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"post-execute-log.d.ts","sourceRoot":"","sources":["../../src/modes/post-execute-log.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAE9D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAE9C;;;;GAIG;AACH,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,WAAW,GAAG,SAAS,EAC/B,IAAI,EAAE,WAAW,EACjB,QAAQ,EAAE,MAAM,GACf,IAAI,CAwBN"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import { AskDbLogEvent } from "../logging/log-events.js";
|
|
2
|
+
/**
|
|
3
|
+
* After a successful read-only execute, record how the mode treats optional
|
|
4
|
+
* second-pass model context (summaries). v1 does not call the model again;
|
|
5
|
+
* modes differ only in contract + structured logs (`branch`).
|
|
6
|
+
*/
|
|
7
|
+
export function logPostExecuteModeBranch(logger, mode, rowCount) {
|
|
8
|
+
if (mode === "schema_only") {
|
|
9
|
+
logger?.info({
|
|
10
|
+
event: AskDbLogEvent.PipelinePostExecute,
|
|
11
|
+
mode,
|
|
12
|
+
branch: "skipped",
|
|
13
|
+
reason: "schema_only_no_row_data_to_model",
|
|
14
|
+
rowCount,
|
|
15
|
+
}, "post-execute: summary path skipped (schema_only)");
|
|
16
|
+
return;
|
|
17
|
+
}
|
|
18
|
+
logger?.info({
|
|
19
|
+
event: AskDbLogEvent.PipelinePostExecute,
|
|
20
|
+
mode,
|
|
21
|
+
branch: "stub",
|
|
22
|
+
rowCount,
|
|
23
|
+
note: "bounded_results_stub_no_second_llm_call",
|
|
24
|
+
}, "post-execute: bounded_results stub branch (no row payload to model in v1)");
|
|
25
|
+
}
|
|
26
|
+
//# sourceMappingURL=post-execute-log.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"post-execute-log.js","sourceRoot":"","sources":["../../src/modes/post-execute-log.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAGzD;;;;GAIG;AACH,MAAM,UAAU,wBAAwB,CACtC,MAA+B,EAC/B,IAAiB,EACjB,QAAgB;IAEhB,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;QAC3B,MAAM,EAAE,IAAI,CACV;YACE,KAAK,EAAE,aAAa,CAAC,mBAAmB;YACxC,IAAI;YACJ,MAAM,EAAE,SAAS;YACjB,MAAM,EAAE,kCAAkC;YAC1C,QAAQ;SACT,EACD,kDAAkD,CACnD,CAAC;QACF,OAAO;IACT,CAAC;IACD,MAAM,EAAE,IAAI,CACV;QACE,KAAK,EAAE,aAAa,CAAC,mBAAmB;QACxC,IAAI;QACJ,MAAM,EAAE,MAAM;QACd,QAAQ;QACR,IAAI,EAAE,yCAAyC;KAChD,EACD,2EAA2E,CAC5E,CAAC;AACJ,CAAC"}
|
package/dist/sql/generate.d.ts
CHANGED
|
@@ -27,6 +27,15 @@ export type GenerateSqlDeps = {
|
|
|
27
27
|
tenantPolicy?: NormalizedTenantPolicy;
|
|
28
28
|
/** Validated tenant scope from the host. Forwarded from ask(). */
|
|
29
29
|
tenantScope?: TenantScope;
|
|
30
|
+
/**
|
|
31
|
+
* Forwarded verbatim to the underlying `generateText` call's `providerOptions`.
|
|
32
|
+
* Resolve provider-portable reasoning/latency effort (e.g. via `@askdb/ai`'s
|
|
33
|
+
* `resolveProviderOptions`) and pass the result here — core stays BYO-model
|
|
34
|
+
* and does not interpret or validate this bag. Omitted entirely (not sent
|
|
35
|
+
* as an empty object) when unset, so existing `generateText` call shapes
|
|
36
|
+
* are unaffected.
|
|
37
|
+
*/
|
|
38
|
+
providerOptions?: Record<string, unknown>;
|
|
30
39
|
};
|
|
31
40
|
/** Result of NL→SQL generation (always includes `sql`; `explain` when {@link GenerateSqlDeps.explain}). */
|
|
32
41
|
export type GenerateSelectSqlResult = {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["../../src/sql/generate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AACzD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAGzD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAE9D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,KAAK,EAAE,sBAAsB,EAAE,WAAW,EAAE,MAAM,+BAA+B,CAAC;AACzF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAIrD,OAAO,EAA4B,KAAK,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAC7F,OAAO,EAGL,KAAK,sBAAsB,EAC5B,MAAM,eAAe,CAAC;AAEvB,MAAM,MAAM,eAAe,GAAG;IAC5B,YAAY,CAAC,EAAE,OAAO,mBAAmB,CAAC;IAC1C,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,qGAAqG;IACrG,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;;OAGG;IACH,yCAAyC,CAAC,EAAE,OAAO,CAAC;IACpD;;;;;OAKG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,+EAA+E;IAC/E,YAAY,CAAC,EAAE,sBAAsB,CAAC;IACtC,kEAAkE;IAClE,WAAW,CAAC,EAAE,WAAW,CAAC;
|
|
1
|
+
{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["../../src/sql/generate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AACzD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAGzD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAE9D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,KAAK,EAAE,sBAAsB,EAAE,WAAW,EAAE,MAAM,+BAA+B,CAAC;AACzF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAIrD,OAAO,EAA4B,KAAK,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAC7F,OAAO,EAGL,KAAK,sBAAsB,EAC5B,MAAM,eAAe,CAAC;AAEvB,MAAM,MAAM,eAAe,GAAG;IAC5B,YAAY,CAAC,EAAE,OAAO,mBAAmB,CAAC;IAC1C,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,qGAAqG;IACrG,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;;OAGG;IACH,yCAAyC,CAAC,EAAE,OAAO,CAAC;IACpD;;;;;OAKG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,+EAA+E;IAC/E,YAAY,CAAC,EAAE,sBAAsB,CAAC;IACtC,kEAAkE;IAClE,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B;;;;;;;OAOG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC3C,CAAC;AAEF,2GAA2G;AAC3G,MAAM,MAAM,uBAAuB,GAAG;IACpC,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,sBAAsB,CAAC;IACjC,eAAe,CAAC,EAAE,qBAAqB,CAAC;IACxC,iGAAiG;IACjG,KAAK,CAAC,EAAE;QAAE,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,WAAW,EAAE,MAAM,GAAG,IAAI,CAAA;KAAE,CAAC;CACtG,CAAC;AAEF;;;;GAIG;AACH,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,WAAW,EACpB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,mBAAmB,EAC3B,KAAK,EAAE,kBAAkB,EACzB,IAAI,GAAE,eAAoB,GACzB,OAAO,CAAC,uBAAuB,CAAC,CAuHlC"}
|
package/dist/sql/generate.js
CHANGED
|
@@ -30,9 +30,15 @@ export async function generateSelectSql(dialect, question, schema, model, deps =
|
|
|
30
30
|
try {
|
|
31
31
|
const result = await generateText({
|
|
32
32
|
model,
|
|
33
|
-
|
|
33
|
+
instructions: buildNlToSqlSystemPrompt(dialect),
|
|
34
34
|
prompt: buildNlToSqlUserPrompt(dialect, question, schema, ambiguityNotes, logger, nlToSqlSchemaOptions, deps.prebuiltDdl, deps.tenantPolicy, deps.tenantScope),
|
|
35
35
|
temperature: 0,
|
|
36
|
+
// Cast: AskDB's public `providerOptions` type is a plain opaque bag
|
|
37
|
+
// (`Record<string, unknown>`) so callers don't need AI SDK JSON types.
|
|
38
|
+
// We don't interpret or validate it — the AI SDK does that.
|
|
39
|
+
...(deps.providerOptions
|
|
40
|
+
? { providerOptions: deps.providerOptions }
|
|
41
|
+
: {}),
|
|
36
42
|
});
|
|
37
43
|
text = result.text;
|
|
38
44
|
const u = result.usage;
|
package/dist/sql/generate.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"generate.js","sourceRoot":"","sources":["../../src/sql/generate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AAEzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAMzD,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAC/E,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAC3F,OAAO,EAAE,wBAAwB,EAA8B,MAAM,uBAAuB,CAAC;AAC7F,OAAO,EACL,+BAA+B,EAC/B,iBAAiB,GAElB,MAAM,eAAe,CAAC;
|
|
1
|
+
{"version":3,"file":"generate.js","sourceRoot":"","sources":["../../src/sql/generate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,mBAAmB,EAAE,MAAM,IAAI,CAAC;AAEzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAMzD,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAC/E,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAC3F,OAAO,EAAE,wBAAwB,EAA8B,MAAM,uBAAuB,CAAC;AAC7F,OAAO,EACL,+BAA+B,EAC/B,iBAAiB,GAElB,MAAM,eAAe,CAAC;AA2CvB;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAoB,EACpB,QAAgB,EAChB,MAA2B,EAC3B,KAAyB,EACzB,OAAwB,EAAE;IAE1B,mBAAmB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACtC,MAAM,cAAc,GAAG,qBAAqB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC/D,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,IAAI,mBAAmB,CAAC;IAC9D,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC3B,MAAM,oBAAoB,GACxB,IAAI,CAAC,yCAAyC,KAAK,IAAI;QACrD,CAAC,CAAC,EAAE,kCAAkC,EAAE,IAAI,EAAE;QAC9C,CAAC,CAAC,SAAS,CAAC;IAEhB,MAAM,EAAE,IAAI,CACV;QACE,KAAK,EAAE,aAAa,CAAC,qBAAqB;QAC1C,cAAc,EAAE,QAAQ,CAAC,MAAM;QAC/B,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;KACjC,EACD,0BAA0B,CAC3B,CAAC;IAEF,IAAI,CAAC;QACH,IAAI,IAAY,CAAC;QACjB,IAAI,KAAuC,CAAC;QAC5C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;gBAChC,KAAK;gBACL,YAAY,EAAE,wBAAwB,CAAC,OAAO,CAAC;gBAC/C,MAAM,EAAE,sBAAsB,CAC5B,OAAO,EACP,QAAQ,EACR,MAAM,EACN,cAAc,EACd,MAAM,EACN,oBAAoB,EACpB,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,WAAW,CACjB;gBACD,WAAW,EAAE,CAAC;gBACd,oEAAoE;gBACpE,uEAAuE;gBACvE,4DAA4D;gBAC5D,GAAG,CAAC,IAAI,CAAC,eAAe;oBACtB,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,eAAwE,EAAE;oBACpG,CAAC,CAAC,EAAE,CAAC;aACR,CAAC,CAAC;YACH,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YACnB,MAAM,CAAC,GAAI,MAQT,CAAC,KAAK,CAAC;YACT,IAAI,CAAC,EAAE,CAAC;gBACN,MAAM,GAAG,GAAG,CAAC,CAAqB,EAAiB,EAAE,CACnD,OAAO,CAAC,KAAK,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAClD,KAAK,GAAG;oBACN,8DAA8D;oBAC9D,6DAA6D;oBAC7D,YAAY,EAAE,GAAG,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,WAAW,CAAC;oBAClD,gBAAgB,EAAE,GAAG,CAAC,CAAC,CAAC,gBAAgB,IAAI,CAAC,CAAC,YAAY,CAAC;oBAC3D,WAAW,EAAE,GAAG,CAAC,CAAC,CAAC,WAAW,CAAC;iBAChC,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,IAAI,kBAAkB,CAAC,sBAAsB,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QACnE,CAAC;QACD,MAAM,SAAS,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,GAAG,GAAG,iBAAiB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAElD,wEAAwE;QACxE,IAAI,eAAkD,CAAC;QACvD,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAC1C,eAAe,GAAG,wBAAwB,CAAC,GAAG,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YACrF,IAAI,eAAe,CAAC,MAAM,EAAE,CAAC;gBAC3B,MAAM,EAAE,IAAI,CACV,EAAE,KAAK,EAAE,aAAa,CAAC,qBAAqB,EAAE,EAC9C,oCAAoC,CACrC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,EAAE,IAAI,CACV;oBACE,KAAK,EAAE,aAAa,CAAC,qBAAqB;oBAC1C,YAAY,EAAE,eAAe,CAAC,QAAQ,CAAC,MAAM;oBAC7C,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,WAAW;iBAC3C,EACD,0CAA0C,CAC3C,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,+BAA+B,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAChF,MAAM,EAAE,IAAI,CACV;YACE,KAAK,EAAE,aAAa,CAAC,wBAAwB;YAC7C,YAAY,EAAE,GAAG,CAAC,MAAM;SACzB,EACD,6BAA6B,CAC9B,CAAC;QACF,MAAM,GAAG,GAA4B,EAAE,GAAG,EAAE,CAAC;QAC7C,IAAI,OAAO,KAAK,SAAS;YAAE,GAAG,CAAC,OAAO,GAAG,OAAO,CAAC;QACjD,IAAI,eAAe,KAAK,SAAS;YAAE,GAAG,CAAC,eAAe,GAAG,eAAe,CAAC;QACzE,IAAI,KAAK,KAAK,SAAS;YAAE,GAAG,CAAC,KAAK,GAAG,KAAK,CAAC;QAC3C,OAAO,GAAG,CAAC;IACb,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,GAAG,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,EAAE,KAAK,CACX;YACE,KAAK,EAAE,aAAa,CAAC,cAAc;YACnC,KAAK,EAAE,UAAU;YACjB,UAAU,EAAE,GAAG;SAChB,EACD,2BAA2B,CAC5B,CAAC;QACF,MAAM,CAAC,CAAC;IACV,CAAC;AACH,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@askdb/core",
|
|
3
|
-
"version": "1.0.0-beta.
|
|
3
|
+
"version": "1.0.0-beta.41",
|
|
4
4
|
"description": "AskDB core: dialect-agnostic NL→SQL pipeline. Provides ask() orchestration, schema/IR types, modes, logging, and retrieval input. Bring your own dialect (e.g. @askdb/postgres).",
|
|
5
5
|
"keywords": [
|
|
6
6
|
"askdb",
|
|
@@ -38,17 +38,17 @@
|
|
|
38
38
|
"NOTICE"
|
|
39
39
|
],
|
|
40
40
|
"engines": {
|
|
41
|
-
"node": ">=
|
|
41
|
+
"node": ">=22"
|
|
42
42
|
},
|
|
43
43
|
"dependencies": {
|
|
44
|
-
"ai": "^
|
|
44
|
+
"ai": "^7.0.51",
|
|
45
45
|
"gray-matter": "^4.0.3",
|
|
46
46
|
"pino": "^10.3.1",
|
|
47
47
|
"zod": "^4.4.3"
|
|
48
48
|
},
|
|
49
49
|
"devDependencies": {
|
|
50
50
|
"typescript": "^6.0.3",
|
|
51
|
-
"vitest": "^4.1.
|
|
51
|
+
"vitest": "^4.1.10"
|
|
52
52
|
},
|
|
53
53
|
"scripts": {
|
|
54
54
|
"build": "tsc -p tsconfig.build.json",
|
package/dist/ai/provider.d.ts
DELETED
|
@@ -1,116 +0,0 @@
|
|
|
1
|
-
import type { EmbeddingModel, LanguageModel } from "ai";
|
|
2
|
-
/**
|
|
3
|
-
* AI provider selector. AskDB is BYO-LanguageModel at the function level
|
|
4
|
-
* (see `ask()`), but the bundled apps (CLI, HTTP API, Studio, TUI) all need
|
|
5
|
-
* to construct one from environment variables. This module is the single
|
|
6
|
-
* place that does that, so adding a new provider only touches one file.
|
|
7
|
-
*
|
|
8
|
-
* - `openai` — OpenAI or any OpenAI-compatible REST endpoint
|
|
9
|
-
* (set `ASKDB_AI_BASE_URL` for compatible endpoints, including
|
|
10
|
-
* Azure AI Foundry's `/openai/v1` route and self-hosted gateways).
|
|
11
|
-
* - `azure` — Azure OpenAI Service / Azure AI Foundry deployments via
|
|
12
|
-
* `@ai-sdk/azure`. Uses the `resourceName` + deployment-name
|
|
13
|
-
* URL shape with the `api-key` header.
|
|
14
|
-
*/
|
|
15
|
-
export type AskDbAiProvider = "openai" | "azure" | "google";
|
|
16
|
-
export type AskDbAiConfig = {
|
|
17
|
-
provider: AskDbAiProvider;
|
|
18
|
-
apiKey: string;
|
|
19
|
-
/** OpenAI: model id (e.g. `gpt-4o-mini`). Azure: deployment name. */
|
|
20
|
-
model: string;
|
|
21
|
-
/** Custom REST base URL. Optional for OpenAI; for Azure, overrides resourceName. */
|
|
22
|
-
baseURL?: string;
|
|
23
|
-
/** Azure only: resource subdomain (e.g. `my-foundry` for `https://my-foundry.openai.azure.com`). */
|
|
24
|
-
resourceName?: string;
|
|
25
|
-
/** Azure only: API version (e.g. `2024-10-21`). */
|
|
26
|
-
apiVersion?: string;
|
|
27
|
-
};
|
|
28
|
-
export type AskDbAiEnv = Record<string, string | undefined>;
|
|
29
|
-
export type ResolveAskDbAiConfigOptions = {
|
|
30
|
-
/** Default model when no env override is set. */
|
|
31
|
-
modelDefault?: string;
|
|
32
|
-
};
|
|
33
|
-
/**
|
|
34
|
-
* Resolves a provider config from environment variables. Returns `undefined`
|
|
35
|
-
* if no API key is configured (callers treat this as "AI is disabled" rather
|
|
36
|
-
* than erroring, so AI features can be optional).
|
|
37
|
-
*
|
|
38
|
-
* BYO-key design: each provider has its native env vars (e.g. `OPENAI_API_KEY`
|
|
39
|
-
* for OpenAI, `AZURE_OPENAI_API_KEY` for Azure / Foundry). The `ASKDB_AI_*`
|
|
40
|
-
* family is a universal alias set that works across providers — useful when
|
|
41
|
-
* you want a single set of names across deployments.
|
|
42
|
-
*
|
|
43
|
-
* Precedence for the API key (within the selected provider):
|
|
44
|
-
* 1. `ASKDB_AI_API_KEY` — universal alias (primary)
|
|
45
|
-
* 2. provider-native primary — `OPENAI_API_KEY` or `AZURE_OPENAI_API_KEY` / `AZURE_API_KEY`
|
|
46
|
-
* 3. provider-native secondary — `OPENAI_API_KEY_SECONDARY` or `AZURE_OPENAI_API_KEY_SECONDARY`
|
|
47
|
-
* 4. `ASKDB_AI_API_KEY_SECONDARY` — universal rotation fallback
|
|
48
|
-
*
|
|
49
|
-
* **Important:** Only `@askdb/config` reads `process.env` (during dotenv load and while evaluating
|
|
50
|
-
* `askdb.config.*`). Pass `getAskDbRuntimeConfig().ai.aiEnv` from `@askdb/config` after
|
|
51
|
-
* `bootstrapAskDbEnv()`, or an explicit plain object in tests.
|
|
52
|
-
*
|
|
53
|
-
* @param env - String map in the shape of canonical AskDB env keys (from the runtime snapshot).
|
|
54
|
-
* @param options - Optional per-app overrides (e.g. a per-app model env var).
|
|
55
|
-
*/
|
|
56
|
-
export declare function resolveAskDbAiConfig(env: AskDbAiEnv, options?: ResolveAskDbAiConfigOptions): AskDbAiConfig | undefined;
|
|
57
|
-
export type ResolveAskDbEmbeddingConfigOptions = {
|
|
58
|
-
/** Default embedding model/deployment when no env override is set. */
|
|
59
|
-
modelDefault?: string;
|
|
60
|
-
/**
|
|
61
|
-
* Embedding model env var (e.g. `ASKDB_RAG_EMBEDDER_MODEL`).
|
|
62
|
-
*/
|
|
63
|
-
modelEnvVar?: string;
|
|
64
|
-
};
|
|
65
|
-
/**
|
|
66
|
-
* Resolves an embedding model config from the same provider/key/base URL
|
|
67
|
-
* connection as `resolveAskDbAiConfig`, but with embedding-specific model
|
|
68
|
-
* precedence so chat model ids (for example `gpt-4o`) are not used as
|
|
69
|
-
* embedding model ids by accident.
|
|
70
|
-
*
|
|
71
|
-
* **Important:** Only `@askdb/config` reads `process.env` (during dotenv load and while evaluating
|
|
72
|
-
* `askdb.config.*`). Pass `getAskDbRuntimeConfig().ai.aiEnv` from `@askdb/config` after
|
|
73
|
-
* `bootstrapAskDbEnv()`, or an explicit plain object in tests.
|
|
74
|
-
*
|
|
75
|
-
* @param env - String map in the shape of canonical AskDB env keys (from the runtime snapshot).
|
|
76
|
-
* @param options - Optional per-app overrides (e.g. a per-app embedding model env var).
|
|
77
|
-
*/
|
|
78
|
-
export declare function resolveAskDbEmbeddingConfig(env: AskDbAiEnv, options?: ResolveAskDbEmbeddingConfigOptions): AskDbAiConfig | undefined;
|
|
79
|
-
/** Build a {@link LanguageModel} from an already-resolved config. */
|
|
80
|
-
export declare function createAskDbLanguageModel(config: AskDbAiConfig): Promise<LanguageModel>;
|
|
81
|
-
export type CreateAskDbEmbeddingModelOptions = {
|
|
82
|
-
/** Optional dimensionality override for providers that support it. */
|
|
83
|
-
dimensions?: number;
|
|
84
|
-
/** Optional end-user id forwarded to providers that support it. */
|
|
85
|
-
user?: string;
|
|
86
|
-
};
|
|
87
|
-
/** Build an AI SDK text embedding model from an already-resolved config. */
|
|
88
|
-
export declare function createAskDbEmbeddingModel(config: AskDbAiConfig, options?: CreateAskDbEmbeddingModelOptions): Promise<EmbeddingModel<string>>;
|
|
89
|
-
/**
|
|
90
|
-
* Convenience wrapper: resolve config from env, then construct a model.
|
|
91
|
-
* Returns `undefined` when no API key is configured.
|
|
92
|
-
*
|
|
93
|
-
* **Important:** Only `@askdb/config` reads `process.env` (during dotenv load and while evaluating
|
|
94
|
-
* `askdb.config.*`). Pass `getAskDbRuntimeConfig().ai.aiEnv` from `@askdb/config` after
|
|
95
|
-
* `bootstrapAskDbEnv()`, or an explicit plain object in tests.
|
|
96
|
-
*
|
|
97
|
-
* @param env - String map in the shape of canonical AskDB env keys (from the runtime snapshot).
|
|
98
|
-
* @param options - Optional per-app overrides (e.g. a per-app model env var).
|
|
99
|
-
*/
|
|
100
|
-
export declare function createAskDbLanguageModelFromEnv(env: AskDbAiEnv, options?: ResolveAskDbAiConfigOptions): Promise<LanguageModel | undefined>;
|
|
101
|
-
/** Resolve embedding config from env, then construct an AI SDK embedding model.
|
|
102
|
-
*
|
|
103
|
-
* **Important:** Only `@askdb/config` reads `process.env` (during dotenv load and while evaluating
|
|
104
|
-
* `askdb.config.*`). Pass `getAskDbRuntimeConfig().ai.aiEnv` after `bootstrapAskDbEnv()`, or
|
|
105
|
-
* an explicit plain object in tests.
|
|
106
|
-
*
|
|
107
|
-
* @param env - String map in the shape of canonical AskDB env keys (from the runtime snapshot).
|
|
108
|
-
* @param options - Optional per-app overrides plus embedding-specific model/dimensions settings.
|
|
109
|
-
*/
|
|
110
|
-
export declare function createAskDbEmbeddingModelFromEnv(env: AskDbAiEnv, options?: ResolveAskDbEmbeddingConfigOptions & CreateAskDbEmbeddingModelOptions): Promise<EmbeddingModel<string> | undefined>;
|
|
111
|
-
/**
|
|
112
|
-
* Human-readable message describing how to configure AI, used by callers
|
|
113
|
-
* when no key is configured.
|
|
114
|
-
*/
|
|
115
|
-
export declare function askDbAiKeyMissingMessage(context: string): string;
|
|
116
|
-
//# sourceMappingURL=provider.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../../src/ai/provider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAGxD;;;;;;;;;;;;GAYG;AACH,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,OAAO,GAAG,QAAQ,CAAC;AAE5D,MAAM,MAAM,aAAa,GAAG;IAC1B,QAAQ,EAAE,eAAe,CAAC;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,qEAAqE;IACrE,KAAK,EAAE,MAAM,CAAC;IACd,oFAAoF;IACpF,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,oGAAoG;IACpG,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,mDAAmD;IACnD,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;AAE5D,MAAM,MAAM,2BAA2B,GAAG;IACxC,iDAAiD;IACjD,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAeF;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAgB,oBAAoB,CAClC,GAAG,EAAE,UAAU,EACf,OAAO,GAAE,2BAAgC,GACxC,aAAa,GAAG,SAAS,CA8D3B;AAED,MAAM,MAAM,kCAAkC,GAAG;IAC/C,sEAAsE;IACtE,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF;;;;;;;;;;;;GAYG;AACH,wBAAgB,2BAA2B,CACzC,GAAG,EAAE,UAAU,EACf,OAAO,GAAE,kCAAuC,GAC/C,aAAa,GAAG,SAAS,CAiE3B;AAED,qEAAqE;AACrE,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,aAAa,GACpB,OAAO,CAAC,aAAa,CAAC,CA0BxB;AAED,MAAM,MAAM,gCAAgC,GAAG;IAC7C,sEAAsE;IACtE,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,mEAAmE;IACnE,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,4EAA4E;AAC5E,wBAAsB,yBAAyB,CAC7C,MAAM,EAAE,aAAa,EACrB,OAAO,GAAE,gCAAqC,GAC7C,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CA8BjC;AAED;;;;;;;;;;GAUG;AACH,wBAAsB,+BAA+B,CACnD,GAAG,EAAE,UAAU,EACf,OAAO,GAAE,2BAAgC,GACxC,OAAO,CAAC,aAAa,GAAG,SAAS,CAAC,CAIpC;AAED;;;;;;;;GAQG;AACH,wBAAsB,gCAAgC,CACpD,GAAG,EAAE,UAAU,EACf,OAAO,GAAE,kCAAkC,GAAG,gCAAqC,GAClF,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC,CAI7C;AAED;;;GAGG;AACH,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAWhE"}
|
package/dist/ai/provider.js
DELETED
|
@@ -1,255 +0,0 @@
|
|
|
1
|
-
import { createOpenAI } from "@ai-sdk/openai";
|
|
2
|
-
const DEFAULT_MODEL = "gpt-4o-mini";
|
|
3
|
-
const DEFAULT_EMBEDDING_MODEL = "text-embedding-3-small";
|
|
4
|
-
function readProvider(env) {
|
|
5
|
-
const raw = (env.ASKDB_AI_PROVIDER ?? "").toLowerCase().trim();
|
|
6
|
-
if (raw === "" || raw === "openai")
|
|
7
|
-
return "openai";
|
|
8
|
-
if (raw === "azure" || raw === "azure-openai" || raw === "foundry")
|
|
9
|
-
return "azure";
|
|
10
|
-
if (raw === "google")
|
|
11
|
-
return "google";
|
|
12
|
-
throw new Error(`Unknown ASKDB_AI_PROVIDER "${env.ASKDB_AI_PROVIDER}". Expected "openai", "azure", or "google".`);
|
|
13
|
-
}
|
|
14
|
-
/**
|
|
15
|
-
* Resolves a provider config from environment variables. Returns `undefined`
|
|
16
|
-
* if no API key is configured (callers treat this as "AI is disabled" rather
|
|
17
|
-
* than erroring, so AI features can be optional).
|
|
18
|
-
*
|
|
19
|
-
* BYO-key design: each provider has its native env vars (e.g. `OPENAI_API_KEY`
|
|
20
|
-
* for OpenAI, `AZURE_OPENAI_API_KEY` for Azure / Foundry). The `ASKDB_AI_*`
|
|
21
|
-
* family is a universal alias set that works across providers — useful when
|
|
22
|
-
* you want a single set of names across deployments.
|
|
23
|
-
*
|
|
24
|
-
* Precedence for the API key (within the selected provider):
|
|
25
|
-
* 1. `ASKDB_AI_API_KEY` — universal alias (primary)
|
|
26
|
-
* 2. provider-native primary — `OPENAI_API_KEY` or `AZURE_OPENAI_API_KEY` / `AZURE_API_KEY`
|
|
27
|
-
* 3. provider-native secondary — `OPENAI_API_KEY_SECONDARY` or `AZURE_OPENAI_API_KEY_SECONDARY`
|
|
28
|
-
* 4. `ASKDB_AI_API_KEY_SECONDARY` — universal rotation fallback
|
|
29
|
-
*
|
|
30
|
-
* **Important:** Only `@askdb/config` reads `process.env` (during dotenv load and while evaluating
|
|
31
|
-
* `askdb.config.*`). Pass `getAskDbRuntimeConfig().ai.aiEnv` from `@askdb/config` after
|
|
32
|
-
* `bootstrapAskDbEnv()`, or an explicit plain object in tests.
|
|
33
|
-
*
|
|
34
|
-
* @param env - String map in the shape of canonical AskDB env keys (from the runtime snapshot).
|
|
35
|
-
* @param options - Optional per-app overrides (e.g. a per-app model env var).
|
|
36
|
-
*/
|
|
37
|
-
export function resolveAskDbAiConfig(env, options = {}) {
|
|
38
|
-
const provider = readProvider(env);
|
|
39
|
-
const providerNativeKey = provider === "azure"
|
|
40
|
-
? env.AZURE_OPENAI_API_KEY || env.AZURE_API_KEY
|
|
41
|
-
: provider === "google"
|
|
42
|
-
? env.GOOGLE_GENERATIVE_AI_API_KEY || env.GOOGLE_AI_API_KEY
|
|
43
|
-
: env.OPENAI_API_KEY;
|
|
44
|
-
const providerNativeKeySecondary = provider === "azure"
|
|
45
|
-
? env.AZURE_OPENAI_API_KEY_SECONDARY || env.AZURE_API_KEY_SECONDARY
|
|
46
|
-
: env.OPENAI_API_KEY_SECONDARY;
|
|
47
|
-
const apiKey = env.ASKDB_AI_API_KEY ||
|
|
48
|
-
providerNativeKey ||
|
|
49
|
-
providerNativeKeySecondary ||
|
|
50
|
-
env.ASKDB_AI_API_KEY_SECONDARY ||
|
|
51
|
-
undefined;
|
|
52
|
-
if (!apiKey)
|
|
53
|
-
return undefined;
|
|
54
|
-
const providerNativeModel = provider === "azure"
|
|
55
|
-
? env.AZURE_OPENAI_DEPLOYMENT || env.AZURE_DEPLOYMENT_NAME
|
|
56
|
-
: provider === "google"
|
|
57
|
-
? env.GOOGLE_AI_MODEL
|
|
58
|
-
: env.OPENAI_MODEL;
|
|
59
|
-
const model = env.ASKDB_AI_MODEL ||
|
|
60
|
-
env.ASKDB_MODEL ||
|
|
61
|
-
providerNativeModel ||
|
|
62
|
-
options.modelDefault ||
|
|
63
|
-
DEFAULT_MODEL;
|
|
64
|
-
const providerNativeBaseURL = provider === "azure"
|
|
65
|
-
? env.AZURE_OPENAI_BASE_URL || env.AZURE_OPENAI_ENDPOINT || env.AZURE_BASE_URL
|
|
66
|
-
: provider === "google"
|
|
67
|
-
? env.GOOGLE_AI_BASE_URL
|
|
68
|
-
: env.OPENAI_BASE_URL;
|
|
69
|
-
const baseURL = env.ASKDB_AI_BASE_URL || providerNativeBaseURL || undefined;
|
|
70
|
-
const resourceName = env.ASKDB_AI_AZURE_RESOURCE_NAME || env.AZURE_RESOURCE_NAME || undefined;
|
|
71
|
-
const apiVersion = env.ASKDB_AI_AZURE_API_VERSION || env.AZURE_OPENAI_API_VERSION || env.AZURE_API_VERSION || undefined;
|
|
72
|
-
if (provider === "azure" && !baseURL && !resourceName) {
|
|
73
|
-
throw new Error("Azure provider requires ASKDB_AI_AZURE_RESOURCE_NAME (e.g. 'my-foundry') " +
|
|
74
|
-
"or ASKDB_AI_BASE_URL pointing at the full endpoint.");
|
|
75
|
-
}
|
|
76
|
-
return {
|
|
77
|
-
provider,
|
|
78
|
-
apiKey,
|
|
79
|
-
model,
|
|
80
|
-
...(baseURL ? { baseURL } : {}),
|
|
81
|
-
...(resourceName ? { resourceName } : {}),
|
|
82
|
-
...(apiVersion ? { apiVersion } : {}),
|
|
83
|
-
};
|
|
84
|
-
}
|
|
85
|
-
/**
|
|
86
|
-
* Resolves an embedding model config from the same provider/key/base URL
|
|
87
|
-
* connection as `resolveAskDbAiConfig`, but with embedding-specific model
|
|
88
|
-
* precedence so chat model ids (for example `gpt-4o`) are not used as
|
|
89
|
-
* embedding model ids by accident.
|
|
90
|
-
*
|
|
91
|
-
* **Important:** Only `@askdb/config` reads `process.env` (during dotenv load and while evaluating
|
|
92
|
-
* `askdb.config.*`). Pass `getAskDbRuntimeConfig().ai.aiEnv` from `@askdb/config` after
|
|
93
|
-
* `bootstrapAskDbEnv()`, or an explicit plain object in tests.
|
|
94
|
-
*
|
|
95
|
-
* @param env - String map in the shape of canonical AskDB env keys (from the runtime snapshot).
|
|
96
|
-
* @param options - Optional per-app overrides (e.g. a per-app embedding model env var).
|
|
97
|
-
*/
|
|
98
|
-
export function resolveAskDbEmbeddingConfig(env, options = {}) {
|
|
99
|
-
const provider = readProvider(env);
|
|
100
|
-
const providerNativeKey = provider === "azure"
|
|
101
|
-
? env.AZURE_OPENAI_API_KEY || env.AZURE_API_KEY
|
|
102
|
-
: provider === "google"
|
|
103
|
-
? env.GOOGLE_GENERATIVE_AI_API_KEY || env.GOOGLE_AI_API_KEY
|
|
104
|
-
: env.OPENAI_API_KEY;
|
|
105
|
-
const providerNativeKeySecondary = provider === "azure"
|
|
106
|
-
? env.AZURE_OPENAI_API_KEY_SECONDARY || env.AZURE_API_KEY_SECONDARY
|
|
107
|
-
: env.OPENAI_API_KEY_SECONDARY;
|
|
108
|
-
const apiKey = env.ASKDB_AI_API_KEY ||
|
|
109
|
-
providerNativeKey ||
|
|
110
|
-
providerNativeKeySecondary ||
|
|
111
|
-
env.ASKDB_AI_API_KEY_SECONDARY ||
|
|
112
|
-
undefined;
|
|
113
|
-
if (!apiKey)
|
|
114
|
-
return undefined;
|
|
115
|
-
const perAppModel = options.modelEnvVar ? env[options.modelEnvVar] : undefined;
|
|
116
|
-
const providerNativeModel = provider === "azure"
|
|
117
|
-
? env.AZURE_OPENAI_EMBEDDING_DEPLOYMENT ||
|
|
118
|
-
env.AZURE_EMBEDDING_DEPLOYMENT_NAME
|
|
119
|
-
: provider === "google"
|
|
120
|
-
? env.GOOGLE_AI_EMBEDDING_MODEL
|
|
121
|
-
: env.OPENAI_EMBEDDING_MODEL;
|
|
122
|
-
const model = perAppModel ||
|
|
123
|
-
env.ASKDB_AI_EMBEDDING_MODEL ||
|
|
124
|
-
env.ASKDB_EMBEDDING_MODEL ||
|
|
125
|
-
providerNativeModel ||
|
|
126
|
-
options.modelDefault ||
|
|
127
|
-
DEFAULT_EMBEDDING_MODEL;
|
|
128
|
-
const providerNativeBaseURL = provider === "azure"
|
|
129
|
-
? env.AZURE_OPENAI_BASE_URL || env.AZURE_OPENAI_ENDPOINT || env.AZURE_BASE_URL
|
|
130
|
-
: provider === "google"
|
|
131
|
-
? env.GOOGLE_AI_BASE_URL
|
|
132
|
-
: env.OPENAI_BASE_URL;
|
|
133
|
-
const baseURL = env.ASKDB_AI_BASE_URL || providerNativeBaseURL || undefined;
|
|
134
|
-
const resourceName = env.ASKDB_AI_AZURE_RESOURCE_NAME || env.AZURE_RESOURCE_NAME || undefined;
|
|
135
|
-
const apiVersion = env.ASKDB_AI_AZURE_API_VERSION || env.AZURE_OPENAI_API_VERSION || env.AZURE_API_VERSION || undefined;
|
|
136
|
-
if (provider === "azure" && !baseURL && !resourceName) {
|
|
137
|
-
throw new Error("Azure provider requires ASKDB_AI_AZURE_RESOURCE_NAME (e.g. 'my-foundry') " +
|
|
138
|
-
"or ASKDB_AI_BASE_URL pointing at the full endpoint.");
|
|
139
|
-
}
|
|
140
|
-
return {
|
|
141
|
-
provider,
|
|
142
|
-
apiKey,
|
|
143
|
-
model,
|
|
144
|
-
...(baseURL ? { baseURL } : {}),
|
|
145
|
-
...(resourceName ? { resourceName } : {}),
|
|
146
|
-
...(apiVersion ? { apiVersion } : {}),
|
|
147
|
-
};
|
|
148
|
-
}
|
|
149
|
-
/** Build a {@link LanguageModel} from an already-resolved config. */
|
|
150
|
-
export async function createAskDbLanguageModel(config) {
|
|
151
|
-
if (config.provider === "azure") {
|
|
152
|
-
// Lazy-loaded so `@ai-sdk/azure` is only required when actually used.
|
|
153
|
-
const { createAzure } = await import("@ai-sdk/azure");
|
|
154
|
-
const azure = createAzure({
|
|
155
|
-
apiKey: config.apiKey,
|
|
156
|
-
...(config.resourceName ? { resourceName: config.resourceName } : {}),
|
|
157
|
-
...(config.baseURL ? { baseURL: config.baseURL } : {}),
|
|
158
|
-
...(config.apiVersion ? { apiVersion: config.apiVersion } : {}),
|
|
159
|
-
});
|
|
160
|
-
return azure(config.model);
|
|
161
|
-
}
|
|
162
|
-
if (config.provider === "google") {
|
|
163
|
-
// Lazy-loaded so `@ai-sdk/google` is only required when actually used.
|
|
164
|
-
const { createGoogleGenerativeAI } = await import("@ai-sdk/google");
|
|
165
|
-
const google = createGoogleGenerativeAI({
|
|
166
|
-
apiKey: config.apiKey,
|
|
167
|
-
...(config.baseURL ? { baseURL: config.baseURL } : {}),
|
|
168
|
-
});
|
|
169
|
-
return google(config.model);
|
|
170
|
-
}
|
|
171
|
-
const openai = createOpenAI({
|
|
172
|
-
apiKey: config.apiKey,
|
|
173
|
-
...(config.baseURL ? { baseURL: config.baseURL } : {}),
|
|
174
|
-
});
|
|
175
|
-
return openai(config.model);
|
|
176
|
-
}
|
|
177
|
-
/** Build an AI SDK text embedding model from an already-resolved config. */
|
|
178
|
-
export async function createAskDbEmbeddingModel(config, options = {}) {
|
|
179
|
-
if (config.provider === "azure") {
|
|
180
|
-
const { createAzure } = await import("@ai-sdk/azure");
|
|
181
|
-
const azure = createAzure({
|
|
182
|
-
apiKey: config.apiKey,
|
|
183
|
-
...(config.resourceName ? { resourceName: config.resourceName } : {}),
|
|
184
|
-
...(config.baseURL ? { baseURL: config.baseURL } : {}),
|
|
185
|
-
...(config.apiVersion ? { apiVersion: config.apiVersion } : {}),
|
|
186
|
-
});
|
|
187
|
-
return azure.embedding(config.model, {
|
|
188
|
-
dimensions: options.dimensions,
|
|
189
|
-
user: options.user,
|
|
190
|
-
});
|
|
191
|
-
}
|
|
192
|
-
if (config.provider === "google") {
|
|
193
|
-
const { createGoogleGenerativeAI } = await import("@ai-sdk/google");
|
|
194
|
-
const google = createGoogleGenerativeAI({
|
|
195
|
-
apiKey: config.apiKey,
|
|
196
|
-
...(config.baseURL ? { baseURL: config.baseURL } : {}),
|
|
197
|
-
});
|
|
198
|
-
return google.textEmbeddingModel(config.model);
|
|
199
|
-
}
|
|
200
|
-
const openai = createOpenAI({
|
|
201
|
-
apiKey: config.apiKey,
|
|
202
|
-
...(config.baseURL ? { baseURL: config.baseURL } : {}),
|
|
203
|
-
});
|
|
204
|
-
return openai.embedding(config.model, {
|
|
205
|
-
dimensions: options.dimensions,
|
|
206
|
-
user: options.user,
|
|
207
|
-
});
|
|
208
|
-
}
|
|
209
|
-
/**
|
|
210
|
-
* Convenience wrapper: resolve config from env, then construct a model.
|
|
211
|
-
* Returns `undefined` when no API key is configured.
|
|
212
|
-
*
|
|
213
|
-
* **Important:** Only `@askdb/config` reads `process.env` (during dotenv load and while evaluating
|
|
214
|
-
* `askdb.config.*`). Pass `getAskDbRuntimeConfig().ai.aiEnv` from `@askdb/config` after
|
|
215
|
-
* `bootstrapAskDbEnv()`, or an explicit plain object in tests.
|
|
216
|
-
*
|
|
217
|
-
* @param env - String map in the shape of canonical AskDB env keys (from the runtime snapshot).
|
|
218
|
-
* @param options - Optional per-app overrides (e.g. a per-app model env var).
|
|
219
|
-
*/
|
|
220
|
-
export async function createAskDbLanguageModelFromEnv(env, options = {}) {
|
|
221
|
-
const config = resolveAskDbAiConfig(env, options);
|
|
222
|
-
if (!config)
|
|
223
|
-
return undefined;
|
|
224
|
-
return createAskDbLanguageModel(config);
|
|
225
|
-
}
|
|
226
|
-
/** Resolve embedding config from env, then construct an AI SDK embedding model.
|
|
227
|
-
*
|
|
228
|
-
* **Important:** Only `@askdb/config` reads `process.env` (during dotenv load and while evaluating
|
|
229
|
-
* `askdb.config.*`). Pass `getAskDbRuntimeConfig().ai.aiEnv` after `bootstrapAskDbEnv()`, or
|
|
230
|
-
* an explicit plain object in tests.
|
|
231
|
-
*
|
|
232
|
-
* @param env - String map in the shape of canonical AskDB env keys (from the runtime snapshot).
|
|
233
|
-
* @param options - Optional per-app overrides plus embedding-specific model/dimensions settings.
|
|
234
|
-
*/
|
|
235
|
-
export async function createAskDbEmbeddingModelFromEnv(env, options = {}) {
|
|
236
|
-
const config = resolveAskDbEmbeddingConfig(env, options);
|
|
237
|
-
if (!config)
|
|
238
|
-
return undefined;
|
|
239
|
-
return createAskDbEmbeddingModel(config, options);
|
|
240
|
-
}
|
|
241
|
-
/**
|
|
242
|
-
* Human-readable message describing how to configure AI, used by callers
|
|
243
|
-
* when no key is configured.
|
|
244
|
-
*/
|
|
245
|
-
export function askDbAiKeyMissingMessage(context) {
|
|
246
|
-
return (`${context}: no AI API key configured. ` +
|
|
247
|
-
`For OpenAI, set OPENAI_API_KEY (or ASKDB_AI_API_KEY). ` +
|
|
248
|
-
`For Azure / Microsoft Foundry, set ASKDB_AI_PROVIDER=azure plus ` +
|
|
249
|
-
`AZURE_OPENAI_API_KEY (or ASKDB_AI_API_KEY), ` +
|
|
250
|
-
`ASKDB_AI_AZURE_RESOURCE_NAME (or ASKDB_AI_BASE_URL), and a deployment name ` +
|
|
251
|
-
`via ASKDB_AI_MODEL. ` +
|
|
252
|
-
`For Google Gemini, set ASKDB_AI_PROVIDER=google plus ` +
|
|
253
|
-
`GOOGLE_GENERATIVE_AI_API_KEY (or ASKDB_AI_API_KEY).`);
|
|
254
|
-
}
|
|
255
|
-
//# sourceMappingURL=provider.js.map
|
package/dist/ai/provider.js.map
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../src/ai/provider.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAqC9C,MAAM,aAAa,GAAG,aAAa,CAAC;AACpC,MAAM,uBAAuB,GAAG,wBAAwB,CAAC;AAEzD,SAAS,YAAY,CAAC,GAAe;IACnC,MAAM,GAAG,GAAG,CAAC,GAAG,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IAC/D,IAAI,GAAG,KAAK,EAAE,IAAI,GAAG,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IACpD,IAAI,GAAG,KAAK,OAAO,IAAI,GAAG,KAAK,cAAc,IAAI,GAAG,KAAK,SAAS;QAAE,OAAO,OAAO,CAAC;IACnF,IAAI,GAAG,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IACtC,MAAM,IAAI,KAAK,CACb,8BAA8B,GAAG,CAAC,iBAAiB,6CAA6C,CACjG,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,UAAU,oBAAoB,CAClC,GAAe,EACf,UAAuC,EAAE;IAEzC,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;IAEnC,MAAM,iBAAiB,GACrB,QAAQ,KAAK,OAAO;QAClB,CAAC,CAAC,GAAG,CAAC,oBAAoB,IAAI,GAAG,CAAC,aAAa;QAC/C,CAAC,CAAC,QAAQ,KAAK,QAAQ;YACrB,CAAC,CAAC,GAAG,CAAC,4BAA4B,IAAI,GAAG,CAAC,iBAAiB;YAC3D,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC;IAC3B,MAAM,0BAA0B,GAC9B,QAAQ,KAAK,OAAO;QAClB,CAAC,CAAC,GAAG,CAAC,8BAA8B,IAAI,GAAG,CAAC,uBAAuB;QACnE,CAAC,CAAC,GAAG,CAAC,wBAAwB,CAAC;IACnC,MAAM,MAAM,GACV,GAAG,CAAC,gBAAgB;QACpB,iBAAiB;QACjB,0BAA0B;QAC1B,GAAG,CAAC,0BAA0B;QAC9B,SAAS,CAAC;IACZ,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAE9B,MAAM,mBAAmB,GACvB,QAAQ,KAAK,OAAO;QAClB,CAAC,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC,qBAAqB;QAC1D,CAAC,CAAC,QAAQ,KAAK,QAAQ;YACrB,CAAC,CAAC,GAAG,CAAC,eAAe;YACrB,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC;IACzB,MAAM,KAAK,GACT,GAAG,CAAC,cAAc;QAClB,GAAG,CAAC,WAAW;QACf,mBAAmB;QACnB,OAAO,CAAC,YAAY;QACpB,aAAa,CAAC;IAEhB,MAAM,qBAAqB,GACzB,QAAQ,KAAK,OAAO;QAClB,CAAC,CAAC,GAAG,CAAC,qBAAqB,IAAI,GAAG,CAAC,qBAAqB,IAAI,GAAG,CAAC,cAAc;QAC9E,CAAC,CAAC,QAAQ,KAAK,QAAQ;YACrB,CAAC,CAAC,GAAG,CAAC,kBAAkB;YACxB,CAAC,CAAC,GAAG,CAAC,eAAe,CAAC;IAC5B,MAAM,OAAO,GAAG,GAAG,CAAC,iBAAiB,IAAI,qBAAqB,IAAI,SAAS,CAAC;IAE5E,MAAM,YAAY,GAChB,GAAG,CAAC,4BAA4B,IAAI,GAAG,CAAC,mBAAmB,IAAI,SAAS,CAAC;IAC3E,MAAM,UAAU,GACd,GAAG,CAAC,0BAA0B,IAAI,GAAG,CAAC,wBAAwB,IAAI,GAAG,CAAC,iBAAiB,IAAI,SAAS,CAAC;IAEvG,IAAI,QAAQ,KAAK,OAAO,IAAI,CAAC,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC;QACtD,MAAM,IAAI,KAAK,CACb,2EAA2E;YACzE,qDAAqD,CACxD,CAAC;IACJ,CAAC;IAED,OAAO;QACL,QAAQ;QACR,MAAM;QACN,KAAK;QACL,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/B,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACtC,CAAC;AACJ,CAAC;AAWD;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,2BAA2B,CACzC,GAAe,EACf,UAA8C,EAAE;IAEhD,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;IAEnC,MAAM,iBAAiB,GACrB,QAAQ,KAAK,OAAO;QAClB,CAAC,CAAC,GAAG,CAAC,oBAAoB,IAAI,GAAG,CAAC,aAAa;QAC/C,CAAC,CAAC,QAAQ,KAAK,QAAQ;YACrB,CAAC,CAAC,GAAG,CAAC,4BAA4B,IAAI,GAAG,CAAC,iBAAiB;YAC3D,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC;IAC3B,MAAM,0BAA0B,GAC9B,QAAQ,KAAK,OAAO;QAClB,CAAC,CAAC,GAAG,CAAC,8BAA8B,IAAI,GAAG,CAAC,uBAAuB;QACnE,CAAC,CAAC,GAAG,CAAC,wBAAwB,CAAC;IACnC,MAAM,MAAM,GACV,GAAG,CAAC,gBAAgB;QACpB,iBAAiB;QACjB,0BAA0B;QAC1B,GAAG,CAAC,0BAA0B;QAC9B,SAAS,CAAC;IACZ,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAE9B,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC/E,MAAM,mBAAmB,GACvB,QAAQ,KAAK,OAAO;QAClB,CAAC,CAAC,GAAG,CAAC,iCAAiC;YACrC,GAAG,CAAC,+BAA+B;QACrC,CAAC,CAAC,QAAQ,KAAK,QAAQ;YACrB,CAAC,CAAC,GAAG,CAAC,yBAAyB;YAC/B,CAAC,CAAC,GAAG,CAAC,sBAAsB,CAAC;IACnC,MAAM,KAAK,GACT,WAAW;QACX,GAAG,CAAC,wBAAwB;QAC5B,GAAG,CAAC,qBAAqB;QACzB,mBAAmB;QACnB,OAAO,CAAC,YAAY;QACpB,uBAAuB,CAAC;IAE1B,MAAM,qBAAqB,GACzB,QAAQ,KAAK,OAAO;QAClB,CAAC,CAAC,GAAG,CAAC,qBAAqB,IAAI,GAAG,CAAC,qBAAqB,IAAI,GAAG,CAAC,cAAc;QAC9E,CAAC,CAAC,QAAQ,KAAK,QAAQ;YACrB,CAAC,CAAC,GAAG,CAAC,kBAAkB;YACxB,CAAC,CAAC,GAAG,CAAC,eAAe,CAAC;IAC5B,MAAM,OAAO,GAAG,GAAG,CAAC,iBAAiB,IAAI,qBAAqB,IAAI,SAAS,CAAC;IAE5E,MAAM,YAAY,GAChB,GAAG,CAAC,4BAA4B,IAAI,GAAG,CAAC,mBAAmB,IAAI,SAAS,CAAC;IAC3E,MAAM,UAAU,GACd,GAAG,CAAC,0BAA0B,IAAI,GAAG,CAAC,wBAAwB,IAAI,GAAG,CAAC,iBAAiB,IAAI,SAAS,CAAC;IAEvG,IAAI,QAAQ,KAAK,OAAO,IAAI,CAAC,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC;QACtD,MAAM,IAAI,KAAK,CACb,2EAA2E;YACzE,qDAAqD,CACxD,CAAC;IACJ,CAAC;IAED,OAAO;QACL,QAAQ;QACR,MAAM;QACN,KAAK;QACL,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/B,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACtC,CAAC;AACJ,CAAC;AAED,qEAAqE;AACrE,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,MAAqB;IAErB,IAAI,MAAM,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QAChC,sEAAsE;QACtE,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,CAAC;QACtD,MAAM,KAAK,GAAG,WAAW,CAAC;YACxB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrE,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAChE,CAAC,CAAC;QACH,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,uEAAuE;QACvE,MAAM,EAAE,wBAAwB,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACpE,MAAM,MAAM,GAAG,wBAAwB,CAAC;YACtC,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACvD,CAAC,CAAC;QACH,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IACD,MAAM,MAAM,GAAG,YAAY,CAAC;QAC1B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvD,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC9B,CAAC;AASD,4EAA4E;AAC5E,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,MAAqB,EACrB,UAA4C,EAAE;IAE9C,IAAI,MAAM,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QAChC,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,CAAC;QACtD,MAAM,KAAK,GAAG,WAAW,CAAC;YACxB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrE,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAChE,CAAC,CAAC;QACH,OAAO,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE;YACnC,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAC,CAAC;IACL,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,MAAM,EAAE,wBAAwB,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACpE,MAAM,MAAM,GAAG,wBAAwB,CAAC;YACtC,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACvD,CAAC,CAAC;QACH,OAAO,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACjD,CAAC;IACD,MAAM,MAAM,GAAG,YAAY,CAAC;QAC1B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvD,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE;QACpC,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,IAAI,EAAE,OAAO,CAAC,IAAI;KACnB,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,CAAC,KAAK,UAAU,+BAA+B,CACnD,GAAe,EACf,UAAuC,EAAE;IAEzC,MAAM,MAAM,GAAG,oBAAoB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAClD,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC9B,OAAO,wBAAwB,CAAC,MAAM,CAAC,CAAC;AAC1C,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,gCAAgC,CACpD,GAAe,EACf,UAAiF,EAAE;IAEnF,MAAM,MAAM,GAAG,2BAA2B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACzD,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC9B,OAAO,yBAAyB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AACpD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,OAAe;IACtD,OAAO,CACL,GAAG,OAAO,8BAA8B;QACxC,wDAAwD;QACxD,kEAAkE;QAClE,8CAA8C;QAC9C,6EAA6E;QAC7E,sBAAsB;QACtB,uDAAuD;QACvD,qDAAqD,CACtD,CAAC;AACJ,CAAC"}
|