@workglow/ai 0.1.1 → 0.2.0

package/dist/task/TextTranslationTask.d.ts

@@ -3,7 +3,8 @@
  * Copyright 2025 Steven Roussey <sroussey@gmail.com>
  * SPDX-License-Identifier: Apache-2.0
  */
-import { CreateWorkflow, TaskConfig } from "@workglow/task-graph";
+import { CreateWorkflow } from "@workglow/task-graph";
+import type { TaskConfig } from "@workglow/task-graph";
 import { DataPortSchema, FromSchema } from "@workglow/util/schema";
 import { StreamingAiTask } from "./base/StreamingAiTask";
 export declare const TextTranslationInputSchema: {
@@ -111,10 +112,11 @@ export declare const TextTranslationOutputSchema: {
 };
 export type TextTranslationTaskInput = FromSchema<typeof TextTranslationInputSchema>;
 export type TextTranslationTaskOutput = FromSchema<typeof TextTranslationOutputSchema>;
+export type TextTranslationTaskConfig = TaskConfig<TextTranslationTaskInput>;
 /**
  * This translates text from one language to another
  */
-export declare class TextTranslationTask extends StreamingAiTask<TextTranslationTaskInput, TextTranslationTaskOutput> {
+export declare class TextTranslationTask extends StreamingAiTask<TextTranslationTaskInput, TextTranslationTaskOutput, TextTranslationTaskConfig> {
     static type: string;
     static category: string;
     static title: string;
@@ -128,32 +130,13 @@ export declare class TextTranslationTask extends StreamingAiTask<TextTranslation
  * @param input The input parameters for text translation (text, model, source_lang, and target_lang)
  * @returns Promise resolving to the translated text output(s)
  */
-export declare const textTranslation: (input: {
-    model: string | {
-        [x: string]: unknown;
-        description?: string | undefined;
-        metadata?: {
-            [x: string]: unknown;
-        } | undefined;
-        model_id?: string | undefined;
-        provider: string;
-        provider_config: {
-            [x: string]: unknown;
-            credential_key?: string | undefined;
-        };
-        tasks?: string[] | undefined;
-        title?: string | undefined;
-    };
-    source_lang: string;
-    target_lang: string;
-    text: string;
-}, config?: TaskConfig | undefined) => Promise<{
+export declare const textTranslation: (input: TextTranslationTaskInput, config?: TextTranslationTaskConfig) => Promise<{
     target_lang: string;
     text: string;
 }>;
 declare module "@workglow/task-graph" {
     interface Workflow {
-        textTranslation: CreateWorkflow<TextTranslationTaskInput, TextTranslationTaskOutput, TaskConfig>;
+        textTranslation: CreateWorkflow<TextTranslationTaskInput, TextTranslationTaskOutput, TextTranslationTaskConfig>;
     }
 }
 //# sourceMappingURL=TextTranslationTask.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"TextTranslationTask.d.ts","sourceRoot":"","sources":["../../src/task/TextTranslationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,UAAU,EAAY,MAAM,sBAAsB,CAAC;AAC5E,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAWzD,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAwBJ,CAAC;AAEpC,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;;;;;;;CAaL,CAAC;AAEpC,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AACrF,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,OAAO,2BAA2B,CAAC,CAAC;AAEvF;;GAEG;AACH,qBAAa,mBAAoB,SAAQ,eAAe,CACtD,wBAAwB,EACxB,yBAAyB,CAC1B;IACC,OAAuB,IAAI,SAAyB;IACpD,OAAuB,QAAQ,SAAmB;IAClD,OAAuB,KAAK,SAAsB;IAClD,OAAuB,WAAW,SAAwE;IAC1G,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;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,UAAU,CACX,CAAC;KACH;CACF"}
+{"version":3,"file":"TextTranslationTask.d.ts","sourceRoot":"","sources":["../../src/task/TextTranslationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAWzD,eAAO,MAAM,0BAA0B;mBAC/B,QAAQ;;iBAEZ,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,uBAAuB;;iBAEtC,WAAW;;;;;;;iBAMX,WAAW;;;;;;;iBAMX,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAI0B,CAAC;AAEpC,eAAO,MAAM,2BAA2B;mBAChC,QAAQ;;iBAEZ,IAAI;2BAnCA,QAAQ;4BACP,MAAM;kCACA,qBAAqB;iCACtB,SAAS;;iBAiCnB,WAAW;;;;;;;;;;CASoB,CAAC;AAEpC,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AACrF,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,OAAO,2BAA2B,CAAC,CAAC;AACvF,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,wBAAwB,CAAC,CAAC;AAE7E;;GAEG;AACH,qBAAa,mBAAoB,SAAQ,eAAe,CACtD,wBAAwB,EACxB,yBAAyB,EACzB,yBAAyB,CAC1B;IACC,OAAuB,IAAI,SAAyB;IACpD,OAAuB,QAAQ,SAAmB;IAClD,OAAuB,KAAK,SAAsB;IAClD,OAAuB,WAAW,SACqC;IACvE,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,eAAe,UACnB,wBAAwB,WACtB,yBAAyB;;;EAGnC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,eAAe,EAAE,cAAc,CAC7B,wBAAwB,EACxB,yBAAyB,EACzB,yBAAyB,CAC1B,CAAC;KACH;CACF"}

package/dist/task/ToolCallingTask.d.ts

@@ -0,0 +1,348 @@
+/**
+ * @license
+ * Copyright 2026 Steven Roussey <sroussey@gmail.com>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { CreateWorkflow } from "@workglow/task-graph";
+import type { TaskConfig } from "@workglow/task-graph";
+import { DataPortSchema, FromSchema } from "@workglow/util/schema";
+import { ServiceRegistry } from "@workglow/util";
+import { StreamingAiTask } from "./base/StreamingAiTask";
+import type { ToolDefinition } from "./ToolCallingUtils";
+export interface ToolDefinitionWithTaskType extends ToolDefinition {
+    /** The task type name this definition was generated from. */
+    readonly taskType: string;
+}
+/**
+ * Converts an allow-list of task type names into {@link ToolDefinitionWithTaskType} objects
+ * suitable for the ToolCallingTask input. Each entry carries the originating
+ * `taskType` so callers don't need to rely on index correspondence.
+ *
+ * Each task's `type`, `description`, `inputSchema()`, and `outputSchema()`
+ * are used to build the tool definition.
+ *
+ * @param taskNames - Array of task type names registered in the task constructors
+ * @param registry - Optional service registry for DI-based lookups
+ * @returns Array of ToolDefinitionWithTaskType objects
+ * @throws Error if a task name is not found in the registry
+ */
+export declare function taskTypesToTools(taskNames: ReadonlyArray<string>, registry?: ServiceRegistry): ToolDefinitionWithTaskType[];
+export declare const ToolDefinitionSchema: {
+    readonly type: "object";
+    readonly properties: {
+        readonly name: {
+            readonly type: "string";
+            readonly title: "Name";
+            readonly description: "The tool name";
+        };
+        readonly description: {
+            readonly type: "string";
+            readonly title: "Description";
+            readonly description: "A description of what the tool does";
+        };
+        readonly inputSchema: {
+            readonly type: "object";
+            readonly title: "Input Schema";
+            readonly description: "JSON Schema describing the tool's input parameters";
+            readonly additionalProperties: true;
+        };
+        readonly outputSchema: {
+            readonly type: "object";
+            readonly title: "Output Schema";
+            readonly description: "JSON Schema describing what the tool returns";
+            readonly additionalProperties: true;
+        };
+        readonly configSchema: {
+            readonly type: "object";
+            readonly title: "Config Schema";
+            readonly description: "JSON Schema describing the task's configuration options (not sent to the LLM)";
+            readonly additionalProperties: true;
+        };
+        readonly config: {
+            readonly type: "object";
+            readonly title: "Config";
+            readonly description: "Concrete configuration values for the backing task (not sent to the LLM)";
+            readonly additionalProperties: true;
+        };
+    };
+    readonly required: readonly ["name", "description", "inputSchema"];
+    readonly additionalProperties: true;
+};
+export declare const ToolCallingInputSchema: {
+    readonly type: "object";
+    readonly properties: {
+        readonly model: {
+            readonly oneOf: readonly [{
+                readonly title: "Model";
+                readonly description: `The model ${string}`;
+            } & {
+                readonly format: import(".").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(".").TypeModelSemantic;
+            }];
+        } & {
+            readonly format: import(".").TypeModelSemantic;
+        };
+        readonly prompt: {
+            readonly oneOf: readonly [{
+                readonly type: "string";
+                readonly title: "Prompt";
+                readonly description: "The prompt to send to the model";
+            }, {
+                readonly type: "array";
+                readonly title: "Prompt";
+                readonly description: "The prompt as an array of strings or content blocks";
+                readonly items: {
+                    readonly oneOf: readonly [{
+                        readonly type: "string";
+                    }, {
+                        readonly type: "object";
+                        readonly properties: {
+                            readonly type: {
+                                readonly type: "string";
+                                readonly enum: readonly ["text", "image", "audio"];
+                            };
+                        };
+                        readonly required: readonly ["type"];
+                        readonly additionalProperties: true;
+                    }];
+                };
+            }];
+            readonly title: "Prompt";
+            readonly description: "The prompt to send to the model";
+        };
+        readonly systemPrompt: {
+            readonly type: "string";
+            readonly title: "System Prompt";
+            readonly description: "Optional system instructions for the model";
+        };
+        readonly messages: {
+            readonly type: "array";
+            readonly title: "Messages";
+            readonly description: "Full conversation history for multi-turn interactions. When provided, used instead of prompt to construct the messages array sent to the provider.";
+            readonly items: {
+                readonly type: "object";
+                readonly properties: {
+                    readonly role: {
+                        readonly type: "string";
+                        readonly enum: readonly ["user", "assistant", "tool"];
+                    };
+                    readonly content: {};
+                };
+                readonly required: readonly ["role", "content"];
+                readonly additionalProperties: true;
+            };
+        };
+        readonly tools: {
+            readonly type: "array";
+            readonly format: "tasks";
+            readonly title: "Tools";
+            readonly description: "Tool definitions available for the model to call";
+            readonly items: {
+                readonly oneOf: readonly [{
+                    readonly type: "string";
+                    readonly format: "tasks";
+                    readonly description: "Task type name";
+                }, {
+                    readonly type: "object";
+                    readonly properties: {
+                        readonly name: {
+                            readonly type: "string";
+                            readonly title: "Name";
+                            readonly description: "The tool name";
+                        };
+                        readonly description: {
+                            readonly type: "string";
+                            readonly title: "Description";
+                            readonly description: "A description of what the tool does";
+                        };
+                        readonly inputSchema: {
+                            readonly type: "object";
+                            readonly title: "Input Schema";
+                            readonly description: "JSON Schema describing the tool's input parameters";
+                            readonly additionalProperties: true;
+                        };
+                        readonly outputSchema: {
+                            readonly type: "object";
+                            readonly title: "Output Schema";
+                            readonly description: "JSON Schema describing what the tool returns";
+                            readonly additionalProperties: true;
+                        };
+                        readonly configSchema: {
+                            readonly type: "object";
+                            readonly title: "Config Schema";
+                            readonly description: "JSON Schema describing the task's configuration options (not sent to the LLM)";
+                            readonly additionalProperties: true;
+                        };
+                        readonly config: {
+                            readonly type: "object";
+                            readonly title: "Config";
+                            readonly description: "Concrete configuration values for the backing task (not sent to the LLM)";
+                            readonly additionalProperties: true;
+                        };
+                    };
+                    readonly required: readonly ["name", "description", "inputSchema"];
+                    readonly additionalProperties: true;
+                }];
+            };
+        };
+        readonly toolChoice: {
+            readonly type: "string";
+            readonly title: "Tool Choice";
+            readonly description: 'Controls tool selection: "auto" (model decides), "none" (no tools), "required" (must call a tool), or a specific tool name';
+            readonly "x-ui-group": "Configuration";
+        };
+        readonly maxTokens: {
+            readonly type: "number";
+            readonly title: "Max Tokens";
+            readonly description: "The maximum number of tokens to generate";
+            readonly minimum: 1;
+            readonly "x-ui-group": "Configuration";
+        };
+        readonly temperature: {
+            readonly type: "number";
+            readonly title: "Temperature";
+            readonly description: "The temperature to use for sampling";
+            readonly minimum: 0;
+            readonly maximum: 2;
+            readonly "x-ui-group": "Configuration";
+        };
+    };
+    readonly required: readonly ["model", "prompt", "tools"];
+    readonly additionalProperties: false;
+};
+export declare const ToolCallingOutputSchema: {
+    readonly type: "object";
+    readonly properties: {
+        readonly text: {
+            readonly type: "string";
+            readonly title: "Text";
+            readonly description: "Any text content generated by the model";
+            readonly "x-stream": "append";
+        };
+        readonly toolCalls: {
+            readonly type: "array";
+            readonly items: {
+                readonly type: "object";
+                readonly properties: {
+                    readonly id: {
+                        readonly type: "string";
+                        readonly title: "ID";
+                        readonly description: "Unique identifier for this tool call";
+                    };
+                    readonly name: {
+                        readonly type: "string";
+                        readonly title: "Name";
+                        readonly description: "The name of the tool to invoke";
+                    };
+                    readonly input: {
+                        readonly type: "object";
+                        readonly title: "Input";
+                        readonly description: "The input arguments for the tool call";
+                        readonly additionalProperties: true;
+                    };
+                };
+                readonly required: readonly ["id", "name", "input"];
+                readonly additionalProperties: false;
+            };
+            readonly title: "Tool Calls";
+            readonly description: "Tool calls requested by the model";
+            readonly "x-stream": "object";
+        };
+    };
+    readonly required: readonly ["text", "toolCalls"];
+    readonly additionalProperties: false;
+};
+/**
+ * Runtime input type for ToolCallingTask.
+ *
+ * The schema uses `oneOf: [string, object]` so the UI can accept both task-name
+ * references and inline tool definitions, but the input resolver converts all
+ * strings to {@link ToolDefinition} objects before execution. The `tools` field
+ * is therefore narrowed to `ToolDefinition[]` here.
+ *
+ * Extends the schema-derived base with the
+ * `messages` field typed explicitly (the loose `content: {}` in the
+ * schema prevents `FromSchema` from producing a useful type).
+ */
+export type ToolCallingTaskInput = Omit<FromSchema<typeof ToolCallingInputSchema>, "tools"> & {
+    readonly tools: ToolDefinition[];
+    readonly messages?: ReadonlyArray<{
+        readonly role: "user" | "assistant" | "tool";
+        readonly content: unknown;
+    }>;
+};
+export type ToolCallingTaskOutput = FromSchema<typeof ToolCallingOutputSchema>;
+export type ToolCallingTaskConfig = TaskConfig<ToolCallingTaskInput>;
+export declare class ToolCallingTask extends StreamingAiTask<ToolCallingTaskInput, ToolCallingTaskOutput, ToolCallingTaskConfig> {
+    static type: string;
+    static category: string;
+    static title: string;
+    static description: string;
+    static inputSchema(): DataPortSchema;
+    static outputSchema(): DataPortSchema;
+}
+/**
+ * Convenience function to run a tool calling task.
+ */
+export declare const toolCalling: (input: ToolCallingTaskInput, config?: ToolCallingTaskConfig) => Promise<{
+    text: string;
+    toolCalls: {
+        id: string;
+        input: {
+            [x: string]: unknown;
+        };
+        name: string;
+    }[];
+}>;
+declare module "@workglow/task-graph" {
+    interface Workflow {
+        toolCalling: CreateWorkflow<ToolCallingTaskInput, ToolCallingTaskOutput, ToolCallingTaskConfig>;
+    }
+}
+//# sourceMappingURL=ToolCallingTask.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ToolCallingTask.d.ts","sourceRoot":"","sources":["../../src/task/ToolCallingTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAiC,MAAM,sBAAsB,CAAC;AAErF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEjD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAMzD,MAAM,WAAW,0BAA2B,SAAQ,cAAc;IAChE,6DAA6D;IAC7D,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;CAC3B;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,aAAa,CAAC,MAAM,CAAC,EAChC,QAAQ,CAAC,EAAE,eAAe,GACzB,0BAA0B,EAAE,CAsB9B;AAMD,eAAO,MAAM,oBAAoB;aAC/B,IAAI,EAAE,QAAQ;aACd,UAAU;iBACR,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,eAAe;;iBAE9B,WAAW;qBACT,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,qCAAqC;;iBAEpD,WAAW;qBACT,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,cAAc;qBACrB,WAAW,EAAE,oDAAoD;qBACjE,oBAAoB;;iBAEtB,YAAY;qBACV,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,8CAA8C;qBAC3D,oBAAoB;;iBAEtB,YAAY;qBACV,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,+EAA+E;qBAC5F,oBAAoB;;iBAEtB,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,QAAQ;qBACf,WAAW,EAAE,0EAA0E;qBACvF,oBAAoB;;;aAGxB,QAAQ,YAAG,MAAM,EAAE,aAAa,EAAE,aAAa;aAC/C,oBAAoB;CACZ,CAAC;AA4BX,eAAO,MAAM,sBAAsB;mBAC3B,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;+BAKA,QAAQ;gCAAS,QAAQ;sCAAe,iCAAiC;;+BAEzE,OAAO;gCACN,QAAQ;sCACF,qDAAqD;;;uCAGtD,QAAQ;;uCAER,QAAQ;;qCAEZ,IAAI;yCAAI,IAAI,EAAE,QAAQ;yCAAE,IAAI;;;;;;;;4BASjC,QAAQ;kCACF,iCAAiC;;;qBAG9C,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,4CAA4C;;;qBAGzD,IAAI,EAAE,OAAO;qBACb,KAAK,EAAE,UAAU;qBACjB,WAAW,EACT,oJAAoJ;qBACtJ,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,UAAU;6BACR,IAAI;iCAAI,IAAI,EAAE,QAAQ;iCAAE,IAAI;;6BAC5B,OAAO;;yBAET,QAAQ;yBACR,oBAAoB;;;;2BAIhB,OAAO;6BACL,OAAO;4BACR,OAAO;kCACD,kDAAkD;;;mCAGnD,QAAQ;qCAAU,OAAO;0CAAe,gBAAgB;;mCA3HlE,QAAQ;;iCAEZ,IAAI;qCACF,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,MAAM;qCACb,WAAW,EAAE,eAAe;;iCAE9B,WAAW;qCACT,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,aAAa;qCACpB,WAAW,EAAE,qCAAqC;;iCAEpD,WAAW;qCACT,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,cAAc;qCACrB,WAAW,EAAE,oDAAoD;qCACjE,oBAAoB;;iCAEtB,YAAY;qCACV,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,eAAe;qCACtB,WAAW,EAAE,8CAA8C;qCAC3D,oBAAoB;;iCAEtB,YAAY;qCACV,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,eAAe;qCACtB,WAAW,EAAE,+EAA+E;qCAC5F,oBAAoB;;iCAEtB,MAAM;qCACJ,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,QAAQ;qCACf,WAAW,EAAE,0EAA0E;qCACvF,oBAAoB;;;iDAGb,MAAM,EAAE,aAAa,EAAE,aAAa;;;;;;qBA4F3C,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EACT,4HAA4H;qBAC9H,YAAY,EAAE,eAAe;;;qBAG7B,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,0CAA0C;qBACvD,OAAO,EAAE,CAAC;qBACV,YAAY,EAAE,eAAe;;;qBAG7B,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,qCAAqC;qBAClD,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,YAAY,EAAE,eAAe;;;;;CAKA,CAAC;AAEpC,eAAO,MAAM,uBAAuB;mBAC5B,QAAQ;;iBAEZ,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,yCAAyC;qBACtD,UAAU,EAAE,QAAQ;;iBAEtB,SAAS;qBACP,IAAI,EAAE,OAAO;qBACb,KAAK;+BA5HH,QAAQ;;6BAEZ,EAAE;iCACA,IAAI,EAAE,QAAQ;iCACd,KAAK,EAAE,IAAI;iCACX,WAAW,EAAE,sCAAsC;;6BAErD,IAAI;iCACF,IAAI,EAAE,QAAQ;iCACd,KAAK,EAAE,MAAM;iCACb,WAAW,EAAE,gCAAgC;;6BAE/C,KAAK;iCACH,IAAI,EAAE,QAAQ;iCACd,KAAK,EAAE,OAAO;iCACd,WAAW,EAAE,uCAAuC;iCACpD,oBAAoB;;;6CAGb,IAAI,EAAE,MAAM,EAAE,OAAO;;;qBA0G5B,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,mCAAmC;qBAChD,UAAU,EAAE,QAAQ;;;;;CAKS,CAAC;AAEpC;;;;;;;;;;;GAWG;AACH,MAAM,MAAM,oBAAoB,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,EAAE,OAAO,CAAC,GAAG;IAC5F,QAAQ,CAAC,KAAK,EAAE,cAAc,EAAE,CAAC;IACjC,QAAQ,CAAC,QAAQ,CAAC,EAAE,aAAa,CAAC;QAChC,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;QAC7C,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;KAC3B,CAAC,CAAC;CACJ,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAC/E,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAMrE,qBAAa,eAAgB,SAAQ,eAAe,CAClD,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,CACtB;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAmB;IAClD,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SACyF;IAC3H,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;CACF;AAED;;GAEG;AACH,eAAO,MAAM,WAAW,UAAW,oBAAoB,WAAW,qBAAqB;;;;;;;;;EAEtF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,qBAAqB,CAAC,CAAC;KACjG;CACF"}

package/dist/task/ToolCallingUtils.d.ts

@@ -0,0 +1,75 @@
+/**
+ * @license
+ * Copyright 2026 Steven Roussey <sroussey@gmail.com>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { JsonSchema } from "@workglow/util/worker";
+/**
+ * A tool definition that can be passed to an LLM for tool calling.
+ * Can be created manually or generated from TaskRegistry entries via `taskTypesToTools` in ToolCallingTask.
+ *
+ * The `name` is used both as the tool name presented to the LLM and as a
+ * lookup key for the backing Task in the TaskRegistry. When a tool is
+ * backed by a configurable task (e.g. `McpToolCallTask`, `JavaScriptTask`),
+ * `configSchema` describes what configuration the task accepts and `config`
+ * provides the concrete values. The LLM never sees `configSchema` or
+ * `config` — they are setup-time concerns used when instantiating the task.
+ */
+export interface ToolDefinition {
+    name: string;
+    description: string;
+    inputSchema: JsonSchema;
+    outputSchema?: JsonSchema;
+    /**
+     * Optional discriminator for tool resolution. When specified, skips
+     * duck-typing heuristics:
+     * - `"function"` — uses the `execute` function directly
+     * - `"task"` — looks up the task by `name` in the TaskRegistry
+     *
+     * When omitted, resolution falls back to the existing heuristic
+     * (check `execute`, then registry lookup, then stub).
+     */
+    type?: "function" | "task";
+    /** JSON Schema describing the task's configuration options. */
+    configSchema?: JsonSchema;
+    /** Concrete configuration values matching {@link configSchema}. */
+    config?: Record<string, unknown>;
+    /**
+     * Optional custom executor function. When provided, the tool is executed
+     * by calling this function directly instead of instantiating a Task.
+     */
+    execute?: (input: Record<string, unknown>) => Promise<Record<string, unknown>>;
+}
+/**
+ * A tool call returned by the LLM, requesting invocation of a specific tool.
+ */
+export interface ToolCall {
+    id: string;
+    name: string;
+    input: Record<string, unknown>;
+}
+export type ToolCalls = Array<ToolCall>;
+/**
+ * Controls which tools the model may call.
+ * - `"auto"` — model decides whether to call tools
+ * - `"none"` — model must not call any tools
+ * - `"required"` — model must call at least one tool
+ * - any other string — model must call the tool with that name
+ */
+export type ToolChoiceOption = "auto" | "none" | "required" | (string & {});
+/**
+ * Builds a tool description string for provider APIs, appending the output
+ * schema when present. Shared across all provider implementations.
+ */
+export declare function buildToolDescription(tool: ToolDefinition): string;
+/**
+ * Validates that a tool call name returned by the LLM matches one of the
+ * allowed tool definitions. Returns true if valid, false otherwise.
+ */
+export declare function isAllowedToolName(name: string, allowedTools: ReadonlyArray<ToolDefinition>): boolean;
+/**
+ * Filters an array of tool calls, removing any whose name does not appear
+ * in the provided tools list. Returns the filtered array.
+ */
+export declare function filterValidToolCalls(toolCalls: ToolCalls, allowedTools: ReadonlyArray<ToolDefinition>): ToolCalls;
+//# sourceMappingURL=ToolCallingUtils.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ToolCallingUtils.d.ts","sourceRoot":"","sources":["../../src/task/ToolCallingUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAExD;;;;;;;;;;GAUG;AACH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,UAAU,CAAC;IACxB,YAAY,CAAC,EAAE,UAAU,CAAC;IAC1B;;;;;;;;OAQG;IACH,IAAI,CAAC,EAAE,UAAU,GAAG,MAAM,CAAC;IAC3B,+DAA+D;IAC/D,YAAY,CAAC,EAAE,UAAU,CAAC;IAC1B,mEAAmE;IACnE,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC;;;OAGG;IACH,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;CAChF;AAED;;GAEG;AACH,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAED,MAAM,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAC;AAExC;;;;;;GAMG;AACH,MAAM,MAAM,gBAAgB,GAAG,MAAM,GAAG,MAAM,GAAG,UAAU,GAAG,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;AAE5E;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,cAAc,GAAG,MAAM,CAMjE;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAC/B,IAAI,EAAE,MAAM,EACZ,YAAY,EAAE,aAAa,CAAC,cAAc,CAAC,GAC1C,OAAO,CAET;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAClC,SAAS,EAAE,SAAS,EACpB,YAAY,EAAE,aAAa,CAAC,cAAc,CAAC,GAC1C,SAAS,CAWX"}

package/dist/task/TopicSegmenterTask.d.ts

@@ -3,7 +3,8 @@
  * Copyright 2025 Steven Roussey <sroussey@gmail.com>
  * SPDX-License-Identifier: Apache-2.0
  */
-import { CreateWorkflow, IExecuteContext, TaskConfig, Task } from "@workglow/task-graph";
+import { CreateWorkflow, IExecuteContext, Task } from "@workglow/task-graph";
+import type { TaskConfig } from "@workglow/task-graph";
 import { DataPortSchema, FromSchema } from "@workglow/util/schema";
 export declare const SegmentationMethod: {
     readonly HEURISTIC: "heuristic";
@@ -87,11 +88,12 @@ declare const outputSchema: {
 };
 export type TopicSegmenterTaskInput = FromSchema<typeof inputSchema>;
 export type TopicSegmenterTaskOutput = FromSchema<typeof outputSchema>;
+export type TopicSegmenterTaskConfig = TaskConfig<TopicSegmenterTaskInput>;
 /**
  * Task for segmenting text into topic-based sections
  * Uses hybrid approach: heuristics + optional embedding similarity
  */
-export declare class TopicSegmenterTask extends Task<TopicSegmenterTaskInput, TopicSegmenterTaskOutput, TaskConfig> {
+export declare class TopicSegmenterTask extends Task<TopicSegmenterTaskInput, TopicSegmenterTaskOutput, TopicSegmenterTaskConfig> {
     static type: string;
     static category: string;
     static title: string;
@@ -131,13 +133,7 @@ export declare class TopicSegmenterTask extends Task<TopicSegmenterTaskInput, To
      */
     private mergeSmallSegments;
 }
-export declare const topicSegmenter: (input: {
-    maxSegmentSize?: number | undefined;
-    method?: "embedding-similarity" | "heuristic" | "hybrid" | undefined;
-    minSegmentSize?: number | undefined;
-    similarityThreshold?: number | undefined;
-    text: string;
-}, config?: TaskConfig | undefined) => Promise<{
+export declare const topicSegmenter: (input: TopicSegmenterTaskInput, config?: TopicSegmenterTaskConfig) => Promise<{
     count: number;
     segments: {
         endOffset: number;
@@ -147,7 +143,7 @@ export declare const topicSegmenter: (input: {
 }>;
 declare module "@workglow/task-graph" {
     interface Workflow {
-        topicSegmenter: CreateWorkflow<TopicSegmenterTaskInput, TopicSegmenterTaskOutput, TaskConfig>;
+        topicSegmenter: CreateWorkflow<TopicSegmenterTaskInput, TopicSegmenterTaskOutput, TopicSegmenterTaskConfig>;
     }
 }
 export {};

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

@@ -1 +1 @@
-{"version":3,"file":"TopicSegmenterTask.d.ts","sourceRoot":"","sources":["../../src/task/TopicSegmenterTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,UAAU,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AACnG,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,eAAO,MAAM,kBAAkB;;;;CAIrB,CAAC;AAEX,MAAM,MAAM,kBAAkB,GAAG,CAAC,OAAO,kBAAkB,CAAC,CAAC,MAAM,OAAO,kBAAkB,CAAC,CAAC;AAE9F,QAAA,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAwCkB,CAAC;AAEpC,QAAA,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA0BiB,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACrE,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAEvE;;;GAGG;AACH,qBAAa,kBAAmB,SAAQ,IAAI,CAC1C,uBAAuB,EACvB,wBAAwB,EACxB,UAAU,CACX;IACC,OAAuB,IAAI,SAAwB;IACnD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAqB;IACjD,OAAuB,WAAW,SAAkE;IACpG,OAAuB,SAAS,UAAQ;IACxC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAO;IAEnD,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,uBAAuB,EAC9B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,wBAAwB,CAAC,CAmCnC;IAED;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAuD7B;;OAEG;IACH,OAAO,CAAC,qBAAqB;IA0D7B;;OAEG;IACH,OAAO,CAAC,cAAc;IA0BtB,OAAO,CAAC,yBAAyB;IAkBjC;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAiB3B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAU3B;;OAEG;IACH,OAAO,CAAC,aAAa;IAarB;;OAEG;IACH,OAAO,CAAC,kBAAkB;CAyC3B;AAED,eAAO,MAAM,cAAc;;;;;;;;;;;;;EAE1B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,cAAc,EAAE,cAAc,CAAC,uBAAuB,EAAE,wBAAwB,EAAE,UAAU,CAAC,CAAC;KAC/F;CACF"}
+{"version":3,"file":"TopicSegmenterTask.d.ts","sourceRoot":"","sources":["../../src/task/TopicSegmenterTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AAEvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,eAAO,MAAM,kBAAkB;aAC7B,SAAS,EAAE,WAAW;aACtB,oBAAoB,EAAE,sBAAsB;aAC5C,MAAM,EAAE,QAAQ;CACR,CAAC;AAEX,MAAM,MAAM,kBAAkB,GAAG,CAAC,OAAO,kBAAkB,CAAC,CAAC,MAAM,OAAO,kBAAkB,CAAC,CAAC;AAE9F,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,iCAAiC;;iBAEhD,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,qBAAqB;qBAC5B,WAAW,EAAE,sCAAsC;qBACnD,OAAO;;iBAET,cAAc;qBACZ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,oCAAoC;qBACjD,OAAO,EAAE,EAAE;qBACX,OAAO,EAAE,GAAG;;iBAEd,cAAc;qBACZ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,oCAAoC;qBACjD,OAAO,EAAE,GAAG;qBACZ,OAAO,EAAE,IAAI;;iBAEf,mBAAmB;qBACjB,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,sBAAsB;qBAC7B,WAAW,EAAE,+DAA+D;qBAC5E,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;;;;;CAKiB,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;iBAEZ,QAAQ;qBACN,IAAI,EAAE,OAAO;qBACb,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,UAAU;6BACR,IAAI;iCAAI,IAAI,EAAE,QAAQ;;6BACtB,WAAW;iCAAI,IAAI,EAAE,QAAQ;;6BAC7B,SAAS;iCAAI,IAAI,EAAE,QAAQ;;;yBAE7B,QAAQ;yBACR,oBAAoB;;qBAEtB,KAAK,EAAE,UAAU;qBACjB,WAAW,EAAE,yBAAyB;;iBAExC,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,6BAA6B;;;;;CAKb,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACrE,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AACvE,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,uBAAuB,CAAC,CAAC;AAE3E;;;GAGG;AACH,qBAAa,kBAAmB,SAAQ,IAAI,CAC1C,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB;IACC,OAAuB,IAAI,SAAwB;IACnD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAqB;IACjD,OAAuB,WAAW,SAC+B;IACjE,OAAuB,SAAS,UAAQ;IACxC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAO;IAEnD,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,uBAAuB,EAC9B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,wBAAwB,CAAC,CAmCnC;IAED;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAuD7B;;OAEG;IACH,OAAO,CAAC,qBAAqB;IA0D7B;;OAEG;IACH,OAAO,CAAC,cAAc;IA0BtB,OAAO,CAAC,yBAAyB;IAkBjC;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAiB3B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAU3B;;OAEG;IACH,OAAO,CAAC,aAAa;IAarB;;OAEG;IACH,OAAO,CAAC,kBAAkB;CAyC3B;AAED,eAAO,MAAM,cAAc,UAClB,uBAAuB,WACrB,wBAAwB;;;;;;;EAGlC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,cAAc,EAAE,cAAc,CAC5B,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB,CAAC;KACH;CACF"}

package/dist/task/UnloadModelTask.d.ts

@@ -3,7 +3,8 @@
  * Copyright 2025 Steven Roussey <sroussey@gmail.com>
  * SPDX-License-Identifier: Apache-2.0
  */
-import { CreateWorkflow, TaskConfig } from "@workglow/task-graph";
+import { CreateWorkflow } from "@workglow/task-graph";
+import type { TaskConfig } from "@workglow/task-graph";
 import { DataPortSchema, FromSchema } from "@workglow/util/schema";
 import { AiTask } from "./base/AiTask";
 declare const UnloadModelInputSchema: {
@@ -136,13 +137,14 @@ declare const UnloadModelOutputSchema: {
 };
 export type UnloadModelTaskRunInput = FromSchema<typeof UnloadModelInputSchema>;
 export type UnloadModelTaskRunOutput = FromSchema<typeof UnloadModelOutputSchema>;
+export type UnloadModelTaskConfig = TaskConfig<UnloadModelTaskRunInput>;
 /**
  * Unload a model from memory and clear its cache.
  *
  * @remarks
  * This task has a side effect of removing the model from memory and deleting cached files
  */
-export declare class UnloadModelTask extends AiTask<UnloadModelTaskRunInput, UnloadModelTaskRunOutput, TaskConfig> {
+export declare class UnloadModelTask extends AiTask<UnloadModelTaskRunInput, UnloadModelTaskRunOutput, UnloadModelTaskConfig> {
     static type: string;
     static category: string;
     static title: string;
@@ -157,23 +159,7 @@ export declare class UnloadModelTask extends AiTask<UnloadModelTaskRunInput, Unl
  * @param input - Input containing model(s) to unload
  * @returns Promise resolving to the unloaded model(s)
  */
-export declare const unloadModel: (input: {
-    model: string | {
-        [x: string]: unknown;
-        description?: string | undefined;
-        metadata?: {
-            [x: string]: unknown;
-        } | undefined;
-        model_id?: string | undefined;
-        provider: string;
-        provider_config: {
-            [x: string]: unknown;
-            credential_key?: string | undefined;
-        };
-        tasks?: string[] | undefined;
-        title?: string | undefined;
-    };
-}, config?: TaskConfig | undefined) => Promise<{
+export declare const unloadModel: (input: UnloadModelTaskRunInput, config?: UnloadModelTaskConfig) => Promise<{
     model: string | {
         [x: string]: unknown;
         description?: string | undefined;
@@ -192,7 +178,7 @@ export declare const unloadModel: (input: {
 }>;
 declare module "@workglow/task-graph" {
     interface Workflow {
-        unloadModel: CreateWorkflow<UnloadModelTaskRunInput, UnloadModelTaskRunOutput, TaskConfig>;
+        unloadModel: CreateWorkflow<UnloadModelTaskRunInput, UnloadModelTaskRunOutput, UnloadModelTaskConfig>;
     }
 }
 export {};

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

@@ -1 +1 @@
-{"version":3,"file":"UnloadModelTask.d.ts","sourceRoot":"","sources":["../../src/task/UnloadModelTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,UAAU,EAAY,MAAM,sBAAsB,CAAC;AAC5E,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,QAAA,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAOO,CAAC;AAEpC,QAAA,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAOM,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAChF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAElF;;;;;GAKG;AACH,qBAAa,eAAgB,SAAQ,MAAM,CACzC,uBAAuB,EACvB,wBAAwB,EACxB,UAAU,CACX;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SAAiE;IACnG,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;IACD,OAAuB,SAAS,UAAS;CAC1C;AAED;;;;;GAKG;AACH,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAEvB,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,uBAAuB,EAAE,wBAAwB,EAAE,UAAU,CAAC,CAAC;KAC5F;CACF"}
+{"version":3,"file":"UnloadModelTask.d.ts","sourceRoot":"","sources":["../../src/task/UnloadModelTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,QAAA,MAAM,sBAAsB;mBACpB,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAI0B,CAAC;AAEpC,QAAA,MAAM,uBAAuB;mBACrB,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAI0B,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAChF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAClF,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,uBAAuB,CAAC,CAAC;AAExE;;;;;GAKG;AACH,qBAAa,eAAgB,SAAQ,MAAM,CACzC,uBAAuB,EACvB,wBAAwB,EACxB,qBAAqB,CACtB;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SAC8B;IAChE,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;IACD,OAAuB,SAAS,UAAS;CAC1C;AAED;;;;;GAKG;AACH,eAAO,MAAM,WAAW,UAAW,uBAAuB,WAAW,qBAAqB;;;;;;;;;;;;;;;;EAEzF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CACzB,uBAAuB,EACvB,wBAAwB,EACxB,qBAAqB,CACtB,CAAC;KACH;CACF"}