[refactor] Update task imports

- Refactored imports in TaskUI and TaskNode components to streamline dependencies.
- Expanded TODO list with new items related to chunk and node handling, model improvements, and documentation updates.
- Removed unnecessary whitespace in several task files for cleaner code.
- Centralized task registration in the index file to improve maintainability and reduce redundancy.

Author: sroussey

TODO.md

@@ -1,8 +1,15 @@
 TODO.md
 
+- [ ] Chunks and nodes are not the same.
+  - [ ] We need to rename the files related to embedding.
+  - [ ] And we may need to save the chunk's node path. Or paths?
+- [ ] Get a better model for question answering.
+- [ ] Get a better model for named entity recognition, the current one recognized everything as a token, not helpful.
+- [ ] Titles are not making it into the chunks.
+- [ ] Tests for CLI commands.
+
 - [ ] Add ability for queues to specify if inputs should be converted to text, binary blob, a transferable object, structured clone, or just passed as is.
 - [ ] Add specialized versions of the task queues for hugging face transformers and tensorflow mediapipe.
 - [ ] Audio conversion like the image conversion
-- [ ] Make all the workflow name start with a lower case letter and change the users of them, particularly in the docs and examples
 - [ ] rename the registration stuff to not look ugly: registerHuggingfaceTransformers() and registerHuggingfaceTransformersUsingWorkers() and registerHuggingfaceTransformersInsideWorker()
 - [ ] fix image transferables

examples/cli/src/components/TaskUI.tsx

@@ -5,7 +5,8 @@
  */
 
 import { DownloadModelTask } from "@workglow/ai";
-import { ArrayTask, ITask, ITaskGraph, TaskStatus } from "@workglow/task-graph";
+import { ITask, ITaskGraph, TaskStatus } from "@workglow/task-graph";
+import { ArrayTask } from "@workglow/tasks";
 import type { FC } from "react";
 import { memo, useEffect, useState } from "react";
 import { Box, Text } from "retuink";

examples/web/src/graph/TaskNode.tsx

@@ -4,7 +4,8 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { ArrayTask, ITask, TaskStatus } from "@workglow/task-graph";
+import { ITask, TaskStatus } from "@workglow/task-graph";
+import { ArrayTask } from "@workglow/tasks";
 import { Node, NodeProps } from "@xyflow/react";
 import { useEffect, useState } from "react";
 import { FiCloud, FiCloudLightning } from "react-icons/fi";

packages/ai/src/task/BackgroundRemovalTask.ts

@@ -61,7 +61,6 @@ export class BackgroundRemovalTask extends AiVisionTask<
   }
 }
 
-
 /**
  * Convenience function to run background removal tasks.
  * Creates and executes a BackgroundRemovalTask with the provided input.

packages/ai/src/task/ChunkToVectorTask.ts

@@ -10,7 +10,6 @@ import {
   IExecuteContext,
   JobQueueTaskConfig,
   Task,
-  TaskRegistry,
   Workflow,
 } from "@workglow/task-graph";
 import {
@@ -160,7 +159,6 @@ export class ChunkToVectorTask extends Task<
   }
 }
 
-
 export const chunkToVector = (input: ChunkToVectorTaskInput, config?: JobQueueTaskConfig) => {
   return new ChunkToVectorTask({} as ChunkToVectorTaskInput, config).run(input);
 };

packages/ai/src/task/ContextBuilderTask.ts

@@ -4,13 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import {
-  CreateWorkflow,
-  JobQueueTaskConfig,
-  Task,
-  TaskRegistry,
-  Workflow,
-} from "@workglow/task-graph";
+import { CreateWorkflow, JobQueueTaskConfig, Task, Workflow } from "@workglow/task-graph";
 import { DataPortSchema, FromSchema } from "@workglow/util";
 
 export const ContextFormat = {
@@ -320,7 +314,6 @@ export class ContextBuilderTask extends Task<
   }
 }
 
-
 export const contextBuilder = (input: ContextBuilderTaskInput, config?: JobQueueTaskConfig) => {
   return new ContextBuilderTask({} as ContextBuilderTaskInput, config).run(input);
 };

packages/ai/src/task/DocumentEnricherTask.ts

@@ -16,7 +16,6 @@ import {
   IExecuteContext,
   JobQueueTaskConfig,
   Task,
-  TaskRegistry,
   Workflow,
 } from "@workglow/task-graph";
 import { DataPortSchema, FromSchema } from "@workglow/util";
@@ -398,7 +397,6 @@ export class DocumentEnricherTask extends Task<
   }
 }
 
-
 export const documentEnricher = (input: DocumentEnricherTaskInput, config?: JobQueueTaskConfig) => {
   return new DocumentEnricherTask({} as DocumentEnricherTaskInput, config).run(input);
 };

packages/ai/src/task/DocumentNodeRetrievalTask.ts

@@ -13,7 +13,6 @@ import {
   IExecuteContext,
   JobQueueTaskConfig,
   Task,
-  TaskRegistry,
   Workflow,
 } from "@workglow/task-graph";
 import {
@@ -231,7 +230,6 @@ export class DocumentNodeRetrievalTask extends Task<
   }
 }
 
-
 export const retrieval = (input: RetrievalTaskInput, config?: JobQueueTaskConfig) => {
   return new DocumentNodeRetrievalTask({} as RetrievalTaskInput, config).run(input);
 };

packages/ai/src/task/DocumentNodeVectorHybridSearchTask.ts

@@ -13,7 +13,6 @@ import {
   IExecuteContext,
   JobQueueTaskConfig,
   Task,
-  TaskRegistry,
   Workflow,
 } from "@workglow/task-graph";
 import {
@@ -217,7 +216,6 @@ export class DocumentNodeVectorHybridSearchTask extends Task<
   }
 }
 
-
 export const hybridSearch = async (
   input: HybridSearchTaskInput,
   config?: JobQueueTaskConfig

packages/ai/src/task/DocumentNodeVectorSearchTask.ts

@@ -13,7 +13,6 @@ import {
   IExecuteContext,
   JobQueueTaskConfig,
   Task,
-  TaskRegistry,
   Workflow,
 } from "@workglow/task-graph";
 import {
@@ -153,7 +152,6 @@ export class DocumentNodeVectorSearchTask extends Task<
   }
 }
 
-
 export const vectorStoreSearch = (
   input: VectorStoreSearchTaskInput,
   config?: JobQueueTaskConfig