@workglow/ai 0.0.121 → 0.0.123
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/README.md +20 -27
- package/dist/browser.js +245 -135
- package/dist/browser.js.map +60 -57
- package/dist/bun.js +245 -135
- package/dist/bun.js.map +60 -57
- package/dist/common.d.ts +2 -0
- package/dist/common.d.ts.map +1 -1
- package/dist/job/AiJob.d.ts +1 -1
- package/dist/job/AiJob.d.ts.map +1 -1
- package/dist/model/InMemoryModelRepository.d.ts.map +1 -1
- package/dist/model/ModelRepository.d.ts +10 -10
- package/dist/model/ModelRepository.d.ts.map +1 -1
- package/dist/model/ModelSchema.d.ts +1 -1
- package/dist/model/ModelSchema.d.ts.map +1 -1
- package/dist/node.js +245 -135
- package/dist/node.js.map +60 -57
- package/dist/provider/AiProvider.d.ts +33 -30
- package/dist/provider/AiProvider.d.ts.map +1 -1
- package/dist/provider/AiProviderRegistry.d.ts +4 -4
- package/dist/provider/AiProviderRegistry.d.ts.map +1 -1
- package/dist/provider/QueuedAiProvider.d.ts +18 -0
- package/dist/provider/QueuedAiProvider.d.ts.map +1 -0
- package/dist/task/AgentTask.d.ts +2 -2
- package/dist/task/AgentTask.d.ts.map +1 -1
- package/dist/task/AgentTypes.d.ts +1 -1
- package/dist/task/AgentTypes.d.ts.map +1 -1
- package/dist/task/AgentUtils.d.ts +1 -1
- package/dist/task/AgentUtils.d.ts.map +1 -1
- package/dist/task/BackgroundRemovalTask.d.ts +32 -2
- package/dist/task/BackgroundRemovalTask.d.ts.map +1 -1
- package/dist/task/ChunkRetrievalTask.d.ts +7 -7
- package/dist/task/ChunkRetrievalTask.d.ts.map +1 -1
- package/dist/task/ChunkToVectorTask.d.ts +26 -4
- package/dist/task/ChunkToVectorTask.d.ts.map +1 -1
- package/dist/task/ChunkVectorHybridSearchTask.d.ts +23 -2
- package/dist/task/ChunkVectorHybridSearchTask.d.ts.map +1 -1
- package/dist/task/ChunkVectorSearchTask.d.ts +13 -5
- package/dist/task/ChunkVectorSearchTask.d.ts.map +1 -1
- package/dist/task/ChunkVectorUpsertTask.d.ts +12 -3
- package/dist/task/ChunkVectorUpsertTask.d.ts.map +1 -1
- package/dist/task/ContextBuilderTask.d.ts +30 -4
- package/dist/task/ContextBuilderTask.d.ts.map +1 -1
- package/dist/task/CountTokensTask.d.ts +19 -2
- package/dist/task/CountTokensTask.d.ts.map +1 -1
- package/dist/task/DocumentEnricherTask.d.ts +39 -18
- package/dist/task/DocumentEnricherTask.d.ts.map +1 -1
- package/dist/task/DownloadModelTask.d.ts +25 -4
- package/dist/task/DownloadModelTask.d.ts.map +1 -1
- package/dist/task/FaceDetectorTask.d.ts +39 -7
- package/dist/task/FaceDetectorTask.d.ts.map +1 -1
- package/dist/task/FaceLandmarkerTask.d.ts +44 -8
- package/dist/task/FaceLandmarkerTask.d.ts.map +1 -1
- package/dist/task/GestureRecognizerTask.d.ts +49 -15
- package/dist/task/GestureRecognizerTask.d.ts.map +1 -1
- package/dist/task/HandLandmarkerTask.d.ts +46 -12
- package/dist/task/HandLandmarkerTask.d.ts.map +1 -1
- package/dist/task/HierarchicalChunkerTask.d.ts +33 -22
- package/dist/task/HierarchicalChunkerTask.d.ts.map +1 -1
- package/dist/task/HierarchyJoinTask.d.ts +36 -11
- package/dist/task/HierarchyJoinTask.d.ts.map +1 -1
- package/dist/task/ImageClassificationTask.d.ts +37 -5
- package/dist/task/ImageClassificationTask.d.ts.map +1 -1
- package/dist/task/ImageEmbeddingTask.d.ts +33 -3
- package/dist/task/ImageEmbeddingTask.d.ts.map +1 -1
- package/dist/task/ImageSegmentationTask.d.ts +37 -5
- package/dist/task/ImageSegmentationTask.d.ts.map +1 -1
- package/dist/task/ImageToTextTask.d.ts +33 -2
- package/dist/task/ImageToTextTask.d.ts.map +1 -1
- package/dist/task/ModelInfoTask.d.ts +36 -10
- package/dist/task/ModelInfoTask.d.ts.map +1 -1
- package/dist/task/ModelSearchTask.d.ts +66 -0
- package/dist/task/ModelSearchTask.d.ts.map +1 -0
- package/dist/task/ObjectDetectionTask.d.ts +39 -7
- package/dist/task/ObjectDetectionTask.d.ts.map +1 -1
- package/dist/task/PoseLandmarkerTask.d.ts +53 -18
- package/dist/task/PoseLandmarkerTask.d.ts.map +1 -1
- package/dist/task/QueryExpanderTask.d.ts +9 -4
- package/dist/task/QueryExpanderTask.d.ts.map +1 -1
- package/dist/task/RerankerTask.d.ts +27 -3
- package/dist/task/RerankerTask.d.ts.map +1 -1
- package/dist/task/StructuralParserTask.d.ts +8 -2
- package/dist/task/StructuralParserTask.d.ts.map +1 -1
- package/dist/task/StructuredGenerationTask.d.ts +24 -2
- package/dist/task/StructuredGenerationTask.d.ts.map +1 -1
- package/dist/task/TextChunkerTask.d.ts +11 -6
- package/dist/task/TextChunkerTask.d.ts.map +1 -1
- package/dist/task/TextClassificationTask.d.ts +24 -5
- package/dist/task/TextClassificationTask.d.ts.map +1 -1
- package/dist/task/TextEmbeddingTask.d.ts +20 -3
- package/dist/task/TextEmbeddingTask.d.ts.map +1 -1
- package/dist/task/TextFillMaskTask.d.ts +23 -6
- package/dist/task/TextFillMaskTask.d.ts.map +1 -1
- package/dist/task/TextGenerationTask.d.ts +24 -2
- package/dist/task/TextGenerationTask.d.ts.map +1 -1
- package/dist/task/TextLanguageDetectionTask.d.ts +23 -5
- package/dist/task/TextLanguageDetectionTask.d.ts.map +1 -1
- package/dist/task/TextNamedEntityRecognitionTask.d.ts +24 -6
- package/dist/task/TextNamedEntityRecognitionTask.d.ts.map +1 -1
- package/dist/task/TextQuestionAnswerTask.d.ts +20 -2
- package/dist/task/TextQuestionAnswerTask.d.ts.map +1 -1
- package/dist/task/TextRewriterTask.d.ts +20 -2
- package/dist/task/TextRewriterTask.d.ts.map +1 -1
- package/dist/task/TextSummaryTask.d.ts +19 -2
- package/dist/task/TextSummaryTask.d.ts.map +1 -1
- package/dist/task/TextTranslationTask.d.ts +21 -2
- package/dist/task/TextTranslationTask.d.ts.map +1 -1
- package/dist/task/ToolCallingTask.d.ts +10 -66
- package/dist/task/ToolCallingTask.d.ts.map +1 -1
- package/dist/task/ToolCallingUtils.d.ts +65 -0
- package/dist/task/ToolCallingUtils.d.ts.map +1 -0
- package/dist/task/TopicSegmenterTask.d.ts +11 -5
- package/dist/task/TopicSegmenterTask.d.ts.map +1 -1
- package/dist/task/UnloadModelTask.d.ts +19 -3
- package/dist/task/UnloadModelTask.d.ts.map +1 -1
- package/dist/task/VectorQuantizeTask.d.ts +12 -8
- package/dist/task/VectorQuantizeTask.d.ts.map +1 -1
- package/dist/task/VectorSimilarityTask.d.ts +10 -5
- package/dist/task/VectorSimilarityTask.d.ts.map +1 -1
- package/dist/task/base/AiTask.d.ts +1 -1
- package/dist/task/base/AiTask.d.ts.map +1 -1
- package/dist/task/base/AiTaskSchemas.d.ts +1 -1
- package/dist/task/base/AiTaskSchemas.d.ts.map +1 -1
- package/dist/task/base/AiVisionTask.d.ts.map +1 -1
- package/dist/task/base/StreamingAiTask.d.ts.map +1 -1
- package/dist/task/index.d.ts +4 -1
- package/dist/task/index.d.ts.map +1 -1
- package/dist/worker.d.ts +21 -0
- package/dist/worker.d.ts.map +1 -0
- package/dist/worker.js +460 -0
- package/dist/worker.js.map +14 -0
- package/package.json +28 -21
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, JobQueueTaskConfig } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
import { AiVisionTask } from "./base/AiVisionTask";
|
|
9
9
|
export declare const ObjectDetectionInputSchema: {
|
|
10
10
|
readonly type: "object";
|
|
@@ -310,26 +310,58 @@ export declare class ObjectDetectionTask extends AiVisionTask<ObjectDetectionTas
|
|
|
310
310
|
* @param input The input parameters for object detection (image, model, and optional labels)
|
|
311
311
|
* @returns Promise resolving to the detected objects with labels, scores, and bounding boxes
|
|
312
312
|
*/
|
|
313
|
-
export declare const objectDetection: (input:
|
|
313
|
+
export declare const objectDetection: (input: {
|
|
314
|
+
image: string | {
|
|
315
|
+
channels: number;
|
|
316
|
+
data: {
|
|
317
|
+
[x: string]: unknown;
|
|
318
|
+
} | {
|
|
319
|
+
channels: number;
|
|
320
|
+
data: number[];
|
|
321
|
+
height: number;
|
|
322
|
+
rawChannels?: number | undefined;
|
|
323
|
+
width: number;
|
|
324
|
+
};
|
|
325
|
+
height: number;
|
|
326
|
+
width: number;
|
|
327
|
+
};
|
|
328
|
+
labels?: string[] | undefined;
|
|
329
|
+
model: string | {
|
|
330
|
+
[x: string]: unknown;
|
|
331
|
+
description?: string | undefined;
|
|
332
|
+
metadata?: {
|
|
333
|
+
[x: string]: unknown;
|
|
334
|
+
} | undefined;
|
|
335
|
+
model_id?: string | undefined;
|
|
336
|
+
provider: string;
|
|
337
|
+
provider_config: {
|
|
338
|
+
[x: string]: unknown;
|
|
339
|
+
credential_key?: string | undefined;
|
|
340
|
+
};
|
|
341
|
+
tasks?: string[] | undefined;
|
|
342
|
+
title?: string | undefined;
|
|
343
|
+
};
|
|
344
|
+
threshold?: number | undefined;
|
|
345
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
314
346
|
detections: {
|
|
315
|
-
score: number;
|
|
316
347
|
box: {
|
|
317
|
-
width: number;
|
|
318
348
|
height: number;
|
|
349
|
+
width: number;
|
|
319
350
|
x: number;
|
|
320
351
|
y: number;
|
|
321
352
|
};
|
|
322
353
|
label: string;
|
|
323
|
-
}[] | {
|
|
324
354
|
score: number;
|
|
355
|
+
}[][] | {
|
|
325
356
|
box: {
|
|
326
|
-
width: number;
|
|
327
357
|
height: number;
|
|
358
|
+
width: number;
|
|
328
359
|
x: number;
|
|
329
360
|
y: number;
|
|
330
361
|
};
|
|
331
362
|
label: string;
|
|
332
|
-
|
|
363
|
+
score: number;
|
|
364
|
+
}[];
|
|
333
365
|
}>;
|
|
334
366
|
declare module "@workglow/task-graph" {
|
|
335
367
|
interface Workflow {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ObjectDetectionTask.d.ts","sourceRoot":"","sources":["../../src/task/ObjectDetectionTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"ObjectDetectionTask.d.ts","sourceRoot":"","sources":["../../src/task/ObjectDetectionTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAyBnD,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2BJ,CAAC;AAEpC,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAcL,CAAC;AAEpC,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AACrF,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,OAAO,2BAA2B,CAAC,CAAC;AAEvF;;;GAGG;AACH,qBAAa,mBAAoB,SAAQ,YAAY,CACnD,wBAAwB,EACxB,yBAAyB,EACzB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAyB;IAC3C,OAAc,QAAQ,SAAqB;IAC3C,OAAc,KAAK,SAAsB;IACzC,OAAc,WAAW,SACiF;IAC1G,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAE3B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,eAAe,EAAE,cAAc,CAC7B,wBAAwB,EACxB,yBAAyB,EACzB,kBAAkB,CACnB,CAAC;KACH;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, JobQueueTaskConfig } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
import { AiVisionTask } from "./base/AiVisionTask";
|
|
9
9
|
export declare const PoseLandmarkerInputSchema: {
|
|
10
10
|
readonly type: "object";
|
|
@@ -445,52 +445,87 @@ export declare class PoseLandmarkerTask extends AiVisionTask<PoseLandmarkerTaskI
|
|
|
445
445
|
* @param input The input parameters for pose landmark detection (image, model, and optional configuration)
|
|
446
446
|
* @returns Promise resolving to the detected pose landmarks and optional segmentation masks
|
|
447
447
|
*/
|
|
448
|
-
export declare const poseLandmarker: (input:
|
|
448
|
+
export declare const poseLandmarker: (input: {
|
|
449
|
+
image: string | {
|
|
450
|
+
channels: number;
|
|
451
|
+
data: {
|
|
452
|
+
[x: string]: unknown;
|
|
453
|
+
} | {
|
|
454
|
+
channels: number;
|
|
455
|
+
data: number[];
|
|
456
|
+
height: number;
|
|
457
|
+
rawChannels?: number | undefined;
|
|
458
|
+
width: number;
|
|
459
|
+
};
|
|
460
|
+
height: number;
|
|
461
|
+
width: number;
|
|
462
|
+
};
|
|
463
|
+
minPoseDetectionConfidence?: number | undefined;
|
|
464
|
+
minPosePresenceConfidence?: number | undefined;
|
|
465
|
+
minTrackingConfidence?: number | undefined;
|
|
466
|
+
model: string | {
|
|
467
|
+
[x: string]: unknown;
|
|
468
|
+
description?: string | undefined;
|
|
469
|
+
metadata?: {
|
|
470
|
+
[x: string]: unknown;
|
|
471
|
+
} | undefined;
|
|
472
|
+
model_id?: string | undefined;
|
|
473
|
+
provider: string;
|
|
474
|
+
provider_config: {
|
|
475
|
+
[x: string]: unknown;
|
|
476
|
+
credential_key?: string | undefined;
|
|
477
|
+
};
|
|
478
|
+
tasks?: string[] | undefined;
|
|
479
|
+
title?: string | undefined;
|
|
480
|
+
};
|
|
481
|
+
numPoses?: number | undefined;
|
|
482
|
+
outputSegmentationMasks?: boolean | undefined;
|
|
483
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
449
484
|
poses: {
|
|
485
|
+
landmarks: {
|
|
486
|
+
presence?: number | undefined;
|
|
487
|
+
visibility?: number | undefined;
|
|
488
|
+
x: number;
|
|
489
|
+
y: number;
|
|
490
|
+
z: number;
|
|
491
|
+
}[];
|
|
450
492
|
segmentationMask?: {
|
|
451
493
|
data: {
|
|
452
494
|
[x: string]: unknown;
|
|
453
495
|
};
|
|
454
|
-
width: number;
|
|
455
496
|
height: number;
|
|
497
|
+
width: number;
|
|
456
498
|
} | undefined;
|
|
457
|
-
|
|
458
|
-
visibility?: number | undefined;
|
|
499
|
+
worldLandmarks: {
|
|
459
500
|
presence?: number | undefined;
|
|
501
|
+
visibility?: number | undefined;
|
|
460
502
|
x: number;
|
|
461
503
|
y: number;
|
|
462
504
|
z: number;
|
|
463
505
|
}[];
|
|
464
|
-
|
|
465
|
-
|
|
506
|
+
}[][] | {
|
|
507
|
+
landmarks: {
|
|
466
508
|
presence?: number | undefined;
|
|
509
|
+
visibility?: number | undefined;
|
|
467
510
|
x: number;
|
|
468
511
|
y: number;
|
|
469
512
|
z: number;
|
|
470
513
|
}[];
|
|
471
|
-
}[] | {
|
|
472
514
|
segmentationMask?: {
|
|
473
515
|
data: {
|
|
474
516
|
[x: string]: unknown;
|
|
475
517
|
};
|
|
476
|
-
width: number;
|
|
477
518
|
height: number;
|
|
519
|
+
width: number;
|
|
478
520
|
} | undefined;
|
|
479
|
-
landmarks: {
|
|
480
|
-
visibility?: number | undefined;
|
|
481
|
-
presence?: number | undefined;
|
|
482
|
-
x: number;
|
|
483
|
-
y: number;
|
|
484
|
-
z: number;
|
|
485
|
-
}[];
|
|
486
521
|
worldLandmarks: {
|
|
487
|
-
visibility?: number | undefined;
|
|
488
522
|
presence?: number | undefined;
|
|
523
|
+
visibility?: number | undefined;
|
|
489
524
|
x: number;
|
|
490
525
|
y: number;
|
|
491
526
|
z: number;
|
|
492
527
|
}[];
|
|
493
|
-
}[]
|
|
528
|
+
}[];
|
|
494
529
|
}>;
|
|
495
530
|
declare module "@workglow/task-graph" {
|
|
496
531
|
interface Workflow {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"PoseLandmarkerTask.d.ts","sourceRoot":"","sources":["../../src/task/PoseLandmarkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"PoseLandmarkerTask.d.ts","sourceRoot":"","sources":["../../src/task/PoseLandmarkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AA0FnD,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAmDH,CAAC;AAEpC,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAcJ,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,yBAAyB,CAAC,CAAC;AACnF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAErF;;;GAGG;AACH,qBAAa,kBAAmB,SAAQ,YAAY,CAClD,uBAAuB,EACvB,wBAAwB,EACxB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAwB;IAC1C,OAAc,QAAQ,SAAqB;IAC3C,OAAc,KAAK,SAAqB;IACxC,OAAc,WAAW,SACyF;IAClH,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,cAAc;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAE1B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,cAAc,EAAE,cAAc,CAC5B,uBAAuB,EACvB,wBAAwB,EACxB,kBAAkB,CACnB,CAAC;KACH;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, IExecuteContext, JobQueueTaskConfig, Task } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
export declare const QueryExpansionMethod: {
|
|
9
9
|
readonly MULTI_QUERY: "multi-query";
|
|
10
10
|
readonly HYDE: "hyde";
|
|
@@ -22,7 +22,7 @@ declare const inputSchema: {
|
|
|
22
22
|
};
|
|
23
23
|
readonly method: {
|
|
24
24
|
readonly type: "string";
|
|
25
|
-
readonly enum: ("
|
|
25
|
+
readonly enum: ("hyde" | "multi-query" | "paraphrase" | "synonyms")[];
|
|
26
26
|
readonly title: "Expansion Method";
|
|
27
27
|
readonly description: "Method to use for query expansion";
|
|
28
28
|
readonly default: "multi-query";
|
|
@@ -114,11 +114,16 @@ export declare class QueryExpanderTask extends Task<QueryExpanderTaskInput, Quer
|
|
|
114
114
|
*/
|
|
115
115
|
private preserveCapitalization;
|
|
116
116
|
}
|
|
117
|
-
export declare const queryExpander: (input:
|
|
117
|
+
export declare const queryExpander: (input: {
|
|
118
|
+
method?: "hyde" | "multi-query" | "paraphrase" | "synonyms" | undefined;
|
|
119
|
+
model?: string | undefined;
|
|
120
|
+
numVariations?: number | undefined;
|
|
121
|
+
query: string;
|
|
122
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
118
123
|
count: number;
|
|
119
|
-
query: string[];
|
|
120
124
|
method: string;
|
|
121
125
|
originalQuery: string;
|
|
126
|
+
query: string[];
|
|
122
127
|
}>;
|
|
123
128
|
declare module "@workglow/task-graph" {
|
|
124
129
|
interface Workflow {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"QueryExpanderTask.d.ts","sourceRoot":"","sources":["../../src/task/QueryExpanderTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"QueryExpanderTask.d.ts","sourceRoot":"","sources":["../../src/task/QueryExpanderTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,eAAO,MAAM,oBAAoB;;;;;CAKvB,CAAC;AAEX,MAAM,MAAM,oBAAoB,GAAG,CAAC,OAAO,oBAAoB,CAAC,CAAC,MAAM,OAAO,oBAAoB,CAAC,CAAC;AAEpG,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA+BkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2BiB,CAAC;AAEpC,MAAM,MAAM,sBAAsB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACpE,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAEtE;;;;;;GAMG;AACH,qBAAa,iBAAkB,SAAQ,IAAI,CACzC,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAuB;IACzC,OAAc,QAAQ,SAAS;IAC/B,OAAc,KAAK,SAAoB;IACvC,OAAc,WAAW,SAAkD;IAC3E,OAAc,SAAS,UAAQ;IAE/B,OAAc,WAAW,IAAI,cAAc,CAE1C;IAED,OAAc,YAAY,IAAI,cAAc,CAE3C;IAEK,OAAO,CACX,KAAK,EAAE,sBAAsB,EAC7B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,uBAAuB,CAAC,CAgClC;IAED;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAyC3B;;OAEG;IACH,OAAO,CAAC,aAAa;IAkBrB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IA6CxB;;;OAGG;IACH,OAAO,CAAC,mBAAmB;IAoB3B;;OAEG;IACH,OAAO,CAAC,sBAAsB;CAM/B;AAED,eAAO,MAAM,aAAa;;;;;;;;;;EAEzB,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,aAAa,EAAE,cAAc,CAC3B,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,CACnB,CAAC;KACH;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, IExecuteContext, JobQueueTaskConfig, Task } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
declare const inputSchema: {
|
|
9
9
|
readonly type: "object";
|
|
10
10
|
readonly properties: {
|
|
@@ -199,14 +199,38 @@ export declare class RerankerTask extends Task<RerankerTaskInput, RerankerTaskOu
|
|
|
199
199
|
*/
|
|
200
200
|
private reciprocalRankFusion;
|
|
201
201
|
}
|
|
202
|
-
export declare const reranker: (input:
|
|
202
|
+
export declare const reranker: (input: {
|
|
203
|
+
chunks: string[];
|
|
203
204
|
metadata?: {
|
|
204
205
|
[x: string]: unknown;
|
|
205
206
|
}[] | undefined;
|
|
207
|
+
method?: "cross-encoder" | "reciprocal-rank-fusion" | "simple" | undefined;
|
|
208
|
+
model?: string | {
|
|
209
|
+
[x: string]: unknown;
|
|
210
|
+
description?: string | undefined;
|
|
211
|
+
metadata?: {
|
|
212
|
+
[x: string]: unknown;
|
|
213
|
+
} | undefined;
|
|
214
|
+
model_id?: string | undefined;
|
|
215
|
+
provider: string;
|
|
216
|
+
provider_config: {
|
|
217
|
+
[x: string]: unknown;
|
|
218
|
+
credential_key?: string | undefined;
|
|
219
|
+
};
|
|
220
|
+
tasks?: string[] | undefined;
|
|
221
|
+
title?: string | undefined;
|
|
222
|
+
} | undefined;
|
|
223
|
+
query: string;
|
|
224
|
+
scores?: number[] | undefined;
|
|
225
|
+
topK?: number | undefined;
|
|
226
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
206
227
|
chunks: string[];
|
|
207
228
|
count: number;
|
|
208
|
-
|
|
229
|
+
metadata?: {
|
|
230
|
+
[x: string]: unknown;
|
|
231
|
+
}[] | undefined;
|
|
209
232
|
originalIndices: number[];
|
|
233
|
+
scores: number[];
|
|
210
234
|
}>;
|
|
211
235
|
declare module "@workglow/task-graph" {
|
|
212
236
|
interface Workflow {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"RerankerTask.d.ts","sourceRoot":"","sources":["../../src/task/RerankerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"RerankerTask.d.ts","sourceRoot":"","sources":["../../src/task/RerankerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAMnE,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAkDkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAuCiB,CAAC;AAEpC,MAAM,MAAM,iBAAiB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAC/D,MAAM,MAAM,kBAAkB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AASjE;;;;;;GAMG;AACH,qBAAa,YAAa,SAAQ,IAAI,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,CAAC;IAC/F,OAAc,IAAI,SAAkB;IACpC,OAAc,QAAQ,SAAS;IAC/B,OAAc,KAAK,SAAc;IACjC,OAAc,WAAW,SAAkD;IAC3E,OAAc,SAAS,UAAQ;IAC/B,OAAO,CAAC,yBAAyB,CAAC,CAAgB;IAElD,OAAc,WAAW,IAAI,cAAc,CAE1C;IAED,OAAc,YAAY,IAAI,cAAc,CAE3C;IAEK,OAAO,CAAC,KAAK,EAAE,iBAAiB,EAAE,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC,CA0C7F;YAEa,kBAAkB;IAyChC,OAAO,CAAC,wBAAwB;IAqBhC;;OAEG;IACH,OAAO,CAAC,YAAY;IAuDpB;;;OAGG;IACH,OAAO,CAAC,oBAAoB;CAqB7B;AAED,eAAO,MAAM,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAEpB,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,QAAQ,EAAE,cAAc,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;KACrF;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, IExecuteContext, JobQueueTaskConfig, Task } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
declare const inputSchema: {
|
|
9
9
|
readonly type: "object";
|
|
10
10
|
readonly properties: {
|
|
@@ -77,7 +77,13 @@ export declare class StructuralParserTask extends Task<StructuralParserTaskInput
|
|
|
77
77
|
execute(input: StructuralParserTaskInput, context: IExecuteContext): Promise<StructuralParserTaskOutput>;
|
|
78
78
|
private countNodes;
|
|
79
79
|
}
|
|
80
|
-
export declare const structuralParser: (input:
|
|
80
|
+
export declare const structuralParser: (input: {
|
|
81
|
+
doc_id?: string | undefined;
|
|
82
|
+
format?: "auto" | "markdown" | "text" | undefined;
|
|
83
|
+
sourceUri?: string | undefined;
|
|
84
|
+
text: string;
|
|
85
|
+
title: string;
|
|
86
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
81
87
|
doc_id: string;
|
|
82
88
|
documentTree: unknown;
|
|
83
89
|
nodeCount: number;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"StructuralParserTask.d.ts","sourceRoot":"","sources":["../../src/task/StructuralParserTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,
|
|
1
|
+
{"version":3,"file":"StructuralParserTask.d.ts","sourceRoot":"","sources":["../../src/task/StructuralParserTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAGnE,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAiCkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;CAoBiB,CAAC;AAEpC,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACvE,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAEzE;;;GAGG;AACH,qBAAa,oBAAqB,SAAQ,IAAI,CAC5C,yBAAyB,EACzB,0BAA0B,EAC1B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA0B;IAC5C,OAAc,QAAQ,SAAc;IACpC,OAAc,KAAK,SAAuB;IAC1C,OAAc,WAAW,SAAsD;IAC/E,OAAc,SAAS,UAAQ;IAE/B,OAAc,WAAW,IAAI,cAAc,CAE1C;IAED,OAAc,YAAY,IAAI,cAAc,CAE3C;IAEK,OAAO,CACX,KAAK,EAAE,yBAAyB,EAChC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,0BAA0B,CAAC,CAyBrC;IAED,OAAO,CAAC,UAAU;CASnB;AAED,eAAO,MAAM,gBAAgB;;;;;;;;;;EAE5B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,gBAAgB,EAAE,cAAc,CAC9B,yBAAyB,EACzB,0BAA0B,EAC1B,kBAAkB,CACnB,CAAC;KACH;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, JobQueueTaskConfig } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
import { StreamingAiTask } from "./base/StreamingAiTask";
|
|
9
9
|
export declare const StructuredGenerationInputSchema: {
|
|
10
10
|
readonly type: "object";
|
|
@@ -125,7 +125,29 @@ export declare class StructuredGenerationTask extends StreamingAiTask<Structured
|
|
|
125
125
|
/**
|
|
126
126
|
* Task for generating structured JSON output using a language model
|
|
127
127
|
*/
|
|
128
|
-
export declare const structuredGeneration: (input:
|
|
128
|
+
export declare const structuredGeneration: (input: {
|
|
129
|
+
maxTokens?: number | undefined;
|
|
130
|
+
model: string | {
|
|
131
|
+
[x: string]: unknown;
|
|
132
|
+
description?: string | undefined;
|
|
133
|
+
metadata?: {
|
|
134
|
+
[x: string]: unknown;
|
|
135
|
+
} | undefined;
|
|
136
|
+
model_id?: string | undefined;
|
|
137
|
+
provider: string;
|
|
138
|
+
provider_config: {
|
|
139
|
+
[x: string]: unknown;
|
|
140
|
+
credential_key?: string | undefined;
|
|
141
|
+
};
|
|
142
|
+
tasks?: string[] | undefined;
|
|
143
|
+
title?: string | undefined;
|
|
144
|
+
};
|
|
145
|
+
outputSchema: {
|
|
146
|
+
[x: string]: unknown;
|
|
147
|
+
};
|
|
148
|
+
prompt: string;
|
|
149
|
+
temperature?: number | undefined;
|
|
150
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
129
151
|
object: {
|
|
130
152
|
[x: string]: unknown;
|
|
131
153
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"StructuredGenerationTask.d.ts","sourceRoot":"","sources":["../../src/task/StructuredGenerationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"StructuredGenerationTask.d.ts","sourceRoot":"","sources":["../../src/task/StructuredGenerationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAIzD,eAAO,MAAM,+BAA+B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAkCT,CAAC;AAEpC,eAAO,MAAM,gCAAgC;;;;;;;;;;;;;;CAcV,CAAC;AAEpC,MAAM,MAAM,6BAA6B,GAAG,UAAU,CAAC,OAAO,+BAA+B,CAAC,CAAC;AAC/F,MAAM,MAAM,8BAA8B,GAAG,UAAU,CAAC,OAAO,gCAAgC,CAAC,CAAC;AAEjG,qBAAa,wBAAyB,SAAQ,eAAe,CAC3D,6BAA6B,EAC7B,8BAA8B,EAC9B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA8B;IAChD,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAA2B;IAC9C,OAAc,WAAW,SACkE;IAC3F,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;CACF;AAED;;GAEG;AACH,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;EAKhC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,oBAAoB,EAAE,cAAc,CAClC,6BAA6B,EAC7B,8BAA8B,EAC9B,kBAAkB,CACnB,CAAC;KACH;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, IExecuteContext, JobQueueTaskConfig, Task } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
export declare const ChunkingStrategy: {
|
|
9
9
|
readonly FIXED: "fixed";
|
|
10
10
|
readonly SENTENCE: "sentence";
|
|
@@ -36,7 +36,7 @@ declare const inputSchema: {
|
|
|
36
36
|
};
|
|
37
37
|
readonly strategy: {
|
|
38
38
|
readonly type: "string";
|
|
39
|
-
readonly enum: ("fixed" | "
|
|
39
|
+
readonly enum: ("fixed" | "paragraph" | "semantic" | "sentence")[];
|
|
40
40
|
readonly title: "Chunking Strategy";
|
|
41
41
|
readonly description: "Strategy to use for chunking text";
|
|
42
42
|
readonly default: "fixed";
|
|
@@ -111,14 +111,19 @@ export declare class TextChunkerTask extends Task<TextChunkerTaskInput, TextChun
|
|
|
111
111
|
*/
|
|
112
112
|
private chunkByParagraph;
|
|
113
113
|
}
|
|
114
|
-
export declare const textChunker: (input:
|
|
114
|
+
export declare const textChunker: (input: {
|
|
115
|
+
chunkOverlap?: number | undefined;
|
|
116
|
+
chunkSize?: number | undefined;
|
|
117
|
+
strategy?: "fixed" | "paragraph" | "semantic" | "sentence" | undefined;
|
|
118
|
+
text: string;
|
|
119
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
120
|
+
chunks: string[];
|
|
115
121
|
metadata: {
|
|
116
|
-
|
|
122
|
+
endChar?: number | undefined;
|
|
117
123
|
index?: number | undefined;
|
|
124
|
+
length?: number | undefined;
|
|
118
125
|
startChar?: number | undefined;
|
|
119
|
-
endChar?: number | undefined;
|
|
120
126
|
}[];
|
|
121
|
-
chunks: string[];
|
|
122
127
|
}>;
|
|
123
128
|
declare module "@workglow/task-graph" {
|
|
124
129
|
interface Workflow {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"TextChunkerTask.d.ts","sourceRoot":"","sources":["../../src/task/TextChunkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"TextChunkerTask.d.ts","sourceRoot":"","sources":["../../src/task/TextChunkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,eAAO,MAAM,gBAAgB;;;;;CAKnB,CAAC;AAEX,MAAM,MAAM,gBAAgB,GAAG,CAAC,OAAO,gBAAgB,CAAC,CAAC,MAAM,OAAO,gBAAgB,CAAC,CAAC;AAExF,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgCkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2BiB,CAAC;AAEpC,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAClE,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AASpE;;;GAGG;AACH,qBAAa,eAAgB,SAAQ,IAAI,CACvC,oBAAoB,EACpB,qBAAqB,EACrB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAqB;IACvC,OAAc,QAAQ,SAAc;IACpC,OAAc,KAAK,SAAkB;IACrC,OAAc,WAAW,SACyD;IAClF,OAAc,SAAS,UAAQ;IAE/B,OAAc,WAAW,IAAI,cAAc,CAE1C;IAED,OAAc,YAAY,IAAI,cAAc,CAE3C;IAEK,OAAO,CACX,KAAK,EAAE,oBAAoB,EAC3B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,qBAAqB,CAAC,CAyBhC;IAED;;OAEG;IACH,OAAO,CAAC,UAAU;IA8BlB;;OAEG;IACH,OAAO,CAAC,eAAe;IAoFvB;;OAEG;IACH,OAAO,CAAC,gBAAgB;CAoEzB;AAED,eAAO,MAAM,WAAW;;;;;;;;;;;;;EAEvB,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,kBAAkB,CAAC,CAAC;KAC9F;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, JobQueueTaskConfig } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
import { AiTask } from "./base/AiTask";
|
|
9
9
|
export declare const TextClassificationInputSchema: {
|
|
10
10
|
readonly type: "object";
|
|
@@ -177,14 +177,33 @@ export declare class TextClassificationTask extends AiTask<TextClassificationTas
|
|
|
177
177
|
* @param input The input parameters for text classification (text and model)
|
|
178
178
|
* @returns Promise resolving to the classification categories with scores
|
|
179
179
|
*/
|
|
180
|
-
export declare const textClassification: (input:
|
|
180
|
+
export declare const textClassification: (input: {
|
|
181
|
+
candidateLabels?: string[] | undefined;
|
|
182
|
+
maxCategories?: number | undefined;
|
|
183
|
+
model: string | {
|
|
184
|
+
[x: string]: unknown;
|
|
185
|
+
description?: string | undefined;
|
|
186
|
+
metadata?: {
|
|
187
|
+
[x: string]: unknown;
|
|
188
|
+
} | undefined;
|
|
189
|
+
model_id?: string | undefined;
|
|
190
|
+
provider: string;
|
|
191
|
+
provider_config: {
|
|
192
|
+
[x: string]: unknown;
|
|
193
|
+
credential_key?: string | undefined;
|
|
194
|
+
};
|
|
195
|
+
tasks?: string[] | undefined;
|
|
196
|
+
title?: string | undefined;
|
|
197
|
+
};
|
|
198
|
+
text: string | string[];
|
|
199
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
181
200
|
categories: {
|
|
182
|
-
score: number;
|
|
183
201
|
label: string;
|
|
184
|
-
}[] | {
|
|
185
202
|
score: number;
|
|
203
|
+
}[][] | {
|
|
186
204
|
label: string;
|
|
187
|
-
|
|
205
|
+
score: number;
|
|
206
|
+
}[];
|
|
188
207
|
}>;
|
|
189
208
|
declare module "@workglow/task-graph" {
|
|
190
209
|
interface Workflow {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"TextClassificationTask.d.ts","sourceRoot":"","sources":["../../src/task/TextClassificationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"TextClassificationTask.d.ts","sourceRoot":"","sources":["../../src/task/TextClassificationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,eAAO,MAAM,6BAA6B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BP,CAAC;AAEpC,eAAO,MAAM,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4BR,CAAC;AAEpC,MAAM,MAAM,2BAA2B,GAAG,UAAU,CAAC,OAAO,6BAA6B,CAAC,CAAC;AAC3F,MAAM,MAAM,4BAA4B,GAAG,UAAU,CAAC,OAAO,8BAA8B,CAAC,CAAC;AAE7F;;;GAGG;AACH,qBAAa,sBAAuB,SAAQ,MAAM,CAChD,2BAA2B,EAC3B,4BAA4B,CAC7B;IACC,OAAc,IAAI,SAA4B;IAC9C,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAqB;IACxC,OAAc,WAAW,SACwG;IACjI,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;;;;;;;;;;;;;;;EAK9B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,kBAAkB,EAAE,cAAc,CAChC,2BAA2B,EAC3B,4BAA4B,EAC5B,kBAAkB,CACnB,CAAC;KACH;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, JobQueueTaskConfig } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema, TypedArraySchemaOptions } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema, TypedArraySchemaOptions } from "@workglow/util/schema";
|
|
8
8
|
import { AiTask } from "./base/AiTask";
|
|
9
9
|
export declare const TextEmbeddingInputSchema: {
|
|
10
10
|
readonly type: "object";
|
|
@@ -129,8 +129,25 @@ export declare class TextEmbeddingTask extends AiTask<TextEmbeddingTaskInput, Te
|
|
|
129
129
|
* @param input - Input containing text(s) and model(s) for embedding
|
|
130
130
|
* @returns Promise resolving to the generated embeddings
|
|
131
131
|
*/
|
|
132
|
-
export declare const textEmbedding: (input:
|
|
133
|
-
|
|
132
|
+
export declare const textEmbedding: (input: {
|
|
133
|
+
model: string | {
|
|
134
|
+
[x: string]: unknown;
|
|
135
|
+
description?: string | undefined;
|
|
136
|
+
metadata?: {
|
|
137
|
+
[x: string]: unknown;
|
|
138
|
+
} | undefined;
|
|
139
|
+
model_id?: string | undefined;
|
|
140
|
+
provider: string;
|
|
141
|
+
provider_config: {
|
|
142
|
+
[x: string]: unknown;
|
|
143
|
+
credential_key?: string | undefined;
|
|
144
|
+
};
|
|
145
|
+
tasks?: string[] | undefined;
|
|
146
|
+
title?: string | undefined;
|
|
147
|
+
};
|
|
148
|
+
text: string | string[];
|
|
149
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
150
|
+
vector: import("@workglow/util/schema").TypedArray[] | import("@workglow/util/schema").TypedArray;
|
|
134
151
|
}>;
|
|
135
152
|
declare module "@workglow/task-graph" {
|
|
136
153
|
interface Workflow {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"TextEmbeddingTask.d.ts","sourceRoot":"","sources":["../../src/task/TextEmbeddingTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EACL,cAAc,EACd,UAAU,EAEV,uBAAuB,EACxB,MAAM,
|
|
1
|
+
{"version":3,"file":"TextEmbeddingTask.d.ts","sourceRoot":"","sources":["../../src/task/TextEmbeddingTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EACL,cAAc,EACd,UAAU,EAEV,uBAAuB,EACxB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAYF,CAAC;AAEpC,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;CAYH,CAAC;AAEpC,MAAM,MAAM,sBAAsB,GAAG,UAAU,CAC7C,OAAO,wBAAwB,EAC/B,uBAAuB,CACxB,CAAC;AACF,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAC9C,OAAO,yBAAyB,EAChC,uBAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,qBAAa,iBAAkB,SAAQ,MAAM,CAAC,sBAAsB,EAAE,uBAAuB,CAAC;IAC5F,OAAc,IAAI,SAAuB;IACzC,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAoB;IACvC,OAAc,WAAW,SAAsE;IAC/F,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;CACF;AAED;;;;GAIG;AACH,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;EAEzB,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,aAAa,EAAE,cAAc,CAC3B,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,CACnB,CAAC;KACH;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, JobQueueTaskConfig } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
import { AiTask } from "./base/AiTask";
|
|
9
9
|
export declare const TextFillMaskInputSchema: {
|
|
10
10
|
readonly type: "object";
|
|
@@ -168,16 +168,33 @@ export declare class TextFillMaskTask extends AiTask<TextFillMaskTaskInput, Text
|
|
|
168
168
|
* @param input The input parameters for fill mask (text with mask token and model)
|
|
169
169
|
* @returns Promise resolving to the predicted tokens with scores and complete sequences
|
|
170
170
|
*/
|
|
171
|
-
export declare const textFillMask: (input:
|
|
171
|
+
export declare const textFillMask: (input: {
|
|
172
|
+
model: string | {
|
|
173
|
+
[x: string]: unknown;
|
|
174
|
+
description?: string | undefined;
|
|
175
|
+
metadata?: {
|
|
176
|
+
[x: string]: unknown;
|
|
177
|
+
} | undefined;
|
|
178
|
+
model_id?: string | undefined;
|
|
179
|
+
provider: string;
|
|
180
|
+
provider_config: {
|
|
181
|
+
[x: string]: unknown;
|
|
182
|
+
credential_key?: string | undefined;
|
|
183
|
+
};
|
|
184
|
+
tasks?: string[] | undefined;
|
|
185
|
+
title?: string | undefined;
|
|
186
|
+
};
|
|
187
|
+
text: string | string[];
|
|
188
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
172
189
|
predictions: {
|
|
173
|
-
score: number;
|
|
174
190
|
entity: string;
|
|
175
|
-
sequence: string;
|
|
176
|
-
}[] | {
|
|
177
191
|
score: number;
|
|
192
|
+
sequence: string;
|
|
193
|
+
}[][] | {
|
|
178
194
|
entity: string;
|
|
195
|
+
score: number;
|
|
179
196
|
sequence: string;
|
|
180
|
-
}[]
|
|
197
|
+
}[];
|
|
181
198
|
}>;
|
|
182
199
|
declare module "@workglow/task-graph" {
|
|
183
200
|
interface Workflow {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"TextFillMaskTask.d.ts","sourceRoot":"","sources":["../../src/task/TextFillMaskTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"TextFillMaskTask.d.ts","sourceRoot":"","sources":["../../src/task/TextFillMaskTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAYD,CAAC;AAEpC,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAiCF,CAAC;AAEpC,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAC/E,MAAM,MAAM,sBAAsB,GAAG,UAAU,CAAC,OAAO,wBAAwB,CAAC,CAAC;AAEjF;;GAEG;AACH,qBAAa,gBAAiB,SAAQ,MAAM,CAAC,qBAAqB,EAAE,sBAAsB,CAAC;IACzF,OAAc,IAAI,SAAsB;IACxC,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAe;IAClC,OAAc,WAAW,SAAiC;IAC1D,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;EAExB,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,YAAY,EAAE,cAAc,CAAC,qBAAqB,EAAE,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;KACjG;CACF"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { CreateWorkflow, JobQueueTaskConfig } from "@workglow/task-graph";
|
|
7
|
-
import { DataPortSchema, FromSchema } from "@workglow/util";
|
|
7
|
+
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
8
|
import { StreamingAiTask } from "./base/StreamingAiTask";
|
|
9
9
|
export declare const TextGenerationInputSchema: {
|
|
10
10
|
readonly type: "object";
|
|
@@ -160,7 +160,29 @@ export declare class TextGenerationTask extends StreamingAiTask<TextGenerationTa
|
|
|
160
160
|
/**
|
|
161
161
|
* Task for generating text using a language model
|
|
162
162
|
*/
|
|
163
|
-
export declare const textGeneration: (input:
|
|
163
|
+
export declare const textGeneration: (input: {
|
|
164
|
+
frequencyPenalty?: number | undefined;
|
|
165
|
+
maxTokens?: number | undefined;
|
|
166
|
+
model: string | {
|
|
167
|
+
[x: string]: unknown;
|
|
168
|
+
description?: string | undefined;
|
|
169
|
+
metadata?: {
|
|
170
|
+
[x: string]: unknown;
|
|
171
|
+
} | undefined;
|
|
172
|
+
model_id?: string | undefined;
|
|
173
|
+
provider: string;
|
|
174
|
+
provider_config: {
|
|
175
|
+
[x: string]: unknown;
|
|
176
|
+
credential_key?: string | undefined;
|
|
177
|
+
};
|
|
178
|
+
tasks?: string[] | undefined;
|
|
179
|
+
title?: string | undefined;
|
|
180
|
+
};
|
|
181
|
+
presencePenalty?: number | undefined;
|
|
182
|
+
prompt: string | string[];
|
|
183
|
+
temperature?: number | undefined;
|
|
184
|
+
topP?: number | undefined;
|
|
185
|
+
}, config?: JobQueueTaskConfig | undefined) => Promise<{
|
|
164
186
|
text: string | string[];
|
|
165
187
|
}>;
|
|
166
188
|
declare module "@workglow/task-graph" {
|