@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.
Files changed (131) hide show
  1. package/README.md +20 -27
  2. package/dist/browser.js +245 -135
  3. package/dist/browser.js.map +60 -57
  4. package/dist/bun.js +245 -135
  5. package/dist/bun.js.map +60 -57
  6. package/dist/common.d.ts +2 -0
  7. package/dist/common.d.ts.map +1 -1
  8. package/dist/job/AiJob.d.ts +1 -1
  9. package/dist/job/AiJob.d.ts.map +1 -1
  10. package/dist/model/InMemoryModelRepository.d.ts.map +1 -1
  11. package/dist/model/ModelRepository.d.ts +10 -10
  12. package/dist/model/ModelRepository.d.ts.map +1 -1
  13. package/dist/model/ModelSchema.d.ts +1 -1
  14. package/dist/model/ModelSchema.d.ts.map +1 -1
  15. package/dist/node.js +245 -135
  16. package/dist/node.js.map +60 -57
  17. package/dist/provider/AiProvider.d.ts +33 -30
  18. package/dist/provider/AiProvider.d.ts.map +1 -1
  19. package/dist/provider/AiProviderRegistry.d.ts +4 -4
  20. package/dist/provider/AiProviderRegistry.d.ts.map +1 -1
  21. package/dist/provider/QueuedAiProvider.d.ts +18 -0
  22. package/dist/provider/QueuedAiProvider.d.ts.map +1 -0
  23. package/dist/task/AgentTask.d.ts +2 -2
  24. package/dist/task/AgentTask.d.ts.map +1 -1
  25. package/dist/task/AgentTypes.d.ts +1 -1
  26. package/dist/task/AgentTypes.d.ts.map +1 -1
  27. package/dist/task/AgentUtils.d.ts +1 -1
  28. package/dist/task/AgentUtils.d.ts.map +1 -1
  29. package/dist/task/BackgroundRemovalTask.d.ts +32 -2
  30. package/dist/task/BackgroundRemovalTask.d.ts.map +1 -1
  31. package/dist/task/ChunkRetrievalTask.d.ts +7 -7
  32. package/dist/task/ChunkRetrievalTask.d.ts.map +1 -1
  33. package/dist/task/ChunkToVectorTask.d.ts +26 -4
  34. package/dist/task/ChunkToVectorTask.d.ts.map +1 -1
  35. package/dist/task/ChunkVectorHybridSearchTask.d.ts +23 -2
  36. package/dist/task/ChunkVectorHybridSearchTask.d.ts.map +1 -1
  37. package/dist/task/ChunkVectorSearchTask.d.ts +13 -5
  38. package/dist/task/ChunkVectorSearchTask.d.ts.map +1 -1
  39. package/dist/task/ChunkVectorUpsertTask.d.ts +12 -3
  40. package/dist/task/ChunkVectorUpsertTask.d.ts.map +1 -1
  41. package/dist/task/ContextBuilderTask.d.ts +30 -4
  42. package/dist/task/ContextBuilderTask.d.ts.map +1 -1
  43. package/dist/task/CountTokensTask.d.ts +19 -2
  44. package/dist/task/CountTokensTask.d.ts.map +1 -1
  45. package/dist/task/DocumentEnricherTask.d.ts +39 -18
  46. package/dist/task/DocumentEnricherTask.d.ts.map +1 -1
  47. package/dist/task/DownloadModelTask.d.ts +25 -4
  48. package/dist/task/DownloadModelTask.d.ts.map +1 -1
  49. package/dist/task/FaceDetectorTask.d.ts +39 -7
  50. package/dist/task/FaceDetectorTask.d.ts.map +1 -1
  51. package/dist/task/FaceLandmarkerTask.d.ts +44 -8
  52. package/dist/task/FaceLandmarkerTask.d.ts.map +1 -1
  53. package/dist/task/GestureRecognizerTask.d.ts +49 -15
  54. package/dist/task/GestureRecognizerTask.d.ts.map +1 -1
  55. package/dist/task/HandLandmarkerTask.d.ts +46 -12
  56. package/dist/task/HandLandmarkerTask.d.ts.map +1 -1
  57. package/dist/task/HierarchicalChunkerTask.d.ts +33 -22
  58. package/dist/task/HierarchicalChunkerTask.d.ts.map +1 -1
  59. package/dist/task/HierarchyJoinTask.d.ts +36 -11
  60. package/dist/task/HierarchyJoinTask.d.ts.map +1 -1
  61. package/dist/task/ImageClassificationTask.d.ts +37 -5
  62. package/dist/task/ImageClassificationTask.d.ts.map +1 -1
  63. package/dist/task/ImageEmbeddingTask.d.ts +33 -3
  64. package/dist/task/ImageEmbeddingTask.d.ts.map +1 -1
  65. package/dist/task/ImageSegmentationTask.d.ts +37 -5
  66. package/dist/task/ImageSegmentationTask.d.ts.map +1 -1
  67. package/dist/task/ImageToTextTask.d.ts +33 -2
  68. package/dist/task/ImageToTextTask.d.ts.map +1 -1
  69. package/dist/task/ModelInfoTask.d.ts +36 -10
  70. package/dist/task/ModelInfoTask.d.ts.map +1 -1
  71. package/dist/task/ModelSearchTask.d.ts +66 -0
  72. package/dist/task/ModelSearchTask.d.ts.map +1 -0
  73. package/dist/task/ObjectDetectionTask.d.ts +39 -7
  74. package/dist/task/ObjectDetectionTask.d.ts.map +1 -1
  75. package/dist/task/PoseLandmarkerTask.d.ts +53 -18
  76. package/dist/task/PoseLandmarkerTask.d.ts.map +1 -1
  77. package/dist/task/QueryExpanderTask.d.ts +9 -4
  78. package/dist/task/QueryExpanderTask.d.ts.map +1 -1
  79. package/dist/task/RerankerTask.d.ts +27 -3
  80. package/dist/task/RerankerTask.d.ts.map +1 -1
  81. package/dist/task/StructuralParserTask.d.ts +8 -2
  82. package/dist/task/StructuralParserTask.d.ts.map +1 -1
  83. package/dist/task/StructuredGenerationTask.d.ts +24 -2
  84. package/dist/task/StructuredGenerationTask.d.ts.map +1 -1
  85. package/dist/task/TextChunkerTask.d.ts +11 -6
  86. package/dist/task/TextChunkerTask.d.ts.map +1 -1
  87. package/dist/task/TextClassificationTask.d.ts +24 -5
  88. package/dist/task/TextClassificationTask.d.ts.map +1 -1
  89. package/dist/task/TextEmbeddingTask.d.ts +20 -3
  90. package/dist/task/TextEmbeddingTask.d.ts.map +1 -1
  91. package/dist/task/TextFillMaskTask.d.ts +23 -6
  92. package/dist/task/TextFillMaskTask.d.ts.map +1 -1
  93. package/dist/task/TextGenerationTask.d.ts +24 -2
  94. package/dist/task/TextGenerationTask.d.ts.map +1 -1
  95. package/dist/task/TextLanguageDetectionTask.d.ts +23 -5
  96. package/dist/task/TextLanguageDetectionTask.d.ts.map +1 -1
  97. package/dist/task/TextNamedEntityRecognitionTask.d.ts +24 -6
  98. package/dist/task/TextNamedEntityRecognitionTask.d.ts.map +1 -1
  99. package/dist/task/TextQuestionAnswerTask.d.ts +20 -2
  100. package/dist/task/TextQuestionAnswerTask.d.ts.map +1 -1
  101. package/dist/task/TextRewriterTask.d.ts +20 -2
  102. package/dist/task/TextRewriterTask.d.ts.map +1 -1
  103. package/dist/task/TextSummaryTask.d.ts +19 -2
  104. package/dist/task/TextSummaryTask.d.ts.map +1 -1
  105. package/dist/task/TextTranslationTask.d.ts +21 -2
  106. package/dist/task/TextTranslationTask.d.ts.map +1 -1
  107. package/dist/task/ToolCallingTask.d.ts +10 -66
  108. package/dist/task/ToolCallingTask.d.ts.map +1 -1
  109. package/dist/task/ToolCallingUtils.d.ts +65 -0
  110. package/dist/task/ToolCallingUtils.d.ts.map +1 -0
  111. package/dist/task/TopicSegmenterTask.d.ts +11 -5
  112. package/dist/task/TopicSegmenterTask.d.ts.map +1 -1
  113. package/dist/task/UnloadModelTask.d.ts +19 -3
  114. package/dist/task/UnloadModelTask.d.ts.map +1 -1
  115. package/dist/task/VectorQuantizeTask.d.ts +12 -8
  116. package/dist/task/VectorQuantizeTask.d.ts.map +1 -1
  117. package/dist/task/VectorSimilarityTask.d.ts +10 -5
  118. package/dist/task/VectorSimilarityTask.d.ts.map +1 -1
  119. package/dist/task/base/AiTask.d.ts +1 -1
  120. package/dist/task/base/AiTask.d.ts.map +1 -1
  121. package/dist/task/base/AiTaskSchemas.d.ts +1 -1
  122. package/dist/task/base/AiTaskSchemas.d.ts.map +1 -1
  123. package/dist/task/base/AiVisionTask.d.ts.map +1 -1
  124. package/dist/task/base/StreamingAiTask.d.ts.map +1 -1
  125. package/dist/task/index.d.ts +4 -1
  126. package/dist/task/index.d.ts.map +1 -1
  127. package/dist/worker.d.ts +21 -0
  128. package/dist/worker.d.ts.map +1 -0
  129. package/dist/worker.js +460 -0
  130. package/dist/worker.js.map +14 -0
  131. 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: ObjectDetectionTaskInput, config?: JobQueueTaskConfig) => Promise<{
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,gBAAgB,CAAC;AAE5D,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;WAC5F,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,eAAe,GAAI,OAAO,wBAAwB,EAAE,SAAS,kBAAkB;;;;;;;;;;;;;;;;;;;;EAE3F,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,eAAe,EAAE,cAAc,CAC7B,wBAAwB,EACxB,yBAAyB,EACzB,kBAAkB,CACnB,CAAC;KACH;CACF"}
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: PoseLandmarkerTaskInput, config?: JobQueueTaskConfig) => Promise<{
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
- landmarks: {
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
- worldLandmarks: {
465
- visibility?: number | undefined;
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,gBAAgB,CAAC;AAE5D,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;WACpG,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,cAAc,GAAI,OAAO,uBAAuB,EAAE,SAAS,kBAAkB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAEzF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,cAAc,EAAE,cAAc,CAC5B,uBAAuB,EACvB,wBAAwB,EACxB,kBAAkB,CACnB,CAAC;KACH;CACF"}
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: ("multi-query" | "hyde" | "synonyms" | "paraphrase")[];
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: QueryExpanderTaskInput, config?: JobQueueTaskConfig) => Promise<{
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,gBAAgB,CAAC;AAE5D,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;WAEjB,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CACX,KAAK,EAAE,sBAAsB,EAC7B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,uBAAuB,CAAC;IAkCnC;;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,GAAI,OAAO,sBAAsB,EAAE,SAAS,kBAAkB;;;;;EAEvF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,aAAa,EAAE,cAAc,CAC3B,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,CACnB,CAAC;KACH;CACF"}
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: RerankerTaskInput, config?: JobQueueTaskConfig) => Promise<{
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
- scores: number[];
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,gBAAgB,CAAC;AAM5D,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;WAEpC,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CAAC,KAAK,EAAE,iBAAiB,EAAE,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC;YA4ChF,kBAAkB;IAyChC,OAAO,CAAC,wBAAwB;IAqBhC;;OAEG;IACH,OAAO,CAAC,YAAY;IAuDpB;;;OAGG;IACH,OAAO,CAAC,oBAAoB;CAqB7B;AAED,eAAO,MAAM,QAAQ,GAAI,OAAO,iBAAiB,EAAE,SAAS,kBAAkB;;;;;;;;EAE7E,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,QAAQ,EAAE,cAAc,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;KACrF;CACF"}
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: StructuralParserTaskInput, config?: JobQueueTaskConfig) => Promise<{
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,EAAS,MAAM,gBAAgB,CAAC;AAEnE,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;WAEjB,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CACX,KAAK,EAAE,yBAAyB,EAChC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,0BAA0B,CAAC;IA2BtC,OAAO,CAAC,UAAU;CASnB;AAED,eAAO,MAAM,gBAAgB,GAAI,OAAO,yBAAyB,EAAE,SAAS,kBAAkB;;;;EAE7F,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,gBAAgB,EAAE,cAAc,CAC9B,yBAAyB,EACzB,0BAA0B,EAC1B,kBAAkB,CACnB,CAAC;KACH;CACF"}
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: StructuredGenerationTaskInput, config?: JobQueueTaskConfig) => Promise<{
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,gBAAgB,CAAC;AAE5D,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;WAC7E,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;GAEG;AACH,eAAO,MAAM,oBAAoB,GAC/B,OAAO,6BAA6B,EACpC,SAAS,kBAAkB;;;;EAG5B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,oBAAoB,EAAE,cAAc,CAClC,6BAA6B,EAC7B,8BAA8B,EAC9B,kBAAkB,CACnB,CAAC;KACH;CACF"}
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" | "sentence" | "paragraph" | "semantic")[];
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: TextChunkerTaskInput, config?: JobQueueTaskConfig) => Promise<{
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
- length?: number | undefined;
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,gBAAgB,CAAC;AAE5D,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;WAEjB,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CACX,KAAK,EAAE,oBAAoB,EAC3B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,qBAAqB,CAAC;IA2BjC;;OAEG;IACH,OAAO,CAAC,UAAU;IA8BlB;;OAEG;IACH,OAAO,CAAC,eAAe;IAoFvB;;OAEG;IACH,OAAO,CAAC,gBAAgB;CAoEzB;AAED,eAAO,MAAM,WAAW,GAAI,OAAO,oBAAoB,EAAE,SAAS,kBAAkB;;;;;;;;EAEnF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,kBAAkB,CAAC,CAAC;KAC9F;CACF"}
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: TextClassificationTaskInput, config?: JobQueueTaskConfig) => Promise<{
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,gBAAgB,CAAC;AAC5D,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;WACnH,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB,GAC7B,OAAO,2BAA2B,EAClC,SAAS,kBAAkB;;;;;;;;EAG5B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,kBAAkB,EAAE,cAAc,CAChC,2BAA2B,EAC3B,4BAA4B,EAC5B,kBAAkB,CACnB,CAAC;KACH;CACF"}
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: TextEmbeddingTaskInput, config?: JobQueueTaskConfig) => Promise<{
133
- vector: import("@workglow/util").TypedArray | import("@workglow/util").TypedArray[];
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,gBAAgB,CAAC;AACxB,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;WACjF,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;GAIG;AACH,eAAO,MAAM,aAAa,GAAU,OAAO,sBAAsB,EAAE,SAAS,kBAAkB;;EAE7F,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,aAAa,EAAE,cAAc,CAC3B,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,CACnB,CAAC;KACH;CACF"}
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: TextFillMaskTaskInput, config?: JobQueueTaskConfig) => Promise<{
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,gBAAgB,CAAC;AAC5D,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;WAC5C,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,YAAY,GAAI,OAAO,qBAAqB,EAAE,SAAS,kBAAkB;;;;;;;;;;EAErF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,YAAY,EAAE,cAAc,CAAC,qBAAqB,EAAE,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;KACjG;CACF"}
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: TextGenerationTaskInput, config?: JobQueueTaskConfig) => Promise<{
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" {