@workglow/ai 0.2.16 → 0.2.18
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 +70 -120
- package/dist/browser.js.map +12 -12
- package/dist/bun.js +70 -120
- package/dist/bun.js.map +12 -12
- package/dist/node.js +70 -120
- package/dist/node.js.map +12 -12
- 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/BackgroundRemovalTask.d.ts +3 -83
- package/dist/task/BackgroundRemovalTask.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/FaceDetectorTask.d.ts +3 -83
- package/dist/task/FaceDetectorTask.d.ts.map +1 -1
- package/dist/task/FaceLandmarkerTask.d.ts +3 -83
- package/dist/task/FaceLandmarkerTask.d.ts.map +1 -1
- package/dist/task/GestureRecognizerTask.d.ts +3 -83
- package/dist/task/GestureRecognizerTask.d.ts.map +1 -1
- package/dist/task/HandLandmarkerTask.d.ts +3 -83
- package/dist/task/HandLandmarkerTask.d.ts.map +1 -1
- package/dist/task/ImageClassificationTask.d.ts +3 -83
- package/dist/task/ImageClassificationTask.d.ts.map +1 -1
- package/dist/task/ImageEmbeddingTask.d.ts +6 -166
- package/dist/task/ImageEmbeddingTask.d.ts.map +1 -1
- package/dist/task/ImageSegmentationTask.d.ts +3 -83
- package/dist/task/ImageSegmentationTask.d.ts.map +1 -1
- package/dist/task/ImageToTextTask.d.ts +3 -83
- package/dist/task/ImageToTextTask.d.ts.map +1 -1
- package/dist/task/ObjectDetectionTask.d.ts +3 -83
- package/dist/task/ObjectDetectionTask.d.ts.map +1 -1
- package/dist/task/PoseLandmarkerTask.d.ts +3 -83
- package/dist/task/PoseLandmarkerTask.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/task/base/AiTaskSchemas.d.ts +4 -84
- package/dist/task/base/AiTaskSchemas.d.ts.map +1 -1
- package/dist/task/base/AiVisionTask.d.ts +2 -5
- package/dist/task/base/AiVisionTask.d.ts.map +1 -1
- package/dist/worker.js +25 -26
- package/dist/worker.js.map +4 -4
- package/package.json +11 -11
|
@@ -11,91 +11,11 @@ export declare const ObjectDetectionInputSchema: {
|
|
|
11
11
|
readonly type: "object";
|
|
12
12
|
readonly properties: {
|
|
13
13
|
readonly image: {
|
|
14
|
-
readonly
|
|
15
|
-
|
|
16
|
-
readonly title: "Image Data";
|
|
17
|
-
readonly description: "Image as data-uri";
|
|
18
|
-
readonly format: "image:data-uri";
|
|
19
|
-
}, {
|
|
20
|
-
readonly type: "object";
|
|
21
|
-
readonly additionalProperties: false;
|
|
22
|
-
readonly properties: {
|
|
23
|
-
readonly data: {
|
|
24
|
-
readonly oneOf: readonly [{
|
|
25
|
-
readonly type: "object";
|
|
26
|
-
readonly format: "image:ImageBitmap";
|
|
27
|
-
readonly title: "ImageBitmap";
|
|
28
|
-
}, {
|
|
29
|
-
readonly type: "object";
|
|
30
|
-
readonly format: "image:OffscreenCanvas";
|
|
31
|
-
readonly title: "OffscreenCanvas";
|
|
32
|
-
}, {
|
|
33
|
-
readonly type: "object";
|
|
34
|
-
readonly format: "image:VideoFrame";
|
|
35
|
-
readonly title: "VideoFrame";
|
|
36
|
-
}, {
|
|
37
|
-
readonly type: "object";
|
|
38
|
-
readonly properties: {
|
|
39
|
-
readonly data: {
|
|
40
|
-
readonly type: "array";
|
|
41
|
-
readonly items: {
|
|
42
|
-
readonly type: "number";
|
|
43
|
-
readonly format: "Uint8Clamped";
|
|
44
|
-
};
|
|
45
|
-
readonly format: "Uint8ClampedArray";
|
|
46
|
-
readonly title: "Data";
|
|
47
|
-
readonly description: "Data of the image";
|
|
48
|
-
};
|
|
49
|
-
readonly width: {
|
|
50
|
-
readonly type: "number";
|
|
51
|
-
readonly title: "Width";
|
|
52
|
-
readonly description: "Width of the image";
|
|
53
|
-
};
|
|
54
|
-
readonly height: {
|
|
55
|
-
readonly type: "number";
|
|
56
|
-
readonly title: "Height";
|
|
57
|
-
readonly description: "Height of the image";
|
|
58
|
-
};
|
|
59
|
-
readonly channels: {
|
|
60
|
-
readonly type: "number";
|
|
61
|
-
readonly title: "Channels";
|
|
62
|
-
readonly description: "Channels of the image";
|
|
63
|
-
};
|
|
64
|
-
readonly rawChannels: {
|
|
65
|
-
readonly type: "number";
|
|
66
|
-
readonly title: "Raw Channels";
|
|
67
|
-
readonly description: "Raw channels of the image";
|
|
68
|
-
};
|
|
69
|
-
};
|
|
70
|
-
readonly additionalProperties: false;
|
|
71
|
-
readonly required: readonly ["data", "width", "height", "channels"];
|
|
72
|
-
readonly format: "image:ImageBinary";
|
|
73
|
-
readonly title: "ImageBinary";
|
|
74
|
-
}];
|
|
75
|
-
};
|
|
76
|
-
readonly width: {
|
|
77
|
-
readonly type: "number";
|
|
78
|
-
readonly title: "Width";
|
|
79
|
-
readonly description: "Width of the image";
|
|
80
|
-
};
|
|
81
|
-
readonly height: {
|
|
82
|
-
readonly type: "number";
|
|
83
|
-
readonly title: "Height";
|
|
84
|
-
readonly description: "Height of the image";
|
|
85
|
-
};
|
|
86
|
-
readonly channels: {
|
|
87
|
-
readonly type: "number";
|
|
88
|
-
readonly title: "Channels";
|
|
89
|
-
readonly description: "Channels of the image";
|
|
90
|
-
readonly minimum: 1;
|
|
91
|
-
readonly maximum: 4;
|
|
92
|
-
};
|
|
93
|
-
};
|
|
94
|
-
readonly required: readonly ["data", "width", "height", "channels"];
|
|
95
|
-
}];
|
|
14
|
+
readonly type: "object";
|
|
15
|
+
readonly properties: {};
|
|
96
16
|
readonly title: "Image";
|
|
17
|
+
readonly description: "Image as data URI, Blob, ImageBitmap, ImageBinary, or GpuImage \u2014 hydrated to GpuImage by the runner";
|
|
97
18
|
readonly format: "image";
|
|
98
|
-
readonly description: "Image as URL or base64-encoded data";
|
|
99
19
|
};
|
|
100
20
|
readonly model: {
|
|
101
21
|
readonly oneOf: readonly [{
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ObjectDetectionTask.d.ts","sourceRoot":"","sources":["../../src/task/ObjectDetectionTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAyBnD,eAAO,MAAM,0BAA0B;mBAC/B,QAAQ;;iBAEZ,KAAK
|
|
1
|
+
{"version":3,"file":"ObjectDetectionTask.d.ts","sourceRoot":"","sources":["../../src/task/ObjectDetectionTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAyBnD,eAAO,MAAM,0BAA0B;mBAC/B,QAAQ;;iBAEZ,KAAK;;;;;;;iBACL,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACL,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;yBACH,IAAI,EAAE,QAAQ;;qBAEhB,KAAK,EAAE,QAAQ;qBACf,WAAW,EACT,kFAAkF;qBACpF,YAAY,EAAE,eAAe;;iBAE/B,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,WAAW;qBAClB,WAAW,EAAE,iDAAiD;qBAC9D,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;qBACZ,YAAY,EAAE,eAAe;;;;;CAKA,CAAC;AAEpC,eAAO,MAAM,2BAA2B;mBAChC,QAAQ;;;;+BAIA,OAAO;;mCAtDf,QAAQ;;iCAEZ,KAAK;qCACH,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,OAAO;qCACd,WAAW,EAAE,kCAAkC;;iCAEjD,KAAK;qCACH,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,kBAAkB;qCACzB,WAAW,EAAE,yCAAyC;qCACtD,OAAO,EAAE,CAAC;qCACV,OAAO,EAAE,CAAC;;iCAEZ,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iDAEM,OAAO,EAAE,OAAO,EAAE,KAAK;;;;+BAuCpB,OAAO;;6BAAW,IAAI,EAAE,OAAO;6BAAE,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;4BAEzC,YAAY;kCACN,oEAAoE;;;;;CAKpD,CAAC;AAEpC,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AACrF,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,OAAO,2BAA2B,CAAC,CAAC;AACvF,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,wBAAwB,CAAC,CAAC;AAE7E;;;GAGG;AACH,qBAAa,mBAAoB,SAAQ,YAAY,CACnD,wBAAwB,EACxB,yBAAyB,EACzB,yBAAyB,CAC1B;IACC,OAAuB,IAAI,SAAyB;IACpD,OAAuB,QAAQ,SAAqB;IACpD,OAAuB,KAAK,SAAsB;IAClD,OAAuB,WAAW,SACwE;IAC1G,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,eAAe,UACnB,wBAAwB,WACtB,yBAAyB;;;;;;;;;;;;;;;;;;;;EAGnC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,eAAe,EAAE,cAAc,CAC7B,wBAAwB,EACxB,yBAAyB,EACzB,yBAAyB,CAC1B,CAAC;KACH;CACF"}
|
|
@@ -11,91 +11,11 @@ export declare const PoseLandmarkerInputSchema: {
|
|
|
11
11
|
readonly type: "object";
|
|
12
12
|
readonly properties: {
|
|
13
13
|
readonly image: {
|
|
14
|
-
readonly
|
|
15
|
-
|
|
16
|
-
readonly title: "Image Data";
|
|
17
|
-
readonly description: "Image as data-uri";
|
|
18
|
-
readonly format: "image:data-uri";
|
|
19
|
-
}, {
|
|
20
|
-
readonly type: "object";
|
|
21
|
-
readonly additionalProperties: false;
|
|
22
|
-
readonly properties: {
|
|
23
|
-
readonly data: {
|
|
24
|
-
readonly oneOf: readonly [{
|
|
25
|
-
readonly type: "object";
|
|
26
|
-
readonly format: "image:ImageBitmap";
|
|
27
|
-
readonly title: "ImageBitmap";
|
|
28
|
-
}, {
|
|
29
|
-
readonly type: "object";
|
|
30
|
-
readonly format: "image:OffscreenCanvas";
|
|
31
|
-
readonly title: "OffscreenCanvas";
|
|
32
|
-
}, {
|
|
33
|
-
readonly type: "object";
|
|
34
|
-
readonly format: "image:VideoFrame";
|
|
35
|
-
readonly title: "VideoFrame";
|
|
36
|
-
}, {
|
|
37
|
-
readonly type: "object";
|
|
38
|
-
readonly properties: {
|
|
39
|
-
readonly data: {
|
|
40
|
-
readonly type: "array";
|
|
41
|
-
readonly items: {
|
|
42
|
-
readonly type: "number";
|
|
43
|
-
readonly format: "Uint8Clamped";
|
|
44
|
-
};
|
|
45
|
-
readonly format: "Uint8ClampedArray";
|
|
46
|
-
readonly title: "Data";
|
|
47
|
-
readonly description: "Data of the image";
|
|
48
|
-
};
|
|
49
|
-
readonly width: {
|
|
50
|
-
readonly type: "number";
|
|
51
|
-
readonly title: "Width";
|
|
52
|
-
readonly description: "Width of the image";
|
|
53
|
-
};
|
|
54
|
-
readonly height: {
|
|
55
|
-
readonly type: "number";
|
|
56
|
-
readonly title: "Height";
|
|
57
|
-
readonly description: "Height of the image";
|
|
58
|
-
};
|
|
59
|
-
readonly channels: {
|
|
60
|
-
readonly type: "number";
|
|
61
|
-
readonly title: "Channels";
|
|
62
|
-
readonly description: "Channels of the image";
|
|
63
|
-
};
|
|
64
|
-
readonly rawChannels: {
|
|
65
|
-
readonly type: "number";
|
|
66
|
-
readonly title: "Raw Channels";
|
|
67
|
-
readonly description: "Raw channels of the image";
|
|
68
|
-
};
|
|
69
|
-
};
|
|
70
|
-
readonly additionalProperties: false;
|
|
71
|
-
readonly required: readonly ["data", "width", "height", "channels"];
|
|
72
|
-
readonly format: "image:ImageBinary";
|
|
73
|
-
readonly title: "ImageBinary";
|
|
74
|
-
}];
|
|
75
|
-
};
|
|
76
|
-
readonly width: {
|
|
77
|
-
readonly type: "number";
|
|
78
|
-
readonly title: "Width";
|
|
79
|
-
readonly description: "Width of the image";
|
|
80
|
-
};
|
|
81
|
-
readonly height: {
|
|
82
|
-
readonly type: "number";
|
|
83
|
-
readonly title: "Height";
|
|
84
|
-
readonly description: "Height of the image";
|
|
85
|
-
};
|
|
86
|
-
readonly channels: {
|
|
87
|
-
readonly type: "number";
|
|
88
|
-
readonly title: "Channels";
|
|
89
|
-
readonly description: "Channels of the image";
|
|
90
|
-
readonly minimum: 1;
|
|
91
|
-
readonly maximum: 4;
|
|
92
|
-
};
|
|
93
|
-
};
|
|
94
|
-
readonly required: readonly ["data", "width", "height", "channels"];
|
|
95
|
-
}];
|
|
14
|
+
readonly type: "object";
|
|
15
|
+
readonly properties: {};
|
|
96
16
|
readonly title: "Image";
|
|
17
|
+
readonly description: "Image as data URI, Blob, ImageBitmap, ImageBinary, or GpuImage \u2014 hydrated to GpuImage by the runner";
|
|
97
18
|
readonly format: "image";
|
|
98
|
-
readonly description: "Image as URL or base64-encoded data";
|
|
99
19
|
};
|
|
100
20
|
readonly model: {
|
|
101
21
|
readonly oneOf: readonly [{
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"PoseLandmarkerTask.d.ts","sourceRoot":"","sources":["../../src/task/PoseLandmarkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AA0FnD,eAAO,MAAM,yBAAyB;mBAC9B,QAAQ;;iBAEZ,KAAK
|
|
1
|
+
{"version":3,"file":"PoseLandmarkerTask.d.ts","sourceRoot":"","sources":["../../src/task/PoseLandmarkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AA0FnD,eAAO,MAAM,yBAAyB;mBAC9B,QAAQ;;iBAEZ,KAAK;;;;;;;iBACL,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACL,QAAQ;qBACN,IAAI,EAAE,QAAQ;qBACd,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,EAAE;qBACX,OAAO,EAAE,CAAC;qBACV,KAAK,EAAE,iBAAiB;qBACxB,WAAW,EAAE,uCAAuC;qBACpD,YAAY,EAAE,eAAe;;iBAE/B,0BAA0B;qBACxB,IAAI,EAAE,QAAQ;qBACd,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;qBACZ,KAAK,EAAE,+BAA+B;qBACtC,WAAW,EAAE,6CAA6C;qBAC1D,YAAY,EAAE,eAAe;;iBAE/B,yBAAyB;qBACvB,IAAI,EAAE,QAAQ;qBACd,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;qBACZ,KAAK,EAAE,8BAA8B;qBACrC,WAAW,EAAE,4CAA4C;qBACzD,YAAY,EAAE,eAAe;;iBAE/B,qBAAqB;qBACnB,IAAI,EAAE,QAAQ;qBACd,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;qBACZ,KAAK,EAAE,yBAAyB;qBAChC,WAAW,EAAE,4CAA4C;qBACzD,YAAY,EAAE,eAAe;;iBAE/B,uBAAuB;qBACrB,IAAI,EAAE,SAAS;qBACf,OAAO;qBACP,KAAK,EAAE,2BAA2B;qBAClC,WAAW,EAAE,yDAAyD;qBACtE,YAAY,EAAE,eAAe;;;;;CAKA,CAAC;AAEpC,eAAO,MAAM,0BAA0B;mBAC/B,QAAQ;;;;+BAIA,OAAO;;mCA9Ef,QAAQ;;iCAEZ,SAAS;qCACP,IAAI,EAAE,OAAO;qCACb,KAAK;+CAlEH,QAAQ;;6CAEZ,CAAC;iDACC,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,cAAc;iDACrB,WAAW,EAAE,uCAAuC;;6CAEtD,CAAC;iDACC,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,cAAc;iDACrB,WAAW,EAAE,uCAAuC;;6CAEtD,CAAC;iDACC,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,cAAc;iDACrB,WAAW,EAAE,sBAAsB;;6CAErC,UAAU;iDACR,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,YAAY;iDACnB,WAAW,EAAE,2DAA2D;;6CAE1E,QAAQ;iDACN,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,UAAU;iDACjB,WAAW,EAAE,uDAAuD;;;6DAG7D,GAAG,EAAE,GAAG,EAAE,GAAG;;;qCAuCpB,KAAK,EAAE,WAAW;qCAClB,WAAW,EAAE,wCAAwC;;iCAEvD,cAAc;qCACZ,IAAI,EAAE,OAAO;qCACb,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCACL,KAAK,EAAE,iBAAiB;qCACxB,WAAW,EAAE,oDAAoD;;iCAEnE,gBAAgB;2CAxCZ,QAAQ;;yCAEZ,IAAI;6CACF,IAAI,EAAE,QAAQ;6CACd,KAAK,EAAE,WAAW;6CAClB,WAAW,EAAE,uDAAuD;;yCAEtE,KAAK;6CACH,IAAI,EAAE,QAAQ;6CACd,KAAK,EAAE,OAAO;6CACd,WAAW,EAAE,gCAAgC;;yCAE/C,MAAM;6CACJ,IAAI,EAAE,QAAQ;6CACd,KAAK,EAAE,QAAQ;6CACf,WAAW,EAAE,iCAAiC;;;yDAGvC,MAAM,EAAE,OAAO,EAAE,QAAQ;;;;iDAwBzB,WAAW,EAAE,gBAAgB;;;;+BA+D1B,OAAO;;6BAAW,IAAI,EAAE,OAAO;6BAAE,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;4BAEzC,iBAAiB;kCACX,+DAA+D;;;;;CAK/C,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,yBAAyB,CAAC,CAAC;AACnF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AACrF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,uBAAuB,CAAC,CAAC;AAE3E;;;GAGG;AACH,qBAAa,kBAAmB,SAAQ,YAAY,CAClD,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB;IACC,OAAuB,IAAI,SAAwB;IACnD,OAAuB,QAAQ,SAAqB;IACpD,OAAuB,KAAK,SAAqB;IACjD,OAAuB,WAAW,SACgF;IAClH,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;CACF;AAED;;;;;GAKG;AACH,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"}
|
|
@@ -151,94 +151,14 @@ export declare function TypeSingleOrArray<const T extends DataPortSchemaNonBoole
|
|
|
151
151
|
};
|
|
152
152
|
export type ImageSource = ImageBitmap | OffscreenCanvas | VideoFrame;
|
|
153
153
|
/**
|
|
154
|
-
* Image input schema
|
|
154
|
+
* Image input schema — hydrated to GpuImage by the runner via the image input resolver.
|
|
155
155
|
*/
|
|
156
156
|
export declare const TypeImageInput: {
|
|
157
|
-
readonly
|
|
158
|
-
|
|
159
|
-
readonly title: "Image Data";
|
|
160
|
-
readonly description: "Image as data-uri";
|
|
161
|
-
readonly format: "image:data-uri";
|
|
162
|
-
}, {
|
|
163
|
-
readonly type: "object";
|
|
164
|
-
readonly additionalProperties: false;
|
|
165
|
-
readonly properties: {
|
|
166
|
-
readonly data: {
|
|
167
|
-
readonly oneOf: readonly [{
|
|
168
|
-
readonly type: "object";
|
|
169
|
-
readonly format: "image:ImageBitmap";
|
|
170
|
-
readonly title: "ImageBitmap";
|
|
171
|
-
}, {
|
|
172
|
-
readonly type: "object";
|
|
173
|
-
readonly format: "image:OffscreenCanvas";
|
|
174
|
-
readonly title: "OffscreenCanvas";
|
|
175
|
-
}, {
|
|
176
|
-
readonly type: "object";
|
|
177
|
-
readonly format: "image:VideoFrame";
|
|
178
|
-
readonly title: "VideoFrame";
|
|
179
|
-
}, {
|
|
180
|
-
readonly type: "object";
|
|
181
|
-
readonly properties: {
|
|
182
|
-
readonly data: {
|
|
183
|
-
readonly type: "array";
|
|
184
|
-
readonly items: {
|
|
185
|
-
readonly type: "number";
|
|
186
|
-
readonly format: "Uint8Clamped";
|
|
187
|
-
};
|
|
188
|
-
readonly format: "Uint8ClampedArray";
|
|
189
|
-
readonly title: "Data";
|
|
190
|
-
readonly description: "Data of the image";
|
|
191
|
-
};
|
|
192
|
-
readonly width: {
|
|
193
|
-
readonly type: "number";
|
|
194
|
-
readonly title: "Width";
|
|
195
|
-
readonly description: "Width of the image";
|
|
196
|
-
};
|
|
197
|
-
readonly height: {
|
|
198
|
-
readonly type: "number";
|
|
199
|
-
readonly title: "Height";
|
|
200
|
-
readonly description: "Height of the image";
|
|
201
|
-
};
|
|
202
|
-
readonly channels: {
|
|
203
|
-
readonly type: "number";
|
|
204
|
-
readonly title: "Channels";
|
|
205
|
-
readonly description: "Channels of the image";
|
|
206
|
-
};
|
|
207
|
-
readonly rawChannels: {
|
|
208
|
-
readonly type: "number";
|
|
209
|
-
readonly title: "Raw Channels";
|
|
210
|
-
readonly description: "Raw channels of the image";
|
|
211
|
-
};
|
|
212
|
-
};
|
|
213
|
-
readonly additionalProperties: false;
|
|
214
|
-
readonly required: readonly ["data", "width", "height", "channels"];
|
|
215
|
-
readonly format: "image:ImageBinary";
|
|
216
|
-
readonly title: "ImageBinary";
|
|
217
|
-
}];
|
|
218
|
-
};
|
|
219
|
-
readonly width: {
|
|
220
|
-
readonly type: "number";
|
|
221
|
-
readonly title: "Width";
|
|
222
|
-
readonly description: "Width of the image";
|
|
223
|
-
};
|
|
224
|
-
readonly height: {
|
|
225
|
-
readonly type: "number";
|
|
226
|
-
readonly title: "Height";
|
|
227
|
-
readonly description: "Height of the image";
|
|
228
|
-
};
|
|
229
|
-
readonly channels: {
|
|
230
|
-
readonly type: "number";
|
|
231
|
-
readonly title: "Channels";
|
|
232
|
-
readonly description: "Channels of the image";
|
|
233
|
-
readonly minimum: 1;
|
|
234
|
-
readonly maximum: 4;
|
|
235
|
-
};
|
|
236
|
-
};
|
|
237
|
-
readonly required: readonly ["data", "width", "height", "channels"];
|
|
238
|
-
}];
|
|
157
|
+
readonly type: "object";
|
|
158
|
+
readonly properties: {};
|
|
239
159
|
readonly title: "Image";
|
|
160
|
+
readonly description: "Image as data URI, Blob, ImageBitmap, ImageBinary, or GpuImage \u2014 hydrated to GpuImage by the runner";
|
|
240
161
|
readonly format: "image";
|
|
241
|
-
readonly description: "Image as URL or base64-encoded data";
|
|
242
162
|
};
|
|
243
163
|
/**
|
|
244
164
|
* Audio input schema supporting URIs and base64-encoded audio in multiple formats
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AiTaskSchemas.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiTaskSchemas.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAc,MAAM,uBAAuB,CAAC;AAG7E,eAAO,MAAM,YAAY,iBAAiB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;mBAEvD,QAAQ;oBACP,UAAU;0BACJ,qBAAqB;wBACvB,CAAC;wBACD,CAAC;CAEH,CAAC;AAEd,MAAM,MAAM,iBAAiB,GAAG,OAAO,GAAG,SAAS,MAAM,EAAE,CAAC;AAE5D,MAAM,MAAM,UAAU,GAAG,wBAAwB,GAAG;IAClD,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;IACxB,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC;CACpC,CAAC;AAEF,wBAAgB,iBAAiB,CAC/B,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;oBAYvC,OAAO;;;;mBAIR,QAAQ;EAEjB;AAED,wBAAgB,iBAAiB,CAC/B,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EASjD;AAED,wBAAgB,SAAS,CACvB,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;;wBAzBvC,OAAO;;;;uBAIR,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA2BjB;AAED,wBAAgB,iBAAiB,CAAC,KAAK,CAAC,CAAC,SAAS,wBAAwB,EAAE,IAAI,EAAE,CAAC;;uBAEzD,OAAO;;;EAEhC;AAED,MAAM,MAAM,WAAW,GAAG,WAAW,GAAG,eAAe,GAAG,UAAU,CAAC;AAErE;;GAEG;AACH,eAAO,MAAM,cAAc
|
|
1
|
+
{"version":3,"file":"AiTaskSchemas.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiTaskSchemas.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAc,MAAM,uBAAuB,CAAC;AAG7E,eAAO,MAAM,YAAY,iBAAiB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;mBAEvD,QAAQ;oBACP,UAAU;0BACJ,qBAAqB;wBACvB,CAAC;wBACD,CAAC;CAEH,CAAC;AAEd,MAAM,MAAM,iBAAiB,GAAG,OAAO,GAAG,SAAS,MAAM,EAAE,CAAC;AAE5D,MAAM,MAAM,UAAU,GAAG,wBAAwB,GAAG;IAClD,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;IACxB,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC;CACpC,CAAC;AAEF,wBAAgB,iBAAiB,CAC/B,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;oBAYvC,OAAO;;;;mBAIR,QAAQ;EAEjB;AAED,wBAAgB,iBAAiB,CAC/B,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EASjD;AAED,wBAAgB,SAAS,CACvB,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;;wBAzBvC,OAAO;;;;uBAIR,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA2BjB;AAED,wBAAgB,iBAAiB,CAAC,KAAK,CAAC,CAAC,SAAS,wBAAwB,EAAE,IAAI,EAAE,CAAC;;uBAEzD,OAAO;;;EAEhC;AAED,MAAM,MAAM,WAAW,GAAG,WAAW,GAAG,eAAe,GAAG,UAAU,CAAC;AAErE;;GAEG;AACH,eAAO,MAAM,cAAc;mBACnB,QAAQ;;oBAEP,OAAO;0BACD,0GAAqG;qBAC1G,OAAO;CACc,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,cAAc;mBACnB,QAAQ;oBACP,OAAO;qBACN,gBAAgB;0BACX,4BAA4B;CACZ,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,eAAe;mBACpB,QAAQ;;iBAEZ,CAAC;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,cAAc;qBAAE,WAAW,EAAE,+BAA+B;;iBACxF,CAAC;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,cAAc;qBAAE,WAAW,EAAE,8BAA8B;;iBACvF,KAAK;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,OAAO;qBAAE,WAAW,EAAE,2BAA2B;;iBACjF,MAAM;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,QAAQ;qBAAE,WAAW,EAAE,4BAA4B;;;;;oBAI/E,cAAc;0BACR,0BAA0B;CACV,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,YAAY;mBACjB,QAAQ;;iBAEZ,KAAK;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,OAAO;qBAAE,WAAW,EAAE,gBAAgB;;iBACtE,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,kCAAkC;qBAC/C,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;;;;;oBAKP,UAAU;0BACJ,8CAA8C;CAC9B,CAAC"}
|
|
@@ -8,14 +8,11 @@ import { AiJobInput } from "../../job/AiJob";
|
|
|
8
8
|
import { AiTask, AiTaskInput } from "./AiTask";
|
|
9
9
|
/**
|
|
10
10
|
* A base class for AI vision tasks.
|
|
11
|
-
*
|
|
11
|
+
* Materializes GpuImage to raw pixels/bitmap at the worker boundary so the
|
|
12
|
+
* worker (which doesn't import GPU code) receives transferable binary data.
|
|
12
13
|
*/
|
|
13
14
|
export declare class AiVisionTask<Input extends AiTaskInput = AiTaskInput, Output extends TaskOutput = TaskOutput, Config extends TaskConfig<Input> = TaskConfig<Input>> extends AiTask<Input, Output, Config> {
|
|
14
15
|
static type: string;
|
|
15
|
-
/**
|
|
16
|
-
* Get the input to submit for execution.
|
|
17
|
-
* Converts image data to a format supported by the target provider.
|
|
18
|
-
*/
|
|
19
16
|
protected getJobInput(input: Input): Promise<AiJobInput<Input>>;
|
|
20
17
|
}
|
|
21
18
|
//# sourceMappingURL=AiVisionTask.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AiVisionTask.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiVisionTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGnE,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAE7C,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAE/C
|
|
1
|
+
{"version":3,"file":"AiVisionTask.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiVisionTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAGnE,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAE7C,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAE/C;;;;GAIG;AACH,qBAAa,YAAY,CACvB,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,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;IACrC,OAAuB,IAAI,EAAE,MAAM,CAAkB;IAErD,UAAyB,WAAW,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CA2B7E;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
|