@workglow/ai 0.0.70 → 0.0.71

package/dist/task/TextNamedEntityRecognitionTask.d.ts

@@ -0,0 +1,212 @@
+/**
+ * @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";
+import { DeReplicateFromSchema } from "./base/AiTaskSchemas";
+export declare const TextNamedEntityRecognitionInputSchema: {
+    readonly type: "object";
+    readonly properties: {
+        readonly text: {
+            readonly "x-replicate": true;
+            readonly format?: string | undefined;
+            readonly oneOf: readonly [{
+                type: "string";
+                title: string;
+                description: string;
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    type: "string";
+                    title: string;
+                    description: string;
+                };
+            }];
+            readonly title: string | undefined;
+            readonly description: string | undefined;
+        };
+        readonly blockList: {
+            readonly type: "array";
+            readonly items: {
+                readonly type: "string";
+            };
+            readonly title: "Block List";
+            readonly description: "The entity types to exclude from results";
+            readonly "x-ui-group": "Configuration";
+            readonly "x-ui-group-open": false;
+        };
+        readonly model: {
+            readonly "x-replicate": true;
+            readonly format?: string | undefined;
+            readonly oneOf: readonly [{
+                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 title: {
+                            readonly type: "string";
+                        };
+                        readonly description: {
+                            readonly type: "string";
+                        };
+                        readonly provider: {
+                            readonly type: "string";
+                        };
+                        readonly providerConfig: {
+                            readonly type: "object";
+                            readonly default: {};
+                        };
+                        readonly metadata: {
+                            readonly type: "object";
+                            readonly default: {};
+                        };
+                    };
+                    readonly required: readonly ["model_id", "tasks", "provider", "title", "description", "providerConfig", "metadata"];
+                    readonly format: "model";
+                    readonly additionalProperties: false;
+                } & {
+                    readonly format: import(".").TypeModelSemantic;
+                }];
+            } & {
+                readonly format: import(".").TypeModelSemantic;
+            }, {
+                readonly type: "array";
+                readonly items: {
+                    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 title: {
+                                readonly type: "string";
+                            };
+                            readonly description: {
+                                readonly type: "string";
+                            };
+                            readonly provider: {
+                                readonly type: "string";
+                            };
+                            readonly providerConfig: {
+                                readonly type: "object";
+                                readonly default: {};
+                            };
+                            readonly metadata: {
+                                readonly type: "object";
+                                readonly default: {};
+                            };
+                        };
+                        readonly required: readonly ["model_id", "tasks", "provider", "title", "description", "providerConfig", "metadata"];
+                        readonly format: "model";
+                        readonly additionalProperties: false;
+                    } & {
+                        readonly format: import(".").TypeModelSemantic;
+                    }];
+                } & {
+                    readonly format: import(".").TypeModelSemantic;
+                };
+            }];
+            readonly title: string | undefined;
+            readonly description: string | undefined;
+        };
+    };
+    readonly required: readonly ["text", "model"];
+    readonly additionalProperties: false;
+};
+export declare const TextNamedEntityRecognitionOutputSchema: {
+    readonly type: "object";
+    readonly properties: {
+        readonly entities: {
+            readonly type: "array";
+            readonly items: {
+                readonly type: "object";
+                readonly properties: {
+                    readonly entity: {
+                        readonly type: "string";
+                        readonly title: "Entity";
+                        readonly description: "The type of the named entity";
+                    };
+                    readonly score: {
+                        readonly type: "number";
+                        readonly title: "Score";
+                        readonly description: "The confidence score for this entity";
+                    };
+                    readonly word: {
+                        readonly type: "string";
+                        readonly title: "Word";
+                        readonly description: "The extracted text of the named entity";
+                    };
+                };
+                readonly required: readonly ["entity", "score", "word"];
+                readonly additionalProperties: false;
+            };
+            readonly title: "Entities";
+            readonly description: "The extracted named entities with their types, scores, and text";
+        };
+    };
+    readonly required: readonly ["entities"];
+    readonly additionalProperties: false;
+};
+export type TextNamedEntityRecognitionTaskInput = FromSchema<typeof TextNamedEntityRecognitionInputSchema>;
+export type TextNamedEntityRecognitionTaskOutput = FromSchema<typeof TextNamedEntityRecognitionOutputSchema>;
+export type TextNamedEntityRecognitionTaskExecuteInput = DeReplicateFromSchema<typeof TextNamedEntityRecognitionInputSchema>;
+export type TextNamedEntityRecognitionTaskExecuteOutput = DeReplicateFromSchema<typeof TextNamedEntityRecognitionOutputSchema>;
+/**
+ * Extracts named entities from text using language models
+ */
+export declare class TextNamedEntityRecognitionTask extends AiTask<TextNamedEntityRecognitionTaskInput, TextNamedEntityRecognitionTaskOutput> {
+    static type: string;
+    static category: string;
+    static title: string;
+    static description: string;
+    static inputSchema(): DataPortSchema;
+    static outputSchema(): DataPortSchema;
+}
+/**
+ * Convenience function to run named entity recognition tasks.
+ * Creates and executes a TextNamedEntityRecognitionTask with the provided input.
+ * @param input The input parameters for named entity recognition (text and model)
+ * @returns Promise resolving to the extracted named entities with types, scores, and text
+ */
+export declare const TextNamedEntityRecognition: (input: TextNamedEntityRecognitionTaskInput, config?: JobQueueTaskConfig) => Promise<{
+    entities: {
+        score: number;
+        entity: string;
+        word: string;
+    }[];
+}>;
+declare module "@workglow/task-graph" {
+    interface Workflow {
+        TextNamedEntityRecognition: CreateWorkflow<TextNamedEntityRecognitionTaskInput, TextNamedEntityRecognitionTaskOutput, JobQueueTaskConfig>;
+    }
+}
+//# sourceMappingURL=TextNamedEntityRecognitionTask.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"TextNamedEntityRecognitionTask.d.ts","sourceRoot":"","sources":["../../src/task/TextNamedEntityRecognitionTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAA0B,MAAM,sBAAsB,CAAC;AAClG,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AACvC,OAAO,EAAE,qBAAqB,EAAiC,MAAM,sBAAsB,CAAC;AAI5F,eAAO,MAAM,qCAAqC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAsBf,CAAC;AAEpC,eAAO,MAAM,sCAAsC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAiChB,CAAC;AAEpC,MAAM,MAAM,mCAAmC,GAAG,UAAU,CAC1D,OAAO,qCAAqC,CAC7C,CAAC;AACF,MAAM,MAAM,oCAAoC,GAAG,UAAU,CAC3D,OAAO,sCAAsC,CAC9C,CAAC;AACF,MAAM,MAAM,0CAA0C,GAAG,qBAAqB,CAC5E,OAAO,qCAAqC,CAC7C,CAAC;AACF,MAAM,MAAM,2CAA2C,GAAG,qBAAqB,CAC7E,OAAO,sCAAsC,CAC9C,CAAC;AAEF;;GAEG;AACH,qBAAa,8BAA+B,SAAQ,MAAM,CACxD,mCAAmC,EACnC,oCAAoC,CACrC;IACC,OAAc,IAAI,SAAoC;IACtD,OAAc,QAAQ,SAAmB;IACzC,OAAc,KAAK,SAAmC;IACtD,OAAc,WAAW,SAA6D;WACxE,WAAW,IAAI,cAAc;WAG7B,YAAY,IAAI,cAAc;CAG7C;AAID;;;;;GAKG;AACH,eAAO,MAAM,0BAA0B,GACrC,OAAO,mCAAmC,EAC1C,SAAS,kBAAkB;;;;;;EAG5B,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC;IACpC,UAAU,QAAQ;QAChB,0BAA0B,EAAE,cAAc,CACxC,mCAAmC,EACnC,oCAAoC,EACpC,kBAAkB,CACnB,CAAC;KACH;CACF"}

package/dist/task/index.d.ts

@@ -9,8 +9,10 @@ export * from "./DocumentSplitterTask";
 export * from "./DownloadModelTask";
 export * from "./TextClassifierTask";
 export * from "./TextEmbeddingTask";
+export * from "./TextFillMaskTask";
 export * from "./TextGenerationTask";
 export * from "./TextLanguageDetectionTask";
+export * from "./TextNamedEntityRecognitionTask";
 export * from "./TextQuestionAnswerTask";
 export * from "./TextRewriterTask";
 export * from "./TextSummaryTask";

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/task/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,eAAe,CAAC;AAC9B,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,0BAA0B,CAAC;AACzC,cAAc,oBAAoB,CAAC;AACnC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mBAAmB,CAAC;AAClC,cAAc,wBAAwB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/task/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,eAAe,CAAC;AAC9B,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,qBAAqB,CAAC;AACpC,cAAc,sBAAsB,CAAC;AACrC,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC;AACnC,cAAc,sBAAsB,CAAC;AACrC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,kCAAkC,CAAC;AACjD,cAAc,0BAA0B,CAAC;AACzC,cAAc,oBAAoB,CAAC;AACnC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mBAAmB,CAAC;AAClC,cAAc,wBAAwB,CAAC"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@workglow/ai",
   "type": "module",
-  "version": "0.0.70",
+  "version": "0.0.71",
   "description": "Core AI functionality for Workglow, including task execution, model management, and AI pipeline orchestration.",
   "scripts": {
     "watch": "concurrently -c 'auto' 'bun:watch-*'",
@@ -36,10 +36,10 @@
     "access": "public"
   },
   "peerDependencies": {
-    "@workglow/job-queue": "0.0.70",
-    "@workglow/storage": "0.0.70",
-    "@workglow/task-graph": "0.0.70",
-    "@workglow/util": "0.0.70"
+    "@workglow/job-queue": "0.0.71",
+    "@workglow/storage": "0.0.71",
+    "@workglow/task-graph": "0.0.71",
+    "@workglow/util": "0.0.71"
   },
   "peerDependenciesMeta": {
     "@workglow/job-queue": {
@@ -56,9 +56,9 @@
     }
   },
   "devDependencies": {
-    "@workglow/job-queue": "0.0.70",
-    "@workglow/storage": "0.0.70",
-    "@workglow/task-graph": "0.0.70",
-    "@workglow/util": "0.0.70"
+    "@workglow/job-queue": "0.0.71",
+    "@workglow/storage": "0.0.71",
+    "@workglow/task-graph": "0.0.71",
+    "@workglow/util": "0.0.71"
   }
 }