@workglow/ai 0.0.107 → 0.0.109

package/dist/bun.js

@@ -3657,9 +3657,63 @@ var imageToText = (input, config) => {
 };
 Workflow23.prototype.imageToText = CreateWorkflow23(ImageToTextTask);
 
-// src/task/ObjectDetectionTask.ts
+// src/task/ModelInfoTask.ts
 import { CreateWorkflow as CreateWorkflow24, Workflow as Workflow24 } from "@workglow/task-graph";
-var modelSchema17 = TypeModel("model:ObjectDetectionTask");
+var modelSchema17 = TypeModel("model");
+var ModelInfoInputSchema = {
+  type: "object",
+  properties: {
+    model: modelSchema17
+  },
+  required: ["model"],
+  additionalProperties: false
+};
+var ModelInfoOutputSchema = {
+  type: "object",
+  properties: {
+    model: modelSchema17,
+    is_local: { type: "boolean" },
+    is_remote: { type: "boolean" },
+    supports_browser: { type: "boolean" },
+    supports_node: { type: "boolean" },
+    is_cached: { type: "boolean" },
+    is_loaded: { type: "boolean" },
+    file_sizes: {}
+  },
+  required: [
+    "model",
+    "is_local",
+    "is_remote",
+    "supports_browser",
+    "supports_node",
+    "is_cached",
+    "is_loaded",
+    "file_sizes"
+  ],
+  additionalProperties: false
+};
+
+class ModelInfoTask extends AiTask {
+  static type = "ModelInfoTask";
+  static category = "Hidden";
+  static title = "Model Info";
+  static description = "Returns runtime information about a model including locality, cache status, and file sizes";
+  static inputSchema() {
+    return ModelInfoInputSchema;
+  }
+  static outputSchema() {
+    return ModelInfoOutputSchema;
+  }
+  static cacheable = false;
+}
+var modelInfo = (input, config) => {
+  return new ModelInfoTask({}, config).run(input);
+};
+Workflow24.prototype.modelInfo = CreateWorkflow24(ModelInfoTask);
+
+// src/task/ObjectDetectionTask.ts
+import { CreateWorkflow as CreateWorkflow25, Workflow as Workflow25 } from "@workglow/task-graph";
+var modelSchema18 = TypeModel("model:ObjectDetectionTask");
 var detectionSchema = {
   type: "object",
   properties: {
@@ -3684,7 +3738,7 @@ var ObjectDetectionInputSchema = {
   type: "object",
   properties: {
     image: TypeImageInput,
-    model: modelSchema17,
+    model: modelSchema18,
     labels: {
       type: "array",
       items: {
@@ -3738,11 +3792,11 @@ class ObjectDetectionTask extends AiVisionTask {
 var objectDetection = (input, config) => {
   return new ObjectDetectionTask({}, config).run(input);
 };
-Workflow24.prototype.objectDetection = CreateWorkflow24(ObjectDetectionTask);
+Workflow25.prototype.objectDetection = CreateWorkflow25(ObjectDetectionTask);
 
 // src/task/PoseLandmarkerTask.ts
-import { CreateWorkflow as CreateWorkflow25, Workflow as Workflow25 } from "@workglow/task-graph";
-var modelSchema18 = TypeModel("model:PoseLandmarkerTask");
+import { CreateWorkflow as CreateWorkflow26, Workflow as Workflow26 } from "@workglow/task-graph";
+var modelSchema19 = TypeModel("model:PoseLandmarkerTask");
 var TypePoseLandmark = {
   type: "object",
   properties: {
@@ -3821,7 +3875,7 @@ var PoseLandmarkerInputSchema = {
   type: "object",
   properties: {
     image: TypeImageInput,
-    model: modelSchema18,
+    model: modelSchema19,
     numPoses: {
       type: "number",
       minimum: 1,
@@ -3900,13 +3954,13 @@ class PoseLandmarkerTask extends AiVisionTask {
 var poseLandmarker = (input, config) => {
   return new PoseLandmarkerTask({}, config).run(input);
 };
-Workflow25.prototype.poseLandmarker = CreateWorkflow25(PoseLandmarkerTask);
+Workflow26.prototype.poseLandmarker = CreateWorkflow26(PoseLandmarkerTask);
 
 // src/task/QueryExpanderTask.ts
 import {
-  CreateWorkflow as CreateWorkflow26,
+  CreateWorkflow as CreateWorkflow27,
   Task as Task10,
-  Workflow as Workflow26
+  Workflow as Workflow27
 } from "@workglow/task-graph";
 var QueryExpansionMethod = {
   MULTI_QUERY: "multi-query",
@@ -4117,18 +4171,18 @@ class QueryExpanderTask extends Task10 {
 var queryExpander = (input, config) => {
   return new QueryExpanderTask({}, config).run(input);
 };
-Workflow26.prototype.queryExpander = CreateWorkflow26(QueryExpanderTask);
+Workflow27.prototype.queryExpander = CreateWorkflow27(QueryExpanderTask);
 
 // src/task/RerankerTask.ts
 import {
-  CreateWorkflow as CreateWorkflow28,
+  CreateWorkflow as CreateWorkflow29,
   Task as Task11,
-  Workflow as Workflow28
+  Workflow as Workflow29
 } from "@workglow/task-graph";
 
 // src/task/TextClassificationTask.ts
-import { CreateWorkflow as CreateWorkflow27, Workflow as Workflow27 } from "@workglow/task-graph";
-var modelSchema19 = TypeModel("model:TextClassificationTask");
+import { CreateWorkflow as CreateWorkflow28, Workflow as Workflow28 } from "@workglow/task-graph";
+var modelSchema20 = TypeModel("model:TextClassificationTask");
 var TextClassificationInputSchema = {
   type: "object",
   properties: {
@@ -4155,7 +4209,7 @@ var TextClassificationInputSchema = {
       description: "The maximum number of categories to return",
       "x-ui-group": "Configuration"
     },
-    model: modelSchema19
+    model: modelSchema20
   },
   required: ["text", "model"],
   additionalProperties: false
@@ -4205,7 +4259,7 @@ class TextClassificationTask extends AiTask {
 var textClassification = (input, config) => {
   return new TextClassificationTask({}, config).run(input);
 };
-Workflow27.prototype.textClassification = CreateWorkflow27(TextClassificationTask);
+Workflow28.prototype.textClassification = CreateWorkflow28(TextClassificationTask);
 
 // src/task/RerankerTask.ts
 var inputSchema11 = {
@@ -4429,14 +4483,14 @@ class RerankerTask extends Task11 {
 var reranker = (input, config) => {
   return new RerankerTask({}, config).run(input);
 };
-Workflow28.prototype.reranker = CreateWorkflow28(RerankerTask);
+Workflow29.prototype.reranker = CreateWorkflow29(RerankerTask);
 
 // src/task/StructuralParserTask.ts
 import { StructuralParser } from "@workglow/dataset";
 import {
-  CreateWorkflow as CreateWorkflow29,
+  CreateWorkflow as CreateWorkflow30,
   Task as Task12,
-  Workflow as Workflow29
+  Workflow as Workflow30
 } from "@workglow/task-graph";
 import { uuid4 as uuid42 } from "@workglow/util";
 var inputSchema12 = {
@@ -4538,15 +4592,15 @@ class StructuralParserTask extends Task12 {
 var structuralParser = (input, config) => {
   return new StructuralParserTask({}, config).run(input);
 };
-Workflow29.prototype.structuralParser = CreateWorkflow29(StructuralParserTask);
+Workflow30.prototype.structuralParser = CreateWorkflow30(StructuralParserTask);
 
 // src/task/StructuredGenerationTask.ts
-import { CreateWorkflow as CreateWorkflow30, Workflow as Workflow30 } from "@workglow/task-graph";
-var modelSchema20 = TypeModel("model:StructuredGenerationTask");
+import { CreateWorkflow as CreateWorkflow31, Workflow as Workflow31 } from "@workglow/task-graph";
+var modelSchema21 = TypeModel("model:StructuredGenerationTask");
 var StructuredGenerationInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema20,
+    model: modelSchema21,
     prompt: {
       type: "string",
       title: "Prompt",
@@ -4609,13 +4663,13 @@ class StructuredGenerationTask extends StreamingAiTask {
 var structuredGeneration = (input, config) => {
   return new StructuredGenerationTask({}, config).run(input);
 };
-Workflow30.prototype.structuredGeneration = CreateWorkflow30(StructuredGenerationTask);
+Workflow31.prototype.structuredGeneration = CreateWorkflow31(StructuredGenerationTask);
 
 // src/task/TextChunkerTask.ts
 import {
-  CreateWorkflow as CreateWorkflow31,
+  CreateWorkflow as CreateWorkflow32,
   Task as Task13,
-  Workflow as Workflow31
+  Workflow as Workflow32
 } from "@workglow/task-graph";
 var ChunkingStrategy = {
   FIXED: "fixed",
@@ -4865,11 +4919,11 @@ class TextChunkerTask extends Task13 {
 var textChunker = (input, config) => {
   return new TextChunkerTask({}, config).run(input);
 };
-Workflow31.prototype.textChunker = CreateWorkflow31(TextChunkerTask);
+Workflow32.prototype.textChunker = CreateWorkflow32(TextChunkerTask);
 
 // src/task/TextFillMaskTask.ts
-import { CreateWorkflow as CreateWorkflow32, Workflow as Workflow32 } from "@workglow/task-graph";
-var modelSchema21 = TypeModel("model:TextFillMaskTask");
+import { CreateWorkflow as CreateWorkflow33, Workflow as Workflow33 } from "@workglow/task-graph";
+var modelSchema22 = TypeModel("model:TextFillMaskTask");
 var TextFillMaskInputSchema = {
   type: "object",
   properties: {
@@ -4878,7 +4932,7 @@ var TextFillMaskInputSchema = {
       title: "Text",
       description: "The text with a mask token to fill"
     },
-    model: modelSchema21
+    model: modelSchema22
   },
   required: ["text", "model"],
   additionalProperties: false
@@ -4933,21 +4987,21 @@ class TextFillMaskTask extends AiTask {
 var textFillMask = (input, config) => {
   return new TextFillMaskTask({}, config).run(input);
 };
-Workflow32.prototype.textFillMask = CreateWorkflow32(TextFillMaskTask);
+Workflow33.prototype.textFillMask = CreateWorkflow33(TextFillMaskTask);
 
 // src/task/TextGenerationTask.ts
-import { CreateWorkflow as CreateWorkflow33, Workflow as Workflow33 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow34, Workflow as Workflow34 } from "@workglow/task-graph";
 var generatedTextSchema2 = {
   type: "string",
   title: "Text",
   description: "The generated text",
   "x-stream": "append"
 };
-var modelSchema22 = TypeModel("model:TextGenerationTask");
+var modelSchema23 = TypeModel("model:TextGenerationTask");
 var TextGenerationInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema22,
+    model: modelSchema23,
     prompt: {
       type: "string",
       title: "Prompt",
@@ -5021,11 +5075,11 @@ class TextGenerationTask extends StreamingAiTask {
 var textGeneration = (input, config) => {
   return new TextGenerationTask({}, config).run(input);
 };
-Workflow33.prototype.textGeneration = CreateWorkflow33(TextGenerationTask);
+Workflow34.prototype.textGeneration = CreateWorkflow34(TextGenerationTask);
 
 // src/task/TextLanguageDetectionTask.ts
-import { CreateWorkflow as CreateWorkflow34, Workflow as Workflow34 } from "@workglow/task-graph";
-var modelSchema23 = TypeModel("model:TextLanguageDetectionTask");
+import { CreateWorkflow as CreateWorkflow35, Workflow as Workflow35 } from "@workglow/task-graph";
+var modelSchema24 = TypeModel("model:TextLanguageDetectionTask");
 var TextLanguageDetectionInputSchema = {
   type: "object",
   properties: {
@@ -5042,7 +5096,7 @@ var TextLanguageDetectionInputSchema = {
       title: "Max Languages",
       description: "The maximum number of languages to return"
     },
-    model: modelSchema23
+    model: modelSchema24
   },
   required: ["text", "model"],
   additionalProperties: false
@@ -5092,10 +5146,10 @@ class TextLanguageDetectionTask extends AiTask {
 var textLanguageDetection = (input, config) => {
   return new TextLanguageDetectionTask({}, config).run(input);
 };
-Workflow34.prototype.textLanguageDetection = CreateWorkflow34(TextLanguageDetectionTask);
+Workflow35.prototype.textLanguageDetection = CreateWorkflow35(TextLanguageDetectionTask);
 
 // src/task/TextQuestionAnswerTask.ts
-import { CreateWorkflow as CreateWorkflow35, Workflow as Workflow35 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow36, Workflow as Workflow36 } from "@workglow/task-graph";
 var contextSchema = {
   type: "string",
   title: "Context",
@@ -5112,13 +5166,13 @@ var textSchema = {
   description: "The generated text",
   "x-stream": "append"
 };
-var modelSchema24 = TypeModel("model:TextQuestionAnswerTask");
+var modelSchema25 = TypeModel("model:TextQuestionAnswerTask");
 var TextQuestionAnswerInputSchema = {
   type: "object",
   properties: {
     context: contextSchema,
     question: questionSchema,
-    model: modelSchema24
+    model: modelSchema25
   },
   required: ["context", "question", "model"],
   additionalProperties: false
@@ -5147,11 +5201,11 @@ class TextQuestionAnswerTask extends StreamingAiTask {
 var textQuestionAnswer = (input, config) => {
   return new TextQuestionAnswerTask({}, config).run(input);
 };
-Workflow35.prototype.textQuestionAnswer = CreateWorkflow35(TextQuestionAnswerTask);
+Workflow36.prototype.textQuestionAnswer = CreateWorkflow36(TextQuestionAnswerTask);
 
 // src/task/TextRewriterTask.ts
-import { CreateWorkflow as CreateWorkflow36, Workflow as Workflow36 } from "@workglow/task-graph";
-var modelSchema25 = TypeModel("model:TextRewriterTask");
+import { CreateWorkflow as CreateWorkflow37, Workflow as Workflow37 } from "@workglow/task-graph";
+var modelSchema26 = TypeModel("model:TextRewriterTask");
 var TextRewriterInputSchema = {
   type: "object",
   properties: {
@@ -5165,7 +5219,7 @@ var TextRewriterInputSchema = {
       title: "Prompt",
       description: "The prompt to direct the rewriting"
     },
-    model: modelSchema25
+    model: modelSchema26
   },
   required: ["text", "prompt", "model"],
   additionalProperties: false
@@ -5199,10 +5253,10 @@ class TextRewriterTask extends StreamingAiTask {
 var textRewriter = (input, config) => {
   return new TextRewriterTask({}, config).run(input);
 };
-Workflow36.prototype.textRewriter = CreateWorkflow36(TextRewriterTask);
+Workflow37.prototype.textRewriter = CreateWorkflow37(TextRewriterTask);
 
 // src/task/ToolCallingTask.ts
-import { CreateWorkflow as CreateWorkflow37, TaskRegistry, Workflow as Workflow37 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow38, TaskRegistry, Workflow as Workflow38 } from "@workglow/task-graph";
 import { getLogger } from "@workglow/util";
 function buildToolDescription(tool) {
   let desc = tool.description;
@@ -5274,11 +5328,11 @@ var ToolDefinitionSchema = {
   required: ["name", "description", "inputSchema"],
   additionalProperties: false
 };
-var modelSchema26 = TypeModel("model:ToolCallingTask");
+var modelSchema27 = TypeModel("model:ToolCallingTask");
 var ToolCallingInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema26,
+    model: modelSchema27,
     prompt: {
       type: "string",
       title: "Prompt",
@@ -5356,11 +5410,11 @@ class ToolCallingTask extends StreamingAiTask {
 var toolCalling = (input, config) => {
   return new ToolCallingTask({}, config).run(input);
 };
-Workflow37.prototype.toolCalling = CreateWorkflow37(ToolCallingTask);
+Workflow38.prototype.toolCalling = CreateWorkflow38(ToolCallingTask);
 
 // src/task/TextTranslationTask.ts
-import { CreateWorkflow as CreateWorkflow38, Workflow as Workflow38 } from "@workglow/task-graph";
-var modelSchema27 = TypeModel("model:TextTranslationTask");
+import { CreateWorkflow as CreateWorkflow39, Workflow as Workflow39 } from "@workglow/task-graph";
+var modelSchema28 = TypeModel("model:TextTranslationTask");
 var translationTextSchema = {
   type: "string",
   title: "Text",
@@ -5387,7 +5441,7 @@ var TextTranslationInputSchema = {
       minLength: 2,
       maxLength: 2
     }),
-    model: modelSchema27
+    model: modelSchema28
   },
   required: ["text", "source_lang", "target_lang", "model"],
   additionalProperties: false
@@ -5422,13 +5476,13 @@ class TextTranslationTask extends StreamingAiTask {
 var textTranslation = (input, config) => {
   return new TextTranslationTask({}, config).run(input);
 };
-Workflow38.prototype.textTranslation = CreateWorkflow38(TextTranslationTask);
+Workflow39.prototype.textTranslation = CreateWorkflow39(TextTranslationTask);
 
 // src/task/TopicSegmenterTask.ts
 import {
-  CreateWorkflow as CreateWorkflow39,
+  CreateWorkflow as CreateWorkflow40,
   Task as Task14,
-  Workflow as Workflow39
+  Workflow as Workflow40
 } from "@workglow/task-graph";
 var SegmentationMethod = {
   HEURISTIC: "heuristic",
@@ -5709,15 +5763,15 @@ class TopicSegmenterTask extends Task14 {
 var topicSegmenter = (input, config) => {
   return new TopicSegmenterTask({}, config).run(input);
 };
-Workflow39.prototype.topicSegmenter = CreateWorkflow39(TopicSegmenterTask);
+Workflow40.prototype.topicSegmenter = CreateWorkflow40(TopicSegmenterTask);
 
 // src/task/UnloadModelTask.ts
-import { CreateWorkflow as CreateWorkflow40, Workflow as Workflow40 } from "@workglow/task-graph";
-var modelSchema28 = TypeModel("model");
+import { CreateWorkflow as CreateWorkflow41, Workflow as Workflow41 } from "@workglow/task-graph";
+var modelSchema29 = TypeModel("model");
 var UnloadModelInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema28
+    model: modelSchema29
   },
   required: ["model"],
   additionalProperties: false
@@ -5725,7 +5779,7 @@ var UnloadModelInputSchema = {
 var UnloadModelOutputSchema = {
   type: "object",
   properties: {
-    model: modelSchema28
+    model: modelSchema29
   },
   required: ["model"],
   additionalProperties: false
@@ -5747,10 +5801,10 @@ class UnloadModelTask extends AiTask {
 var unloadModel = (input, config) => {
   return new UnloadModelTask({}, config).run(input);
 };
-Workflow40.prototype.unloadModel = CreateWorkflow40(UnloadModelTask);
+Workflow41.prototype.unloadModel = CreateWorkflow41(UnloadModelTask);
 
 // src/task/VectorQuantizeTask.ts
-import { CreateWorkflow as CreateWorkflow41, Task as Task15, Workflow as Workflow41 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow42, Task as Task15, Workflow as Workflow42 } from "@workglow/task-graph";
 import {
   normalizeNumberArray,
   TensorType,
@@ -5930,10 +5984,10 @@ class VectorQuantizeTask extends Task15 {
 var vectorQuantize = (input, config) => {
   return new VectorQuantizeTask({}, config).run(input);
 };
-Workflow41.prototype.vectorQuantize = CreateWorkflow41(VectorQuantizeTask);
+Workflow42.prototype.vectorQuantize = CreateWorkflow42(VectorQuantizeTask);
 
 // src/task/VectorSimilarityTask.ts
-import { CreateWorkflow as CreateWorkflow42, GraphAsTask, Workflow as Workflow42 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow43, GraphAsTask, Workflow as Workflow43 } from "@workglow/task-graph";
 import {
   cosineSimilarity,
   hammingSimilarity,
@@ -6039,7 +6093,7 @@ class VectorSimilarityTask extends GraphAsTask {
 var similarity = (input, config) => {
   return new VectorSimilarityTask({}, config).run(input);
 };
-Workflow42.prototype.similarity = CreateWorkflow42(VectorSimilarityTask);
+Workflow43.prototype.similarity = CreateWorkflow43(VectorSimilarityTask);
 
 // src/task/index.ts
 var registerAiTasks = () => {
@@ -6064,6 +6118,7 @@ var registerAiTasks = () => {
     ImageEmbeddingTask,
     ImageSegmentationTask,
     ImageToTextTask,
+    ModelInfoTask,
     ObjectDetectionTask,
     PoseLandmarkerTask,
     QueryExpanderTask,
@@ -6118,6 +6173,7 @@ export {
   queryExpander,
   poseLandmarker,
   objectDetection,
+  modelInfo,
   isAllowedToolName,
   imageToText,
   imageSegmentation,
@@ -6208,6 +6264,7 @@ export {
   ModelRepository,
   ModelRecordSchema,
   ModelPrimaryKeyNames,
+  ModelInfoTask,
   ModelConfigSchema,
   MODEL_REPOSITORY,
   InMemoryModelRepository,
@@ -6259,4 +6316,4 @@ export {
   AiJob
 };
 
-//# debugId=63F068BFF8E5F20864756E2164756E21
+//# debugId=30EA03AC63C4C0B564756E2164756E21