@workglow/ai 0.0.104 → 0.0.106

package/dist/bun.js

@@ -420,7 +420,7 @@ class AiProvider {
   }
 }
 // src/task/index.ts
-import { TaskRegistry } from "@workglow/task-graph";
+import { TaskRegistry as TaskRegistry2 } from "@workglow/task-graph";
 
 // src/task/BackgroundRemovalTask.ts
 import { CreateWorkflow, Workflow } from "@workglow/task-graph";
@@ -5201,9 +5201,166 @@ var textRewriter = (input, config) => {
 };
 Workflow36.prototype.textRewriter = CreateWorkflow36(TextRewriterTask);
 
+// src/task/ToolCallingTask.ts
+import { CreateWorkflow as CreateWorkflow37, TaskRegistry, Workflow as Workflow37 } from "@workglow/task-graph";
+import { getLogger } from "@workglow/util";
+function buildToolDescription(tool) {
+  let desc = tool.description;
+  if (tool.outputSchema && typeof tool.outputSchema === "object") {
+    desc += `
+
+Returns: ${JSON.stringify(tool.outputSchema)}`;
+  }
+  return desc;
+}
+function isAllowedToolName(name, allowedTools) {
+  return allowedTools.some((t) => t.name === name);
+}
+function filterValidToolCalls(toolCalls, allowedTools) {
+  const filtered = {};
+  for (const [key, value] of Object.entries(toolCalls)) {
+    const tc = value;
+    if (tc.name && isAllowedToolName(tc.name, allowedTools)) {
+      filtered[key] = value;
+    } else {
+      getLogger().warn(`Filtered out tool call with unknown name "${tc.name ?? "(missing)"}"`, {
+        callId: key,
+        toolName: tc.name
+      });
+    }
+  }
+  return filtered;
+}
+function taskTypesToTools(taskNames) {
+  return taskNames.map((name) => {
+    const ctor = TaskRegistry.all.get(name);
+    if (!ctor) {
+      throw new Error(`taskTypesToTools: Unknown task type "${name}" \u2014 not found in TaskRegistry`);
+    }
+    return {
+      name: ctor.type,
+      description: ctor.description ?? "",
+      inputSchema: ctor.inputSchema(),
+      outputSchema: ctor.outputSchema()
+    };
+  });
+}
+var ToolDefinitionSchema = {
+  type: "object",
+  properties: {
+    name: {
+      type: "string",
+      title: "Name",
+      description: "The tool name"
+    },
+    description: {
+      type: "string",
+      title: "Description",
+      description: "A description of what the tool does"
+    },
+    inputSchema: {
+      type: "object",
+      title: "Input Schema",
+      description: "JSON Schema describing the tool's input parameters",
+      additionalProperties: true
+    },
+    outputSchema: {
+      type: "object",
+      title: "Output Schema",
+      description: "JSON Schema describing what the tool returns",
+      additionalProperties: true
+    }
+  },
+  required: ["name", "description", "inputSchema"],
+  additionalProperties: false
+};
+var modelSchema26 = TypeModel("model:ToolCallingTask");
+var ToolCallingInputSchema = {
+  type: "object",
+  properties: {
+    model: modelSchema26,
+    prompt: {
+      type: "string",
+      title: "Prompt",
+      description: "The prompt to send to the model"
+    },
+    systemPrompt: {
+      type: "string",
+      title: "System Prompt",
+      description: "Optional system instructions for the model"
+    },
+    tools: {
+      type: "array",
+      title: "Tools",
+      description: "Tool definitions available for the model to call",
+      items: ToolDefinitionSchema
+    },
+    toolChoice: {
+      type: "string",
+      title: "Tool Choice",
+      description: 'Controls tool selection: "auto" (model decides), "none" (no tools), "required" (must call a tool), or a specific tool name',
+      "x-ui-group": "Configuration"
+    },
+    maxTokens: {
+      type: "number",
+      title: "Max Tokens",
+      description: "The maximum number of tokens to generate",
+      minimum: 1,
+      "x-ui-group": "Configuration"
+    },
+    temperature: {
+      type: "number",
+      title: "Temperature",
+      description: "The temperature to use for sampling",
+      minimum: 0,
+      maximum: 2,
+      "x-ui-group": "Configuration"
+    }
+  },
+  required: ["model", "prompt", "tools"],
+  additionalProperties: false
+};
+var ToolCallingOutputSchema = {
+  type: "object",
+  properties: {
+    text: {
+      type: "string",
+      title: "Text",
+      description: "Any text content generated by the model",
+      "x-stream": "append"
+    },
+    toolCalls: {
+      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
+};
+
+class ToolCallingTask extends StreamingAiTask {
+  static type = "ToolCallingTask";
+  static category = "AI Text Model";
+  static title = "Tool Calling";
+  static description = "Sends a prompt with tool definitions to a language model and returns text along with any tool calls the model requests";
+  static inputSchema() {
+    return ToolCallingInputSchema;
+  }
+  static outputSchema() {
+    return ToolCallingOutputSchema;
+  }
+}
+var toolCalling = (input, config) => {
+  return new ToolCallingTask({}, config).run(input);
+};
+Workflow37.prototype.toolCalling = CreateWorkflow37(ToolCallingTask);
+
 // src/task/TextTranslationTask.ts
-import { CreateWorkflow as CreateWorkflow37, Workflow as Workflow37 } from "@workglow/task-graph";
-var modelSchema26 = TypeModel("model:TextTranslationTask");
+import { CreateWorkflow as CreateWorkflow38, Workflow as Workflow38 } from "@workglow/task-graph";
+var modelSchema27 = TypeModel("model:TextTranslationTask");
 var translationTextSchema = {
   type: "string",
   title: "Text",
@@ -5230,7 +5387,7 @@ var TextTranslationInputSchema = {
       minLength: 2,
       maxLength: 2
     }),
-    model: modelSchema26
+    model: modelSchema27
   },
   required: ["text", "source_lang", "target_lang", "model"],
   additionalProperties: false
@@ -5265,13 +5422,13 @@ class TextTranslationTask extends StreamingAiTask {
 var textTranslation = (input, config) => {
   return new TextTranslationTask({}, config).run(input);
 };
-Workflow37.prototype.textTranslation = CreateWorkflow37(TextTranslationTask);
+Workflow38.prototype.textTranslation = CreateWorkflow38(TextTranslationTask);
 
 // src/task/TopicSegmenterTask.ts
 import {
-  CreateWorkflow as CreateWorkflow38,
+  CreateWorkflow as CreateWorkflow39,
   Task as Task14,
-  Workflow as Workflow38
+  Workflow as Workflow39
 } from "@workglow/task-graph";
 var SegmentationMethod = {
   HEURISTIC: "heuristic",
@@ -5552,15 +5709,15 @@ class TopicSegmenterTask extends Task14 {
 var topicSegmenter = (input, config) => {
   return new TopicSegmenterTask({}, config).run(input);
 };
-Workflow38.prototype.topicSegmenter = CreateWorkflow38(TopicSegmenterTask);
+Workflow39.prototype.topicSegmenter = CreateWorkflow39(TopicSegmenterTask);
 
 // src/task/UnloadModelTask.ts
-import { CreateWorkflow as CreateWorkflow39, Workflow as Workflow39 } from "@workglow/task-graph";
-var modelSchema27 = TypeModel("model");
+import { CreateWorkflow as CreateWorkflow40, Workflow as Workflow40 } from "@workglow/task-graph";
+var modelSchema28 = TypeModel("model");
 var UnloadModelInputSchema = {
   type: "object",
   properties: {
-    model: modelSchema27
+    model: modelSchema28
   },
   required: ["model"],
   additionalProperties: false
@@ -5568,7 +5725,7 @@ var UnloadModelInputSchema = {
 var UnloadModelOutputSchema = {
   type: "object",
   properties: {
-    model: modelSchema27
+    model: modelSchema28
   },
   required: ["model"],
   additionalProperties: false
@@ -5590,10 +5747,10 @@ class UnloadModelTask extends AiTask {
 var unloadModel = (input, config) => {
   return new UnloadModelTask({}, config).run(input);
 };
-Workflow39.prototype.unloadModel = CreateWorkflow39(UnloadModelTask);
+Workflow40.prototype.unloadModel = CreateWorkflow40(UnloadModelTask);
 
 // src/task/VectorQuantizeTask.ts
-import { CreateWorkflow as CreateWorkflow40, Task as Task15, Workflow as Workflow40 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow41, Task as Task15, Workflow as Workflow41 } from "@workglow/task-graph";
 import {
   normalizeNumberArray,
   TensorType,
@@ -5773,10 +5930,10 @@ class VectorQuantizeTask extends Task15 {
 var vectorQuantize = (input, config) => {
   return new VectorQuantizeTask({}, config).run(input);
 };
-Workflow40.prototype.vectorQuantize = CreateWorkflow40(VectorQuantizeTask);
+Workflow41.prototype.vectorQuantize = CreateWorkflow41(VectorQuantizeTask);
 
 // src/task/VectorSimilarityTask.ts
-import { CreateWorkflow as CreateWorkflow41, GraphAsTask, Workflow as Workflow41 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow42, GraphAsTask, Workflow as Workflow42 } from "@workglow/task-graph";
 import {
   cosineSimilarity,
   hammingSimilarity,
@@ -5882,7 +6039,7 @@ class VectorSimilarityTask extends GraphAsTask {
 var similarity = (input, config) => {
   return new VectorSimilarityTask({}, config).run(input);
 };
-Workflow41.prototype.similarity = CreateWorkflow41(VectorSimilarityTask);
+Workflow42.prototype.similarity = CreateWorkflow42(VectorSimilarityTask);
 
 // src/task/index.ts
 var registerAiTasks = () => {
@@ -5924,12 +6081,13 @@ var registerAiTasks = () => {
     TextRewriterTask,
     TextSummaryTask,
     TextTranslationTask,
+    ToolCallingTask,
     TopicSegmenterTask,
     UnloadModelTask,
     VectorQuantizeTask,
     VectorSimilarityTask
   ];
-  tasks.map(TaskRegistry.registerTask);
+  tasks.map(TaskRegistry2.registerTask);
   return tasks;
 };
 export {
@@ -5937,6 +6095,7 @@ export {
   vectorQuantize,
   unloadModel,
   topicSegmenter,
+  toolCalling,
   textTranslation,
   textSummary,
   textRewriter,
@@ -5948,6 +6107,7 @@ export {
   textEmbedding,
   textClassification,
   textChunker,
+  taskTypesToTools,
   structuredGeneration,
   structuralParser,
   similarity,
@@ -5958,6 +6118,7 @@ export {
   queryExpander,
   poseLandmarker,
   objectDetection,
+  isAllowedToolName,
   imageToText,
   imageSegmentation,
   imageEmbedding,
@@ -5969,6 +6130,7 @@ export {
   getGlobalModelRepository,
   getAiProviderRegistry,
   gestureRecognizer,
+  filterValidToolCalls,
   faceLandmarker,
   faceDetector,
   downloadModel,
@@ -5978,6 +6140,7 @@ export {
   chunkVectorUpsert,
   chunkToVector,
   chunkRetrieval,
+  buildToolDescription,
   backgroundRemoval,
   VectorSimilarityTask,
   VectorQuantizeTask,
@@ -5992,6 +6155,9 @@ export {
   TypeBoundingBox,
   TypeAudioInput,
   TopicSegmenterTask,
+  ToolCallingTask,
+  ToolCallingOutputSchema,
+  ToolCallingInputSchema,
   TextTranslationTask,
   TextTranslationOutputSchema,
   TextTranslationInputSchema,
@@ -6093,4 +6259,4 @@ export {
   AiJob
 };
 
-//# debugId=074733AA4A9EEEC264756E2164756E21
+//# debugId=63F068BFF8E5F20864756E2164756E21