@workglow/ai 0.0.120 → 0.0.122

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 (128) hide show
  1. package/README.md +20 -27
  2. package/dist/browser.js +234 -124
  3. package/dist/browser.js.map +18 -15
  4. package/dist/bun.js +234 -124
  5. package/dist/bun.js.map +18 -15
  6. package/dist/common.d.ts +2 -0
  7. package/dist/common.d.ts.map +1 -1
  8. package/dist/job/AiJob.d.ts.map +1 -1
  9. package/dist/model/InMemoryModelRepository.d.ts.map +1 -1
  10. package/dist/model/ModelRepository.d.ts +10 -10
  11. package/dist/model/ModelRepository.d.ts.map +1 -1
  12. package/dist/model/ModelSchema.d.ts +1 -1
  13. package/dist/model/ModelSchema.d.ts.map +1 -1
  14. package/dist/node.js +234 -124
  15. package/dist/node.js.map +18 -15
  16. package/dist/provider/AiProvider.d.ts +33 -30
  17. package/dist/provider/AiProvider.d.ts.map +1 -1
  18. package/dist/provider/AiProviderRegistry.d.ts +4 -4
  19. package/dist/provider/AiProviderRegistry.d.ts.map +1 -1
  20. package/dist/provider/QueuedAiProvider.d.ts +18 -0
  21. package/dist/provider/QueuedAiProvider.d.ts.map +1 -0
  22. package/dist/task/AgentTask.d.ts +2 -2
  23. package/dist/task/AgentTask.d.ts.map +1 -1
  24. package/dist/task/AgentTypes.d.ts +1 -1
  25. package/dist/task/AgentTypes.d.ts.map +1 -1
  26. package/dist/task/AgentUtils.d.ts +1 -1
  27. package/dist/task/AgentUtils.d.ts.map +1 -1
  28. package/dist/task/BackgroundRemovalTask.d.ts +31 -1
  29. package/dist/task/BackgroundRemovalTask.d.ts.map +1 -1
  30. package/dist/task/ChunkRetrievalTask.d.ts +6 -6
  31. package/dist/task/ChunkRetrievalTask.d.ts.map +1 -1
  32. package/dist/task/ChunkToVectorTask.d.ts +25 -3
  33. package/dist/task/ChunkToVectorTask.d.ts.map +1 -1
  34. package/dist/task/ChunkVectorHybridSearchTask.d.ts +22 -1
  35. package/dist/task/ChunkVectorHybridSearchTask.d.ts.map +1 -1
  36. package/dist/task/ChunkVectorSearchTask.d.ts +12 -4
  37. package/dist/task/ChunkVectorSearchTask.d.ts.map +1 -1
  38. package/dist/task/ChunkVectorUpsertTask.d.ts +11 -2
  39. package/dist/task/ChunkVectorUpsertTask.d.ts.map +1 -1
  40. package/dist/task/ContextBuilderTask.d.ts +29 -3
  41. package/dist/task/ContextBuilderTask.d.ts.map +1 -1
  42. package/dist/task/CountTokensTask.d.ts +18 -1
  43. package/dist/task/CountTokensTask.d.ts.map +1 -1
  44. package/dist/task/DocumentEnricherTask.d.ts +38 -17
  45. package/dist/task/DocumentEnricherTask.d.ts.map +1 -1
  46. package/dist/task/DownloadModelTask.d.ts +24 -3
  47. package/dist/task/DownloadModelTask.d.ts.map +1 -1
  48. package/dist/task/FaceDetectorTask.d.ts +38 -6
  49. package/dist/task/FaceDetectorTask.d.ts.map +1 -1
  50. package/dist/task/FaceLandmarkerTask.d.ts +43 -7
  51. package/dist/task/FaceLandmarkerTask.d.ts.map +1 -1
  52. package/dist/task/GestureRecognizerTask.d.ts +48 -14
  53. package/dist/task/GestureRecognizerTask.d.ts.map +1 -1
  54. package/dist/task/HandLandmarkerTask.d.ts +45 -11
  55. package/dist/task/HandLandmarkerTask.d.ts.map +1 -1
  56. package/dist/task/HierarchicalChunkerTask.d.ts +32 -21
  57. package/dist/task/HierarchicalChunkerTask.d.ts.map +1 -1
  58. package/dist/task/HierarchyJoinTask.d.ts +35 -10
  59. package/dist/task/HierarchyJoinTask.d.ts.map +1 -1
  60. package/dist/task/ImageClassificationTask.d.ts +36 -4
  61. package/dist/task/ImageClassificationTask.d.ts.map +1 -1
  62. package/dist/task/ImageEmbeddingTask.d.ts +31 -1
  63. package/dist/task/ImageEmbeddingTask.d.ts.map +1 -1
  64. package/dist/task/ImageSegmentationTask.d.ts +36 -4
  65. package/dist/task/ImageSegmentationTask.d.ts.map +1 -1
  66. package/dist/task/ImageToTextTask.d.ts +32 -1
  67. package/dist/task/ImageToTextTask.d.ts.map +1 -1
  68. package/dist/task/ModelInfoTask.d.ts +35 -9
  69. package/dist/task/ModelInfoTask.d.ts.map +1 -1
  70. package/dist/task/ModelSearchTask.d.ts +66 -0
  71. package/dist/task/ModelSearchTask.d.ts.map +1 -0
  72. package/dist/task/ObjectDetectionTask.d.ts +38 -6
  73. package/dist/task/ObjectDetectionTask.d.ts.map +1 -1
  74. package/dist/task/PoseLandmarkerTask.d.ts +52 -17
  75. package/dist/task/PoseLandmarkerTask.d.ts.map +1 -1
  76. package/dist/task/QueryExpanderTask.d.ts +8 -3
  77. package/dist/task/QueryExpanderTask.d.ts.map +1 -1
  78. package/dist/task/RerankerTask.d.ts +26 -2
  79. package/dist/task/RerankerTask.d.ts.map +1 -1
  80. package/dist/task/StructuralParserTask.d.ts +7 -1
  81. package/dist/task/StructuralParserTask.d.ts.map +1 -1
  82. package/dist/task/StructuredGenerationTask.d.ts +23 -1
  83. package/dist/task/StructuredGenerationTask.d.ts.map +1 -1
  84. package/dist/task/TextChunkerTask.d.ts +10 -5
  85. package/dist/task/TextChunkerTask.d.ts.map +1 -1
  86. package/dist/task/TextClassificationTask.d.ts +23 -4
  87. package/dist/task/TextClassificationTask.d.ts.map +1 -1
  88. package/dist/task/TextEmbeddingTask.d.ts +19 -2
  89. package/dist/task/TextEmbeddingTask.d.ts.map +1 -1
  90. package/dist/task/TextFillMaskTask.d.ts +22 -5
  91. package/dist/task/TextFillMaskTask.d.ts.map +1 -1
  92. package/dist/task/TextGenerationTask.d.ts +23 -1
  93. package/dist/task/TextGenerationTask.d.ts.map +1 -1
  94. package/dist/task/TextLanguageDetectionTask.d.ts +22 -4
  95. package/dist/task/TextLanguageDetectionTask.d.ts.map +1 -1
  96. package/dist/task/TextNamedEntityRecognitionTask.d.ts +23 -5
  97. package/dist/task/TextNamedEntityRecognitionTask.d.ts.map +1 -1
  98. package/dist/task/TextQuestionAnswerTask.d.ts +19 -1
  99. package/dist/task/TextQuestionAnswerTask.d.ts.map +1 -1
  100. package/dist/task/TextRewriterTask.d.ts +19 -1
  101. package/dist/task/TextRewriterTask.d.ts.map +1 -1
  102. package/dist/task/TextSummaryTask.d.ts +18 -1
  103. package/dist/task/TextSummaryTask.d.ts.map +1 -1
  104. package/dist/task/TextTranslationTask.d.ts +20 -1
  105. package/dist/task/TextTranslationTask.d.ts.map +1 -1
  106. package/dist/task/ToolCallingTask.d.ts +9 -66
  107. package/dist/task/ToolCallingTask.d.ts.map +1 -1
  108. package/dist/task/ToolCallingUtils.d.ts +65 -0
  109. package/dist/task/ToolCallingUtils.d.ts.map +1 -0
  110. package/dist/task/TopicSegmenterTask.d.ts +10 -4
  111. package/dist/task/TopicSegmenterTask.d.ts.map +1 -1
  112. package/dist/task/UnloadModelTask.d.ts +18 -2
  113. package/dist/task/UnloadModelTask.d.ts.map +1 -1
  114. package/dist/task/VectorQuantizeTask.d.ts +11 -7
  115. package/dist/task/VectorQuantizeTask.d.ts.map +1 -1
  116. package/dist/task/VectorSimilarityTask.d.ts +8 -3
  117. package/dist/task/VectorSimilarityTask.d.ts.map +1 -1
  118. package/dist/task/base/AiTask.d.ts.map +1 -1
  119. package/dist/task/base/AiTaskSchemas.d.ts.map +1 -1
  120. package/dist/task/base/AiVisionTask.d.ts.map +1 -1
  121. package/dist/task/base/StreamingAiTask.d.ts.map +1 -1
  122. package/dist/task/index.d.ts +4 -1
  123. package/dist/task/index.d.ts.map +1 -1
  124. package/dist/worker.d.ts +21 -0
  125. package/dist/worker.d.ts.map +1 -0
  126. package/dist/worker.js +460 -0
  127. package/dist/worker.js.map +14 -0
  128. package/package.json +28 -21
@@ -1 +1 @@
1
- {"version":3,"file":"HierarchyJoinTask.d.ts","sourceRoot":"","sources":["../../src/task/HierarchyJoinTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAQH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAE5D,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAyCkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BiB,CAAC;AAEpC,MAAM,MAAM,sBAAsB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACpE,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAEtE;;;GAGG;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,SAA2D;IACpF,OAAc,SAAS,UAAS;WAElB,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CACX,KAAK,EAAE,sBAAsB,EAC7B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,uBAAuB,CAAC;CA8FpC;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":"HierarchyJoinTask.d.ts","sourceRoot":"","sources":["../../src/task/HierarchyJoinTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAQH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAE5D,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAyCkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BiB,CAAC;AAEpC,MAAM,MAAM,sBAAsB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACpE,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAEtE;;;GAGG;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,SAA2D;IACpF,OAAc,SAAS,UAAS;IAEhC,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,CA6FlC;CACF;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"}
@@ -256,14 +256,46 @@ export declare class ImageClassificationTask extends AiVisionTask<ImageClassific
256
256
  * @param input The input parameters for image classification (image, model, and optional categories)
257
257
  * @returns Promise resolving to the classification categories with scores
258
258
  */
259
- export declare const imageClassification: (input: ImageClassificationTaskInput, config?: JobQueueTaskConfig) => Promise<{
259
+ export declare const imageClassification: (input: {
260
+ categories?: string[] | undefined;
261
+ image: string | {
262
+ channels: number;
263
+ data: {
264
+ [x: string]: unknown;
265
+ } | {
266
+ channels: number;
267
+ data: number[];
268
+ height: number;
269
+ rawChannels?: number | undefined;
270
+ width: number;
271
+ };
272
+ height: number;
273
+ width: number;
274
+ };
275
+ maxCategories?: number | undefined;
276
+ model: string | {
277
+ [x: string]: unknown;
278
+ description?: string | undefined;
279
+ metadata?: {
280
+ [x: string]: unknown;
281
+ } | undefined;
282
+ model_id?: string | undefined;
283
+ provider: string;
284
+ provider_config: {
285
+ [x: string]: unknown;
286
+ credential_key?: string | undefined;
287
+ };
288
+ tasks?: string[] | undefined;
289
+ title?: string | undefined;
290
+ };
291
+ }, config?: JobQueueTaskConfig | undefined) => Promise<{
260
292
  categories: {
261
- score: number;
262
293
  label: string;
263
- }[] | {
264
294
  score: number;
295
+ }[][] | {
265
296
  label: string;
266
- }[][];
297
+ score: number;
298
+ }[];
267
299
  }>;
268
300
  declare module "@workglow/task-graph" {
269
301
  interface Workflow {
@@ -1 +1 @@
1
- {"version":3,"file":"ImageClassificationTask.d.ts","sourceRoot":"","sources":["../../src/task/ImageClassificationTask.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;AAInD,eAAO,MAAM,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2BR,CAAC;AAEpC,eAAO,MAAM,+BAA+B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAcT,CAAC;AAEpC,MAAM,MAAM,4BAA4B,GAAG,UAAU,CAAC,OAAO,8BAA8B,CAAC,CAAC;AAC7F,MAAM,MAAM,6BAA6B,GAAG,UAAU,CAAC,OAAO,+BAA+B,CAAC,CAAC;AAE/F;;;GAGG;AACH,qBAAa,uBAAwB,SAAQ,YAAY,CACvD,4BAA4B,EAC5B,6BAA6B,EAC7B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA6B;IAC/C,OAAc,QAAQ,SAAqB;IAC3C,OAAc,KAAK,SAA0B;IAC7C,OAAc,WAAW,SACkG;WAC7G,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,mBAAmB,GAC9B,OAAO,4BAA4B,EACnC,SAAS,kBAAkB;;;;;;;;EAG5B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,mBAAmB,EAAE,cAAc,CACjC,4BAA4B,EAC5B,6BAA6B,EAC7B,kBAAkB,CACnB,CAAC;KACH;CACF"}
1
+ {"version":3,"file":"ImageClassificationTask.d.ts","sourceRoot":"","sources":["../../src/task/ImageClassificationTask.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;AAInD,eAAO,MAAM,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2BR,CAAC;AAEpC,eAAO,MAAM,+BAA+B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAcT,CAAC;AAEpC,MAAM,MAAM,4BAA4B,GAAG,UAAU,CAAC,OAAO,8BAA8B,CAAC,CAAC;AAC7F,MAAM,MAAM,6BAA6B,GAAG,UAAU,CAAC,OAAO,+BAA+B,CAAC,CAAC;AAE/F;;;GAGG;AACH,qBAAa,uBAAwB,SAAQ,YAAY,CACvD,4BAA4B,EAC5B,6BAA6B,EAC7B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA6B;IAC/C,OAAc,QAAQ,SAAqB;IAC3C,OAAc,KAAK,SAA0B;IAC7C,OAAc,WAAW,SACkG;IAC3H,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAK/B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,mBAAmB,EAAE,cAAc,CACjC,4BAA4B,EAC5B,6BAA6B,EAC7B,kBAAkB,CACnB,CAAC;KACH;CACF"}
@@ -189,7 +189,37 @@ export declare class ImageEmbeddingTask extends AiVisionTask<ImageEmbeddingTaskI
189
189
  * @param input The input parameters for image embedding (image and model)
190
190
  * @returns Promise resolving to the image embedding vector
191
191
  */
192
- export declare const imageEmbedding: (input: ImageEmbeddingTaskInput, config?: JobQueueTaskConfig) => Promise<{
192
+ export declare const imageEmbedding: (input: {
193
+ image: string | {
194
+ channels: number;
195
+ data: {
196
+ [x: string]: unknown;
197
+ } | {
198
+ channels: number;
199
+ data: number[];
200
+ height: number;
201
+ rawChannels?: number | undefined;
202
+ width: number;
203
+ };
204
+ height: number;
205
+ width: number;
206
+ };
207
+ model: string | {
208
+ [x: string]: unknown;
209
+ description?: string | undefined;
210
+ metadata?: {
211
+ [x: string]: unknown;
212
+ } | undefined;
213
+ model_id?: string | undefined;
214
+ provider: string;
215
+ provider_config: {
216
+ [x: string]: unknown;
217
+ credential_key?: string | undefined;
218
+ };
219
+ tasks?: string[] | undefined;
220
+ title?: string | undefined;
221
+ };
222
+ }, config?: JobQueueTaskConfig | undefined) => Promise<{
193
223
  vector: import("@workglow/util").TypedArray;
194
224
  }>;
195
225
  declare module "@workglow/task-graph" {
@@ -1 +1 @@
1
- {"version":3,"file":"ImageEmbeddingTask.d.ts","sourceRoot":"","sources":["../../src/task/ImageEmbeddingTask.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;AAExB,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AASnD,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAQH,CAAC;AAEpC,eAAO,MAAM,0BAA0B;;;;;;;;;;;;CAOJ,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAC9C,OAAO,yBAAyB,EAChC,uBAAuB,CACxB,CAAC;AACF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAC/C,OAAO,0BAA0B,EACjC,uBAAuB,CACxB,CAAC;AAEF;;GAEG;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,SAA0D;WACrE,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":"ImageEmbeddingTask.d.ts","sourceRoot":"","sources":["../../src/task/ImageEmbeddingTask.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;AAExB,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AASnD,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAQH,CAAC;AAEpC,eAAO,MAAM,0BAA0B;;;;;;;;;;;;CAOJ,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAC9C,OAAO,yBAAyB,EAChC,uBAAuB,CACxB,CAAC;AACF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAC/C,OAAO,0BAA0B,EACjC,uBAAuB,CACxB,CAAC;AAEF;;GAEG;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,SAA0D;IACnF,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"}
@@ -263,20 +263,52 @@ export declare class ImageSegmentationTask extends AiVisionTask<ImageSegmentatio
263
263
  * @param input The input parameters for image segmentation (image and model)
264
264
  * @returns Promise resolving to the segmentation masks with labels and scores
265
265
  */
266
- export declare const imageSegmentation: (input: ImageSegmentationTaskInput, config?: JobQueueTaskConfig) => Promise<{
266
+ export declare const imageSegmentation: (input: {
267
+ image: string | {
268
+ channels: number;
269
+ data: {
270
+ [x: string]: unknown;
271
+ } | {
272
+ channels: number;
273
+ data: number[];
274
+ height: number;
275
+ rawChannels?: number | undefined;
276
+ width: number;
277
+ };
278
+ height: number;
279
+ width: number;
280
+ };
281
+ maskThreshold?: number | undefined;
282
+ model: string | {
283
+ [x: string]: unknown;
284
+ description?: string | undefined;
285
+ metadata?: {
286
+ [x: string]: unknown;
287
+ } | undefined;
288
+ model_id?: string | undefined;
289
+ provider: string;
290
+ provider_config: {
291
+ [x: string]: unknown;
292
+ credential_key?: string | undefined;
293
+ };
294
+ tasks?: string[] | undefined;
295
+ title?: string | undefined;
296
+ };
297
+ threshold?: number | undefined;
298
+ }, config?: JobQueueTaskConfig | undefined) => Promise<{
267
299
  masks: {
268
- score: number;
269
300
  label: string;
270
301
  mask: {
271
302
  [x: string]: unknown;
272
303
  };
273
- }[] | {
274
304
  score: number;
305
+ }[][] | {
275
306
  label: string;
276
307
  mask: {
277
308
  [x: string]: unknown;
278
309
  };
279
- }[][];
310
+ score: number;
311
+ }[];
280
312
  }>;
281
313
  declare module "@workglow/task-graph" {
282
314
  interface Workflow {
@@ -1 +1 @@
1
- {"version":3,"file":"ImageSegmentationTask.d.ts","sourceRoot":"","sources":["../../src/task/ImageSegmentationTask.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;AAInD,eAAO,MAAM,4BAA4B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA0BN,CAAC;AA4BpC,eAAO,MAAM,6BAA6B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAcP,CAAC;AAEpC,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,OAAO,4BAA4B,CAAC,CAAC;AACzF,MAAM,MAAM,2BAA2B,GAAG,UAAU,CAAC,OAAO,6BAA6B,CAAC,CAAC;AAE3F;;GAEG;AACH,qBAAa,qBAAsB,SAAQ,YAAY,CACrD,0BAA0B,EAC1B,2BAA2B,EAC3B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA2B;IAC7C,OAAc,QAAQ,SAAqB;IAC3C,OAAc,KAAK,SAAwB;IAC3C,OAAc,WAAW,SACiD;WAC5D,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,iBAAiB,GAC5B,OAAO,0BAA0B,EACjC,SAAS,kBAAkB;;;;;;;;;;;;;;EAG5B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,iBAAiB,EAAE,cAAc,CAC/B,0BAA0B,EAC1B,2BAA2B,EAC3B,kBAAkB,CACnB,CAAC;KACH;CACF"}
1
+ {"version":3,"file":"ImageSegmentationTask.d.ts","sourceRoot":"","sources":["../../src/task/ImageSegmentationTask.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;AAInD,eAAO,MAAM,4BAA4B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA0BN,CAAC;AA4BpC,eAAO,MAAM,6BAA6B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAcP,CAAC;AAEpC,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,OAAO,4BAA4B,CAAC,CAAC;AACzF,MAAM,MAAM,2BAA2B,GAAG,UAAU,CAAC,OAAO,6BAA6B,CAAC,CAAC;AAE3F;;GAEG;AACH,qBAAa,qBAAsB,SAAQ,YAAY,CACrD,0BAA0B,EAC1B,2BAA2B,EAC3B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA2B;IAC7C,OAAc,QAAQ,SAAqB;IAC3C,OAAc,KAAK,SAAwB;IAC3C,OAAc,WAAW,SACiD;IAC1E,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAK7B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,iBAAiB,EAAE,cAAc,CAC/B,0BAA0B,EAC1B,2BAA2B,EAC3B,kBAAkB,CACnB,CAAC;KACH;CACF"}
@@ -207,7 +207,38 @@ export declare class ImageToTextTask extends AiVisionTask<ImageToTextTaskInput,
207
207
  * @param input The input parameters for image to text (image and model)
208
208
  * @returns Promise resolving to the generated text description
209
209
  */
210
- export declare const imageToText: (input: ImageToTextTaskInput, config?: JobQueueTaskConfig) => Promise<{
210
+ export declare const imageToText: (input: {
211
+ image: string | {
212
+ channels: number;
213
+ data: {
214
+ [x: string]: unknown;
215
+ } | {
216
+ channels: number;
217
+ data: number[];
218
+ height: number;
219
+ rawChannels?: number | undefined;
220
+ width: number;
221
+ };
222
+ height: number;
223
+ width: number;
224
+ };
225
+ maxTokens?: number | undefined;
226
+ model: string | {
227
+ [x: string]: unknown;
228
+ description?: string | undefined;
229
+ metadata?: {
230
+ [x: string]: unknown;
231
+ } | undefined;
232
+ model_id?: string | undefined;
233
+ provider: string;
234
+ provider_config: {
235
+ [x: string]: unknown;
236
+ credential_key?: string | undefined;
237
+ };
238
+ tasks?: string[] | undefined;
239
+ title?: string | undefined;
240
+ };
241
+ }, config?: JobQueueTaskConfig | undefined) => Promise<{
211
242
  text: string | string[];
212
243
  }>;
213
244
  declare module "@workglow/task-graph" {
@@ -1 +1 @@
1
- {"version":3,"file":"ImageToTextTask.d.ts","sourceRoot":"","sources":["../../src/task/ImageToTextTask.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;AAUnD,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgBA,CAAC;AAEpC,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;CAWD,CAAC;AAEpC,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAC7E,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAE/E;;GAEG;AACH,qBAAa,eAAgB,SAAQ,YAAY,CAC/C,oBAAoB,EACpB,qBAAqB,EACrB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAqB;IACvC,OAAc,QAAQ,SAAqB;IAC3C,OAAc,KAAK,SAAmB;IACtC,OAAc,WAAW,SACgD;WAC3D,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,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":"ImageToTextTask.d.ts","sourceRoot":"","sources":["../../src/task/ImageToTextTask.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;AAUnD,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgBA,CAAC;AAEpC,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;CAWD,CAAC;AAEpC,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAC7E,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAE/E;;GAEG;AACH,qBAAa,eAAgB,SAAQ,YAAY,CAC/C,oBAAoB,EACpB,qBAAqB,EACrB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAqB;IACvC,OAAc,QAAQ,SAAqB;IAC3C,OAAc,KAAK,SAAmB;IACtC,OAAc,WAAW,SACgD;IACzE,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;CACF;AAED;;;;;GAKG;AACH,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"}
@@ -3,7 +3,7 @@
3
3
  * Copyright 2025 Steven Roussey <sroussey@gmail.com>
4
4
  * SPDX-License-Identifier: Apache-2.0
5
5
  */
6
- import { CreateWorkflow, JobQueueTaskConfig } from "@workglow/task-graph";
6
+ import { CreateWorkflow, type IExecuteContext, JobQueueTaskConfig } from "@workglow/task-graph";
7
7
  import { DataPortSchema, FromSchema } from "@workglow/util";
8
8
  import { AiTask } from "./base/AiTask";
9
9
  declare const ModelInfoInputSchema: {
@@ -154,6 +154,13 @@ declare const ModelInfoOutputSchema: {
154
154
  readonly type: "boolean";
155
155
  };
156
156
  readonly file_sizes: {};
157
+ readonly quantizations: {
158
+ readonly type: "array";
159
+ readonly items: {
160
+ readonly type: "string";
161
+ };
162
+ readonly description: "Available quantization variants (e.g. fp32, fp16, q8). Only present for models with quantization options.";
163
+ };
157
164
  };
158
165
  readonly required: readonly ["model", "is_local", "is_remote", "supports_browser", "supports_node", "is_cached", "is_loaded", "file_sizes"];
159
166
  readonly additionalProperties: false;
@@ -166,11 +173,12 @@ export type ModelInfoTaskOutput = FromSchema<typeof ModelInfoOutputSchema>;
166
173
  export declare class ModelInfoTask extends AiTask<ModelInfoTaskInput, ModelInfoTaskOutput, JobQueueTaskConfig> {
167
174
  static type: string;
168
175
  static category: string;
176
+ static cacheable: boolean;
169
177
  static title: string;
170
178
  static description: string;
171
179
  static inputSchema(): DataPortSchema;
172
180
  static outputSchema(): DataPortSchema;
173
- static cacheable: boolean;
181
+ execute(input: ModelInfoTaskInput, context: IExecuteContext): Promise<ModelInfoTaskOutput>;
174
182
  }
175
183
  /**
176
184
  * Retrieve runtime metadata about a model.
@@ -178,29 +186,47 @@ export declare class ModelInfoTask extends AiTask<ModelInfoTaskInput, ModelInfoT
178
186
  * @param input - Input containing the model to query
179
187
  * @returns Promise resolving to model info including locality and cache status
180
188
  */
181
- export declare const modelInfo: (input: ModelInfoTaskInput, config?: JobQueueTaskConfig) => Promise<{
189
+ export declare const modelInfo: (input: {
190
+ detail?: "cached_status" | "files" | "files_with_metadata" | undefined;
182
191
  model: string | {
183
192
  [x: string]: unknown;
184
- title?: string | undefined;
185
193
  description?: string | undefined;
186
- model_id?: string | undefined;
187
- tasks?: string[] | undefined;
188
194
  metadata?: {
189
195
  [x: string]: unknown;
190
196
  } | undefined;
197
+ model_id?: string | undefined;
191
198
  provider: string;
192
199
  provider_config: {
193
200
  [x: string]: unknown;
194
201
  credential_key?: string | undefined;
195
202
  };
203
+ tasks?: string[] | undefined;
204
+ title?: string | undefined;
196
205
  };
206
+ }, config?: JobQueueTaskConfig | undefined) => Promise<{
207
+ file_sizes: unknown;
208
+ is_cached: boolean;
209
+ is_loaded: boolean;
197
210
  is_local: boolean;
198
211
  is_remote: boolean;
212
+ model: string | {
213
+ [x: string]: unknown;
214
+ description?: string | undefined;
215
+ metadata?: {
216
+ [x: string]: unknown;
217
+ } | undefined;
218
+ model_id?: string | undefined;
219
+ provider: string;
220
+ provider_config: {
221
+ [x: string]: unknown;
222
+ credential_key?: string | undefined;
223
+ };
224
+ tasks?: string[] | undefined;
225
+ title?: string | undefined;
226
+ };
227
+ quantizations?: string[] | undefined;
199
228
  supports_browser: boolean;
200
229
  supports_node: boolean;
201
- is_cached: boolean;
202
- is_loaded: boolean;
203
- file_sizes: unknown;
204
230
  }>;
205
231
  declare module "@workglow/task-graph" {
206
232
  interface Workflow {
@@ -1 +1 @@
1
- {"version":3,"file":"ModelInfoTask.d.ts","sourceRoot":"","sources":["../../src/task/ModelInfoTask.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,QAAA,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAYS,CAAC;AAEpC,QAAA,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAuBQ,CAAC;AAEpC,MAAM,MAAM,kBAAkB,GAAG,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC;AACzE,MAAM,MAAM,mBAAmB,GAAG,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAE3E;;GAEG;AACH,qBAAa,aAAc,SAAQ,MAAM,CACvC,kBAAkB,EAClB,mBAAmB,EACnB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAmB;IACrC,OAAc,QAAQ,SAAY;IAClC,OAAc,KAAK,SAAgB;IACnC,OAAc,WAAW,SACsE;WACjF,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;IAG5C,OAAc,SAAS,UAAS;CACjC;AAED;;;;;GAKG;AACH,eAAO,MAAM,SAAS,GAAI,OAAO,kBAAkB,EAAE,SAAS,kBAAkB;;;;;;;;;;;;;;;;;;;;;;;EAE/E,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,SAAS,EAAE,cAAc,CAAC,kBAAkB,EAAE,mBAAmB,EAAE,kBAAkB,CAAC,CAAC;KACxF;CACF"}
1
+ {"version":3,"file":"ModelInfoTask.d.ts","sourceRoot":"","sources":["../../src/task/ModelInfoTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,KAAK,eAAe,EACpB,kBAAkB,EAEnB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAG5D,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,QAAA,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAYS,CAAC;AAEpC,QAAA,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA6BQ,CAAC;AAEpC,MAAM,MAAM,kBAAkB,GAAG,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC;AACzE,MAAM,MAAM,mBAAmB,GAAG,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAE3E;;GAEG;AACH,qBAAa,aAAc,SAAQ,MAAM,CACvC,kBAAkB,EAClB,mBAAmB,EACnB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAmB;IACrC,OAAc,QAAQ,SAAc;IACpC,OAAc,SAAS,UAAS;IAChC,OAAc,KAAK,SAAgB;IACnC,OAAc,WAAW,SACsE;IAC/F,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;IAEK,OAAO,CAAC,KAAK,EAAE,kBAAkB,EAAE,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAS/F;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAErB,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,SAAS,EAAE,cAAc,CAAC,kBAAkB,EAAE,mBAAmB,EAAE,kBAAkB,CAAC,CAAC;KACxF;CACF"}
@@ -0,0 +1,66 @@
1
+ /**
2
+ * @license
3
+ * Copyright 2025 Steven Roussey <sroussey@gmail.com>
4
+ * SPDX-License-Identifier: Apache-2.0
5
+ */
6
+ import { CreateWorkflow, IExecuteContext, Task, TaskConfig } from "@workglow/task-graph";
7
+ import { DataPortSchema, FromSchema } from "@workglow/util";
8
+ import type { ModelRecord } from "../model/ModelSchema";
9
+ /**
10
+ * A single result item from a model search.
11
+ */
12
+ export interface ModelSearchResultItem {
13
+ readonly id: string;
14
+ readonly label: string;
15
+ readonly description: string;
16
+ readonly record: Partial<ModelRecord>;
17
+ readonly raw: unknown;
18
+ }
19
+ declare const ModelSearchInputSchema: {
20
+ readonly type: "object";
21
+ readonly properties: {
22
+ readonly provider: {
23
+ readonly type: "string";
24
+ readonly title: "Provider";
25
+ readonly description: "The model provider to search (e.g. ANTHROPIC, OPENAI, HF_TRANSFORMERS_ONNX)";
26
+ };
27
+ readonly query: {
28
+ readonly type: "string";
29
+ readonly title: "Query";
30
+ readonly description: "Search query string";
31
+ };
32
+ };
33
+ readonly required: readonly ["provider", "query"];
34
+ readonly additionalProperties: false;
35
+ };
36
+ export type ModelSearchTaskInput = FromSchema<typeof ModelSearchInputSchema>;
37
+ export type ModelSearchTaskOutput = {
38
+ results: ModelSearchResultItem[];
39
+ };
40
+ /**
41
+ * Search for models using a provider-specific run function from the AiProviderRegistry.
42
+ */
43
+ export declare class ModelSearchTask extends Task<ModelSearchTaskInput, ModelSearchTaskOutput, TaskConfig> {
44
+ static type: string;
45
+ static category: string;
46
+ static title: string;
47
+ static description: string;
48
+ static cacheable: boolean;
49
+ static inputSchema(): DataPortSchema;
50
+ static outputSchema(): DataPortSchema;
51
+ execute(input: ModelSearchTaskInput, context: IExecuteContext): Promise<ModelSearchTaskOutput>;
52
+ }
53
+ /**
54
+ * Search for models using a provider-specific search function.
55
+ */
56
+ export declare const modelSearch: (input: {
57
+ provider: string;
58
+ query: string;
59
+ }, config?: TaskConfig | undefined) => Promise<ModelSearchTaskOutput>;
60
+ declare module "@workglow/task-graph" {
61
+ interface Workflow {
62
+ modelSearch: CreateWorkflow<ModelSearchTaskInput, ModelSearchTaskOutput, TaskConfig>;
63
+ }
64
+ }
65
+ export {};
66
+ //# sourceMappingURL=ModelSearchTask.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ModelSearchTask.d.ts","sourceRoot":"","sources":["../../src/task/ModelSearchTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAE,UAAU,EAAY,MAAM,sBAAsB,CAAC;AACnG,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAE5D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IACtC,QAAQ,CAAC,GAAG,EAAE,OAAO,CAAC;CACvB;AAED,QAAA,MAAM,sBAAsB;;;;;;;;;;;;;;;;CAgBO,CAAC;AAyBpC,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAC7E,MAAM,MAAM,qBAAqB,GAAG;IAAE,OAAO,EAAE,qBAAqB,EAAE,CAAA;CAAE,CAAC;AAEzE;;GAEG;AACH,qBAAa,eAAgB,SAAQ,IAAI,CACvC,oBAAoB,EACpB,qBAAqB,EACrB,UAAU,CACX;IACC,OAAc,IAAI,SAAqB;IACvC,OAAc,QAAQ,SAAc;IACpC,OAAc,KAAK,SAAkB;IACrC,OAAc,WAAW,SAAgE;IACzF,OAAc,SAAS,UAAS;IAEhC,OAAc,WAAW,IAAI,cAAc,CAE1C;IACD,OAAc,YAAY,IAAI,cAAc,CAE3C;IAEK,OAAO,CACX,KAAK,EAAE,oBAAoB,EAC3B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,qBAAqB,CAAC,CAQhC;CACF;AAED;;GAEG;AACH,eAAO,MAAM,WAAW;;;qEAEvB,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,UAAU,CAAC,CAAC;KACtF;CACF"}
@@ -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,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;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"}
@@ -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,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;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"}
@@ -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,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;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"}