@workglow/ai 0.0.69 → 0.0.71

package/dist/node.js

@@ -828,16 +828,10 @@ var TextClassifierInputSchema = {
       description: "The text to classify"
     }),
     maxCategories: {
-      oneOf: [
-        {
-          type: "number",
-          minimum: 1,
-          maximum: 1000
-        },
-        {
-          type: "null"
-        }
-      ],
+      type: "number",
+      minimum: 1,
+      maximum: 1000,
+      default: 5,
       title: "Max Categories",
       description: "The maximum number of categories to return"
     },
@@ -938,18 +932,86 @@ var TextEmbedding = async (input, config) => {
   return new TextEmbeddingTask(input, config).run();
 };
 Workflow4.prototype.TextEmbedding = CreateWorkflow4(TextEmbeddingTask);
-// src/task/TextGenerationTask.ts
+// src/task/TextFillMaskTask.ts
 import { CreateWorkflow as CreateWorkflow5, TaskRegistry as TaskRegistry5, Workflow as Workflow5 } from "@workglow/task-graph";
+var modelSchema4 = TypeReplicateArray(TypeModel("model:TextFillMaskTask"));
+var TextFillMaskInputSchema = {
+  type: "object",
+  properties: {
+    text: TypeReplicateArray({
+      type: "string",
+      title: "Text",
+      description: "The text with a mask token to fill"
+    }),
+    model: modelSchema4
+  },
+  required: ["text", "model"],
+  additionalProperties: false
+};
+var TextFillMaskOutputSchema = {
+  type: "object",
+  properties: {
+    predictions: {
+      type: "array",
+      items: {
+        type: "object",
+        properties: {
+          entity: {
+            type: "string",
+            title: "Entity",
+            description: "The token that was predicted to fill the mask"
+          },
+          score: {
+            type: "number",
+            title: "Score",
+            description: "The confidence score for this prediction"
+          },
+          sequence: {
+            type: "string",
+            title: "Sequence",
+            description: "The complete text with the mask filled"
+          }
+        },
+        required: ["entity", "score", "sequence"],
+        additionalProperties: false
+      },
+      title: "Predictions",
+      description: "The predicted tokens to fill the mask with their scores and complete sequences"
+    }
+  },
+  required: ["predictions"],
+  additionalProperties: false
+};
+
+class TextFillMaskTask extends AiTask {
+  static type = "TextFillMaskTask";
+  static category = "AI Text Model";
+  static title = "Text Fill Mask";
+  static description = "Fills masked tokens in text using language models";
+  static inputSchema() {
+    return TextFillMaskInputSchema;
+  }
+  static outputSchema() {
+    return TextFillMaskOutputSchema;
+  }
+}
+TaskRegistry5.registerTask(TextFillMaskTask);
+var TextFillMask = (input, config) => {
+  return new TextFillMaskTask(input, config).run();
+};
+Workflow5.prototype.TextFillMask = CreateWorkflow5(TextFillMaskTask);
+// src/task/TextGenerationTask.ts
+import { CreateWorkflow as CreateWorkflow6, TaskRegistry as TaskRegistry6, Workflow as Workflow6 } from "@workglow/task-graph";
 var generatedTextSchema = {
   type: "string",
   title: "Text",
   description: "The generated text"
 };
-var modelSchema4 = TypeReplicateArray(TypeModel("model:TextGenerationTask"));
+var modelSchema5 = TypeReplicateArray(TypeModel("model:TextGenerationTask"));
 var TextGenerationInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema4,
+    model: modelSchema5,
     prompt: TypeReplicateArray({
       type: "string",
       title: "Prompt",
@@ -1024,14 +1086,14 @@ class TextGenerationTask extends AiTask {
     return TextGenerationOutputSchema;
   }
 }
-TaskRegistry5.registerTask(TextGenerationTask);
+TaskRegistry6.registerTask(TextGenerationTask);
 var TextGeneration = (input, config) => {
   return new TextGenerationTask(input, config).run();
 };
-Workflow5.prototype.TextGeneration = CreateWorkflow5(TextGenerationTask);
+Workflow6.prototype.TextGeneration = CreateWorkflow6(TextGenerationTask);
 // src/task/TextLanguageDetectionTask.ts
-import { CreateWorkflow as CreateWorkflow6, TaskRegistry as TaskRegistry6, Workflow as Workflow6 } from "@workglow/task-graph";
-var modelSchema5 = TypeReplicateArray(TypeModel("model:TextLanguageDetectionTask"));
+import { CreateWorkflow as CreateWorkflow7, TaskRegistry as TaskRegistry7, Workflow as Workflow7 } from "@workglow/task-graph";
+var modelSchema6 = TypeReplicateArray(TypeModel("model:TextLanguageDetectionTask"));
 var TextLanguageDetectionInputSchema = {
   type: "object",
   properties: {
@@ -1041,20 +1103,14 @@ var TextLanguageDetectionInputSchema = {
       description: "The text to detect the language of"
     }),
     maxLanguages: {
-      oneOf: [
-        {
-          type: "number",
-          minimum: 1,
-          maximum: 1000
-        },
-        {
-          type: "null"
-        }
-      ],
+      type: "number",
+      minimum: 0,
+      maximum: 100,
+      default: 5,
       title: "Max Languages",
       description: "The maximum number of languages to return"
     },
-    model: modelSchema5
+    model: modelSchema6
   },
   required: ["text", "model"],
   additionalProperties: false
@@ -1101,13 +1157,91 @@ class TextLanguageDetectionTask extends AiTask {
     return TextLanguageDetectionOutputSchema;
   }
 }
-TaskRegistry6.registerTask(TextLanguageDetectionTask);
+TaskRegistry7.registerTask(TextLanguageDetectionTask);
 var TextLanguageDetection = (input, config) => {
   return new TextLanguageDetectionTask(input, config).run();
 };
-Workflow6.prototype.TextLanguageDetection = CreateWorkflow6(TextLanguageDetectionTask);
+Workflow7.prototype.TextLanguageDetection = CreateWorkflow7(TextLanguageDetectionTask);
+// src/task/TextNamedEntityRecognitionTask.ts
+import { CreateWorkflow as CreateWorkflow8, TaskRegistry as TaskRegistry8, Workflow as Workflow8 } from "@workglow/task-graph";
+var modelSchema7 = TypeReplicateArray(TypeModel("model:NamedEntityRecognitionTask"));
+var TextNamedEntityRecognitionInputSchema = {
+  type: "object",
+  properties: {
+    text: TypeReplicateArray({
+      type: "string",
+      title: "Text",
+      description: "The text to extract named entities from"
+    }),
+    blockList: {
+      type: "array",
+      items: {
+        type: "string"
+      },
+      title: "Block List",
+      description: "The entity types to exclude from results",
+      "x-ui-group": "Configuration",
+      "x-ui-group-open": false
+    },
+    model: modelSchema7
+  },
+  required: ["text", "model"],
+  additionalProperties: false
+};
+var TextNamedEntityRecognitionOutputSchema = {
+  type: "object",
+  properties: {
+    entities: {
+      type: "array",
+      items: {
+        type: "object",
+        properties: {
+          entity: {
+            type: "string",
+            title: "Entity",
+            description: "The type of the named entity"
+          },
+          score: {
+            type: "number",
+            title: "Score",
+            description: "The confidence score for this entity"
+          },
+          word: {
+            type: "string",
+            title: "Word",
+            description: "The extracted text of the named entity"
+          }
+        },
+        required: ["entity", "score", "word"],
+        additionalProperties: false
+      },
+      title: "Entities",
+      description: "The extracted named entities with their types, scores, and text"
+    }
+  },
+  required: ["entities"],
+  additionalProperties: false
+};
+
+class TextNamedEntityRecognitionTask extends AiTask {
+  static type = "TextNamedEntityRecognitionTask";
+  static category = "AI Text Model";
+  static title = "Text Named Entity Recognition";
+  static description = "Extracts named entities from text using language models";
+  static inputSchema() {
+    return TextNamedEntityRecognitionInputSchema;
+  }
+  static outputSchema() {
+    return TextNamedEntityRecognitionOutputSchema;
+  }
+}
+TaskRegistry8.registerTask(TextNamedEntityRecognitionTask);
+var TextNamedEntityRecognition = (input, config) => {
+  return new TextNamedEntityRecognitionTask(input, config).run();
+};
+Workflow8.prototype.TextNamedEntityRecognition = CreateWorkflow8(TextNamedEntityRecognitionTask);
 // src/task/TextQuestionAnswerTask.ts
-import { CreateWorkflow as CreateWorkflow7, TaskRegistry as TaskRegistry7, Workflow as Workflow7 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow9, TaskRegistry as TaskRegistry9, Workflow as Workflow9 } from "@workglow/task-graph";
 var contextSchema = {
   type: "string",
   title: "Context",
@@ -1123,13 +1257,13 @@ var textSchema = {
   title: "Text",
   description: "The generated text"
 };
-var modelSchema6 = TypeReplicateArray(TypeModel("model:TextQuestionAnswerTask"));
+var modelSchema8 = TypeReplicateArray(TypeModel("model:TextQuestionAnswerTask"));
 var TextQuestionAnswerInputSchema = {
   type: "object",
   properties: {
     context: TypeReplicateArray(contextSchema),
     question: TypeReplicateArray(questionSchema),
-    model: modelSchema6
+    model: modelSchema8
   },
   required: ["context", "question", "model"],
   additionalProperties: false
@@ -1159,14 +1293,14 @@ class TextQuestionAnswerTask extends AiTask {
     return TextQuestionAnswerOutputSchema;
   }
 }
-TaskRegistry7.registerTask(TextQuestionAnswerTask);
+TaskRegistry9.registerTask(TextQuestionAnswerTask);
 var TextQuestionAnswer = (input, config) => {
   return new TextQuestionAnswerTask(input, config).run();
 };
-Workflow7.prototype.TextQuestionAnswer = CreateWorkflow7(TextQuestionAnswerTask);
+Workflow9.prototype.TextQuestionAnswer = CreateWorkflow9(TextQuestionAnswerTask);
 // src/task/TextRewriterTask.ts
-import { CreateWorkflow as CreateWorkflow8, TaskRegistry as TaskRegistry8, Workflow as Workflow8 } from "@workglow/task-graph";
-var modelSchema7 = TypeReplicateArray(TypeModel("model:TextRewriterTask"));
+import { CreateWorkflow as CreateWorkflow10, TaskRegistry as TaskRegistry10, Workflow as Workflow10 } from "@workglow/task-graph";
+var modelSchema9 = TypeReplicateArray(TypeModel("model:TextRewriterTask"));
 var TextRewriterInputSchema = {
   type: "object",
   properties: {
@@ -1180,7 +1314,7 @@ var TextRewriterInputSchema = {
       title: "Prompt",
       description: "The prompt to direct the rewriting"
     }),
-    model: modelSchema7
+    model: modelSchema9
   },
   required: ["text", "prompt", "model"],
   additionalProperties: false
@@ -1210,14 +1344,14 @@ class TextRewriterTask extends AiTask {
     return TextRewriterOutputSchema;
   }
 }
-TaskRegistry8.registerTask(TextRewriterTask);
+TaskRegistry10.registerTask(TextRewriterTask);
 var TextRewriter = (input, config) => {
   return new TextRewriterTask(input, config).run();
 };
-Workflow8.prototype.TextRewriter = CreateWorkflow8(TextRewriterTask);
+Workflow10.prototype.TextRewriter = CreateWorkflow10(TextRewriterTask);
 // src/task/TextSummaryTask.ts
-import { CreateWorkflow as CreateWorkflow9, TaskRegistry as TaskRegistry9, Workflow as Workflow9 } from "@workglow/task-graph";
-var modelSchema8 = TypeReplicateArray(TypeModel("model:TextSummaryTask"));
+import { CreateWorkflow as CreateWorkflow11, TaskRegistry as TaskRegistry11, Workflow as Workflow11 } from "@workglow/task-graph";
+var modelSchema10 = TypeReplicateArray(TypeModel("model:TextSummaryTask"));
 var TextSummaryInputSchema = {
   type: "object",
   properties: {
@@ -1226,7 +1360,7 @@ var TextSummaryInputSchema = {
       title: "Text",
       description: "The text to summarize"
     }),
-    model: modelSchema8
+    model: modelSchema10
   },
   required: ["text", "model"],
   additionalProperties: false
@@ -1256,14 +1390,14 @@ class TextSummaryTask extends AiTask {
     return TextSummaryOutputSchema;
   }
 }
-TaskRegistry9.registerTask(TextSummaryTask);
+TaskRegistry11.registerTask(TextSummaryTask);
 var TextSummary = async (input, config) => {
   return new TextSummaryTask(input, config).run();
 };
-Workflow9.prototype.TextSummary = CreateWorkflow9(TextSummaryTask);
+Workflow11.prototype.TextSummary = CreateWorkflow11(TextSummaryTask);
 // src/task/TextTranslationTask.ts
-import { CreateWorkflow as CreateWorkflow10, TaskRegistry as TaskRegistry10, Workflow as Workflow10 } from "@workglow/task-graph";
-var modelSchema9 = TypeReplicateArray(TypeModel("model:TextTranslationTask"));
+import { CreateWorkflow as CreateWorkflow12, TaskRegistry as TaskRegistry12, Workflow as Workflow12 } from "@workglow/task-graph";
+var modelSchema11 = TypeReplicateArray(TypeModel("model:TextTranslationTask"));
 var translationTextSchema = {
   type: "string",
   title: "Text",
@@ -1289,7 +1423,7 @@ var TextTranslationInputSchema = {
       minLength: 2,
       maxLength: 2
     })),
-    model: modelSchema9
+    model: modelSchema11
   },
   required: ["text", "source_lang", "target_lang", "model"],
   additionalProperties: false
@@ -1325,18 +1459,18 @@ class TextTranslationTask extends AiTask {
     return TextTranslationOutputSchema;
   }
 }
-TaskRegistry10.registerTask(TextTranslationTask);
+TaskRegistry12.registerTask(TextTranslationTask);
 var TextTranslation = (input, config) => {
   return new TextTranslationTask(input, config).run();
 };
-Workflow10.prototype.TextTranslation = CreateWorkflow10(TextTranslationTask);
+Workflow12.prototype.TextTranslation = CreateWorkflow12(TextTranslationTask);
 // src/task/UnloadModelTask.ts
-import { CreateWorkflow as CreateWorkflow11, TaskRegistry as TaskRegistry11, Workflow as Workflow11 } from "@workglow/task-graph";
-var modelSchema10 = TypeReplicateArray(TypeModel("model"));
+import { CreateWorkflow as CreateWorkflow13, TaskRegistry as TaskRegistry13, Workflow as Workflow13 } from "@workglow/task-graph";
+var modelSchema12 = TypeReplicateArray(TypeModel("model"));
 var UnloadModelInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema10
+    model: modelSchema12
   },
   required: ["model"],
   additionalProperties: false
@@ -1344,7 +1478,7 @@ var UnloadModelInputSchema = {
 var UnloadModelOutputSchema = {
   type: "object",
   properties: {
-    model: modelSchema10
+    model: modelSchema12
   },
   required: ["model"],
   additionalProperties: false
@@ -1363,18 +1497,18 @@ class UnloadModelTask extends AiTask {
   }
   static cacheable = false;
 }
-TaskRegistry11.registerTask(UnloadModelTask);
+TaskRegistry13.registerTask(UnloadModelTask);
 var UnloadModel = (input, config) => {
   return new UnloadModelTask(input, config).run();
 };
-Workflow11.prototype.UnloadModel = CreateWorkflow11(UnloadModelTask);
+Workflow13.prototype.UnloadModel = CreateWorkflow13(UnloadModelTask);
 // src/task/VectorSimilarityTask.ts
 import {
   ArrayTask,
-  CreateWorkflow as CreateWorkflow12,
+  CreateWorkflow as CreateWorkflow14,
   TaskError,
-  TaskRegistry as TaskRegistry12,
-  Workflow as Workflow12
+  TaskRegistry as TaskRegistry14,
+  Workflow as Workflow14
 } from "@workglow/task-graph";
 var SimilarityFn = {
   COSINE: "cosine",
@@ -1468,11 +1602,11 @@ class VectorSimilarityTask extends ArrayTask {
     };
   }
 }
-TaskRegistry12.registerTask(VectorSimilarityTask);
+TaskRegistry14.registerTask(VectorSimilarityTask);
 var Similarity = (input, config) => {
   return new VectorSimilarityTask(input, config).run();
 };
-Workflow12.prototype.Similarity = CreateWorkflow12(VectorSimilarityTask);
+Workflow14.prototype.Similarity = CreateWorkflow14(VectorSimilarityTask);
 function inner(arr1, arr2) {
   return 1 - arr1.reduce((acc, val, i) => acc + val * arr2[i], 0);
 }
@@ -1532,6 +1666,10 @@ export {
   TextQuestionAnswerOutputSchema,
   TextQuestionAnswerInputSchema,
   TextQuestionAnswer,
+  TextNamedEntityRecognitionTask,
+  TextNamedEntityRecognitionOutputSchema,
+  TextNamedEntityRecognitionInputSchema,
+  TextNamedEntityRecognition,
   TextLanguageDetectionTask,
   TextLanguageDetectionOutputSchema,
   TextLanguageDetectionInputSchema,
@@ -1541,6 +1679,10 @@ export {
   TextGenerationInputSchema,
   TextGeneration,
   TextFragment,
+  TextFillMaskTask,
+  TextFillMaskOutputSchema,
+  TextFillMaskInputSchema,
+  TextFillMask,
   TextEmbeddingTask,
   TextEmbeddingOutputSchema,
   TextEmbeddingInputSchema,
@@ -1572,4 +1714,4 @@ export {
   AiJob
 };
 
-//# debugId=A1B6F27F96F1B91464756E2164756E21
+//# debugId=4EF87B775AAC634764756E2164756E21