@workglow/ai 0.0.108 → 0.0.110

package/dist/provider/AiProvider.d.ts

@@ -64,6 +64,10 @@ export interface AiProviderRegisterOptions {
 export declare abstract class AiProvider<TModelConfig extends ModelConfig = ModelConfig> {
     /** Unique provider identifier (e.g., "HF_TRANSFORMERS_ONNX") */
     abstract readonly name: string;
+    /** Whether this provider runs models locally (on the same machine). */
+    abstract readonly isLocal: boolean;
+    /** Whether this provider can run in a browser environment. */
+    abstract readonly supportsBrowser: boolean;
     /**
      * List of task type names this provider supports.
      * This is lightweight metadata -- no heavy library imports needed.

package/dist/provider/AiProvider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AiProvider.d.ts","sourceRoot":"","sources":["../../src/provider/AiProvider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EAAyC,KAAK,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC1F,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,OAAO,EACL,KAAK,uBAAuB,EAC5B,KAAK,eAAe,EACpB,KAAK,kBAAkB,EAExB,MAAM,sBAAsB,CAAC;AAE9B;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,QAAQ,GAAG,QAAQ,CAAC;AAEjD;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACxC,sEAAsE;IACtE,IAAI,EAAE,cAAc,CAAC;IACrB,+DAA+D;IAC/D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,8BAA8B;IAC9B,KAAK,CAAC,EAAE;QACN,wDAAwD;QACxD,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,uEAAuE;QACvE,UAAU,CAAC,EAAE,OAAO,CAAC;KACtB,CAAC;CACH;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AACH,8BAAsB,UAAU,CAAC,YAAY,SAAS,WAAW,GAAG,WAAW;IAC7E,gEAAgE;IAChE,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAE/B;;;OAGG;IACH,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAAE,SAAS,MAAM,EAAE,CAAC;IAE/C;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;IAEnF;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;IAE5F;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CACvC,MAAM,EACN,uBAAuB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAChD,CAAC;gBAGA,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,EAC/D,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,EACxE,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,uBAAuB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC;IAOjF,qDAAqD;IACrD,IAAI,kBAAkB,IAAI,SAAS,MAAM,EAAE,CAE1C;IAED;;;;OAIG;IACH,QAAQ,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EACzE,QAAQ,EAAE,MAAM,GACf,eAAe,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,CAAC,GAAG,SAAS;IAIlD;;;;OAIG;IACH,WAAW,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EAC5E,QAAQ,EAAE,MAAM,GACf,kBAAkB,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,CAAC,GAAG,SAAS;IAIrD;;;;OAIG;IACH,gBAAgB,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EACjF,QAAQ,EAAE,MAAM,GACf,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,CAAC,GAAG,SAAS;IAM1D;;;;;;;;;;;;OAYG;IACG,QAAQ,CAAC,OAAO,GAAE,yBAA8C,GAAG,OAAO,CAAC,IAAI,CAAC;IAsDtF;;;;;;;;OAQG;IACH,sBAAsB,CAAC,YAAY,EAAE,YAAY,GAAG,IAAI;IAsBxD;;;;OAIG;cACa,YAAY,CAAC,QAAQ,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC;IAEhF;;;OAGG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAE9B;;;OAGG;cACa,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAGhE"}
+{"version":3,"file":"AiProvider.d.ts","sourceRoot":"","sources":["../../src/provider/AiProvider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EAAyC,KAAK,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC1F,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,OAAO,EACL,KAAK,uBAAuB,EAC5B,KAAK,eAAe,EACpB,KAAK,kBAAkB,EAExB,MAAM,sBAAsB,CAAC;AAE9B;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,QAAQ,GAAG,QAAQ,CAAC;AAEjD;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACxC,sEAAsE;IACtE,IAAI,EAAE,cAAc,CAAC;IACrB,+DAA+D;IAC/D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,8BAA8B;IAC9B,KAAK,CAAC,EAAE;QACN,wDAAwD;QACxD,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,uEAAuE;QACvE,UAAU,CAAC,EAAE,OAAO,CAAC;KACtB,CAAC;CACH;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AACH,8BAAsB,UAAU,CAAC,YAAY,SAAS,WAAW,GAAG,WAAW;IAC7E,gEAAgE;IAChE,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAE/B,uEAAuE;IACvE,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAEnC,8DAA8D;IAC9D,QAAQ,CAAC,QAAQ,CAAC,eAAe,EAAE,OAAO,CAAC;IAE3C;;;OAGG;IACH,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAAE,SAAS,MAAM,EAAE,CAAC;IAE/C;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;IAEnF;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;IAE5F;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CACvC,MAAM,EACN,uBAAuB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAChD,CAAC;gBAGA,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,EAC/D,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,EACxE,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,uBAAuB,CAAC,GAAG,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC;IAOjF,qDAAqD;IACrD,IAAI,kBAAkB,IAAI,SAAS,MAAM,EAAE,CAE1C;IAED;;;;OAIG;IACH,QAAQ,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EACzE,QAAQ,EAAE,MAAM,GACf,eAAe,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,CAAC,GAAG,SAAS;IAIlD;;;;OAIG;IACH,WAAW,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EAC5E,QAAQ,EAAE,MAAM,GACf,kBAAkB,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,CAAC,GAAG,SAAS;IAIrD;;;;OAIG;IACH,gBAAgB,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EACjF,QAAQ,EAAE,MAAM,GACf,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAAE,YAAY,CAAC,GAAG,SAAS;IAM1D;;;;;;;;;;;;OAYG;IACG,QAAQ,CAAC,OAAO,GAAE,yBAA8C,GAAG,OAAO,CAAC,IAAI,CAAC;IAsDtF;;;;;;;;OAQG;IACH,sBAAsB,CAAC,YAAY,EAAE,YAAY,GAAG,IAAI;IAsBxD;;;;OAIG;cACa,YAAY,CAAC,QAAQ,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC;IAEhF;;;OAGG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAE9B;;;OAGG;cACa,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAGhE"}

package/dist/task/CountTokensTask.d.ts

@@ -10,9 +10,18 @@ export declare const CountTokensInputSchema: {
     readonly type: "object";
     readonly properties: {
         readonly text: {
-            readonly type: "string";
-            readonly title: "Text";
-            readonly description: "The text to count tokens for";
+            readonly anyOf: readonly [{
+                readonly type: "string";
+                readonly title: "Text";
+                readonly description: "The text to count tokens for";
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "string";
+                    readonly title: "Text";
+                    readonly description: "The text to count tokens for";
+                };
+            }];
         };
         readonly model: {
             readonly oneOf: readonly [{
@@ -79,9 +88,18 @@ export declare const CountTokensOutputSchema: {
     readonly type: "object";
     readonly properties: {
         readonly count: {
-            readonly type: "number";
-            readonly title: "Token Count";
-            readonly description: "The number of tokens in the text";
+            readonly anyOf: readonly [{
+                readonly type: "number";
+                readonly title: "Token Count";
+                readonly description: "The number of tokens in the text";
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "number";
+                    readonly title: "Token Count";
+                    readonly description: "The number of tokens in the text";
+                };
+            }];
         };
     };
     readonly required: readonly ["count"];
@@ -111,7 +129,7 @@ export declare class CountTokensTask extends AiTask<CountTokensTaskInput, CountT
  * @returns Promise resolving to the token count
  */
 export declare const countTokens: (input: CountTokensTaskInput, config?: JobQueueTaskConfig) => Promise<{
-    count: number;
+    count: number | number[];
 }>;
 declare module "@workglow/task-graph" {
     interface Workflow {

package/dist/task/CountTokensTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"CountTokensTask.d.ts","sourceRoot":"","sources":["../../src/task/CountTokensTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAYA,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;;;;;;GAMG;AACH,qBAAa,eAAgB,SAAQ,MAAM,CAAC,oBAAoB,EAAE,qBAAqB,CAAC;IACtF,OAAc,IAAI,SAAqB;IACvC,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAkB;IACrC,OAAc,WAAW,SACoD;IAC7E,OAAc,SAAS,UAAQ;WACjB,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;GAIG;AACH,eAAO,MAAM,WAAW,GAAU,OAAO,oBAAoB,EAAE,SAAS,kBAAkB;;EAEzF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,kBAAkB,CAAC,CAAC;KAC9F;CACF"}
+{"version":3,"file":"CountTokensTask.d.ts","sourceRoot":"","sources":["../../src/task/CountTokensTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAYA,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;;;;;;GAMG;AACH,qBAAa,eAAgB,SAAQ,MAAM,CAAC,oBAAoB,EAAE,qBAAqB,CAAC;IACtF,OAAc,IAAI,SAAqB;IACvC,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAkB;IACrC,OAAc,WAAW,SACoD;IAC7E,OAAc,SAAS,UAAQ;WACjB,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;GAIG;AACH,eAAO,MAAM,WAAW,GAAU,OAAO,oBAAoB,EAAE,SAAS,kBAAkB;;EAEzF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,kBAAkB,CAAC,CAAC;KAC9F;CACF"}

package/dist/task/DocumentEnricherTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"DocumentEnricherTask.d.ts","sourceRoot":"","sources":["../../src/task/DocumentEnricherTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAM5D,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAyCkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;CAyBiB,CAAC;AAEpC,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACvE,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAEzE;;;GAGG;AACH,qBAAa,oBAAqB,SAAQ,IAAI,CAC5C,yBAAyB,EACzB,0BAA0B,EAC1B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA0B;IAC5C,OAAc,QAAQ,SAAc;IACpC,OAAc,KAAK,SAAuB;IAC1C,OAAc,WAAW,SAAuD;IAChF,OAAc,SAAS,UAAQ;WAEjB,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CACX,KAAK,EAAE,yBAAyB,EAChC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,0BAA0B,CAAC;IAkDtC;;OAEG;YACW,UAAU;IA6ExB;;OAEG;YACW,SAAS;IASvB;;OAEG;YACW,eAAe;IAgE7B;;OAEG;YACW,mBAAmB;IAoBjC;;OAEG;YACW,eAAe;CAqC9B;AAED,eAAO,MAAM,gBAAgB,GAAI,OAAO,yBAAyB,EAAE,SAAS,kBAAkB;;;;;EAE7F,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,gBAAgB,EAAE,cAAc,CAC9B,yBAAyB,EACzB,0BAA0B,EAC1B,kBAAkB,CACnB,CAAC;KACH;CACF"}
+{"version":3,"file":"DocumentEnricherTask.d.ts","sourceRoot":"","sources":["../../src/task/DocumentEnricherTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAM5D,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAyCkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;CAyBiB,CAAC;AAEpC,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACvE,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAEzE;;;GAGG;AACH,qBAAa,oBAAqB,SAAQ,IAAI,CAC5C,yBAAyB,EACzB,0BAA0B,EAC1B,kBAAkB,CACnB;IACC,OAAc,IAAI,SAA0B;IAC5C,OAAc,QAAQ,SAAc;IACpC,OAAc,KAAK,SAAuB;IAC1C,OAAc,WAAW,SAAuD;IAChF,OAAc,SAAS,UAAQ;WAEjB,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CACX,KAAK,EAAE,yBAAyB,EAChC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,0BAA0B,CAAC;IAoDtC;;OAEG;YACW,UAAU;IA6ExB;;OAEG;YACW,SAAS;IASvB;;OAEG;YACW,eAAe;IAgE7B;;OAEG;YACW,mBAAmB;IAoBjC;;OAEG;YACW,eAAe;CAqC9B;AAED,eAAO,MAAM,gBAAgB,GAAI,OAAO,yBAAyB,EAAE,SAAS,kBAAkB;;;;;EAE7F,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,gBAAgB,EAAE,cAAc,CAC9B,yBAAyB,EACzB,0BAA0B,EAC1B,kBAAkB,CACnB,CAAC;KACH;CACF"}

package/dist/task/ModelInfoTask.d.ts

@@ -0,0 +1,211 @@
+/**
+ * @license
+ * Copyright 2025 Steven Roussey <sroussey@gmail.com>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { CreateWorkflow, JobQueueTaskConfig } from "@workglow/task-graph";
+import { DataPortSchema, FromSchema } from "@workglow/util";
+import { AiTask } from "./base/AiTask";
+declare const ModelInfoInputSchema: {
+    readonly type: "object";
+    readonly properties: {
+        readonly model: {
+            readonly oneOf: readonly [{
+                readonly title: "Model";
+                readonly description: `The model ${string}`;
+            } & {
+                readonly format: import("./base/AiTaskSchemas").TypeModelSemantic;
+                readonly type: "string";
+            }, {
+                readonly type: "object";
+                readonly properties: {
+                    readonly model_id: {
+                        readonly type: "string";
+                    };
+                    readonly tasks: {
+                        readonly type: "array";
+                        readonly items: {
+                            readonly type: "string";
+                        };
+                        readonly "x-ui-editor": "multiselect";
+                    };
+                    readonly title: {
+                        readonly type: "string";
+                    };
+                    readonly description: {
+                        readonly type: "string";
+                        readonly "x-ui-editor": "textarea";
+                    };
+                    readonly provider: {
+                        readonly type: "string";
+                    };
+                    readonly provider_config: {
+                        readonly type: "object";
+                        readonly properties: {
+                            readonly credential_key: {
+                                readonly type: "string";
+                                readonly format: "credential";
+                                readonly "x-ui-hidden": true;
+                            };
+                        };
+                        readonly additionalProperties: true;
+                        readonly default: {};
+                    };
+                    readonly metadata: {
+                        readonly type: "object";
+                        readonly default: {};
+                        readonly "x-ui-hidden": true;
+                    };
+                };
+                readonly required: readonly ["provider", "provider_config"];
+                readonly format: "model";
+                readonly additionalProperties: true;
+            } & {
+                readonly format: import("./base/AiTaskSchemas").TypeModelSemantic;
+            }];
+        } & {
+            readonly format: import("./base/AiTaskSchemas").TypeModelSemantic;
+        };
+        readonly detail: {
+            readonly type: "string";
+            readonly enum: readonly ["cached_status", "files", "files_with_metadata"];
+            readonly default: "files_with_metadata";
+        };
+    };
+    readonly required: readonly ["model"];
+    readonly additionalProperties: false;
+};
+declare const ModelInfoOutputSchema: {
+    readonly type: "object";
+    readonly properties: {
+        readonly model: {
+            readonly oneOf: readonly [{
+                readonly title: "Model";
+                readonly description: `The model ${string}`;
+            } & {
+                readonly format: import("./base/AiTaskSchemas").TypeModelSemantic;
+                readonly type: "string";
+            }, {
+                readonly type: "object";
+                readonly properties: {
+                    readonly model_id: {
+                        readonly type: "string";
+                    };
+                    readonly tasks: {
+                        readonly type: "array";
+                        readonly items: {
+                            readonly type: "string";
+                        };
+                        readonly "x-ui-editor": "multiselect";
+                    };
+                    readonly title: {
+                        readonly type: "string";
+                    };
+                    readonly description: {
+                        readonly type: "string";
+                        readonly "x-ui-editor": "textarea";
+                    };
+                    readonly provider: {
+                        readonly type: "string";
+                    };
+                    readonly provider_config: {
+                        readonly type: "object";
+                        readonly properties: {
+                            readonly credential_key: {
+                                readonly type: "string";
+                                readonly format: "credential";
+                                readonly "x-ui-hidden": true;
+                            };
+                        };
+                        readonly additionalProperties: true;
+                        readonly default: {};
+                    };
+                    readonly metadata: {
+                        readonly type: "object";
+                        readonly default: {};
+                        readonly "x-ui-hidden": true;
+                    };
+                };
+                readonly required: readonly ["provider", "provider_config"];
+                readonly format: "model";
+                readonly additionalProperties: true;
+            } & {
+                readonly format: import("./base/AiTaskSchemas").TypeModelSemantic;
+            }];
+        } & {
+            readonly format: import("./base/AiTaskSchemas").TypeModelSemantic;
+        };
+        readonly is_local: {
+            readonly type: "boolean";
+        };
+        readonly is_remote: {
+            readonly type: "boolean";
+        };
+        readonly supports_browser: {
+            readonly type: "boolean";
+        };
+        readonly supports_node: {
+            readonly type: "boolean";
+        };
+        readonly is_cached: {
+            readonly type: "boolean";
+        };
+        readonly is_loaded: {
+            readonly type: "boolean";
+        };
+        readonly file_sizes: {};
+    };
+    readonly required: readonly ["model", "is_local", "is_remote", "supports_browser", "supports_node", "is_cached", "is_loaded", "file_sizes"];
+    readonly additionalProperties: false;
+};
+export type ModelInfoTaskInput = FromSchema<typeof ModelInfoInputSchema>;
+export type ModelInfoTaskOutput = FromSchema<typeof ModelInfoOutputSchema>;
+/**
+ * Retrieve runtime metadata about a model: locality, browser support, cache status, and file sizes.
+ */
+export declare class ModelInfoTask extends AiTask<ModelInfoTaskInput, ModelInfoTaskOutput, JobQueueTaskConfig> {
+    static type: string;
+    static category: string;
+    static title: string;
+    static description: string;
+    static inputSchema(): DataPortSchema;
+    static outputSchema(): DataPortSchema;
+    static cacheable: boolean;
+}
+/**
+ * Retrieve runtime metadata about a model.
+ *
+ * @param input - Input containing the model to query
+ * @returns Promise resolving to model info including locality and cache status
+ */
+export declare const modelInfo: (input: ModelInfoTaskInput, config?: JobQueueTaskConfig) => Promise<{
+    model: string | {
+        [x: string]: unknown;
+        title?: string | undefined;
+        description?: string | undefined;
+        model_id?: string | undefined;
+        tasks?: string[] | undefined;
+        metadata?: {
+            [x: string]: unknown;
+        } | undefined;
+        provider: string;
+        provider_config: {
+            [x: string]: unknown;
+            credential_key?: string | undefined;
+        };
+    };
+    is_local: boolean;
+    is_remote: boolean;
+    supports_browser: boolean;
+    supports_node: boolean;
+    is_cached: boolean;
+    is_loaded: boolean;
+    file_sizes: unknown;
+}>;
+declare module "@workglow/task-graph" {
+    interface Workflow {
+        modelInfo: CreateWorkflow<ModelInfoTaskInput, ModelInfoTaskOutput, JobQueueTaskConfig>;
+    }
+}
+export {};
+//# sourceMappingURL=ModelInfoTask.d.ts.map

package/dist/task/ModelInfoTask.d.ts.map

@@ -0,0 +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"}

package/dist/task/RerankerTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"RerankerTask.d.ts","sourceRoot":"","sources":["../../src/task/RerankerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAM5D,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAkDkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAuCiB,CAAC;AAEpC,MAAM,MAAM,iBAAiB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAC/D,MAAM,MAAM,kBAAkB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AASjE;;;;;;GAMG;AACH,qBAAa,YAAa,SAAQ,IAAI,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,CAAC;IAC/F,OAAc,IAAI,SAAkB;IACpC,OAAc,QAAQ,SAAS;IAC/B,OAAc,KAAK,SAAc;IACjC,OAAc,WAAW,SAAkD;IAC3E,OAAc,SAAS,UAAQ;IAC/B,OAAO,CAAC,yBAAyB,CAAC,CAAgB;WAEpC,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CAAC,KAAK,EAAE,iBAAiB,EAAE,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC;YA4ChF,kBAAkB;IAuChC,OAAO,CAAC,wBAAwB;IAqBhC;;OAEG;IACH,OAAO,CAAC,YAAY;IAuDpB;;;OAGG;IACH,OAAO,CAAC,oBAAoB;CAqB7B;AAED,eAAO,MAAM,QAAQ,GAAI,OAAO,iBAAiB,EAAE,SAAS,kBAAkB;;;;;;;;EAE7E,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,QAAQ,EAAE,cAAc,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;KACrF;CACF"}
+{"version":3,"file":"RerankerTask.d.ts","sourceRoot":"","sources":["../../src/task/RerankerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,cAAc,EACd,eAAe,EACf,kBAAkB,EAClB,IAAI,EAEL,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAM5D,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAkDkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAuCiB,CAAC;AAEpC,MAAM,MAAM,iBAAiB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAC/D,MAAM,MAAM,kBAAkB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AASjE;;;;;;GAMG;AACH,qBAAa,YAAa,SAAQ,IAAI,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,CAAC;IAC/F,OAAc,IAAI,SAAkB;IACpC,OAAc,QAAQ,SAAS;IAC/B,OAAc,KAAK,SAAc;IACjC,OAAc,WAAW,SAAkD;IAC3E,OAAc,SAAS,UAAQ;IAC/B,OAAO,CAAC,yBAAyB,CAAC,CAAgB;WAEpC,WAAW,IAAI,cAAc;WAI7B,YAAY,IAAI,cAAc;IAItC,OAAO,CAAC,KAAK,EAAE,iBAAiB,EAAE,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC;YA4ChF,kBAAkB;IAyChC,OAAO,CAAC,wBAAwB;IAqBhC;;OAEG;IACH,OAAO,CAAC,YAAY;IAuDpB;;;OAGG;IACH,OAAO,CAAC,oBAAoB;CAqB7B;AAED,eAAO,MAAM,QAAQ,GAAI,OAAO,iBAAiB,EAAE,SAAS,kBAAkB;;;;;;;;EAE7E,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,QAAQ,EAAE,cAAc,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;KACrF;CACF"}

package/dist/task/TextClassificationTask.d.ts

@@ -10,9 +10,18 @@ export declare const TextClassificationInputSchema: {
     readonly type: "object";
     readonly properties: {
         readonly text: {
-            readonly type: "string";
-            readonly title: "Text";
-            readonly description: "The text to classify";
+            readonly anyOf: readonly [{
+                readonly type: "string";
+                readonly title: "Text";
+                readonly description: "The text to classify";
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "string";
+                    readonly title: "Text";
+                    readonly description: "The text to classify";
+                };
+            }];
         };
         readonly candidateLabels: {
             readonly type: "array";
@@ -97,26 +106,52 @@ export declare const TextClassificationOutputSchema: {
     readonly type: "object";
     readonly properties: {
         readonly categories: {
-            readonly type: "array";
-            readonly items: {
-                readonly type: "object";
-                readonly properties: {
-                    readonly label: {
-                        readonly type: "string";
-                        readonly title: "Label";
-                        readonly description: "The name of the category";
+            readonly anyOf: readonly [{
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "object";
+                    readonly properties: {
+                        readonly label: {
+                            readonly type: "string";
+                            readonly title: "Label";
+                            readonly description: "The name of the category";
+                        };
+                        readonly score: {
+                            readonly type: "number";
+                            readonly title: "Score";
+                            readonly description: "The confidence score for this category";
+                        };
                     };
-                    readonly score: {
-                        readonly type: "number";
-                        readonly title: "Score";
-                        readonly description: "The confidence score for this category";
+                    readonly required: readonly ["label", "score"];
+                    readonly additionalProperties: false;
+                };
+                readonly title: "Categories";
+                readonly description: "The classification categories with their scores";
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "array";
+                    readonly items: {
+                        readonly type: "object";
+                        readonly properties: {
+                            readonly label: {
+                                readonly type: "string";
+                                readonly title: "Label";
+                                readonly description: "The name of the category";
+                            };
+                            readonly score: {
+                                readonly type: "number";
+                                readonly title: "Score";
+                                readonly description: "The confidence score for this category";
+                            };
+                        };
+                        readonly required: readonly ["label", "score"];
+                        readonly additionalProperties: false;
                     };
+                    readonly title: "Categories";
+                    readonly description: "The classification categories with their scores";
                 };
-                readonly required: readonly ["label", "score"];
-                readonly additionalProperties: false;
-            };
-            readonly title: "Categories";
-            readonly description: "The classification categories with their scores";
+            }];
         };
     };
     readonly required: readonly ["categories"];
@@ -146,7 +181,10 @@ export declare const textClassification: (input: TextClassificationTaskInput, co
     categories: {
         score: number;
         label: string;
-    }[];
+    }[] | {
+        score: number;
+        label: string;
+    }[][];
 }>;
 declare module "@workglow/task-graph" {
     interface Workflow {

package/dist/task/TextClassificationTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TextClassificationTask.d.ts","sourceRoot":"","sources":["../../src/task/TextClassificationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,eAAO,MAAM,6BAA6B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BP,CAAC;AAEpC,eAAO,MAAM,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4BR,CAAC;AAEpC,MAAM,MAAM,2BAA2B,GAAG,UAAU,CAAC,OAAO,6BAA6B,CAAC,CAAC;AAC3F,MAAM,MAAM,4BAA4B,GAAG,UAAU,CAAC,OAAO,8BAA8B,CAAC,CAAC;AAE7F;;;GAGG;AACH,qBAAa,sBAAuB,SAAQ,MAAM,CAChD,2BAA2B,EAC3B,4BAA4B,CAC7B;IACC,OAAc,IAAI,SAA4B;IAC9C,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAqB;IACxC,OAAc,WAAW,SACwG;WACnH,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB,GAC7B,OAAO,2BAA2B,EAClC,SAAS,kBAAkB;;;;;EAG5B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,kBAAkB,EAAE,cAAc,CAChC,2BAA2B,EAC3B,4BAA4B,EAC5B,kBAAkB,CACnB,CAAC;KACH;CACF"}
+{"version":3,"file":"TextClassificationTask.d.ts","sourceRoot":"","sources":["../../src/task/TextClassificationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,eAAO,MAAM,6BAA6B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8BP,CAAC;AAEpC,eAAO,MAAM,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4BR,CAAC;AAEpC,MAAM,MAAM,2BAA2B,GAAG,UAAU,CAAC,OAAO,6BAA6B,CAAC,CAAC;AAC3F,MAAM,MAAM,4BAA4B,GAAG,UAAU,CAAC,OAAO,8BAA8B,CAAC,CAAC;AAE7F;;;GAGG;AACH,qBAAa,sBAAuB,SAAQ,MAAM,CAChD,2BAA2B,EAC3B,4BAA4B,CAC7B;IACC,OAAc,IAAI,SAA4B;IAC9C,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAqB;IACxC,OAAc,WAAW,SACwG;WACnH,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB,GAC7B,OAAO,2BAA2B,EAClC,SAAS,kBAAkB;;;;;;;;EAG5B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,kBAAkB,EAAE,cAAc,CAChC,2BAA2B,EAC3B,4BAA4B,EAC5B,kBAAkB,CACnB,CAAC;KACH;CACF"}

package/dist/task/TextFillMaskTask.d.ts

@@ -10,9 +10,18 @@ export declare const TextFillMaskInputSchema: {
     readonly type: "object";
     readonly properties: {
         readonly text: {
-            readonly type: "string";
-            readonly title: "Text";
-            readonly description: "The text with a mask token to fill";
+            readonly anyOf: readonly [{
+                readonly type: "string";
+                readonly title: "Text";
+                readonly description: "The text with a mask token to fill";
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "string";
+                    readonly title: "Text";
+                    readonly description: "The text with a mask token to fill";
+                };
+            }];
         };
         readonly model: {
             readonly oneOf: readonly [{
@@ -79,31 +88,62 @@ export declare const TextFillMaskOutputSchema: {
     readonly type: "object";
     readonly properties: {
         readonly predictions: {
-            readonly type: "array";
-            readonly items: {
-                readonly type: "object";
-                readonly properties: {
-                    readonly entity: {
-                        readonly type: "string";
-                        readonly title: "Entity";
-                        readonly description: "The token that was predicted to fill the mask";
-                    };
-                    readonly score: {
-                        readonly type: "number";
-                        readonly title: "Score";
-                        readonly description: "The confidence score for this prediction";
+            readonly anyOf: readonly [{
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "object";
+                    readonly properties: {
+                        readonly entity: {
+                            readonly type: "string";
+                            readonly title: "Entity";
+                            readonly description: "The token that was predicted to fill the mask";
+                        };
+                        readonly score: {
+                            readonly type: "number";
+                            readonly title: "Score";
+                            readonly description: "The confidence score for this prediction";
+                        };
+                        readonly sequence: {
+                            readonly type: "string";
+                            readonly title: "Sequence";
+                            readonly description: "The complete text with the mask filled";
+                        };
                     };
-                    readonly sequence: {
-                        readonly type: "string";
-                        readonly title: "Sequence";
-                        readonly description: "The complete text with the mask filled";
+                    readonly required: readonly ["entity", "score", "sequence"];
+                    readonly additionalProperties: false;
+                };
+                readonly title: "Predictions";
+                readonly description: "The predicted tokens to fill the mask with their scores and complete sequences";
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "array";
+                    readonly items: {
+                        readonly type: "object";
+                        readonly properties: {
+                            readonly entity: {
+                                readonly type: "string";
+                                readonly title: "Entity";
+                                readonly description: "The token that was predicted to fill the mask";
+                            };
+                            readonly score: {
+                                readonly type: "number";
+                                readonly title: "Score";
+                                readonly description: "The confidence score for this prediction";
+                            };
+                            readonly sequence: {
+                                readonly type: "string";
+                                readonly title: "Sequence";
+                                readonly description: "The complete text with the mask filled";
+                            };
+                        };
+                        readonly required: readonly ["entity", "score", "sequence"];
+                        readonly additionalProperties: false;
                     };
+                    readonly title: "Predictions";
+                    readonly description: "The predicted tokens to fill the mask with their scores and complete sequences";
                 };
-                readonly required: readonly ["entity", "score", "sequence"];
-                readonly additionalProperties: false;
-            };
-            readonly title: "Predictions";
-            readonly description: "The predicted tokens to fill the mask with their scores and complete sequences";
+            }];
         };
     };
     readonly required: readonly ["predictions"];
@@ -133,7 +173,11 @@ export declare const textFillMask: (input: TextFillMaskTaskInput, config?: JobQu
         score: number;
         entity: string;
         sequence: string;
-    }[];
+    }[] | {
+        score: number;
+        entity: string;
+        sequence: string;
+    }[][];
 }>;
 declare module "@workglow/task-graph" {
     interface Workflow {

package/dist/task/TextFillMaskTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TextFillMaskTask.d.ts","sourceRoot":"","sources":["../../src/task/TextFillMaskTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAYD,CAAC;AAEpC,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAiCF,CAAC;AAEpC,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAC/E,MAAM,MAAM,sBAAsB,GAAG,UAAU,CAAC,OAAO,wBAAwB,CAAC,CAAC;AAEjF;;GAEG;AACH,qBAAa,gBAAiB,SAAQ,MAAM,CAAC,qBAAqB,EAAE,sBAAsB,CAAC;IACzF,OAAc,IAAI,SAAsB;IACxC,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAe;IAClC,OAAc,WAAW,SAAiC;WAC5C,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,YAAY,GAAI,OAAO,qBAAqB,EAAE,SAAS,kBAAkB;;;;;;EAErF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,YAAY,EAAE,cAAc,CAAC,qBAAqB,EAAE,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;KACjG;CACF"}
+{"version":3,"file":"TextFillMaskTask.d.ts","sourceRoot":"","sources":["../../src/task/TextFillMaskTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAYD,CAAC;AAEpC,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAiCF,CAAC;AAEpC,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAC/E,MAAM,MAAM,sBAAsB,GAAG,UAAU,CAAC,OAAO,wBAAwB,CAAC,CAAC;AAEjF;;GAEG;AACH,qBAAa,gBAAiB,SAAQ,MAAM,CAAC,qBAAqB,EAAE,sBAAsB,CAAC;IACzF,OAAc,IAAI,SAAsB;IACxC,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAe;IAClC,OAAc,WAAW,SAAiC;WAC5C,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;;;;GAKG;AACH,eAAO,MAAM,YAAY,GAAI,OAAO,qBAAqB,EAAE,SAAS,kBAAkB;;;;;;;;;;EAErF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,YAAY,EAAE,cAAc,CAAC,qBAAqB,EAAE,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;KACjG;CACF"}

package/dist/task/TextGenerationTask.d.ts

@@ -67,9 +67,18 @@ export declare const TextGenerationInputSchema: {
             readonly format: import(".").TypeModelSemantic;
         };
         readonly prompt: {
-            readonly type: "string";
-            readonly title: "Prompt";
-            readonly description: "The prompt to generate text from";
+            readonly anyOf: readonly [{
+                readonly type: "string";
+                readonly title: "Prompt";
+                readonly description: "The prompt to generate text from";
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "string";
+                    readonly title: "Prompt";
+                    readonly description: "The prompt to generate text from";
+                };
+            }];
         };
         readonly maxTokens: {
             readonly type: "number";
@@ -119,10 +128,20 @@ export declare const TextGenerationOutputSchema: {
     readonly type: "object";
     readonly properties: {
         readonly text: {
-            readonly type: "string";
-            readonly title: "Text";
-            readonly description: "The generated text";
-            readonly "x-stream": "append";
+            readonly anyOf: readonly [{
+                readonly type: "string";
+                readonly title: "Text";
+                readonly description: "The generated text";
+                readonly "x-stream": "append";
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    readonly type: "string";
+                    readonly title: "Text";
+                    readonly description: "The generated text";
+                    readonly "x-stream": "append";
+                };
+            }];
         };
     };
     readonly required: readonly ["text"];
@@ -142,7 +161,7 @@ export declare class TextGenerationTask extends StreamingAiTask<TextGenerationTa
  * Task for generating text using a language model
  */
 export declare const textGeneration: (input: TextGenerationTaskInput, config?: JobQueueTaskConfig) => Promise<{
-    text: string;
+    text: string | string[];
 }>;
 declare module "@workglow/task-graph" {
     interface Workflow {

package/dist/task/TextGenerationTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TextGenerationTask.d.ts","sourceRoot":"","sources":["../../src/task/TextGenerationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAE5D,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAWzD,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAoDH,CAAC;AAEpC,eAAO,MAAM,0BAA0B;;;;;;;;;;;;CAOJ,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,yBAAyB,CAAC,CAAC;AACnF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAErF,qBAAa,kBAAmB,SAAQ,eAAe,CACrD,uBAAuB,EACvB,wBAAwB,EACxB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAwB;IAC1C,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAqB;IACxC,OAAc,WAAW,SAC2D;WACtE,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;GAEG;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"}
+{"version":3,"file":"TextGenerationTask.d.ts","sourceRoot":"","sources":["../../src/task/TextGenerationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAY,MAAM,sBAAsB,CAAC;AACpF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAE5D,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAWzD,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAoDH,CAAC;AAEpC,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;CAOJ,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,yBAAyB,CAAC,CAAC;AACnF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAErF,qBAAa,kBAAmB,SAAQ,eAAe,CACrD,uBAAuB,EACvB,wBAAwB,EACxB,kBAAkB,CACnB;IACC,OAAc,IAAI,SAAwB;IAC1C,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAqB;IACxC,OAAc,WAAW,SAC2D;WACtE,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAED;;GAEG;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"}