@workglow/task-graph 0.1.1 → 0.1.2

package/dist/bun.js

@@ -3659,10 +3659,11 @@ class WorkflowTask extends GraphAsTask {
 }
 
 class Workflow {
-  constructor(cache, parent, iteratorTask) {
+  constructor(cache, parent, iteratorTask, registry) {
     this._outputCache = cache;
     this._parentWorkflow = parent;
     this._iteratorTask = iteratorTask;
+    this._registry = registry ?? parent?._registry;
     this._graph = new TaskGraph({ outputCache: this._outputCache });
     if (!parent) {
       this._onChanged = this._onChanged.bind(this);
@@ -3673,6 +3674,7 @@ class Workflow {
   _dataFlows = [];
   _error = "";
   _outputCache;
+  _registry;
   _abortController;
   _parentWorkflow;
   _iteratorTask;
@@ -3786,7 +3788,7 @@ class Workflow {
       const output = await this.graph.run(input, {
         parentSignal: this._abortController.signal,
         outputCache: this._outputCache,
-        registry: config?.registry
+        registry: config?.registry ?? this._registry
       });
       const results = this.graph.mergeExecuteOutputsToRunOutput(output, PROPERTY_ARRAY);
       this.events.emit("complete");
@@ -3947,7 +3949,7 @@ class Workflow {
     return this;
   }
   addTaskToGraph(taskClass, input, config) {
-    const task = new taskClass(input, config);
+    const task = new taskClass(input, config, this._registry ? { registry: this._registry } : undefined);
     const id = this.graph.addTask(task);
     this.events.emit("changed", id);
     return task;
@@ -3973,7 +3975,7 @@ class Workflow {
       });
       this._dataFlows = [];
     }
-    const loopBuilder = new Workflow(this.outputCache(), this, task);
+    const loopBuilder = new Workflow(this.outputCache(), this, task, this._registry);
     if (parent) {
       loopBuilder._pendingLoopConnect = { parent, iteratorTask: task };
     }
@@ -4559,7 +4561,18 @@ function extractLoopConfig(task) {
   }
   return config;
 }
+function stripUndefined(obj) {
+  if (!obj)
+    return obj;
+  const result = {};
+  for (const [k, v] of Object.entries(obj)) {
+    if (v !== undefined)
+      result[k] = v;
+  }
+  return Object.keys(result).length > 0 ? result : undefined;
+}
 function buildMethodArgs(defaults, config, baseIndent = "", columnOffset = 0) {
+  defaults = stripUndefined(defaults);
   const hasDefaults = defaults && Object.keys(defaults).length > 0;
   const hasConfig = Object.keys(config).length > 0;
   if (!hasDefaults && !hasConfig)
@@ -4834,6 +4847,68 @@ queueMicrotask(() => {
   };
   Workflow.prototype.endFallbackWith = CreateEndLoopWorkflow("endFallbackWith");
 });
+// src/task/InputCompactor.ts
+import { getInputCompactors } from "@workglow/util";
+function schemaAllowsString(schema) {
+  if (typeof schema !== "object" || schema === null)
+    return false;
+  const s = schema;
+  if (s.type === "string")
+    return true;
+  const variants = s.oneOf ?? s.anyOf;
+  if (Array.isArray(variants)) {
+    for (const variant of variants) {
+      if (schemaAllowsString(variant))
+        return true;
+    }
+  }
+  return false;
+}
+async function compactSchemaInputs(input, schema, config) {
+  if (typeof schema === "boolean")
+    return input;
+  const properties = schema.properties;
+  if (!properties || typeof properties !== "object")
+    return input;
+  const compactors = getInputCompactors();
+  const compacted = { ...input };
+  for (const [key, propSchema] of Object.entries(properties)) {
+    let value = compacted[key];
+    const format = getSchemaFormat(propSchema);
+    if (format) {
+      let compactor = compactors.get(format);
+      if (!compactor) {
+        const prefix = getFormatPrefix(format);
+        compactor = compactors.get(prefix);
+      }
+      if (compactor) {
+        if (value !== null && value !== undefined && typeof value === "object" && !Array.isArray(value) && schemaAllowsString(propSchema)) {
+          const id = await compactor(value, format, config.registry);
+          if (id !== undefined) {
+            compacted[key] = id;
+            continue;
+          }
+        } else if (Array.isArray(value)) {
+          compacted[key] = await Promise.all(value.map(async (item) => {
+            if (item !== null && item !== undefined && typeof item === "object" && !Array.isArray(item)) {
+              const id = await compactor(item, format, config.registry);
+              return id !== undefined ? id : item;
+            }
+            return item;
+          }));
+          continue;
+        }
+      }
+    }
+    if (value !== null && value !== undefined && typeof value === "object" && !Array.isArray(value)) {
+      const objectSchema = getObjectSchema(propSchema);
+      if (objectSchema) {
+        compacted[key] = await compactSchemaInputs(value, objectSchema, config);
+      }
+    }
+  }
+  return compacted;
+}
 // src/task/IteratorTaskRunner.ts
 import { uuid4 as uuid46 } from "@workglow/util";
 class IteratorTaskRunner extends GraphAsTaskRunner {
@@ -6335,6 +6410,7 @@ queueMicrotask(() => {
 import {
   createServiceToken as createServiceToken3,
   globalServiceRegistry as globalServiceRegistry4,
+  registerInputCompactor,
   registerInputResolver
 } from "@workglow/util";
 var taskConstructors = new Map;
@@ -6377,6 +6453,17 @@ function resolveTaskFromRegistry(id, _format, registry) {
   };
 }
 registerInputResolver("tasks", resolveTaskFromRegistry);
+registerInputCompactor("tasks", (value, _format, registry) => {
+  if (typeof value === "object" && value !== null && "name" in value) {
+    const name = value.name;
+    if (typeof name !== "string")
+      return;
+    const constructors = getTaskConstructors(registry);
+    const ctor = constructors.get(name);
+    return ctor ? name : undefined;
+  }
+  return;
+});
 
 // src/task/TaskJSON.ts
 var createSingleTaskFromJSON = (item, registry, options) => {
@@ -6685,8 +6772,10 @@ export {
   getTaskConstructors,
   getStructuredOutputSchemas,
   getStreamingPorts,
+  getSchemaFormat,
   getPortStreamMode,
   getOutputStreamMode,
+  getObjectSchema,
   getObjectPortId,
   getNestedValue,
   getLastTask,
@@ -6694,6 +6783,7 @@ export {
   getIterationContextSchemaForType,
   getInputModeFromSchema,
   getGlobalTaskConstructors,
+  getFormatPrefix,
   getAppendPortId,
   formatValue,
   findArrayPorts,
@@ -6715,6 +6805,7 @@ export {
   conditionalTaskConfigSchema,
   computeGraphOutputSchema,
   computeGraphInputSchema,
+  compactSchemaInputs,
   calculateNodeDepths,
   buildIterationInputSchema,
   addIterationContextToSchema,
@@ -6778,4 +6869,4 @@ export {
   ConditionalTask
 };
 
-//# debugId=8BC5BADB1209138664756E2164756E21
+//# debugId=BDC0F4EAE393712864756E2164756E21