@workglow/ai 0.0.109 → 0.0.110

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,
@@ -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
@@ -5333,11 +5339,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",
@@ -5377,19 +5383,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
@@ -5415,20 +5421,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",
@@ -6316,4 +6322,4 @@ export {
   AiJob
 };
 
-//# debugId=30EA03AC63C4C0B564756E2164756E21
+//# debugId=E73E8DF34D71240E64756E2164756E21