@workglow/ai 0.0.109 → 0.0.111

package/dist/bun.js

@@ -322,7 +322,7 @@ async function createDefaultQueue(providerName, concurrency) {
   const server = new JobQueueServer(AiJob, {
     storage,
     queueName: providerName,
-    limiter: new ConcurrencyLimiter(concurrency, 100)
+    limiter: new ConcurrencyLimiter(concurrency)
   });
   const client = new JobQueueClient({
     storage,
@@ -420,7 +420,7 @@ class AiProvider {
   }
 }
 // src/task/index.ts
-import { TaskRegistry as TaskRegistry2 } from "@workglow/task-graph";
+import { TaskRegistry } from "@workglow/task-graph";
 
 // src/task/BackgroundRemovalTask.ts
 import { CreateWorkflow, Workflow } from "@workglow/task-graph";
@@ -601,6 +601,7 @@ class AiTask extends JobQueueTask {
     const modelLabel = typeof input.model === "string" ? input.model : typeof input.model === "object" && input.model ? input.model.model_id || input.model.title || input.model.provider : undefined;
     config.title ||= `${new.target.type || new.target.name}${modelLabel ? " with model " + modelLabel : ""}`;
     super(input, config);
+    this.jobClass = AiJob;
   }
   async getJobInput(input) {
     const model = input.model;
@@ -1581,11 +1582,11 @@ var modelSchema3 = TypeModel("model");
 var CountTokensInputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The text to count tokens for"
-    },
+    }),
     model: modelSchema3
   },
   required: ["text", "model"],
@@ -1594,11 +1595,11 @@ var CountTokensInputSchema = {
 var CountTokensOutputSchema = {
   type: "object",
   properties: {
-    count: {
+    count: TypeSingleOrArray({
       type: "number",
       title: "Token Count",
       description: "The number of tokens in the text"
-    }
+    })
   },
   required: ["count"],
   additionalProperties: false
@@ -1939,11 +1940,11 @@ var modelSchema5 = TypeModel("model:TextNamedEntityRecognitionTask");
 var TextNamedEntityRecognitionInputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The text to extract named entities from"
-    },
+    }),
     blockList: {
       type: "array",
       items: {
@@ -1962,7 +1963,7 @@ var TextNamedEntityRecognitionInputSchema = {
 var TextNamedEntityRecognitionOutputSchema = {
   type: "object",
   properties: {
-    entities: {
+    entities: TypeSingleOrArray({
       type: "array",
       items: {
         type: "object",
@@ -1988,7 +1989,7 @@ var TextNamedEntityRecognitionOutputSchema = {
       },
       title: "Entities",
       description: "The extracted named entities with their types, scores, and text"
-    }
+    })
   },
   required: ["entities"],
   additionalProperties: false
@@ -2050,11 +2051,11 @@ var modelSchema6 = TypeModel("model:TextSummaryTask");
 var TextSummaryInputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The text to summarize"
-    },
+    }),
     model: modelSchema6
   },
   required: ["text", "model"],
@@ -2063,12 +2064,12 @@ var TextSummaryInputSchema = {
 var TextSummaryOutputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The summarized text",
       "x-stream": "append"
-    }
+    })
   },
   required: ["text"],
   additionalProperties: false
@@ -3663,7 +3664,12 @@ var modelSchema17 = TypeModel("model");
 var ModelInfoInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema17
+    model: modelSchema17,
+    detail: {
+      type: "string",
+      enum: ["cached_status", "files", "files_with_metadata"],
+      default: "files_with_metadata"
+    }
   },
   required: ["model"],
   additionalProperties: false
@@ -4186,11 +4192,11 @@ var modelSchema20 = TypeModel("model:TextClassificationTask");
 var TextClassificationInputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The text to classify"
-    },
+    }),
     candidateLabels: {
       type: "array",
       items: {
@@ -4217,7 +4223,7 @@ var TextClassificationInputSchema = {
 var TextClassificationOutputSchema = {
   type: "object",
   properties: {
-    categories: {
+    categories: TypeSingleOrArray({
       type: "array",
       items: {
         type: "object",
@@ -4238,7 +4244,7 @@ var TextClassificationOutputSchema = {
       },
       title: "Categories",
       description: "The classification categories with their scores"
-    }
+    })
   },
   required: ["categories"],
   additionalProperties: false
@@ -4927,11 +4933,11 @@ var modelSchema22 = TypeModel("model:TextFillMaskTask");
 var TextFillMaskInputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The text with a mask token to fill"
-    },
+    }),
     model: modelSchema22
   },
   required: ["text", "model"],
@@ -4940,7 +4946,7 @@ var TextFillMaskInputSchema = {
 var TextFillMaskOutputSchema = {
   type: "object",
   properties: {
-    predictions: {
+    predictions: TypeSingleOrArray({
       type: "array",
       items: {
         type: "object",
@@ -4966,7 +4972,7 @@ var TextFillMaskOutputSchema = {
       },
       title: "Predictions",
       description: "The predicted tokens to fill the mask with their scores and complete sequences"
-    }
+    })
   },
   required: ["predictions"],
   additionalProperties: false
@@ -4991,22 +4997,22 @@ Workflow33.prototype.textFillMask = CreateWorkflow33(TextFillMaskTask);
 
 // src/task/TextGenerationTask.ts
 import { CreateWorkflow as CreateWorkflow34, Workflow as Workflow34 } from "@workglow/task-graph";
-var generatedTextSchema2 = {
+var generatedTextSchema2 = TypeSingleOrArray({
   type: "string",
   title: "Text",
   description: "The generated text",
   "x-stream": "append"
-};
+});
 var modelSchema23 = TypeModel("model:TextGenerationTask");
 var TextGenerationInputSchema = {
   type: "object",
   properties: {
     model: modelSchema23,
-    prompt: {
+    prompt: TypeSingleOrArray({
       type: "string",
       title: "Prompt",
       description: "The prompt to generate text from"
-    },
+    }),
     maxTokens: {
       type: "number",
       title: "Max Tokens",
@@ -5083,11 +5089,11 @@ var modelSchema24 = TypeModel("model:TextLanguageDetectionTask");
 var TextLanguageDetectionInputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The text to detect the language of"
-    },
+    }),
     maxLanguages: {
       type: "number",
       minimum: 0,
@@ -5104,7 +5110,7 @@ var TextLanguageDetectionInputSchema = {
 var TextLanguageDetectionOutputSchema = {
   type: "object",
   properties: {
-    languages: {
+    languages: TypeSingleOrArray({
       type: "array",
       items: {
         type: "object",
@@ -5125,7 +5131,7 @@ var TextLanguageDetectionOutputSchema = {
       },
       title: "Languages",
       description: "The languages with their scores"
-    }
+    })
   },
   required: ["languages"],
   additionalProperties: false
@@ -5150,22 +5156,22 @@ Workflow35.prototype.textLanguageDetection = CreateWorkflow35(TextLanguageDetect
 
 // src/task/TextQuestionAnswerTask.ts
 import { CreateWorkflow as CreateWorkflow36, Workflow as Workflow36 } from "@workglow/task-graph";
-var contextSchema = {
+var contextSchema = TypeSingleOrArray({
   type: "string",
   title: "Context",
   description: "The context of the question"
-};
-var questionSchema = {
+});
+var questionSchema = TypeSingleOrArray({
   type: "string",
   title: "Question",
   description: "The question to answer"
-};
-var textSchema = {
+});
+var textSchema = TypeSingleOrArray({
   type: "string",
   title: "Text",
   description: "The generated text",
   "x-stream": "append"
-};
+});
 var modelSchema25 = TypeModel("model:TextQuestionAnswerTask");
 var TextQuestionAnswerInputSchema = {
   type: "object",
@@ -5209,11 +5215,11 @@ var modelSchema26 = TypeModel("model:TextRewriterTask");
 var TextRewriterInputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The text to rewrite"
-    },
+    }),
     prompt: {
       type: "string",
       title: "Prompt",
@@ -5227,12 +5233,12 @@ var TextRewriterInputSchema = {
 var TextRewriterOutputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The rewritten text",
       "x-stream": "append"
-    }
+    })
   },
   required: ["text"],
   additionalProperties: false
@@ -5255,8 +5261,78 @@ var textRewriter = (input, config) => {
 };
 Workflow37.prototype.textRewriter = CreateWorkflow37(TextRewriterTask);
 
+// src/task/TextTranslationTask.ts
+import { CreateWorkflow as CreateWorkflow38, Workflow as Workflow38 } from "@workglow/task-graph";
+var modelSchema27 = TypeModel("model:TextTranslationTask");
+var translationTextSchema = TypeSingleOrArray({
+  type: "string",
+  title: "Text",
+  description: "The translated text",
+  "x-stream": "replace"
+});
+var TextTranslationInputSchema = {
+  type: "object",
+  properties: {
+    text: TypeSingleOrArray({
+      type: "string",
+      title: "Text",
+      description: "The text to translate"
+    }),
+    source_lang: TypeLanguage({
+      title: "Source Language",
+      description: "The source language",
+      minLength: 2,
+      maxLength: 2
+    }),
+    target_lang: TypeLanguage({
+      title: "Target Language",
+      description: "The target language",
+      minLength: 2,
+      maxLength: 2
+    }),
+    model: modelSchema27
+  },
+  required: ["text", "source_lang", "target_lang", "model"],
+  additionalProperties: false
+};
+var TextTranslationOutputSchema = {
+  type: "object",
+  properties: {
+    text: translationTextSchema,
+    target_lang: TypeLanguage({
+      title: "Output Language",
+      description: "The output language",
+      minLength: 2,
+      maxLength: 2
+    })
+  },
+  required: ["text", "target_lang"],
+  additionalProperties: false
+};
+
+class TextTranslationTask extends StreamingAiTask {
+  static type = "TextTranslationTask";
+  static category = "AI Text Model";
+  static title = "Text Translation";
+  static description = "Translates text from one language to another using language models";
+  static inputSchema() {
+    return TextTranslationInputSchema;
+  }
+  static outputSchema() {
+    return TextTranslationOutputSchema;
+  }
+}
+var textTranslation = (input, config) => {
+  return new TextTranslationTask({}, config).run(input);
+};
+Workflow38.prototype.textTranslation = CreateWorkflow38(TextTranslationTask);
+
 // src/task/ToolCallingTask.ts
-import { CreateWorkflow as CreateWorkflow38, TaskRegistry, Workflow as Workflow38 } from "@workglow/task-graph";
+import {
+  CreateWorkflow as CreateWorkflow39,
+  getTaskConstructors,
+  Workflow as Workflow39
+} from "@workglow/task-graph";
 import { getLogger } from "@workglow/util";
 function buildToolDescription(tool) {
   let desc = tool.description;
@@ -5285,11 +5361,12 @@ function filterValidToolCalls(toolCalls, allowedTools) {
   }
   return filtered;
 }
-function taskTypesToTools(taskNames) {
+function taskTypesToTools(taskNames, registry) {
+  const constructors = getTaskConstructors(registry);
   return taskNames.map((name) => {
-    const ctor = TaskRegistry.all.get(name);
+    const ctor = constructors.get(name);
     if (!ctor) {
-      throw new Error(`taskTypesToTools: Unknown task type "${name}" \u2014 not found in TaskRegistry`);
+      throw new Error(`taskTypesToTools: Unknown task type "${name}" \u2014 not found in task constructors registry (ServiceRegistry: ${registry ? "custom" : "default"})`);
     }
     return {
       name: ctor.type,
@@ -5328,16 +5405,16 @@ var ToolDefinitionSchema = {
   required: ["name", "description", "inputSchema"],
   additionalProperties: false
 };
-var modelSchema27 = TypeModel("model:ToolCallingTask");
+var modelSchema28 = TypeModel("model:ToolCallingTask");
 var ToolCallingInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema27,
-    prompt: {
+    model: modelSchema28,
+    prompt: TypeSingleOrArray({
       type: "string",
       title: "Prompt",
       description: "The prompt to send to the model"
-    },
+    }),
     systemPrompt: {
       type: "string",
       title: "System Prompt",
@@ -5345,9 +5422,15 @@ var ToolCallingInputSchema = {
     },
     tools: {
       type: "array",
+      format: "tasks",
       title: "Tools",
       description: "Tool definitions available for the model to call",
-      items: ToolDefinitionSchema
+      items: {
+        oneOf: [
+          { type: "string", format: "tasks", description: "Task type name" },
+          ToolDefinitionSchema
+        ]
+      }
     },
     toolChoice: {
       type: "string",
@@ -5377,19 +5460,19 @@ var ToolCallingInputSchema = {
 var ToolCallingOutputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "Any text content generated by the model",
       "x-stream": "append"
-    },
-    toolCalls: {
+    }),
+    toolCalls: TypeSingleOrArray({
       type: "object",
       title: "Tool Calls",
       description: "Tool invocations requested by the model, keyed by call id",
       additionalProperties: true,
       "x-stream": "object"
-    }
+    })
   },
   required: ["text", "toolCalls"],
   additionalProperties: false
@@ -5410,73 +5493,7 @@ class ToolCallingTask extends StreamingAiTask {
 var toolCalling = (input, config) => {
   return new ToolCallingTask({}, config).run(input);
 };
-Workflow38.prototype.toolCalling = CreateWorkflow38(ToolCallingTask);
-
-// src/task/TextTranslationTask.ts
-import { CreateWorkflow as CreateWorkflow39, Workflow as Workflow39 } from "@workglow/task-graph";
-var modelSchema28 = TypeModel("model:TextTranslationTask");
-var translationTextSchema = {
-  type: "string",
-  title: "Text",
-  description: "The translated text",
-  "x-stream": "replace"
-};
-var TextTranslationInputSchema = {
-  type: "object",
-  properties: {
-    text: {
-      type: "string",
-      title: "Text",
-      description: "The text to translate"
-    },
-    source_lang: TypeLanguage({
-      title: "Source Language",
-      description: "The source language",
-      minLength: 2,
-      maxLength: 2
-    }),
-    target_lang: TypeLanguage({
-      title: "Target Language",
-      description: "The target language",
-      minLength: 2,
-      maxLength: 2
-    }),
-    model: modelSchema28
-  },
-  required: ["text", "source_lang", "target_lang", "model"],
-  additionalProperties: false
-};
-var TextTranslationOutputSchema = {
-  type: "object",
-  properties: {
-    text: translationTextSchema,
-    target_lang: TypeLanguage({
-      title: "Output Language",
-      description: "The output language",
-      minLength: 2,
-      maxLength: 2
-    })
-  },
-  required: ["text", "target_lang"],
-  additionalProperties: false
-};
-
-class TextTranslationTask extends StreamingAiTask {
-  static type = "TextTranslationTask";
-  static category = "AI Text Model";
-  static title = "Text Translation";
-  static description = "Translates text from one language to another using language models";
-  static inputSchema() {
-    return TextTranslationInputSchema;
-  }
-  static outputSchema() {
-    return TextTranslationOutputSchema;
-  }
-}
-var textTranslation = (input, config) => {
-  return new TextTranslationTask({}, config).run(input);
-};
-Workflow39.prototype.textTranslation = CreateWorkflow39(TextTranslationTask);
+Workflow39.prototype.toolCalling = CreateWorkflow39(ToolCallingTask);
 
 // src/task/TopicSegmenterTask.ts
 import {
@@ -6142,7 +6159,7 @@ var registerAiTasks = () => {
     VectorQuantizeTask,
     VectorSimilarityTask
   ];
-  tasks.map(TaskRegistry2.registerTask);
+  tasks.map(TaskRegistry.registerTask);
   return tasks;
 };
 export {
@@ -6211,6 +6228,7 @@ export {
   TypeBoundingBox,
   TypeAudioInput,
   TopicSegmenterTask,
+  ToolDefinitionSchema,
   ToolCallingTask,
   ToolCallingOutputSchema,
   ToolCallingInputSchema,
@@ -6316,4 +6334,4 @@ export {
   AiJob
 };
 
-//# debugId=30EA03AC63C4C0B564756E2164756E21
+//# debugId=0AADE5DC48AE458464756E2164756E21