@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
@@ -1 +1 @@
1
- {"version":3,"file":"HandLandmarkerTask.d.ts","sourceRoot":"","sources":["../../src/task/HandLandmarkerTask.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;AAgFnD,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4CH,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,SAC+E;WAC1F,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":"HandLandmarkerTask.d.ts","sourceRoot":"","sources":["../../src/task/HandLandmarkerTask.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;AAgFnD,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4CH,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,SAC+E;IACxG,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
  declare const inputSchema: {
9
9
  readonly type: "object";
10
10
  readonly properties: {
@@ -255,47 +255,58 @@ export declare class HierarchicalChunkerTask extends Task<HierarchicalChunkerTas
255
255
  static inputSchema(): DataPortSchema;
256
256
  static outputSchema(): DataPortSchema;
257
257
  execute(input: HierarchicalChunkerTaskInput, context: IExecuteContext): Promise<HierarchicalChunkerTaskOutput>;
258
- /**
259
- * Hierarchical chunking that respects document structure
260
- */
261
258
  private chunkHierarchically;
262
- /**
263
- * Chunk a single text string, using countFn for token counting.
264
- * countFn always returns a number -- it falls back to estimation internally
265
- * when no real tokenizer is available.
266
- */
267
259
  private chunkText;
268
- /**
269
- * Flat chunking (ignores hierarchy)
270
- */
271
260
  private chunkFlat;
272
261
  /**
273
262
  * Collect all text from a node and its descendants
274
263
  */
275
264
  private collectAllText;
276
265
  }
277
- export declare const hierarchicalChunker: (input: HierarchicalChunkerTaskInput, config?: JobQueueTaskConfig) => Promise<{
266
+ export declare const hierarchicalChunker: (input: {
278
267
  doc_id: string;
268
+ documentTree: unknown;
269
+ maxTokens?: number | undefined;
270
+ model?: string | {
271
+ [x: string]: unknown;
272
+ description?: string | undefined;
273
+ metadata?: {
274
+ [x: string]: unknown;
275
+ } | undefined;
276
+ model_id?: string | undefined;
277
+ provider: string;
278
+ provider_config: {
279
+ [x: string]: unknown;
280
+ credential_key?: string | undefined;
281
+ };
282
+ tasks?: string[] | undefined;
283
+ title?: string | undefined;
284
+ } | undefined;
285
+ overlap?: number | undefined;
286
+ reservedTokens?: number | undefined;
287
+ strategy?: "flat" | "hierarchical" | "sentence" | undefined;
288
+ }, config?: JobQueueTaskConfig | undefined) => Promise<{
279
289
  chunks: {
280
290
  [x: string]: unknown;
291
+ chunkId: string;
292
+ depth: number;
293
+ doc_id: string;
281
294
  doc_title?: string | undefined;
282
- leafNodeId?: string | undefined;
283
- summary?: string | undefined;
284
295
  entities?: {
285
- type: string;
286
- text: string;
287
296
  score: number;
297
+ text: string;
298
+ type: string;
288
299
  }[] | undefined;
300
+ leafNodeId?: string | undefined;
301
+ nodePath: string[];
289
302
  parentSummaries?: string[] | undefined;
290
303
  sectionTitles?: string[] | undefined;
291
- doc_id: string;
292
- chunkId: string;
304
+ summary?: string | undefined;
293
305
  text: string;
294
- nodePath: string[];
295
- depth: number;
296
306
  }[];
297
- text: string[];
298
307
  count: number;
308
+ doc_id: string;
309
+ text: string[];
299
310
  }>;
300
311
  declare module "@workglow/task-graph" {
301
312
  interface Workflow {
@@ -1 +1 @@
1
- {"version":3,"file":"HierarchicalChunkerTask.d.ts","sourceRoot":"","sources":["../../src/task/HierarchicalChunkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAWH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAS,MAAM,gBAAgB,CAAC;AASnE,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4CkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4BiB,CAAC;AAEpC,MAAM,MAAM,4BAA4B,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAC1E,MAAM,MAAM,6BAA6B,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAE5E;;;;;GAKG;AACH,qBAAa,uBAAwB,SAAQ,IAAI,CAC/C,4BAA4B,EAC5B,6BAA6B,EAC7B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA6B;IAC/C,OAAc,QAAQ,SAAc;IACpC,OAAc,KAAK,SAA0B;IAC7C,OAAc,WAAW,SAA6D;IACtF,OAAc,SAAS,UAAQ;WAEjB,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CACX,KAAK,EAAE,4BAA4B,EACnC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,6BAA6B,CAAC;IAqDzC;;OAEG;YACW,mBAAmB;IA6BjC;;;;OAIG;YACW,SAAS;IA+EvB;;OAEG;YACW,SAAS;IAWvB;;OAEG;IACH,OAAO,CAAC,cAAc;CAgBvB;AAED,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":"HierarchicalChunkerTask.d.ts","sourceRoot":"","sources":["../../src/task/HierarchicalChunkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAWH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAUnE,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4CkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4BiB,CAAC;AAEpC,MAAM,MAAM,4BAA4B,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAC1E,MAAM,MAAM,6BAA6B,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAE5E;;;;;GAKG;AACH,qBAAa,uBAAwB,SAAQ,IAAI,CAC/C,4BAA4B,EAC5B,6BAA6B,EAC7B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA6B;IAC/C,OAAc,QAAQ,SAAc;IACpC,OAAc,KAAK,SAA0B;IAC7C,OAAc,WAAW,SAA6D;IACtF,OAAc,SAAS,UAAQ;IAE/B,OAAc,WAAW,IAAI,cAAc,CAE1C;IAED,OAAc,YAAY,IAAI,cAAc,CAE3C;IAEK,OAAO,CACX,KAAK,EAAE,4BAA4B,EACnC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,6BAA6B,CAAC,CAmDxC;YAKa,mBAAmB;YAkCnB,SAAS;YAkFT,SAAS;IAWvB;;OAEG;IACH,OAAO,CAAC,cAAc;CAgBvB;AAED,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"}
@@ -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: {
@@ -312,29 +312,54 @@ export declare class HierarchyJoinTask extends Task<HierarchyJoinTaskInput, Hier
312
312
  static outputSchema(): DataPortSchema;
313
313
  execute(input: HierarchyJoinTaskInput, context: IExecuteContext): Promise<HierarchyJoinTaskOutput>;
314
314
  }
315
- export declare const hierarchyJoin: (input: HierarchyJoinTaskInput, config?: JobQueueTaskConfig) => Promise<{
315
+ export declare const hierarchyJoin: (input: {
316
+ chunk_ids: string[];
317
+ chunks: string[];
318
+ includeEntities?: boolean | undefined;
319
+ includeParentSummaries?: boolean | undefined;
320
+ knowledgeBase: unknown;
316
321
  metadata: {
317
322
  [x: string]: unknown;
323
+ chunkId: string;
324
+ depth: number;
325
+ doc_id: string;
318
326
  doc_title?: string | undefined;
319
- leafNodeId?: string | undefined;
320
- summary?: string | undefined;
321
327
  entities?: {
322
- type: string;
323
- text: string;
324
328
  score: number;
329
+ text: string;
330
+ type: string;
325
331
  }[] | undefined;
332
+ leafNodeId?: string | undefined;
333
+ nodePath: string[];
326
334
  parentSummaries?: string[] | undefined;
327
335
  sectionTitles?: string[] | undefined;
328
- doc_id: string;
329
- chunkId: string;
336
+ summary?: string | undefined;
330
337
  text: string;
331
- nodePath: string[];
332
- depth: number;
333
338
  }[];
339
+ scores: number[];
340
+ }, config?: JobQueueTaskConfig | undefined) => Promise<{
341
+ chunk_ids: string[];
334
342
  chunks: string[];
335
343
  count: number;
344
+ metadata: {
345
+ [x: string]: unknown;
346
+ chunkId: string;
347
+ depth: number;
348
+ doc_id: string;
349
+ doc_title?: string | undefined;
350
+ entities?: {
351
+ score: number;
352
+ text: string;
353
+ type: string;
354
+ }[] | undefined;
355
+ leafNodeId?: string | undefined;
356
+ nodePath: string[];
357
+ parentSummaries?: string[] | undefined;
358
+ sectionTitles?: string[] | undefined;
359
+ summary?: string | undefined;
360
+ text: string;
361
+ }[];
336
362
  scores: number[];
337
- chunk_ids: string[];
338
363
  }>;
339
364
  declare module "@workglow/task-graph" {
340
365
  interface Workflow {
@@ -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,uBAAuB,CAAC;AAEnE,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"}
@@ -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 ImageClassificationInputSchema: {
10
10
  readonly type: "object";
@@ -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,uBAAuB,CAAC;AAEnE,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"}
@@ -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 { AiVisionTask } from "./base/AiVisionTask";
9
9
  export declare const ImageEmbeddingInputSchema: {
10
10
  readonly type: "object";
@@ -189,8 +189,38 @@ 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<{
193
- vector: import("@workglow/util").TypedArray;
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<{
223
+ vector: import("@workglow/util/schema").TypedArray;
194
224
  }>;
195
225
  declare module "@workglow/task-graph" {
196
226
  interface Workflow {
@@ -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,uBAAuB,CAAC;AAE/B,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"}
@@ -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 ImageSegmentationInputSchema: {
10
10
  readonly type: "object";
@@ -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,uBAAuB,CAAC;AAEnE,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"}
@@ -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 ImageToTextInputSchema: {
10
10
  readonly type: "object";
@@ -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,uBAAuB,CAAC;AAEnE,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,8 +3,8 @@
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";
7
- import { DataPortSchema, FromSchema } from "@workglow/util";
6
+ import { CreateWorkflow, type IExecuteContext, JobQueueTaskConfig } from "@workglow/task-graph";
7
+ import { DataPortSchema, FromSchema } from "@workglow/util/schema";
8
8
  import { AiTask } from "./base/AiTask";
9
9
  declare const ModelInfoInputSchema: {
10
10
  readonly type: "object";
@@ -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,uBAAuB,CAAC;AAGnE,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/schema";
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,uBAAuB,CAAC;AAEnE,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"}