@workglow/ai 0.0.110 → 0.0.111

package/dist/bun.js

@@ -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";
@@ -5261,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;
@@ -5291,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,
@@ -5334,11 +5405,11 @@ 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,
+    model: modelSchema28,
     prompt: TypeSingleOrArray({
       type: "string",
       title: "Prompt",
@@ -5351,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",
@@ -5416,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 = 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: 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 {
@@ -6148,7 +6159,7 @@ var registerAiTasks = () => {
     VectorQuantizeTask,
     VectorSimilarityTask
   ];
-  tasks.map(TaskRegistry2.registerTask);
+  tasks.map(TaskRegistry.registerTask);
   return tasks;
 };
 export {
@@ -6217,6 +6228,7 @@ export {
   TypeBoundingBox,
   TypeAudioInput,
   TopicSegmenterTask,
+  ToolDefinitionSchema,
   ToolCallingTask,
   ToolCallingOutputSchema,
   ToolCallingInputSchema,
@@ -6322,4 +6334,4 @@ export {
   AiJob
 };
 
-//# debugId=E73E8DF34D71240E64756E2164756E21
+//# debugId=0AADE5DC48AE458464756E2164756E21