feat: Task constructor signature, ToolCallingTask and AgentTask (#353)

* feat(ai): restore ToolCallingTask and AgentTask for development
* refactor: simplify type handling by removing unnecessary casts
* refactor(ai): apply TypeScript's override keyword across agent and tool task classes
* refactor(ai): update timeout handling and enhance tool parsing functionality
* refactor(logging): update testing logger initialization and configuration
* refactor: enhance error handling and diagnostics across job queue and execution strategies
* feat(ai-provider): introduce ToolCallParser and enhance tool calling functionality
* feat(ai-provider): enhance tool call parsing and message handling
* feat(ai-provider): implement shared streaming helper for text chunk generation
* feat(ai-provider): enhance tool call parsing and message handling
* feat(ai-provider): enhance logging and add integration tests for LlamaCpp models
* refactor: change Task constructor to take only config, not input
* refactor: update test files for new Task constructor signature
* refactor: update remaining test files for new Task constructor
* refactor: update TaskJSON test for new Task constructor
* refactor: update all task/ test files for new Task constructor
* fix: resolve remaining constructor signature issues
* fix: fix LambdaTask helper and ArrayTask test constructor patterns
* fix: fix ConditionalTask and TaskRunnerStreaming test constructor patterns
* fix: fix ArrayTask inner task cloning to use new constructor signature
* refactor: update task configurations to use new TaskConfig structure
* refactor: update task configurations to use generic TaskConfig types

Author: sroussey

bunfig.toml

@@ -2,6 +2,8 @@
 # Explicitly set the linker to isolated
 linker = "isolated"
 [test]
-retry = 1
+retry = 0
 root = "./packages/test/src/test"
 preload = ["./scripts/configure-hft-test-cache.ts"]
+[console]
+depth = 5

examples/cli/src/commands/agent.ts

@@ -8,6 +8,7 @@ import {
   computeGraphInputSchema,
   createGraphFromGraphJSON,
   type TaskGraphJson,
+  type TaskDeserializationOptions,
 } from "@workglow/task-graph";
 import type { DataPortSchemaObject } from "@workglow/util/schema";
 import type { Command } from "commander";
@@ -233,9 +234,24 @@ export function registerAgentCommand(program: Command): void {
         process.exit(1);
       }
 
+      // Restrict which task types can be instantiated from untrusted agent JSON.
+      // This allowlist should include only task types that are safe and expected in agent graphs.
+      const deserializationOptions: TaskDeserializationOptions = {
+        allowedTypes: new Set<string>([
+          // Core control-flow / composition tasks
+          "GraphAsTask",
+          "ConditionalTask",
+          "IteratorTask",
+          "MapTask",
+          "ReduceTask",
+          "WhileTask",
+          // Add additional allowed task type names here as needed.
+        ]),
+      };
+
       let graph;
       try {
-        graph = createGraphFromGraphJSON(json);
+        graph = createGraphFromGraphJSON(json, undefined, deserializationOptions);
       } catch (e) {
         console.error(`Invalid agent graph: ${formatError(e)}`);
         process.exit(1);

examples/web/src/App.tsx

@@ -218,7 +218,7 @@ export const App = () => {
 
   const setNewJson = useCallback(
     (json: string) => {
-      const task = new JsonTask({ json });
+      const task = new JsonTask({ defaults: { json } });
       if (task.hasChildren()) {
         workflow.graph = task.subGraph;
       } else {

examples/web/src/editor/JsonEditor.tsx

@@ -40,7 +40,7 @@ export const JsonEditor: React.FC<PopupProps> = ({
       // this will throw an error if the JSON is invalid
       JSON.parse(jsonString);
       // this will throw an error if the JSON is not a valid task graph
-      new JsonTask({ json: jsonString }, { title: "Test JSON" });
+      new JsonTask({ title: "Test JSON", defaults: { json: jsonString } });
 
       setIsValidJSON(true);
       setCode(jsonString);

packages/ai-provider/package.json

@@ -20,7 +20,10 @@
     "test": "bun test"
   },
   "exports": {
-    "./src/*": "./src/*",
+    "./test": {
+      "types": "./src/common/ToolCallParsers.ts",
+      "import": "./src/common/ToolCallParsers.ts"
+    },
     "./anthropic": {
       "types": "./dist/provider-anthropic/index.d.ts",
       "import": "./dist/provider-anthropic/index.js"
@@ -213,4 +216,4 @@
     "tiktoken": "catalog:",
     "js-tiktoken": "catalog:"
   }
-}
+}