@workglow/ai 0.2.16 → 0.2.17
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/browser.js +52 -35
- package/dist/browser.js.map +10 -10
- package/dist/bun.js +52 -35
- package/dist/bun.js.map +10 -10
- package/dist/node.js +52 -35
- package/dist/node.js.map +10 -10
- package/dist/provider/AiProvider.d.ts +8 -8
- package/dist/provider/AiProvider.d.ts.map +1 -1
- package/dist/provider/AiProviderRegistry.d.ts +16 -16
- package/dist/provider/AiProviderRegistry.d.ts.map +1 -1
- package/dist/task/ChunkVectorUpsertTask.d.ts.map +1 -1
- package/dist/task/ContextBuilderTask.d.ts +3 -2
- package/dist/task/ContextBuilderTask.d.ts.map +1 -1
- package/dist/task/TextChunkerTask.d.ts.map +1 -1
- package/dist/task/VectorQuantizeTask.d.ts +2 -1
- package/dist/task/VectorQuantizeTask.d.ts.map +1 -1
- package/dist/task/VectorSimilarityTask.d.ts +2 -4
- package/dist/task/VectorSimilarityTask.d.ts.map +1 -1
- package/dist/task/base/AiTask.d.ts +4 -4
- package/dist/task/base/AiTask.d.ts.map +1 -1
- package/dist/worker.js +25 -26
- package/dist/worker.js.map +4 -4
- package/package.json +11 -11
|
@@ -6,7 +6,7 @@
|
|
|
6
6
|
import { TaskInput, TaskOutput } from "@workglow/task-graph";
|
|
7
7
|
import type { WorkerServerBase as WorkerServer } from "@workglow/util/worker";
|
|
8
8
|
import type { ModelConfig } from "../model/ModelSchema";
|
|
9
|
-
import type {
|
|
9
|
+
import type { AiProviderPreviewRunFn, AiProviderRunFn, AiProviderStreamFn } from "./AiProviderRegistry";
|
|
10
10
|
/**
|
|
11
11
|
* Job queue concurrency: one limit for the primary ({@link QueuedAiProvider} hardware) queue,
|
|
12
12
|
* or per-slot limits. Hugging Face Transformers ONNX uses `gpu` and `cpu` for its two queues.
|
|
@@ -115,12 +115,12 @@ export declare abstract class AiProvider<TModelConfig extends ModelConfig = Mode
|
|
|
115
115
|
*/
|
|
116
116
|
protected readonly streamTasks?: Record<string, AiProviderStreamFn<any, any, TModelConfig>>;
|
|
117
117
|
/**
|
|
118
|
-
* Map of task type names to their
|
|
118
|
+
* Map of task type names to their preview run functions.
|
|
119
119
|
* Injected via constructor alongside `tasks`. Only needed for tasks that
|
|
120
|
-
* provide lightweight
|
|
120
|
+
* provide lightweight previews via executePreview().
|
|
121
121
|
*/
|
|
122
|
-
protected readonly
|
|
123
|
-
constructor(tasks?: Record<string, AiProviderRunFn<any, any, TModelConfig>>, streamTasks?: Record<string, AiProviderStreamFn<any, any, TModelConfig>>,
|
|
122
|
+
protected readonly previewTasks?: Record<string, AiProviderPreviewRunFn<any, any, TModelConfig>>;
|
|
123
|
+
constructor(tasks?: Record<string, AiProviderRunFn<any, any, TModelConfig>>, streamTasks?: Record<string, AiProviderStreamFn<any, any, TModelConfig>>, previewTasks?: Record<string, AiProviderPreviewRunFn<any, any, TModelConfig>>);
|
|
124
124
|
/** Get all task type names this provider supports */
|
|
125
125
|
get supportedTaskTypes(): readonly string[];
|
|
126
126
|
/**
|
|
@@ -136,11 +136,11 @@ export declare abstract class AiProvider<TModelConfig extends ModelConfig = Mode
|
|
|
136
136
|
*/
|
|
137
137
|
getStreamFn<I extends TaskInput = TaskInput, O extends TaskOutput = TaskOutput>(taskType: string): AiProviderStreamFn<I, O, TModelConfig> | undefined;
|
|
138
138
|
/**
|
|
139
|
-
* Get the
|
|
139
|
+
* Get the preview run function for a specific task type.
|
|
140
140
|
* @param taskType - The task type name (e.g., "CountTokensTask")
|
|
141
|
-
* @returns The
|
|
141
|
+
* @returns The preview function, or undefined if not supported for this task type
|
|
142
142
|
*/
|
|
143
|
-
|
|
143
|
+
getPreviewRunFn<I extends TaskInput = TaskInput, O extends TaskOutput = TaskOutput>(taskType: string): AiProviderPreviewRunFn<I, O, TModelConfig> | undefined;
|
|
144
144
|
/**
|
|
145
145
|
* Register this provider on the main thread.
|
|
146
146
|
*
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AiProvider.d.ts","sourceRoot":"","sources":["../../src/provider/AiProvider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,KAAK,EAAE,gBAAgB,IAAI,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAE9E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,KAAK,EACV,
|
|
1
|
+
{"version":3,"file":"AiProvider.d.ts","sourceRoot":"","sources":["../../src/provider/AiProvider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,KAAK,EAAE,gBAAgB,IAAI,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAE9E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,KAAK,EACV,sBAAsB,EACtB,eAAe,EACf,kBAAkB,EACnB,MAAM,sBAAsB,CAAC;AAG9B;;;GAGG;AACH,MAAM,MAAM,0BAA0B,GAAG,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AACzE,eAAO,MAAM,0CAA0C,QAAiB,CAAC;AAEzE;;;GAGG;AACH,wBAAgB,oCAAoC,CAClD,WAAW,EAAE,0BAA0B,GAAG,SAAS,GAClD,MAAM,CAQR;AAED;;;;;;;GAOG;AACH,MAAM,WAAW,yBAAyB;IACxC;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,CAAC;IACjC;;;OAGG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,8BAA8B;IAC9B,KAAK,CAAC,EAAE;QACN;;;;WAIG;QACH,WAAW,CAAC,EAAE,0BAA0B,CAAC;QACzC,uEAAuE;QACvE,UAAU,CAAC,EAAE,OAAO,CAAC;KACtB,CAAC;CACH;AAED;;;GAGG;AACH,MAAM,WAAW,yBAA0B,SAAQ,yBAAyB;IAC1E,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;CAC5B;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,8BAAsB,UAAU,CAAC,YAAY,SAAS,WAAW,GAAG,WAAW;IAC7E,gEAAgE;IAChE,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAE/B;;OAEG;IACH,QAAQ,CAAC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAEtC,uEAAuE;IACvE,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAEnC,8DAA8D;IAC9D,QAAQ,CAAC,QAAQ,CAAC,eAAe,EAAE,OAAO,CAAC;IAE3C;;;OAGG;IACH,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAAE,SAAS,MAAM,EAAE,CAAC;IAE/C;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;IAEnF;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;IAE5F;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,sBAAsB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;IAEjG,YACE,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,EAC/D,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,EACxE,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,sBAAsB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,EAK9E;IAED,qDAAqD;IACrD,IAAI,kBAAkB,IAAI,SAAS,MAAM,EAAE,CAE1C;IAED;;;;OAIG;IACH,QAAQ,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EACzE,QAAQ,EAAE,MAAM,GACf,eAAe,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,CAAC,GAAG,SAAS,CAEjD;IAED;;;;OAIG;IACH,WAAW,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EAC5E,QAAQ,EAAE,MAAM,GACf,kBAAkB,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,CAAC,GAAG,SAAS,CAEpD;IAED;;;;OAIG;IACH,eAAe,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EAChF,QAAQ,EAAE,MAAM,GACf,sBAAsB,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,CAAC,GAAG,SAAS,CAExD;IAED;;;;;;;;;OASG;IACG,QAAQ,CAAC,OAAO,GAAE,yBAA8B,GAAG,OAAO,CAAC,IAAI,CAAC,CAgErE;IAED;;;;;;;;OAQG;IACH,sBAAsB,CAAC,YAAY,EAAE,YAAY,GAAG,IAAI,CAoBvD;IAED;;;;OAIG;IACH,UAAgB,YAAY,CAAC,QAAQ,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAG;IAEnF;;;OAGG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAG;IAEjC;;;;;;;OAOG;IACH,aAAa,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM,CAEzC;IAED;;;;OAIG;IACG,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAG;IAE1D;;;;OAIG;IACH,UAAgB,aAAa,CAAC,QAAQ,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAG;CACrF"}
|
|
@@ -17,11 +17,11 @@ import type { AiProvider } from "./AiProvider";
|
|
|
17
17
|
*/
|
|
18
18
|
export type AiProviderRunFn<Input extends TaskInput = TaskInput, Output extends TaskOutput = TaskOutput, Model extends ModelConfig = ModelConfig> = (input: Input, model: Model | undefined, update_progress: (progress: number, message?: string, details?: any) => void, signal: AbortSignal, outputSchema?: JsonSchema, sessionId?: string) => Promise<Output>;
|
|
19
19
|
/**
|
|
20
|
-
* Type for the
|
|
21
|
-
*
|
|
22
|
-
* No `signal` or `update_progress` --
|
|
20
|
+
* Type for the preview run function for AiTask.executePreview().
|
|
21
|
+
* Computes a fast preview from input alone -- no prior output needed.
|
|
22
|
+
* No `signal` or `update_progress` -- preview execution is lightweight and synchronous-ish.
|
|
23
23
|
*/
|
|
24
|
-
export type
|
|
24
|
+
export type AiProviderPreviewRunFn<Input extends TaskInput = TaskInput, Output extends TaskOutput = TaskOutput, Model extends ModelConfig = ModelConfig> = (input: Input, model: Model | undefined) => Promise<Output | undefined>;
|
|
25
25
|
/**
|
|
26
26
|
* Type for the streaming run function for the AiJob.
|
|
27
27
|
* Returns an AsyncIterable of StreamEvents instead of a Promise.
|
|
@@ -53,7 +53,7 @@ export type AiProviderStreamFn<Input extends TaskInput = TaskInput, Output exten
|
|
|
53
53
|
export declare class AiProviderRegistry {
|
|
54
54
|
runFnRegistry: Map<string, Map<string, AiProviderRunFn<any, any>>>;
|
|
55
55
|
streamFnRegistry: Map<string, Map<string, AiProviderStreamFn<any, any>>>;
|
|
56
|
-
|
|
56
|
+
previewRunFnRegistry: Map<string, Map<string, AiProviderPreviewRunFn<any, any>>>;
|
|
57
57
|
private providers;
|
|
58
58
|
private strategyResolvers;
|
|
59
59
|
private defaultStrategy;
|
|
@@ -63,7 +63,7 @@ export declare class AiProviderRegistry {
|
|
|
63
63
|
*/
|
|
64
64
|
registerProvider(provider: AiProvider<any>): void;
|
|
65
65
|
/**
|
|
66
|
-
* Removes a previously registered provider and all of its run/stream/
|
|
66
|
+
* Removes a previously registered provider and all of its run/stream/preview functions.
|
|
67
67
|
* Used for cleanup when provider initialization fails partway through.
|
|
68
68
|
* @param name - The provider name to unregister
|
|
69
69
|
*/
|
|
@@ -142,21 +142,21 @@ export declare class AiProviderRegistry {
|
|
|
142
142
|
*/
|
|
143
143
|
getStreamFn<Input extends TaskInput = TaskInput, Output extends TaskOutput = TaskOutput>(modelProvider: string, taskType: string): AiProviderStreamFn<Input, Output> | undefined;
|
|
144
144
|
/**
|
|
145
|
-
* Registers a worker-proxied
|
|
146
|
-
* Creates a proxy that delegates
|
|
147
|
-
* Returns undefined (non-throwing) if the worker has no
|
|
145
|
+
* Registers a worker-proxied preview function for a specific task type and model provider.
|
|
146
|
+
* Creates a proxy that delegates preview execution to a Web Worker via WorkerManager.
|
|
147
|
+
* Returns undefined (non-throwing) if the worker has no preview function for the task type.
|
|
148
148
|
*/
|
|
149
|
-
|
|
149
|
+
registerAsWorkerPreviewRunFn<Input extends TaskInput = TaskInput, Output extends TaskOutput = TaskOutput>(modelProvider: string, taskType: string): void;
|
|
150
150
|
/**
|
|
151
|
-
* Registers a
|
|
152
|
-
* Called by AiTask.
|
|
151
|
+
* Registers a preview execution function for a specific task type and model provider.
|
|
152
|
+
* Called by AiTask.executePreview() to provide a fast, lightweight preview without a network call.
|
|
153
153
|
*/
|
|
154
|
-
|
|
154
|
+
registerPreviewRunFn<Input extends TaskInput = TaskInput, Output extends TaskOutput = TaskOutput>(modelProvider: string, taskType: string, previewRunFn: AiProviderPreviewRunFn<Input, Output>): void;
|
|
155
155
|
/**
|
|
156
|
-
* Retrieves the
|
|
157
|
-
* Returns undefined if no
|
|
156
|
+
* Retrieves the preview execution function for a task type and model provider.
|
|
157
|
+
* Returns undefined if no preview function is registered (fallback to default behavior).
|
|
158
158
|
*/
|
|
159
|
-
|
|
159
|
+
getPreviewRunFn<Input extends TaskInput = TaskInput, Output extends TaskOutput = TaskOutput>(modelProvider: string, taskType: string): AiProviderPreviewRunFn<Input, Output> | undefined;
|
|
160
160
|
/**
|
|
161
161
|
* Retrieves the direct execution function for a task type and model
|
|
162
162
|
* Bypasses the job queue system for immediate execution
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AiProviderRegistry.d.ts","sourceRoot":"","sources":["../../src/provider/AiProviderRegistry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAExD,OAAO,KAAK,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,mCAAmC,CAAC;AAClG,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAE/C;;;;;GAKG;AACH,MAAM,MAAM,eAAe,CACzB,KAAK,SAAS,SAAS,GAAG,SAAS,EACnC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,KAAK,SAAS,WAAW,GAAG,WAAW,IACrC,CACF,KAAK,EAAE,KAAK,EACZ,KAAK,EAAE,KAAK,GAAG,SAAS,EACxB,eAAe,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,KAAK,IAAI,EAC5E,MAAM,EAAE,WAAW,EACnB,YAAY,CAAC,EAAE,UAAU,EACzB,SAAS,CAAC,EAAE,MAAM,KACf,OAAO,CAAC,MAAM,CAAC,CAAC;AAErB;;;;GAIG;AACH,MAAM,MAAM,
|
|
1
|
+
{"version":3,"file":"AiProviderRegistry.d.ts","sourceRoot":"","sources":["../../src/provider/AiProviderRegistry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAExD,OAAO,KAAK,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,mCAAmC,CAAC;AAClG,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAE/C;;;;;GAKG;AACH,MAAM,MAAM,eAAe,CACzB,KAAK,SAAS,SAAS,GAAG,SAAS,EACnC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,KAAK,SAAS,WAAW,GAAG,WAAW,IACrC,CACF,KAAK,EAAE,KAAK,EACZ,KAAK,EAAE,KAAK,GAAG,SAAS,EACxB,eAAe,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,KAAK,IAAI,EAC5E,MAAM,EAAE,WAAW,EACnB,YAAY,CAAC,EAAE,UAAU,EACzB,SAAS,CAAC,EAAE,MAAM,KACf,OAAO,CAAC,MAAM,CAAC,CAAC;AAErB;;;;GAIG;AACH,MAAM,MAAM,sBAAsB,CAChC,KAAK,SAAS,SAAS,GAAG,SAAS,EACnC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,KAAK,SAAS,WAAW,GAAG,WAAW,IACrC,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,GAAG,SAAS,KAAK,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;AAE5E;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,MAAM,kBAAkB,CAC5B,KAAK,SAAS,SAAS,GAAG,SAAS,EACnC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,KAAK,SAAS,WAAW,GAAG,WAAW,IACrC,CACF,KAAK,EAAE,KAAK,EACZ,KAAK,EAAE,KAAK,GAAG,SAAS,EACxB,MAAM,EAAE,WAAW,EACnB,YAAY,CAAC,EAAE,UAAU,EACzB,SAAS,CAAC,EAAE,MAAM,KACf,aAAa,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;AAExC;;;;GAIG;AACH,qBAAa,kBAAkB;IAC7B,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAa;IAC/E,gBAAgB,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAa;IACrF,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,sBAAsB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAa;IAC7F,OAAO,CAAC,SAAS,CAA2C;IAC5D,OAAO,CAAC,iBAAiB,CAA8C;IACvE,OAAO,CAAC,eAAe,CAAmC;IAE1D;;;OAGG;IACH,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAAC,GAAG,CAAC,GAAG,IAAI,CAEhD;IAED;;;;OAIG;IACH,kBAAkB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAarC;IAED;;;;OAIG;IACH,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,SAAS,CAErD;IAED;;OAEG;IACH,YAAY,IAAI,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,CAE3C;IAED;;;OAGG;IACH,uBAAuB,IAAI,MAAM,EAAE,CAElC;IAED;;;;OAIG;IACH,wBAAwB,CAAC,YAAY,EAAE,MAAM,EAAE,QAAQ,EAAE,kBAAkB,GAAG,IAAI,CAEjF;IAED;;;OAGG;IACH,WAAW,CAAC,KAAK,EAAE,WAAW,GAAG,oBAAoB,CAOpD;IAED;;;;;;OAMG;IACH,aAAa,CAAC,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,GAAG,MAAM,CAQ9D;IAED;;;;OAIG;IACG,cAAc,CAAC,YAAY,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAK3E;IAED;;;;OAIG;IACH,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,CAIhD;IAED;;;;;OAKG;IACH,aAAa,CAAC,KAAK,SAAS,SAAS,GAAG,SAAS,EAAE,MAAM,SAAS,UAAU,GAAG,UAAU,EACvF,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,QAMtC;IAED,qBAAqB,CACnB,KAAK,SAAS,SAAS,GAAG,SAAS,EACnC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,aAAa,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,QAsBxC;IAED;;;;;OAKG;IACH,gBAAgB,CAAC,KAAK,SAAS,SAAS,GAAG,SAAS,EAAE,MAAM,SAAS,UAAU,GAAG,UAAU,EAC1F,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,kBAAkB,CAAC,KAAK,EAAE,MAAM,CAAC,QAM5C;IAED;;;;;OAKG;IACH,wBAAwB,CACtB,KAAK,SAAS,SAAS,GAAG,SAAS,EACnC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,aAAa,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,QAiBxC;IAED;;;OAGG;IACH,WAAW,CAAC,KAAK,SAAS,SAAS,GAAG,SAAS,EAAE,MAAM,SAAS,UAAU,GAAG,UAAU,EACrF,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,MAAM,GACf,kBAAkB,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,SAAS,CAG/C;IAED;;;;OAIG;IACH,4BAA4B,CAC1B,KAAK,SAAS,SAAS,GAAG,SAAS,EACnC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,aAAa,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,QAYxC;IAED;;;OAGG;IACH,oBAAoB,CAAC,KAAK,SAAS,SAAS,GAAG,SAAS,EAAE,MAAM,SAAS,UAAU,GAAG,UAAU,EAC9F,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,sBAAsB,CAAC,KAAK,EAAE,MAAM,CAAC,QAMpD;IAED;;;OAGG;IACH,eAAe,CAAC,KAAK,SAAS,SAAS,GAAG,SAAS,EAAE,MAAM,SAAS,UAAU,GAAG,UAAU,EACzF,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,MAAM,GACf,sBAAsB,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,SAAS,CAGnD;IAED;;;OAGG;IACH,cAAc,CAAC,KAAK,SAAS,SAAS,GAAG,SAAS,EAAE,MAAM,SAAS,UAAU,GAAG,UAAU,EACxF,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,MAAM;;;;;;;;;;;;;;;;OAkBjB;CACF;AAID,wBAAgB,qBAAqB,uBAEpC;AACD,wBAAgB,qBAAqB,CAAC,EAAE,EAAE,kBAAkB,QAE3D"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ChunkVectorUpsertTask.d.ts","sourceRoot":"","sources":["../../src/task/ChunkVectorUpsertTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AACvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EACL,cAAc,EACd,UAAU,EAGV,uBAAuB,EACxB,MAAM,uBAAuB,CAAC;AAG/B,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,aAAa;;;;;;;;;;;;;;;;iBAIb,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACN,MAAM;;;;;;;;;;;;;;;;iBAMN,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,gBAAgB;qBACvB,WAAW,EAAE,kEAAkE;;;;;CAKlD,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;iBAEZ,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,4BAA4B;;iBAE3C,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,6CAA6C;;iBAE5D,SAAS;qBACP,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,WAAW;qBAClB,WAAW,EAAE,+BAA+B;;;;;CAKf,CAAC;AAEpC,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,OAAO,WAAW,EAAE,uBAAuB,CAAC,CAAC;AACjG,MAAM,MAAM,2BAA2B,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAC1E,MAAM,MAAM,2BAA2B,GAAG,UAAU,CAAC,0BAA0B,CAAC,CAAC;AAEjF;;;;GAIG;AACH,qBAAa,qBAAsB,SAAQ,IAAI,CAC7C,0BAA0B,EAC1B,2BAA2B,EAC3B,2BAA2B,CAC5B;IACC,OAAuB,IAAI,SAA2B;IACtD,OAAuB,QAAQ,SAAkB;IACjD,OAAuB,KAAK,SAAyB;IACrD,OAAuB,WAAW,SACoC;IACtE,OAAuB,SAAS,UAAS;IAEzC,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,0BAA0B,EACjC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,2BAA2B,CAAC,
|
|
1
|
+
{"version":3,"file":"ChunkVectorUpsertTask.d.ts","sourceRoot":"","sources":["../../src/task/ChunkVectorUpsertTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AACvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EACL,cAAc,EACd,UAAU,EAGV,uBAAuB,EACxB,MAAM,uBAAuB,CAAC;AAG/B,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,aAAa;;;;;;;;;;;;;;;;iBAIb,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACN,MAAM;;;;;;;;;;;;;;;;iBAMN,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,gBAAgB;qBACvB,WAAW,EAAE,kEAAkE;;;;;CAKlD,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;iBAEZ,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,4BAA4B;;iBAE3C,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,6CAA6C;;iBAE5D,SAAS;qBACP,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,WAAW;qBAClB,WAAW,EAAE,+BAA+B;;;;;CAKf,CAAC;AAEpC,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,OAAO,WAAW,EAAE,uBAAuB,CAAC,CAAC;AACjG,MAAM,MAAM,2BAA2B,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAC1E,MAAM,MAAM,2BAA2B,GAAG,UAAU,CAAC,0BAA0B,CAAC,CAAC;AAEjF;;;;GAIG;AACH,qBAAa,qBAAsB,SAAQ,IAAI,CAC7C,0BAA0B,EAC1B,2BAA2B,EAC3B,2BAA2B,CAC5B;IACC,OAAuB,IAAI,SAA2B;IACtD,OAAuB,QAAQ,SAAkB;IACjD,OAAuB,KAAK,SAAyB;IACrD,OAAuB,WAAW,SACoC;IACtE,OAAuB,SAAS,UAAS;IAEzC,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,0BAA0B,EACjC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,2BAA2B,CAAC,CAsDtC;CACF;AAED,eAAO,MAAM,iBAAiB,UACrB,0BAA0B,WACxB,2BAA2B;;;;EAGrC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,iBAAiB,EAAE,cAAc,CAC/B,0BAA0B,EAC1B,2BAA2B,EAC3B,2BAA2B,CAC5B,CAAC;KACH;CACF"}
|
|
@@ -3,7 +3,7 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow,
|
|
6
|
+
import { CreateWorkflow, IExecuteContext, IExecutePreviewContext, Task } from "@workglow/task-graph";
|
|
7
7
|
import type { TaskConfig } from "@workglow/task-graph";
|
|
8
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
9
9
|
export declare const ContextFormat: {
|
|
@@ -200,7 +200,8 @@ export declare class ContextBuilderTask extends Task<ContextBuilderTaskInput, Co
|
|
|
200
200
|
static cacheable: boolean;
|
|
201
201
|
static inputSchema(): DataPortSchema;
|
|
202
202
|
static outputSchema(): DataPortSchema;
|
|
203
|
-
|
|
203
|
+
execute(input: ContextBuilderTaskInput, context: IExecuteContext): Promise<ContextBuilderTaskOutput>;
|
|
204
|
+
executePreview(input: ContextBuilderTaskInput, context: IExecutePreviewContext): Promise<ContextBuilderTaskOutput>;
|
|
204
205
|
private formatChunk;
|
|
205
206
|
private formatNumbered;
|
|
206
207
|
private formatXML;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ContextBuilderTask.d.ts","sourceRoot":"","sources":["../../src/task/ContextBuilderTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,
|
|
1
|
+
{"version":3,"file":"ContextBuilderTask.d.ts","sourceRoot":"","sources":["../../src/task/ContextBuilderTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EACL,cAAc,EACd,eAAe,EACf,sBAAsB,EACtB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAInE,eAAO,MAAM,aAAa;aACxB,MAAM,EAAE,QAAQ;aAChB,QAAQ,EAAE,UAAU;aACpB,GAAG,EAAE,KAAK;aACV,QAAQ,EAAE,UAAU;aACpB,IAAI,EAAE,MAAM;CACJ,CAAC;AAEX,MAAM,MAAM,aAAa,GAAG,CAAC,OAAO,aAAa,CAAC,CAAC,MAAM,OAAO,aAAa,CAAC,CAAC;AAO/E,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,iCAAiC;;iBAEhD,QAAQ;qBACN,IAAI,EAAE,OAAO;qBACb,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,KAAK,EAAE,UAAU;yBACjB,WAAW,EAAE,yBAAyB;;qBAExC,KAAK,EAAE,UAAU;qBACjB,WAAW,EAAE,oCAAoC;;iBAEnD,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,QAAQ;qBACf,WAAW,EAAE,4CAA4C;;iBAE3D,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,QAAQ;qBACf,WAAW,EAAE,+BAA+B;qBAC5C,OAAO;;iBAET,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,yDAAyD;qBACtE,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;;iBAEZ,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EACT,gGAAgG;qBAClG,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;;iBAEZ,eAAe;qBACb,IAAI,EAAE,SAAS;qBACf,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,4CAA4C;qBACzD,OAAO;;iBAET,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,WAAW;qBAClB,WAAW,EAAE,0BAA0B;qBACvC,OAAO,EAAE,MAAM;;iBAEjB,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAI0B,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;iBAEZ,OAAO;qBACL,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,SAAS;qBAChB,WAAW,EAAE,kCAAkC;;iBAEjD,UAAU;qBACR,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,sCAAsC;;iBAErD,WAAW;qBACT,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,cAAc;qBACrB,WAAW,EAAE,uCAAuC;;iBAEtD,WAAW;qBACT,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,cAAc;qBACrB,WAAW,EAAE,sCAAsC;;;;;CAKtB,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACrE,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AACvE,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,uBAAuB,CAAC,CAAC;AAE3E;;;;;;;GAOG;AACH,qBAAa,kBAAmB,SAAQ,IAAI,CAC1C,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB;IACC,OAAuB,IAAI,SAAwB;IACnD,OAAuB,QAAQ,SAAS;IACxC,OAAuB,KAAK,SAAqB;IACjD,OAAuB,WAAW,SAA0D;IAC5F,OAAuB,SAAS,UAAQ;IAExC,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,uBAAuB,EAC9B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,wBAAwB,CAAC,CAGnC;IAEc,cAAc,CAC3B,KAAK,EAAE,uBAAuB,EAC9B,OAAO,EAAE,sBAAsB,GAC9B,OAAO,CAAC,wBAAwB,CAAC,CAgFnC;IAED,OAAO,CAAC,WAAW;IAuBnB,OAAO,CAAC,cAAc;IAiBtB,OAAO,CAAC,SAAS;IA2BjB,OAAO,CAAC,cAAc;IAwBtB,OAAO,CAAC,UAAU;IAsBlB,OAAO,CAAC,oBAAoB;IAa5B,OAAO,CAAC,SAAS;CAQlB;AAED,eAAO,MAAM,cAAc,UAClB,uBAAuB,WACrB,wBAAwB;;;;;EAGlC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,cAAc,EAAE,cAAc,CAC5B,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB,CAAC;KACH;CACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"TextChunkerTask.d.ts","sourceRoot":"","sources":["../../src/task/TextChunkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AAEvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,eAAO,MAAM,gBAAgB;aAC3B,KAAK,EAAE,OAAO;aACd,QAAQ,EAAE,UAAU;aACpB,SAAS,EAAE,WAAW;aACtB,QAAQ,EAAE,UAAU;CACZ,CAAC;AAEX,MAAM,MAAM,gBAAgB,GAAG,CAAC,OAAO,gBAAgB,CAAC,CAAC,MAAM,OAAO,gBAAgB,CAAC,CAAC;AAExF,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,mBAAmB;;iBAElC,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,
|
|
1
|
+
{"version":3,"file":"TextChunkerTask.d.ts","sourceRoot":"","sources":["../../src/task/TextChunkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AAEvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,eAAO,MAAM,gBAAgB;aAC3B,KAAK,EAAE,OAAO;aACd,QAAQ,EAAE,UAAU;aACpB,SAAS,EAAE,WAAW;aACtB,QAAQ,EAAE,UAAU;CACZ,CAAC;AAEX,MAAM,MAAM,gBAAgB,GAAG,CAAC,OAAO,gBAAgB,CAAC,CAAC,MAAM,OAAO,gBAAgB,CAAC,CAAC;AAExF,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,mBAAmB;;iBAElC,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EACT,oIAAoI;;iBAExI,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,0CAA0C;qBACvD,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;;iBAEd,YAAY;qBACV,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,gDAAgD;qBAC7D,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,EAAE;;iBAEb,QAAQ;qBACN,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,mBAAmB;qBAC1B,WAAW,EAAE,mCAAmC;qBAChD,OAAO;;;;;CAKsB,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;iBAEZ,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,uDAAuD;;iBAEtE,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACN,IAAI;qBACF,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,qCAAqC;;iBAEpD,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,4BAA4B;;;;;CAKZ,CAAC;AAEpC,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAClE,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AACpE,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAQrE;;;;;;;GAOG;AACH,qBAAa,eAAgB,SAAQ,IAAI,CACvC,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,CACtB;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SACgD;IAClF,OAAuB,SAAS,UAAQ;IAExC,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,oBAAoB,EAC3B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,qBAAqB,CAAC,CA6ChC;IAED,uCAAuC;IACvC,OAAO,CAAC,UAAU;IAkBlB,gEAAgE;IAChE,OAAO,CAAC,eAAe;IAgEvB,kEAAkE;IAClE,OAAO,CAAC,gBAAgB;CAoDzB;AAED,eAAO,MAAM,WAAW,UAAW,oBAAoB,WAAW,qBAAqB;;;;;;;;;;;;;;;;;;;;;;EAEtF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,qBAAqB,CAAC,CAAC;KACjG;CACF"}
|
|
@@ -96,7 +96,8 @@ export declare class VectorQuantizeTask extends Task<VectorQuantizeTaskInput, Ve
|
|
|
96
96
|
static cacheable: boolean;
|
|
97
97
|
static inputSchema(): DataPortSchema;
|
|
98
98
|
static outputSchema(): DataPortSchema;
|
|
99
|
-
|
|
99
|
+
execute(input: VectorQuantizeTaskInput): Promise<VectorQuantizeTaskOutput>;
|
|
100
|
+
executePreview(input: VectorQuantizeTaskInput): Promise<VectorQuantizeTaskOutput>;
|
|
100
101
|
private getVectorType;
|
|
101
102
|
private vectorQuantize;
|
|
102
103
|
/**
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"VectorQuantizeTask.d.ts","sourceRoot":"","sources":["../../src/task/VectorQuantizeTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AACtE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EACL,cAAc,EACd,UAAU,EAGV,UAAU,EAEV,uBAAuB,EACxB,MAAM,uBAAuB,CAAC;AAE/B,QAAA,MAAM,WAAW;mBACT,QAAQ;;;;;;;;;+BASA,OAAO;;;;;;;;4BAOV,iBAAiB;kCACX,wCAAwC;;;qBAGrD,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,0BAA0B;qBACvC,OAAO;;;qBAGP,IAAI,EAAE,SAAS;qBACf,KAAK,EAAE,WAAW;qBAClB,WAAW,EAAE,sCAAsC;qBACnD,OAAO;;;;;CAKsB,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;;;;;;;;+BASA,OAAO;;;;;;;;4BAOV,kBAAkB;kCACZ,sCAAsC;;;qBAGnD,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,sBAAsB;;;qBAGnC,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,0BAA0B;;;;;CAKV,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,WAAW,EAAE,uBAAuB,CAAC,CAAC;AAC9F,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,YAAY,EAAE,uBAAuB,CAAC,CAAC;AAChG,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,uBAAuB,CAAC,CAAC;AAE3E;;;GAGG;AACH,qBAAa,kBAAmB,SAAQ,IAAI,CAC1C,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB;IACC,OAAuB,IAAI,SAAwB;IACnD,OAAuB,QAAQ,SAAY;IAC3C,OAAuB,KAAK,SAAc;IAC1C,OAAuB,WAAW,SAAgE;IAClG,OAAuB,SAAS,UAAQ;IAExC,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,
|
|
1
|
+
{"version":3,"file":"VectorQuantizeTask.d.ts","sourceRoot":"","sources":["../../src/task/VectorQuantizeTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AACtE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EACL,cAAc,EACd,UAAU,EAGV,UAAU,EAEV,uBAAuB,EACxB,MAAM,uBAAuB,CAAC;AAE/B,QAAA,MAAM,WAAW;mBACT,QAAQ;;;;;;;;;+BASA,OAAO;;;;;;;;4BAOV,iBAAiB;kCACX,wCAAwC;;;qBAGrD,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,0BAA0B;qBACvC,OAAO;;;qBAGP,IAAI,EAAE,SAAS;qBACf,KAAK,EAAE,WAAW;qBAClB,WAAW,EAAE,sCAAsC;qBACnD,OAAO;;;;;CAKsB,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;;;;;;;;+BASA,OAAO;;;;;;;;4BAOV,kBAAkB;kCACZ,sCAAsC;;;qBAGnD,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,sBAAsB;;;qBAGnC,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,0BAA0B;;;;;CAKV,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,WAAW,EAAE,uBAAuB,CAAC,CAAC;AAC9F,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,YAAY,EAAE,uBAAuB,CAAC,CAAC;AAChG,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,uBAAuB,CAAC,CAAC;AAE3E;;;GAGG;AACH,qBAAa,kBAAmB,SAAQ,IAAI,CAC1C,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB;IACC,OAAuB,IAAI,SAAwB;IACnD,OAAuB,QAAQ,SAAY;IAC3C,OAAuB,KAAK,SAAc;IAC1C,OAAuB,WAAW,SAAgE;IAClG,OAAuB,SAAS,UAAQ;IAExC,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAGxF;IAEc,cAAc,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAa/F;IAED,OAAO,CAAC,aAAa;IAWrB,OAAO,CAAC,cAAc;IAuCtB;;OAEG;IACH,OAAO,CAAC,UAAU;IAiBlB,OAAO,CAAC,cAAc;IAMtB,OAAO,CAAC,eAAe;IASvB,OAAO,CAAC,eAAe;IAMvB,OAAO,CAAC,gBAAgB;CAQzB;AAED,eAAO,MAAM,cAAc,UAClB,uBAAuB,WACrB,wBAAwB;;;;EAGlC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,cAAc,EAAE,cAAc,CAC5B,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB,CAAC;KACH;CACF"}
|
|
@@ -83,10 +83,8 @@ export declare class VectorSimilarityTask extends GraphAsTask<VectorSimilarityTa
|
|
|
83
83
|
static readonly cacheable = true;
|
|
84
84
|
static inputSchema(): DataPortSchema;
|
|
85
85
|
static outputSchema(): DataPortSchema;
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
score: number[];
|
|
89
|
-
}>;
|
|
86
|
+
execute(input: VectorSimilarityTaskInput): Promise<VectorSimilarityTaskOutput>;
|
|
87
|
+
executePreview({ query, vectors, method, topK }: VectorSimilarityTaskInput): Promise<VectorSimilarityTaskOutput>;
|
|
90
88
|
}
|
|
91
89
|
export declare const similarity: (input: VectorSimilarityTaskInput, config?: VectorSimilarityTaskConfig) => Promise<{
|
|
92
90
|
output: import("@workglow/util/schema").TypedArray[];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"VectorSimilarityTask.d.ts","sourceRoot":"","sources":["../../src/task/VectorSimilarityTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,WAAW,EAAY,MAAM,sBAAsB,CAAC;AAC7E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAEL,cAAc,EACd,UAAU,EAIV,uBAAuB,EACxB,MAAM,uBAAuB,CAAC;AAE/B,eAAO,MAAM,YAAY;aACvB,MAAM,EAAE,QAAQ;aAChB,OAAO,EAAE,SAAS;aAClB,OAAO,EAAE,SAAS;CACV,CAAC;AAQX,MAAM,MAAM,YAAY,GAAG,CAAC,OAAO,YAAY,CAAC,CAAC,MAAM,OAAO,YAAY,CAAC,CAAC;AAE5E,QAAA,MAAM,qBAAqB;mBACnB,QAAQ;;iBAEZ,KAAK;;;;;;iBAIL,OAAO;qBACL,IAAI,EAAE,OAAO;qBACb,KAAK;;;;;;;iBAKP,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,iCAAiC;qBAC9C,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,EAAE;;iBAEb,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,yBAAe;qBACtB,WAAW,EAAE,4CAA4C;qBACzD,OAAO;;;;;CAKsB,CAAC;AAEpC,QAAA,MAAM,sBAAsB;mBACpB,QAAQ;;iBAEZ,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;;;;;;;iBAKP,KAAK;qBACH,IAAI,EAAE,OAAO;qBACb,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,KAAK,EAAE,OAAO;yBACd,WAAW,EAAE,0CAA0C;;;;;;CAM5B,CAAC;AAEpC,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAChD,OAAO,qBAAqB,EAC5B,uBAAuB,CACxB,CAAC;AACF,MAAM,MAAM,0BAA0B,GAAG,UAAU,CACjD,OAAO,sBAAsB,EAC7B,uBAAuB,CACxB,CAAC;AACF,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,yBAAyB,CAAC,CAAC;AAE/E,qBAAa,oBAAqB,SAAQ,WAAW,CACnD,yBAAyB,EACzB,0BAA0B,EAC1B,0BAA0B,CAC3B;IACC,gBAAyB,IAAI,0BAA0B;IACvD,gBAAyB,QAAQ,YAAY;IAC7C,gBAAyB,KAAK,uBAAuB;IACrD,OAAuB,WAAW,SAC+C;IACjF,gBAAyB,SAAS,QAAQ;IAE1C,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,
|
|
1
|
+
{"version":3,"file":"VectorSimilarityTask.d.ts","sourceRoot":"","sources":["../../src/task/VectorSimilarityTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,WAAW,EAAY,MAAM,sBAAsB,CAAC;AAC7E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAEL,cAAc,EACd,UAAU,EAIV,uBAAuB,EACxB,MAAM,uBAAuB,CAAC;AAE/B,eAAO,MAAM,YAAY;aACvB,MAAM,EAAE,QAAQ;aAChB,OAAO,EAAE,SAAS;aAClB,OAAO,EAAE,SAAS;CACV,CAAC;AAQX,MAAM,MAAM,YAAY,GAAG,CAAC,OAAO,YAAY,CAAC,CAAC,MAAM,OAAO,YAAY,CAAC,CAAC;AAE5E,QAAA,MAAM,qBAAqB;mBACnB,QAAQ;;iBAEZ,KAAK;;;;;;iBAIL,OAAO;qBACL,IAAI,EAAE,OAAO;qBACb,KAAK;;;;;;;iBAKP,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,iCAAiC;qBAC9C,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,EAAE;;iBAEb,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,yBAAe;qBACtB,WAAW,EAAE,4CAA4C;qBACzD,OAAO;;;;;CAKsB,CAAC;AAEpC,QAAA,MAAM,sBAAsB;mBACpB,QAAQ;;iBAEZ,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;;;;;;;iBAKP,KAAK;qBACH,IAAI,EAAE,OAAO;qBACb,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,KAAK,EAAE,OAAO;yBACd,WAAW,EAAE,0CAA0C;;;;;;CAM5B,CAAC;AAEpC,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAChD,OAAO,qBAAqB,EAC5B,uBAAuB,CACxB,CAAC;AACF,MAAM,MAAM,0BAA0B,GAAG,UAAU,CACjD,OAAO,sBAAsB,EAC7B,uBAAuB,CACxB,CAAC;AACF,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,yBAAyB,CAAC,CAAC;AAE/E,qBAAa,oBAAqB,SAAQ,WAAW,CACnD,yBAAyB,EACzB,0BAA0B,EAC1B,0BAA0B,CAC3B;IACC,gBAAyB,IAAI,0BAA0B;IACvD,gBAAyB,QAAQ,YAAY;IAC7C,gBAAyB,KAAK,uBAAuB;IACrD,OAAuB,WAAW,SAC+C;IACjF,gBAAyB,SAAS,QAAQ;IAE1C,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CAAC,KAAK,EAAE,yBAAyB,GAAG,OAAO,CAAC,0BAA0B,CAAC,CAE5F;IAEc,cAAc,CAAC,EAC5B,KAAK,EACL,OAAO,EACP,MAAM,EACN,IAAI,EACL,EAAE,yBAAyB,GAAG,OAAO,CAAC,0BAA0B,CAAC,CAmBjE;CACF;AAED,eAAO,MAAM,UAAU,UACd,yBAAyB,WACvB,0BAA0B;;;EAGpC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,UAAU,EAAE,cAAc,CACxB,yBAAyB,EACzB,0BAA0B,EAC1B,0BAA0B,CAC3B,CAAC;KACH;CACF"}
|
|
@@ -8,7 +8,7 @@
|
|
|
8
8
|
* provider-registered strategy (direct or queued).
|
|
9
9
|
*/
|
|
10
10
|
import { Job } from "@workglow/job-queue";
|
|
11
|
-
import type { IExecuteContext,
|
|
11
|
+
import type { IExecuteContext, IExecutePreviewContext, TaskConfig, TaskEntitlements, TaskOutput } from "@workglow/task-graph";
|
|
12
12
|
import { Task, TaskInput } from "@workglow/task-graph";
|
|
13
13
|
import type { ServiceRegistry } from "@workglow/util";
|
|
14
14
|
import type { DataPortSchema } from "@workglow/util/schema";
|
|
@@ -45,10 +45,10 @@ export declare class AiTask<Input extends AiTaskInput = AiTaskInput, Output exte
|
|
|
45
45
|
*/
|
|
46
46
|
protected getDefaultQueueName(input: Input): Promise<string | undefined>;
|
|
47
47
|
/**
|
|
48
|
-
* Delegates to a provider-registered
|
|
49
|
-
* otherwise falls back to the default Task.
|
|
48
|
+
* Delegates to a provider-registered preview run function if one exists,
|
|
49
|
+
* otherwise falls back to the default Task.executePreview().
|
|
50
50
|
*/
|
|
51
|
-
|
|
51
|
+
executePreview(input: Input, context: IExecutePreviewContext): Promise<Output | undefined>;
|
|
52
52
|
/**
|
|
53
53
|
* Validates that model inputs are valid ModelConfig objects.
|
|
54
54
|
*/
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AiTask.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;GAGG;AAEH,OAAO,EAAE,GAAG,EAAE,MAAM,qBAAqB,CAAC;AAC1C,OAAO,KAAK,EACV,eAAe,EACf,
|
|
1
|
+
{"version":3,"file":"AiTask.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;GAGG;AAEH,OAAO,EAAE,GAAG,EAAE,MAAM,qBAAqB,CAAC;AAC1C,OAAO,KAAK,EACV,eAAe,EACf,sBAAsB,EACtB,UAAU,EAEV,gBAAgB,EAChB,UAAU,EACX,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAEL,IAAI,EAGJ,SAAS,EAEV,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,KAAK,EAAE,cAAc,EAAc,MAAM,uBAAuB,CAAC;AAExE,OAAO,EAAS,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAGpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAiB3D,MAAM,WAAW,WAAY,SAAQ,SAAS;IAC5C,KAAK,EAAE,MAAM,GAAG,WAAW,CAAC;CAC7B;AAED;;;;;;GAMG;AACH,qBAAa,MAAM,CACjB,KAAK,SAAS,WAAW,GAAG,WAAW,EACvC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,MAAM,SAAS,UAAU,CAAC,KAAK,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,CACpD,SAAQ,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;IACnC,OAAuB,IAAI,EAAE,MAAM,CAAY;IAC/C,OAAuB,sBAAsB,EAAE,OAAO,CAAQ;IAE9D,OAAuB,YAAY,IAAI,gBAAgB,CAItD;IAEe,YAAY,IAAI,gBAAgB,CAoB/C;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAMc,OAAO,CACpB,KAAK,EAAE,KAAK,EACZ,cAAc,EAAE,eAAe,GAC9B,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CA4B7B;IAMD;;;OAGG;IACH,UAAgB,WAAW,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAsCpE;IAED;;OAEG;IACG,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,CAAC,CAYzF;IAED;;OAEG;IACH,UAAgB,mBAAmB,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAG7E;IAMD;;;OAGG;IACY,cAAc,CAC3B,KAAK,EAAE,KAAK,EACZ,OAAO,EAAE,sBAAsB,GAC9B,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAa7B;IAMD;;OAEG;IACmB,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,CA+ClE;IAEqB,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,eAAe,GAAG,OAAO,CAAC,KAAK,CAAC,CAqCzF;CACF"}
|
package/dist/worker.js
CHANGED
|
@@ -175,7 +175,7 @@ class DirectExecutionStrategy {
|
|
|
175
175
|
class AiProviderRegistry {
|
|
176
176
|
runFnRegistry = new Map;
|
|
177
177
|
streamFnRegistry = new Map;
|
|
178
|
-
|
|
178
|
+
previewRunFnRegistry = new Map;
|
|
179
179
|
providers = new Map;
|
|
180
180
|
strategyResolvers = new Map;
|
|
181
181
|
defaultStrategy;
|
|
@@ -191,7 +191,7 @@ class AiProviderRegistry {
|
|
|
191
191
|
for (const [, providerMap] of this.streamFnRegistry) {
|
|
192
192
|
providerMap.delete(name);
|
|
193
193
|
}
|
|
194
|
-
for (const [, providerMap] of this.
|
|
194
|
+
for (const [, providerMap] of this.previewRunFnRegistry) {
|
|
195
195
|
providerMap.delete(name);
|
|
196
196
|
}
|
|
197
197
|
}
|
|
@@ -269,25 +269,24 @@ class AiProviderRegistry {
|
|
|
269
269
|
const taskTypeMap = this.streamFnRegistry.get(taskType);
|
|
270
270
|
return taskTypeMap?.get(modelProvider);
|
|
271
271
|
}
|
|
272
|
-
|
|
273
|
-
const
|
|
272
|
+
registerAsWorkerPreviewRunFn(modelProvider, taskType) {
|
|
273
|
+
const previewFn = async (input, model) => {
|
|
274
274
|
const workerManager = globalServiceRegistry.get(WORKER_MANAGER);
|
|
275
|
-
return workerManager.
|
|
275
|
+
return workerManager.callWorkerPreviewFunction(modelProvider, taskType, [
|
|
276
276
|
input,
|
|
277
|
-
output,
|
|
278
277
|
model
|
|
279
278
|
]);
|
|
280
279
|
};
|
|
281
|
-
this.
|
|
280
|
+
this.registerPreviewRunFn(modelProvider, taskType, previewFn);
|
|
282
281
|
}
|
|
283
|
-
|
|
284
|
-
if (!this.
|
|
285
|
-
this.
|
|
282
|
+
registerPreviewRunFn(modelProvider, taskType, previewRunFn) {
|
|
283
|
+
if (!this.previewRunFnRegistry.has(taskType)) {
|
|
284
|
+
this.previewRunFnRegistry.set(taskType, new Map);
|
|
286
285
|
}
|
|
287
|
-
this.
|
|
286
|
+
this.previewRunFnRegistry.get(taskType).set(modelProvider, previewRunFn);
|
|
288
287
|
}
|
|
289
|
-
|
|
290
|
-
const taskTypeMap = this.
|
|
288
|
+
getPreviewRunFn(modelProvider, taskType) {
|
|
289
|
+
const taskTypeMap = this.previewRunFnRegistry.get(taskType);
|
|
291
290
|
return taskTypeMap?.get(modelProvider);
|
|
292
291
|
}
|
|
293
292
|
getDirectRunFn(modelProvider, taskType) {
|
|
@@ -325,11 +324,11 @@ function resolveAiProviderGpuQueueConcurrency(concurrency) {
|
|
|
325
324
|
class AiProvider {
|
|
326
325
|
tasks;
|
|
327
326
|
streamTasks;
|
|
328
|
-
|
|
329
|
-
constructor(tasks, streamTasks,
|
|
327
|
+
previewTasks;
|
|
328
|
+
constructor(tasks, streamTasks, previewTasks) {
|
|
330
329
|
this.tasks = tasks;
|
|
331
330
|
this.streamTasks = streamTasks;
|
|
332
|
-
this.
|
|
331
|
+
this.previewTasks = previewTasks;
|
|
333
332
|
}
|
|
334
333
|
get supportedTaskTypes() {
|
|
335
334
|
return this.taskTypes;
|
|
@@ -340,8 +339,8 @@ class AiProvider {
|
|
|
340
339
|
getStreamFn(taskType) {
|
|
341
340
|
return this.streamTasks?.[taskType];
|
|
342
341
|
}
|
|
343
|
-
|
|
344
|
-
return this.
|
|
342
|
+
getPreviewRunFn(taskType) {
|
|
343
|
+
return this.previewTasks?.[taskType];
|
|
345
344
|
}
|
|
346
345
|
async register(options = {}) {
|
|
347
346
|
const isInline = !!this.tasks;
|
|
@@ -369,7 +368,7 @@ class AiProvider {
|
|
|
369
368
|
for (const taskType of this.taskTypes) {
|
|
370
369
|
registry.registerAsWorkerRunFn(this.name, taskType);
|
|
371
370
|
registry.registerAsWorkerStreamFn(this.name, taskType);
|
|
372
|
-
registry.
|
|
371
|
+
registry.registerAsWorkerPreviewRunFn(this.name, taskType);
|
|
373
372
|
}
|
|
374
373
|
} else {
|
|
375
374
|
for (const [taskType, fn] of Object.entries(this.tasks)) {
|
|
@@ -381,9 +380,9 @@ class AiProvider {
|
|
|
381
380
|
}
|
|
382
381
|
}
|
|
383
382
|
}
|
|
384
|
-
if (this.
|
|
385
|
-
for (const [taskType, fn] of Object.entries(this.
|
|
386
|
-
registry.
|
|
383
|
+
if (this.previewTasks) {
|
|
384
|
+
for (const [taskType, fn] of Object.entries(this.previewTasks)) {
|
|
385
|
+
registry.registerPreviewRunFn(this.name, taskType, fn);
|
|
387
386
|
}
|
|
388
387
|
}
|
|
389
388
|
registry.registerProvider(this);
|
|
@@ -406,9 +405,9 @@ class AiProvider {
|
|
|
406
405
|
workerServer.registerStreamFunction(taskType, fn);
|
|
407
406
|
}
|
|
408
407
|
}
|
|
409
|
-
if (this.
|
|
410
|
-
for (const [taskType, fn] of Object.entries(this.
|
|
411
|
-
workerServer.
|
|
408
|
+
if (this.previewTasks) {
|
|
409
|
+
for (const [taskType, fn] of Object.entries(this.previewTasks)) {
|
|
410
|
+
workerServer.registerPreviewFunction(taskType, fn);
|
|
412
411
|
}
|
|
413
412
|
}
|
|
414
413
|
}
|
|
@@ -659,4 +658,4 @@ export {
|
|
|
659
658
|
AiProvider
|
|
660
659
|
};
|
|
661
660
|
|
|
662
|
-
//# debugId=
|
|
661
|
+
//# debugId=D0A6C3AF35B09B5364756E2164756E21
|