@workglow/ai 0.0.109 → 0.0.110

package/dist/node.js

@@ -321,7 +321,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,
@@ -600,6 +600,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;
@@ -1580,11 +1581,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"],
@@ -1593,11 +1594,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
@@ -1938,11 +1939,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: {
@@ -1961,7 +1962,7 @@ var TextNamedEntityRecognitionInputSchema = {
 var TextNamedEntityRecognitionOutputSchema = {
   type: "object",
   properties: {
-    entities: {
+    entities: TypeSingleOrArray({
       type: "array",
       items: {
         type: "object",
@@ -1987,7 +1988,7 @@ var TextNamedEntityRecognitionOutputSchema = {
       },
       title: "Entities",
       description: "The extracted named entities with their types, scores, and text"
-    }
+    })
   },
   required: ["entities"],
   additionalProperties: false
@@ -2049,11 +2050,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"],
@@ -2062,12 +2063,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
@@ -3662,7 +3663,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
@@ -4185,11 +4191,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: {
@@ -4216,7 +4222,7 @@ var TextClassificationInputSchema = {
 var TextClassificationOutputSchema = {
   type: "object",
   properties: {
-    categories: {
+    categories: TypeSingleOrArray({
       type: "array",
       items: {
         type: "object",
@@ -4237,7 +4243,7 @@ var TextClassificationOutputSchema = {
       },
       title: "Categories",
       description: "The classification categories with their scores"
-    }
+    })
   },
   required: ["categories"],
   additionalProperties: false
@@ -4926,11 +4932,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"],
@@ -4939,7 +4945,7 @@ var TextFillMaskInputSchema = {
 var TextFillMaskOutputSchema = {
   type: "object",
   properties: {
-    predictions: {
+    predictions: TypeSingleOrArray({
       type: "array",
       items: {
         type: "object",
@@ -4965,7 +4971,7 @@ var TextFillMaskOutputSchema = {
       },
       title: "Predictions",
       description: "The predicted tokens to fill the mask with their scores and complete sequences"
-    }
+    })
   },
   required: ["predictions"],
   additionalProperties: false
@@ -4990,22 +4996,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",
@@ -5082,11 +5088,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,
@@ -5103,7 +5109,7 @@ var TextLanguageDetectionInputSchema = {
 var TextLanguageDetectionOutputSchema = {
   type: "object",
   properties: {
-    languages: {
+    languages: TypeSingleOrArray({
       type: "array",
       items: {
         type: "object",
@@ -5124,7 +5130,7 @@ var TextLanguageDetectionOutputSchema = {
       },
       title: "Languages",
       description: "The languages with their scores"
-    }
+    })
   },
   required: ["languages"],
   additionalProperties: false
@@ -5149,22 +5155,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",
@@ -5208,11 +5214,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",
@@ -5226,12 +5232,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
@@ -5332,11 +5338,11 @@ var ToolCallingInputSchema = {
   type: "object",
   properties: {
     model: modelSchema27,
-    prompt: {
+    prompt: TypeSingleOrArray({
       type: "string",
       title: "Prompt",
       description: "The prompt to send to the model"
-    },
+    }),
     systemPrompt: {
       type: "string",
       title: "System Prompt",
@@ -5376,19 +5382,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
@@ -5414,20 +5420,20 @@ 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 = {
+var translationTextSchema = TypeSingleOrArray({
   type: "string",
   title: "Text",
   description: "The translated text",
   "x-stream": "replace"
-};
+});
 var TextTranslationInputSchema = {
   type: "object",
   properties: {
-    text: {
+    text: TypeSingleOrArray({
       type: "string",
       title: "Text",
       description: "The text to translate"
-    },
+    }),
     source_lang: TypeLanguage({
       title: "Source Language",
       description: "The source language",
@@ -6315,4 +6321,4 @@ export {
   AiJob
 };
 
-//# debugId=860B8462B40C292264756E2164756E21
+//# debugId=D821FA1B2BABFC6264756E2164756E21