@workglow/ai 0.0.65 → 0.0.67

package/dist/bun.js

@@ -22,6 +22,9 @@ class ModelRepository {
     this.modelTabularRepository = modelTabularRepository;
   }
   events = new EventEmitter;
+  async setupDatabase() {
+    await this.modelTabularRepository.setupDatabase?.();
+  }
   on(name, fn) {
     this.events.on(name, fn);
   }
@@ -814,9 +817,86 @@ var DownloadModel = (input, config) => {
   return new DownloadModelTask(input, config).run();
 };
 Workflow2.prototype.DownloadModel = CreateWorkflow2(DownloadModelTask);
-// src/task/TextEmbeddingTask.ts
+// src/task/TextClassifierTask.ts
 import { CreateWorkflow as CreateWorkflow3, TaskRegistry as TaskRegistry3, Workflow as Workflow3 } from "@workglow/task-graph";
-var modelSchema2 = TypeReplicateArray(TypeModel("model:TextEmbeddingTask"));
+var modelSchema2 = TypeReplicateArray(TypeModel("model:TextClassifierTask"));
+var TextClassifierInputSchema = {
+  type: "object",
+  properties: {
+    text: TypeReplicateArray({
+      type: "string",
+      title: "Text",
+      description: "The text to classify"
+    }),
+    maxCategories: {
+      oneOf: [
+        {
+          type: "number",
+          minimum: 1,
+          maximum: 1000
+        },
+        {
+          type: "null"
+        }
+      ],
+      title: "Max Categories",
+      description: "The maximum number of categories to return"
+    },
+    model: modelSchema2
+  },
+  required: ["text", "model"],
+  additionalProperties: false
+};
+var TextClassifierOutputSchema = {
+  type: "object",
+  properties: {
+    categories: {
+      type: "array",
+      items: {
+        type: "object",
+        properties: {
+          label: {
+            type: "string",
+            title: "Label",
+            description: "The name of the category"
+          },
+          score: {
+            type: "number",
+            title: "Score",
+            description: "The confidence score for this category"
+          }
+        },
+        required: ["label", "score"],
+        additionalProperties: false
+      },
+      title: "Categories",
+      description: "The classification categories with their scores"
+    }
+  },
+  required: ["categories"],
+  additionalProperties: false
+};
+
+class TextClassifierTask extends AiTask {
+  static type = "TextClassifierTask";
+  static category = "AI Text Model";
+  static title = "Text Classifier";
+  static description = "Classifies text into predefined categories using language models";
+  static inputSchema() {
+    return TextClassifierInputSchema;
+  }
+  static outputSchema() {
+    return TextClassifierOutputSchema;
+  }
+}
+TaskRegistry3.registerTask(TextClassifierTask);
+var TextClassifier = (input, config) => {
+  return new TextClassifierTask(input, config).run();
+};
+Workflow3.prototype.TextClassifier = CreateWorkflow3(TextClassifierTask);
+// src/task/TextEmbeddingTask.ts
+import { CreateWorkflow as CreateWorkflow4, TaskRegistry as TaskRegistry4, Workflow as Workflow4 } from "@workglow/task-graph";
+var modelSchema3 = TypeReplicateArray(TypeModel("model:TextEmbeddingTask"));
 var TextEmbeddingInputSchema = {
   type: "object",
   properties: {
@@ -825,7 +905,7 @@ var TextEmbeddingInputSchema = {
       title: "Text",
       description: "The text to embed"
     }),
-    model: modelSchema2
+    model: modelSchema3
   },
   required: ["text", "model"],
   additionalProperties: false
@@ -854,23 +934,23 @@ class TextEmbeddingTask extends AiTask {
     return TextEmbeddingOutputSchema;
   }
 }
-TaskRegistry3.registerTask(TextEmbeddingTask);
+TaskRegistry4.registerTask(TextEmbeddingTask);
 var TextEmbedding = async (input, config) => {
   return new TextEmbeddingTask(input, config).run();
 };
-Workflow3.prototype.TextEmbedding = CreateWorkflow3(TextEmbeddingTask);
+Workflow4.prototype.TextEmbedding = CreateWorkflow4(TextEmbeddingTask);
 // src/task/TextGenerationTask.ts
-import { CreateWorkflow as CreateWorkflow4, TaskRegistry as TaskRegistry4, Workflow as Workflow4 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow5, TaskRegistry as TaskRegistry5, Workflow as Workflow5 } from "@workglow/task-graph";
 var generatedTextSchema = {
   type: "string",
   title: "Text",
   description: "The generated text"
 };
-var modelSchema3 = TypeReplicateArray(TypeModel("model:TextGenerationTask"));
+var modelSchema4 = TypeReplicateArray(TypeModel("model:TextGenerationTask"));
 var TextGenerationInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema3,
+    model: modelSchema4,
     prompt: TypeReplicateArray({
       type: "string",
       title: "Prompt",
@@ -945,13 +1025,90 @@ class TextGenerationTask extends AiTask {
     return TextGenerationOutputSchema;
   }
 }
-TaskRegistry4.registerTask(TextGenerationTask);
+TaskRegistry5.registerTask(TextGenerationTask);
 var TextGeneration = (input, config) => {
   return new TextGenerationTask(input, config).run();
 };
-Workflow4.prototype.TextGeneration = CreateWorkflow4(TextGenerationTask);
+Workflow5.prototype.TextGeneration = CreateWorkflow5(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"));
+var TextLanguageDetectionInputSchema = {
+  type: "object",
+  properties: {
+    text: TypeReplicateArray({
+      type: "string",
+      title: "Text",
+      description: "The text to detect the language of"
+    }),
+    maxLanguages: {
+      oneOf: [
+        {
+          type: "number",
+          minimum: 1,
+          maximum: 1000
+        },
+        {
+          type: "null"
+        }
+      ],
+      title: "Max Languages",
+      description: "The maximum number of languages to return"
+    },
+    model: modelSchema5
+  },
+  required: ["text", "model"],
+  additionalProperties: false
+};
+var TextLanguageDetectionOutputSchema = {
+  type: "object",
+  properties: {
+    languages: {
+      type: "array",
+      items: {
+        type: "object",
+        properties: {
+          language: {
+            type: "string",
+            title: "Language",
+            description: "The language"
+          },
+          score: {
+            type: "number",
+            title: "Score",
+            description: "The confidence score for this language"
+          }
+        },
+        required: ["language", "score"],
+        additionalProperties: false
+      },
+      title: "Languages",
+      description: "The languages with their scores"
+    }
+  },
+  required: ["languages"],
+  additionalProperties: false
+};
+
+class TextLanguageDetectionTask extends AiTask {
+  static type = "TextLanguageDetectionTask";
+  static category = "AI Text Model";
+  static title = "Language Detection";
+  static description = "Detects the language of text using language models";
+  static inputSchema() {
+    return TextLanguageDetectionInputSchema;
+  }
+  static outputSchema() {
+    return TextLanguageDetectionOutputSchema;
+  }
+}
+TaskRegistry6.registerTask(TextLanguageDetectionTask);
+var TextLanguageDetection = (input, config) => {
+  return new TextLanguageDetectionTask(input, config).run();
+};
+Workflow6.prototype.TextLanguageDetection = CreateWorkflow6(TextLanguageDetectionTask);
 // src/task/TextQuestionAnswerTask.ts
-import { CreateWorkflow as CreateWorkflow5, TaskRegistry as TaskRegistry5, Workflow as Workflow5 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow7, TaskRegistry as TaskRegistry7, Workflow as Workflow7 } from "@workglow/task-graph";
 var contextSchema = {
   type: "string",
   title: "Context",
@@ -967,13 +1124,13 @@ var textSchema = {
   title: "Text",
   description: "The generated text"
 };
-var modelSchema4 = TypeReplicateArray(TypeModel("model:TextQuestionAnswerTask"));
+var modelSchema6 = TypeReplicateArray(TypeModel("model:TextQuestionAnswerTask"));
 var TextQuestionAnswerInputSchema = {
   type: "object",
   properties: {
     context: TypeReplicateArray(contextSchema),
     question: TypeReplicateArray(questionSchema),
-    model: modelSchema4
+    model: modelSchema6
   },
   required: ["context", "question", "model"],
   additionalProperties: false
@@ -1003,14 +1160,14 @@ class TextQuestionAnswerTask extends AiTask {
     return TextQuestionAnswerOutputSchema;
   }
 }
-TaskRegistry5.registerTask(TextQuestionAnswerTask);
+TaskRegistry7.registerTask(TextQuestionAnswerTask);
 var TextQuestionAnswer = (input, config) => {
   return new TextQuestionAnswerTask(input, config).run();
 };
-Workflow5.prototype.TextQuestionAnswer = CreateWorkflow5(TextQuestionAnswerTask);
+Workflow7.prototype.TextQuestionAnswer = CreateWorkflow7(TextQuestionAnswerTask);
 // src/task/TextRewriterTask.ts
-import { CreateWorkflow as CreateWorkflow6, TaskRegistry as TaskRegistry6, Workflow as Workflow6 } from "@workglow/task-graph";
-var modelSchema5 = TypeReplicateArray(TypeModel("model:TextRewriterTask"));
+import { CreateWorkflow as CreateWorkflow8, TaskRegistry as TaskRegistry8, Workflow as Workflow8 } from "@workglow/task-graph";
+var modelSchema7 = TypeReplicateArray(TypeModel("model:TextRewriterTask"));
 var TextRewriterInputSchema = {
   type: "object",
   properties: {
@@ -1024,7 +1181,7 @@ var TextRewriterInputSchema = {
       title: "Prompt",
       description: "The prompt to direct the rewriting"
     }),
-    model: modelSchema5
+    model: modelSchema7
   },
   required: ["text", "prompt", "model"],
   additionalProperties: false
@@ -1054,14 +1211,14 @@ class TextRewriterTask extends AiTask {
     return TextRewriterOutputSchema;
   }
 }
-TaskRegistry6.registerTask(TextRewriterTask);
+TaskRegistry8.registerTask(TextRewriterTask);
 var TextRewriter = (input, config) => {
   return new TextRewriterTask(input, config).run();
 };
-Workflow6.prototype.TextRewriter = CreateWorkflow6(TextRewriterTask);
+Workflow8.prototype.TextRewriter = CreateWorkflow8(TextRewriterTask);
 // src/task/TextSummaryTask.ts
-import { CreateWorkflow as CreateWorkflow7, TaskRegistry as TaskRegistry7, Workflow as Workflow7 } from "@workglow/task-graph";
-var modelSchema6 = TypeReplicateArray(TypeModel("model:TextSummaryTask"));
+import { CreateWorkflow as CreateWorkflow9, TaskRegistry as TaskRegistry9, Workflow as Workflow9 } from "@workglow/task-graph";
+var modelSchema8 = TypeReplicateArray(TypeModel("model:TextSummaryTask"));
 var TextSummaryInputSchema = {
   type: "object",
   properties: {
@@ -1070,7 +1227,7 @@ var TextSummaryInputSchema = {
       title: "Text",
       description: "The text to summarize"
     }),
-    model: modelSchema6
+    model: modelSchema8
   },
   required: ["text", "model"],
   additionalProperties: false
@@ -1100,14 +1257,14 @@ class TextSummaryTask extends AiTask {
     return TextSummaryOutputSchema;
   }
 }
-TaskRegistry7.registerTask(TextSummaryTask);
+TaskRegistry9.registerTask(TextSummaryTask);
 var TextSummary = async (input, config) => {
   return new TextSummaryTask(input, config).run();
 };
-Workflow7.prototype.TextSummary = CreateWorkflow7(TextSummaryTask);
+Workflow9.prototype.TextSummary = CreateWorkflow9(TextSummaryTask);
 // src/task/TextTranslationTask.ts
-import { CreateWorkflow as CreateWorkflow8, TaskRegistry as TaskRegistry8, Workflow as Workflow8 } from "@workglow/task-graph";
-var modelSchema7 = TypeReplicateArray(TypeModel("model:TextTranslationTask"));
+import { CreateWorkflow as CreateWorkflow10, TaskRegistry as TaskRegistry10, Workflow as Workflow10 } from "@workglow/task-graph";
+var modelSchema9 = TypeReplicateArray(TypeModel("model:TextTranslationTask"));
 var translationTextSchema = {
   type: "string",
   title: "Text",
@@ -1133,7 +1290,7 @@ var TextTranslationInputSchema = {
       minLength: 2,
       maxLength: 2
     })),
-    model: modelSchema7
+    model: modelSchema9
   },
   required: ["text", "source_lang", "target_lang", "model"],
   additionalProperties: false
@@ -1169,18 +1326,56 @@ class TextTranslationTask extends AiTask {
     return TextTranslationOutputSchema;
   }
 }
-TaskRegistry8.registerTask(TextTranslationTask);
+TaskRegistry10.registerTask(TextTranslationTask);
 var TextTranslation = (input, config) => {
   return new TextTranslationTask(input, config).run();
 };
-Workflow8.prototype.TextTranslation = CreateWorkflow8(TextTranslationTask);
+Workflow10.prototype.TextTranslation = CreateWorkflow10(TextTranslationTask);
+// src/task/UnloadModelTask.ts
+import { CreateWorkflow as CreateWorkflow11, TaskRegistry as TaskRegistry11, Workflow as Workflow11 } from "@workglow/task-graph";
+var modelSchema10 = TypeReplicateArray(TypeModel("model"));
+var UnloadModelInputSchema = {
+  type: "object",
+  properties: {
+    model: modelSchema10
+  },
+  required: ["model"],
+  additionalProperties: false
+};
+var UnloadModelOutputSchema = {
+  type: "object",
+  properties: {
+    model: modelSchema10
+  },
+  required: ["model"],
+  additionalProperties: false
+};
+
+class UnloadModelTask extends AiTask {
+  static type = "UnloadModelTask";
+  static category = "Hidden";
+  static title = "Unload Model";
+  static description = "Unloads and clears cached AI models from memory and storage";
+  static inputSchema() {
+    return UnloadModelInputSchema;
+  }
+  static outputSchema() {
+    return UnloadModelOutputSchema;
+  }
+  static cacheable = false;
+}
+TaskRegistry11.registerTask(UnloadModelTask);
+var UnloadModel = (input, config) => {
+  return new UnloadModelTask(input, config).run();
+};
+Workflow11.prototype.UnloadModel = CreateWorkflow11(UnloadModelTask);
 // src/task/VectorSimilarityTask.ts
 import {
   ArrayTask,
-  CreateWorkflow as CreateWorkflow9,
+  CreateWorkflow as CreateWorkflow12,
   TaskError,
-  TaskRegistry as TaskRegistry9,
-  Workflow as Workflow9
+  TaskRegistry as TaskRegistry12,
+  Workflow as Workflow12
 } from "@workglow/task-graph";
 var SimilarityFn = {
   COSINE: "cosine",
@@ -1274,11 +1469,11 @@ class VectorSimilarityTask extends ArrayTask {
     };
   }
 }
-TaskRegistry9.registerTask(VectorSimilarityTask);
+TaskRegistry12.registerTask(VectorSimilarityTask);
 var Similarity = (input, config) => {
   return new VectorSimilarityTask(input, config).run();
 };
-Workflow9.prototype.Similarity = CreateWorkflow9(VectorSimilarityTask);
+Workflow12.prototype.Similarity = CreateWorkflow12(VectorSimilarityTask);
 function inner(arr1, arr2) {
   return 1 - arr1.reduce((acc, val, i) => acc + val * arr2[i], 0);
 }
@@ -1314,6 +1509,8 @@ export {
   getGlobalModelRepository,
   getAiProviderRegistry,
   VectorSimilarityTask,
+  UnloadModelTask,
+  UnloadModel,
   TypedArraySchema,
   TypeReplicateArray,
   TypeModelByDetail,
@@ -1336,6 +1533,10 @@ export {
   TextQuestionAnswerOutputSchema,
   TextQuestionAnswerInputSchema,
   TextQuestionAnswer,
+  TextLanguageDetectionTask,
+  TextLanguageDetectionOutputSchema,
+  TextLanguageDetectionInputSchema,
+  TextLanguageDetection,
   TextGenerationTask,
   TextGenerationOutputSchema,
   TextGenerationInputSchema,
@@ -1345,6 +1546,10 @@ export {
   TextEmbeddingOutputSchema,
   TextEmbeddingInputSchema,
   TextEmbedding,
+  TextClassifierTask,
+  TextClassifierOutputSchema,
+  TextClassifierInputSchema,
+  TextClassifier,
   TableFragment,
   SimilarityFn,
   Similarity,
@@ -1368,4 +1573,4 @@ export {
   AiJob
 };
 
-//# debugId=C2CC732C6401A8CC64756E2164756E21
+//# debugId=9EADFBC77D1F6A3764756E2164756E21