@workglow/ai 0.2.5 → 0.2.7

package/dist/node.js

@@ -454,7 +454,15 @@ var ModelConfigSchema = {
     provider_config: {
       type: "object",
       properties: {
-        credential_key: { type: "string", format: "credential", "x-ui-hidden": true }
+        credential_key: { type: "string", format: "credential", "x-ui-hidden": true },
+        native_dimensions: {
+          type: "integer",
+          description: "Native output vector dimensions for embedding models"
+        },
+        mrl: {
+          type: "boolean",
+          description: "Whether the model supports Matryoshka Representation Learning"
+        }
       },
       additionalProperties: true,
       default: {}
@@ -648,6 +656,7 @@ registerInputCompactor("model", async (value, _format, registry) => {
 
 // src/provider/AiProvider.ts
 import { globalServiceRegistry as globalServiceRegistry3, WORKER_MANAGER as WORKER_MANAGER2 } from "@workglow/util/worker";
+var DEFAULT_AI_PROVIDER_WORKER_IDLE_TIMEOUT_MS = 15 * 60 * 1000;
 function resolveAiProviderGpuQueueConcurrency(concurrency) {
   if (concurrency === undefined) {
     return 1;
@@ -695,7 +704,13 @@ class AiProvider {
     const registry = getAiProviderRegistry();
     if (!isInline && options.worker) {
       const workerManager = globalServiceRegistry3.get(WORKER_MANAGER2);
-      workerManager.registerWorker(this.name, options.worker);
+      if (typeof options.worker === "function") {
+        workerManager.registerWorker(this.name, options.worker, {
+          idleTimeoutMs: options.workerIdleTimeoutMs ?? DEFAULT_AI_PROVIDER_WORKER_IDLE_TIMEOUT_MS
+        });
+      } else {
+        workerManager.registerWorker(this.name, options.worker);
+      }
       for (const taskType of this.taskTypes) {
         registry.registerAsWorkerRunFn(this.name, taskType);
         registry.registerAsWorkerStreamFn(this.name, taskType);
@@ -1049,6 +1064,18 @@ class AiTask extends Task {
     const jobInput = await this.getJobInput(input);
     const strategy = getAiProviderRegistry().getStrategy(model);
     const output = await strategy.execute(jobInput, executeContext, this.runConfig.runnerId);
+    if (executeContext.resourceScope) {
+      const registry = getAiProviderRegistry();
+      const provider = registry.getProvider(model.provider);
+      const unloadFn = provider?.getRunFn("UnloadModelTask");
+      if (unloadFn) {
+        const modelPath = model.provider_config?.model_path ?? model.model;
+        const resourceKey = `ai:${model.provider}:${modelPath}`;
+        executeContext.resourceScope.register(resourceKey, async () => {
+          await unloadFn({ model }, model, () => {}, AbortSignal.timeout(30000));
+        });
+      }
+    }
     return output;
   }
   async getJobInput(input) {
@@ -1144,7 +1171,7 @@ class AiTask extends Task {
     if (modelTaskProperties.length > 0) {
       const modelRepo = registry.get(MODEL_REPOSITORY);
       const taskModels = await modelRepo.findModelsByTask(this.type) ?? [];
-      for (const [key, propSchema] of modelTaskProperties) {
+      for (const [key] of modelTaskProperties) {
         const requestedModel = input[key];
         if (typeof requestedModel === "string") {
           const found = taskModels?.find((m) => m.model_id === requestedModel);
@@ -3205,6 +3232,8 @@ var inputSchema7 = {
       description: "The document ID"
     },
     documentTree: {
+      type: "object",
+      additionalProperties: true,
       title: "Document Tree",
       description: "The hierarchical document tree to enrich"
     },
@@ -4207,6 +4236,8 @@ var inputSchema9 = {
       description: "The ID of the document"
     },
     documentTree: {
+      type: "object",
+      additionalProperties: true,
       title: "Document Tree",
       description: "The hierarchical document tree to chunk"
     },
@@ -4592,8 +4623,97 @@ var hierarchyJoin = (input, config) => {
 };
 Workflow22.prototype.hierarchyJoin = CreateWorkflow22(HierarchyJoinTask);
 
+// src/task/KbToDocumentsTask.ts
+import { TypeKnowledgeBase as TypeKnowledgeBase7 } from "@workglow/knowledge-base";
+import { CreateWorkflow as CreateWorkflow23, Task as Task13, Workflow as Workflow23 } from "@workglow/task-graph";
+var inputSchema11 = {
+  type: "object",
+  properties: {
+    knowledgeBase: TypeKnowledgeBase7({
+      title: "Knowledge Base",
+      description: "The knowledge base instance to list documents from"
+    }),
+    onlyStale: {
+      type: "boolean",
+      title: "Only Stale",
+      description: "If true, only return documents that have no chunks (need embedding)",
+      default: true
+    }
+  },
+  required: ["knowledgeBase"],
+  additionalProperties: false
+};
+var outputSchema11 = {
+  type: "object",
+  properties: {
+    doc_id: {
+      type: "array",
+      items: { type: "string" },
+      title: "Document IDs",
+      description: "Array of document IDs"
+    },
+    documentTree: {
+      type: "array",
+      items: { type: "object", additionalProperties: true },
+      title: "Document Trees",
+      description: "Array of document root nodes (parallel to doc_id)"
+    },
+    title: {
+      type: "array",
+      items: { type: "string" },
+      title: "Titles",
+      description: "Array of document titles (parallel to doc_id)"
+    }
+  },
+  required: ["doc_id", "documentTree", "title"],
+  additionalProperties: false
+};
+
+class KbToDocumentsTask extends Task13 {
+  static type = "KbToDocumentsTask";
+  static category = "Vector Store";
+  static title = "Knowledge Base to Documents";
+  static description = "List documents from a knowledge base, optionally filtering to only those that need embedding";
+  static cacheable = false;
+  static inputSchema() {
+    return inputSchema11;
+  }
+  static outputSchema() {
+    return outputSchema11;
+  }
+  async execute(input, context) {
+    const { knowledgeBase, onlyStale = true } = input;
+    const kb = knowledgeBase;
+    await context.updateProgress(1, "Listing documents");
+    const allDocIds = await kb.listDocuments();
+    const doc_id = [];
+    const documentTree = [];
+    const title = [];
+    for (const id of allDocIds) {
+      if (onlyStale) {
+        const chunks = await kb.getChunksForDocument(id);
+        if (chunks.length > 0) {
+          continue;
+        }
+      }
+      const doc = await kb.getDocument(id);
+      if (!doc) {
+        continue;
+      }
+      doc_id.push(id);
+      documentTree.push(doc.root);
+      title.push(doc.metadata.title);
+    }
+    return { doc_id, documentTree, title };
+  }
+}
+var kbToDocuments = (input, config) => {
+  return new KbToDocumentsTask(config).run(input);
+};
+Workflow23.prototype.kbToDocuments = CreateWorkflow23(KbToDocumentsTask);
+
 // src/task/ImageClassificationTask.ts
-import { CreateWorkflow as CreateWorkflow23, Workflow as Workflow23 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow24, Workflow as Workflow24 } from "@workglow/task-graph";
 var modelSchema15 = TypeModel("model:ImageClassificationTask");
 var ImageClassificationInputSchema = {
   type: "object",
@@ -4653,10 +4773,10 @@ class ImageClassificationTask extends AiVisionTask {
 var imageClassification = (input, config) => {
   return new ImageClassificationTask(config).run(input);
 };
-Workflow23.prototype.imageClassification = CreateWorkflow23(ImageClassificationTask);
+Workflow24.prototype.imageClassification = CreateWorkflow24(ImageClassificationTask);
 
 // src/task/ImageEmbeddingTask.ts
-import { CreateWorkflow as CreateWorkflow24, Workflow as Workflow24 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow25, Workflow as Workflow25 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema7
 } from "@workglow/util/schema";
@@ -4697,10 +4817,10 @@ class ImageEmbeddingTask extends AiVisionTask {
 var imageEmbedding = (input, config) => {
   return new ImageEmbeddingTask(config).run(input);
 };
-Workflow24.prototype.imageEmbedding = CreateWorkflow24(ImageEmbeddingTask);
+Workflow25.prototype.imageEmbedding = CreateWorkflow25(ImageEmbeddingTask);
 
 // src/task/ImageSegmentationTask.ts
-import { CreateWorkflow as CreateWorkflow25, Workflow as Workflow25 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow26, Workflow as Workflow26 } from "@workglow/task-graph";
 var modelSchema17 = TypeModel("model:ImageSegmentationTask");
 var ImageSegmentationInputSchema = {
   type: "object",
@@ -4785,10 +4905,10 @@ class ImageSegmentationTask extends AiVisionTask {
 var imageSegmentation = (input, config) => {
   return new ImageSegmentationTask(config).run(input);
 };
-Workflow25.prototype.imageSegmentation = CreateWorkflow25(ImageSegmentationTask);
+Workflow26.prototype.imageSegmentation = CreateWorkflow26(ImageSegmentationTask);
 
 // src/task/ImageToTextTask.ts
-import { CreateWorkflow as CreateWorkflow26, Workflow as Workflow26 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow27, Workflow as Workflow27 } from "@workglow/task-graph";
 var modelSchema18 = TypeModel("model:ImageToTextTask");
 var generatedTextSchema = {
   type: "string",
@@ -4840,10 +4960,10 @@ class ImageToTextTask extends AiVisionTask {
 var imageToText = (input, config) => {
   return new ImageToTextTask(config).run(input);
 };
-Workflow26.prototype.imageToText = CreateWorkflow26(ImageToTextTask);
+Workflow27.prototype.imageToText = CreateWorkflow27(ImageToTextTask);
 
 // src/task/ModelInfoTask.ts
-import { CreateWorkflow as CreateWorkflow27, Workflow as Workflow27 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow28, Workflow as Workflow28 } from "@workglow/task-graph";
 var modelSchema19 = TypeModel("model");
 var ModelInfoInputSchema = {
   type: "object",
@@ -4851,7 +4971,7 @@ var ModelInfoInputSchema = {
     model: modelSchema19,
     detail: {
       type: "string",
-      enum: ["cached_status", "files", "files_with_metadata"],
+      enum: ["cached_status", "files", "files_with_metadata", "dimensions"],
       default: "files_with_metadata"
     }
   },
@@ -4873,6 +4993,14 @@ var ModelInfoOutputSchema = {
       type: "array",
       items: { type: "string" },
       description: "Available quantization variants (e.g. fp32, fp16, q8). Only present for models with quantization options."
+    },
+    native_dimensions: {
+      type: "integer",
+      description: "Native output vector dimensions. Only present when detail is 'dimensions' and the provider can determine them."
+    },
+    mrl: {
+      type: "boolean",
+      description: "Whether the model supports Matryoshka Representation Learning. Only present when detail is 'dimensions'."
     }
   },
   required: [
@@ -4911,10 +5039,10 @@ class ModelInfoTask extends AiTask {
 var modelInfo = (input, config) => {
   return new ModelInfoTask(config).run(input);
 };
-Workflow27.prototype.modelInfo = CreateWorkflow27(ModelInfoTask);
+Workflow28.prototype.modelInfo = CreateWorkflow28(ModelInfoTask);
 
 // src/task/ModelSearchTask.ts
-import { CreateWorkflow as CreateWorkflow28, Task as Task13, Workflow as Workflow28 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow29, Task as Task14, Workflow as Workflow29 } from "@workglow/task-graph";
 var ModelSearchInputSchema = {
   type: "object",
   properties: {
@@ -4979,7 +5107,7 @@ var ModelSearchOutputSchema = {
   additionalProperties: false
 };
 
-class ModelSearchTask extends Task13 {
+class ModelSearchTask extends Task14 {
   static type = "ModelSearchTask";
   static category = "AI Model";
   static title = "Model Search";
@@ -5005,10 +5133,10 @@ class ModelSearchTask extends Task13 {
 var modelSearch = (input, config) => {
   return new ModelSearchTask(config).run(input);
 };
-Workflow28.prototype.modelSearch = CreateWorkflow28(ModelSearchTask);
+Workflow29.prototype.modelSearch = CreateWorkflow29(ModelSearchTask);
 
 // src/task/ObjectDetectionTask.ts
-import { CreateWorkflow as CreateWorkflow29, Workflow as Workflow29 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow30, Workflow as Workflow30 } from "@workglow/task-graph";
 var modelSchema20 = TypeModel("model:ObjectDetectionTask");
 var detectionSchema = {
   type: "object",
@@ -5088,10 +5216,10 @@ class ObjectDetectionTask extends AiVisionTask {
 var objectDetection = (input, config) => {
   return new ObjectDetectionTask(config).run(input);
 };
-Workflow29.prototype.objectDetection = CreateWorkflow29(ObjectDetectionTask);
+Workflow30.prototype.objectDetection = CreateWorkflow30(ObjectDetectionTask);
 
 // src/task/PoseLandmarkerTask.ts
-import { CreateWorkflow as CreateWorkflow30, Workflow as Workflow30 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow31, Workflow as Workflow31 } from "@workglow/task-graph";
 var modelSchema21 = TypeModel("model:PoseLandmarkerTask");
 var TypePoseLandmark = {
   type: "object",
@@ -5250,17 +5378,17 @@ class PoseLandmarkerTask extends AiVisionTask {
 var poseLandmarker = (input, config) => {
   return new PoseLandmarkerTask(config).run(input);
 };
-Workflow30.prototype.poseLandmarker = CreateWorkflow30(PoseLandmarkerTask);
+Workflow31.prototype.poseLandmarker = CreateWorkflow31(PoseLandmarkerTask);
 
 // src/task/QueryExpanderTask.ts
-import { CreateWorkflow as CreateWorkflow31, Task as Task14, Workflow as Workflow31 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow32, Task as Task15, Workflow as Workflow32 } from "@workglow/task-graph";
 var QueryExpansionMethod = {
   MULTI_QUERY: "multi-query",
   HYDE: "hyde",
   SYNONYMS: "synonyms",
   PARAPHRASE: "paraphrase"
 };
-var inputSchema11 = {
+var inputSchema12 = {
   type: "object",
   properties: {
     query: {
@@ -5292,7 +5420,7 @@ var inputSchema11 = {
   required: ["query"],
   additionalProperties: false
 };
-var outputSchema11 = {
+var outputSchema12 = {
   type: "object",
   properties: {
     query: {
@@ -5321,17 +5449,17 @@ var outputSchema11 = {
   additionalProperties: false
 };
 
-class QueryExpanderTask extends Task14 {
+class QueryExpanderTask extends Task15 {
   static type = "QueryExpanderTask";
   static category = "RAG";
   static title = "Query Expander";
   static description = "Expand queries to improve retrieval coverage";
   static cacheable = true;
   static inputSchema() {
-    return inputSchema11;
+    return inputSchema12;
   }
   static outputSchema() {
-    return outputSchema11;
+    return outputSchema12;
   }
   async execute(input, context) {
     const { query, method = QueryExpansionMethod.MULTI_QUERY, numVariations = 3 } = input;
@@ -5463,13 +5591,13 @@ class QueryExpanderTask extends Task14 {
 var queryExpander = (input, config) => {
   return new QueryExpanderTask(config).run(input);
 };
-Workflow31.prototype.queryExpander = CreateWorkflow31(QueryExpanderTask);
+Workflow32.prototype.queryExpander = CreateWorkflow32(QueryExpanderTask);
 
 // src/task/RerankerTask.ts
-import { CreateWorkflow as CreateWorkflow33, Task as Task15, Workflow as Workflow33 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow34, Task as Task16, Workflow as Workflow34 } from "@workglow/task-graph";
 
 // src/task/TextClassificationTask.ts
-import { CreateWorkflow as CreateWorkflow32, Workflow as Workflow32 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow33, Workflow as Workflow33 } from "@workglow/task-graph";
 var modelSchema22 = TypeModel("model:TextClassificationTask");
 var TextClassificationInputSchema = {
   type: "object",
@@ -5547,10 +5675,10 @@ class TextClassificationTask extends AiTask {
 var textClassification = (input, config) => {
   return new TextClassificationTask(config).run(input);
 };
-Workflow32.prototype.textClassification = CreateWorkflow32(TextClassificationTask);
+Workflow33.prototype.textClassification = CreateWorkflow33(TextClassificationTask);
 
 // src/task/RerankerTask.ts
-var inputSchema12 = {
+var inputSchema13 = {
   type: "object",
   properties: {
     query: {
@@ -5601,7 +5729,7 @@ var inputSchema12 = {
   required: ["query", "chunks"],
   additionalProperties: false
 };
-var outputSchema12 = {
+var outputSchema13 = {
   type: "object",
   properties: {
     chunks: {
@@ -5642,18 +5770,17 @@ var outputSchema12 = {
   additionalProperties: false
 };
 
-class RerankerTask extends Task15 {
+class RerankerTask extends Task16 {
   static type = "RerankerTask";
   static category = "RAG";
   static title = "Reranker";
   static description = "Rerank retrieved chunks to improve relevance";
   static cacheable = true;
-  resolvedCrossEncoderModel;
   static inputSchema() {
-    return inputSchema12;
+    return inputSchema13;
   }
   static outputSchema() {
-    return outputSchema12;
+    return outputSchema13;
   }
   async execute(input, context) {
     const { query, chunks, scores = [], metadata = [], topK, method = "simple", model } = input;
@@ -5771,13 +5898,13 @@ class RerankerTask extends Task15 {
 var reranker = (input, config) => {
   return new RerankerTask(config).run(input);
 };
-Workflow33.prototype.reranker = CreateWorkflow33(RerankerTask);
+Workflow34.prototype.reranker = CreateWorkflow34(RerankerTask);
 
 // src/task/StructuralParserTask.ts
 import { StructuralParser } from "@workglow/knowledge-base";
-import { CreateWorkflow as CreateWorkflow34, Task as Task16, Workflow as Workflow34 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow35, Task as Task17, Workflow as Workflow35 } from "@workglow/task-graph";
 import { uuid4 as uuid42 } from "@workglow/util";
-var inputSchema13 = {
+var inputSchema14 = {
   type: "object",
   properties: {
     text: {
@@ -5811,7 +5938,7 @@ var inputSchema13 = {
   required: ["text", "title"],
   additionalProperties: false
 };
-var outputSchema13 = {
+var outputSchema14 = {
   type: "object",
   properties: {
     doc_id: {
@@ -5820,8 +5947,10 @@ var outputSchema13 = {
       description: "Generated or provided document ID"
     },
     documentTree: {
+      type: "object",
       title: "Document Tree",
-      description: "Parsed hierarchical document tree"
+      description: "Parsed hierarchical document tree",
+      additionalProperties: true
     },
     nodeCount: {
       type: "number",
@@ -5833,21 +5962,21 @@ var outputSchema13 = {
   additionalProperties: false
 };
 
-class StructuralParserTask extends Task16 {
+class StructuralParserTask extends Task17 {
   static type = "StructuralParserTask";
   static category = "Document";
   static title = "Structural Parser";
   static description = "Parse documents into hierarchical tree structure";
   static cacheable = true;
   static inputSchema() {
-    return inputSchema13;
+    return inputSchema14;
   }
   static outputSchema() {
-    return outputSchema13;
+    return outputSchema14;
   }
   async execute(input, context) {
     const { text, title, format = "auto", sourceUri, doc_id: providedDocId } = input;
-    const doc_id = providedDocId || uuid42();
+    const doc_id = providedDocId || sourceUri || uuid42();
     let documentTree;
     if (format === "markdown") {
       documentTree = await StructuralParser.parseMarkdown(doc_id, text, title);
@@ -5876,10 +6005,10 @@ class StructuralParserTask extends Task16 {
 var structuralParser = (input, config) => {
   return new StructuralParserTask(config).run(input);
 };
-Workflow34.prototype.structuralParser = CreateWorkflow34(StructuralParserTask);
+Workflow35.prototype.structuralParser = CreateWorkflow35(StructuralParserTask);
 
 // src/task/StructuredGenerationTask.ts
-import { CreateWorkflow as CreateWorkflow35, Workflow as Workflow35 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow36, Workflow as Workflow36 } from "@workglow/task-graph";
 var modelSchema23 = TypeModel("model:StructuredGenerationTask");
 var StructuredGenerationInputSchema = {
   type: "object",
@@ -5947,17 +6076,17 @@ class StructuredGenerationTask extends StreamingAiTask {
 var structuredGeneration = (input, config) => {
   return new StructuredGenerationTask(config).run(input);
 };
-Workflow35.prototype.structuredGeneration = CreateWorkflow35(StructuredGenerationTask);
+Workflow36.prototype.structuredGeneration = CreateWorkflow36(StructuredGenerationTask);
 
 // src/task/TextChunkerTask.ts
-import { CreateWorkflow as CreateWorkflow36, Task as Task17, Workflow as Workflow36 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow37, Task as Task18, Workflow as Workflow37 } from "@workglow/task-graph";
 var ChunkingStrategy = {
   FIXED: "fixed",
   SENTENCE: "sentence",
   PARAGRAPH: "paragraph",
   SEMANTIC: "semantic"
 };
-var inputSchema14 = {
+var inputSchema15 = {
   type: "object",
   properties: {
     text: {
@@ -5990,7 +6119,7 @@ var inputSchema14 = {
   required: ["text"],
   additionalProperties: false
 };
-var outputSchema14 = {
+var outputSchema15 = {
   type: "object",
   properties: {
     chunks: {
@@ -6019,17 +6148,17 @@ var outputSchema14 = {
   additionalProperties: false
 };
 
-class TextChunkerTask extends Task17 {
+class TextChunkerTask extends Task18 {
   static type = "TextChunkerTask";
   static category = "Document";
   static title = "Text Chunker";
   static description = "Splits text into chunks using various strategies (fixed, sentence, paragraph)";
   static cacheable = true;
   static inputSchema() {
-    return inputSchema14;
+    return inputSchema15;
   }
   static outputSchema() {
-    return outputSchema14;
+    return outputSchema15;
   }
   async execute(input, context) {
     const { text, chunkSize = 512, chunkOverlap = 50, strategy = ChunkingStrategy.FIXED } = input;
@@ -6199,10 +6328,10 @@ class TextChunkerTask extends Task17 {
 var textChunker = (input, config) => {
   return new TextChunkerTask(config).run(input);
 };
-Workflow36.prototype.textChunker = CreateWorkflow36(TextChunkerTask);
+Workflow37.prototype.textChunker = CreateWorkflow37(TextChunkerTask);
 
 // src/task/TextFillMaskTask.ts
-import { CreateWorkflow as CreateWorkflow37, Workflow as Workflow37 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow38, Workflow as Workflow38 } from "@workglow/task-graph";
 var modelSchema24 = TypeModel("model:TextFillMaskTask");
 var TextFillMaskInputSchema = {
   type: "object",
@@ -6267,10 +6396,10 @@ class TextFillMaskTask extends AiTask {
 var textFillMask = (input, config) => {
   return new TextFillMaskTask(config).run(input);
 };
-Workflow37.prototype.textFillMask = CreateWorkflow37(TextFillMaskTask);
+Workflow38.prototype.textFillMask = CreateWorkflow38(TextFillMaskTask);
 
 // src/task/TextGenerationTask.ts
-import { CreateWorkflow as CreateWorkflow38, Workflow as Workflow38 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow39, Workflow as Workflow39 } from "@workglow/task-graph";
 var generatedTextSchema2 = {
   type: "string",
   title: "Text",
@@ -6355,10 +6484,10 @@ class TextGenerationTask extends StreamingAiTask {
 var textGeneration = (input, config) => {
   return new TextGenerationTask(config).run(input);
 };
-Workflow38.prototype.textGeneration = CreateWorkflow38(TextGenerationTask);
+Workflow39.prototype.textGeneration = CreateWorkflow39(TextGenerationTask);
 
 // src/task/TextLanguageDetectionTask.ts
-import { CreateWorkflow as CreateWorkflow39, Workflow as Workflow39 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow40, Workflow as Workflow40 } from "@workglow/task-graph";
 var modelSchema26 = TypeModel("model:TextLanguageDetectionTask");
 var TextLanguageDetectionInputSchema = {
   type: "object",
@@ -6426,10 +6555,10 @@ class TextLanguageDetectionTask extends AiTask {
 var textLanguageDetection = (input, config) => {
   return new TextLanguageDetectionTask(config).run(input);
 };
-Workflow39.prototype.textLanguageDetection = CreateWorkflow39(TextLanguageDetectionTask);
+Workflow40.prototype.textLanguageDetection = CreateWorkflow40(TextLanguageDetectionTask);
 
 // src/task/TextQuestionAnswerTask.ts
-import { CreateWorkflow as CreateWorkflow40, Workflow as Workflow40 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow41, Workflow as Workflow41 } from "@workglow/task-graph";
 var contextSchema = {
   type: "string",
   title: "Context",
@@ -6481,10 +6610,10 @@ class TextQuestionAnswerTask extends StreamingAiTask {
 var textQuestionAnswer = (input, config) => {
   return new TextQuestionAnswerTask(config).run(input);
 };
-Workflow40.prototype.textQuestionAnswer = CreateWorkflow40(TextQuestionAnswerTask);
+Workflow41.prototype.textQuestionAnswer = CreateWorkflow41(TextQuestionAnswerTask);
 
 // src/task/TextRewriterTask.ts
-import { CreateWorkflow as CreateWorkflow41, Workflow as Workflow41 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow42, Workflow as Workflow42 } from "@workglow/task-graph";
 var modelSchema28 = TypeModel("model:TextRewriterTask");
 var TextRewriterInputSchema = {
   type: "object",
@@ -6533,10 +6662,10 @@ class TextRewriterTask extends StreamingAiTask {
 var textRewriter = (input, config) => {
   return new TextRewriterTask(config).run(input);
 };
-Workflow41.prototype.textRewriter = CreateWorkflow41(TextRewriterTask);
+Workflow42.prototype.textRewriter = CreateWorkflow42(TextRewriterTask);
 
 // src/task/TextTranslationTask.ts
-import { CreateWorkflow as CreateWorkflow42, Workflow as Workflow42 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow43, Workflow as Workflow43 } from "@workglow/task-graph";
 var modelSchema29 = TypeModel("model:TextTranslationTask");
 var translationTextSchema = {
   type: "string",
@@ -6599,16 +6728,16 @@ class TextTranslationTask extends StreamingAiTask {
 var textTranslation = (input, config) => {
   return new TextTranslationTask(config).run(input);
 };
-Workflow42.prototype.textTranslation = CreateWorkflow42(TextTranslationTask);
+Workflow43.prototype.textTranslation = CreateWorkflow43(TextTranslationTask);
 
 // src/task/TopicSegmenterTask.ts
-import { CreateWorkflow as CreateWorkflow43, Task as Task18, Workflow as Workflow43 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow44, Task as Task19, Workflow as Workflow44 } from "@workglow/task-graph";
 var SegmentationMethod = {
   HEURISTIC: "heuristic",
   EMBEDDING_SIMILARITY: "embedding-similarity",
   HYBRID: "hybrid"
 };
-var inputSchema15 = {
+var inputSchema16 = {
   type: "object",
   properties: {
     text: {
@@ -6649,7 +6778,7 @@ var inputSchema15 = {
   required: ["text"],
   additionalProperties: false
 };
-var outputSchema15 = {
+var outputSchema16 = {
   type: "object",
   properties: {
     segments: {
@@ -6677,7 +6806,7 @@ var outputSchema15 = {
   additionalProperties: false
 };
 
-class TopicSegmenterTask extends Task18 {
+class TopicSegmenterTask extends Task19 {
   static type = "TopicSegmenterTask";
   static category = "Document";
   static title = "Topic Segmenter";
@@ -6685,10 +6814,10 @@ class TopicSegmenterTask extends Task18 {
   static cacheable = true;
   static EMBEDDING_DIMENSIONS = 256;
   static inputSchema() {
-    return inputSchema15;
+    return inputSchema16;
   }
   static outputSchema() {
-    return outputSchema15;
+    return outputSchema16;
   }
   async execute(input, context) {
     const {
@@ -6882,10 +7011,10 @@ class TopicSegmenterTask extends Task18 {
 var topicSegmenter = (input, config) => {
   return new TopicSegmenterTask(config).run(input);
 };
-Workflow43.prototype.topicSegmenter = CreateWorkflow43(TopicSegmenterTask);
+Workflow44.prototype.topicSegmenter = CreateWorkflow44(TopicSegmenterTask);
 
 // src/task/UnloadModelTask.ts
-import { CreateWorkflow as CreateWorkflow44, Workflow as Workflow44 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow45, Workflow as Workflow45 } from "@workglow/task-graph";
 var modelSchema30 = TypeModel("model");
 var UnloadModelInputSchema = {
   type: "object",
@@ -6920,16 +7049,16 @@ class UnloadModelTask extends AiTask {
 var unloadModel = (input, config) => {
   return new UnloadModelTask(config).run(input);
 };
-Workflow44.prototype.unloadModel = CreateWorkflow44(UnloadModelTask);
+Workflow45.prototype.unloadModel = CreateWorkflow45(UnloadModelTask);
 
 // src/task/VectorQuantizeTask.ts
-import { CreateWorkflow as CreateWorkflow45, Task as Task19, Workflow as Workflow45 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow46, Task as Task20, Workflow as Workflow46 } from "@workglow/task-graph";
 import {
   normalizeNumberArray,
   TensorType,
   TypedArraySchema as TypedArraySchema8
 } from "@workglow/util/schema";
-var inputSchema16 = {
+var inputSchema17 = {
   type: "object",
   properties: {
     vector: {
@@ -6966,7 +7095,7 @@ var inputSchema16 = {
   required: ["vector", "targetType"],
   additionalProperties: false
 };
-var outputSchema16 = {
+var outputSchema17 = {
   type: "object",
   properties: {
     vector: {
@@ -7003,17 +7132,17 @@ var outputSchema16 = {
   additionalProperties: false
 };
 
-class VectorQuantizeTask extends Task19 {
+class VectorQuantizeTask extends Task20 {
   static type = "VectorQuantizeTask";
   static category = "Vector";
   static title = "Quantize";
   static description = "Quantize vectors to reduce storage and improve performance";
   static cacheable = true;
   static inputSchema() {
-    return inputSchema16;
+    return inputSchema17;
   }
   static outputSchema() {
-    return outputSchema16;
+    return outputSchema17;
   }
   async executeReactive(input) {
     const { vector, targetType, normalize = true } = input;
@@ -7103,10 +7232,10 @@ class VectorQuantizeTask extends Task19 {
 var vectorQuantize = (input, config) => {
   return new VectorQuantizeTask(config).run(input);
 };
-Workflow45.prototype.vectorQuantize = CreateWorkflow45(VectorQuantizeTask);
+Workflow46.prototype.vectorQuantize = CreateWorkflow46(VectorQuantizeTask);
 
 // src/task/VectorSimilarityTask.ts
-import { CreateWorkflow as CreateWorkflow46, GraphAsTask, Workflow as Workflow46 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow47, GraphAsTask, Workflow as Workflow47 } from "@workglow/task-graph";
 import {
   cosineSimilarity,
   hammingSimilarity,
@@ -7212,7 +7341,7 @@ class VectorSimilarityTask extends GraphAsTask {
 var similarity = (input, config) => {
   return new VectorSimilarityTask(config).run(input);
 };
-Workflow46.prototype.similarity = CreateWorkflow46(VectorSimilarityTask);
+Workflow47.prototype.similarity = CreateWorkflow47(VectorSimilarityTask);
 // src/task/MessageConversion.ts
 function getInputMessages(input) {
   const messages = input.messages;
@@ -7431,6 +7560,7 @@ var registerAiTasks = () => {
     HandLandmarkerTask,
     HierarchicalChunkerTask,
     HierarchyJoinTask,
+    KbToDocumentsTask,
     ImageClassificationTask,
     ImageEmbeddingTask,
     ImageSegmentationTask,
@@ -7499,6 +7629,7 @@ export {
   objectDetection,
   modelSearch,
   modelInfo,
+  kbToDocuments,
   isAllowedToolName,
   imageToText,
   imageSegmentation,
@@ -7609,6 +7740,7 @@ export {
   ModelInfoTask,
   ModelConfigSchema,
   MODEL_REPOSITORY,
+  KbToDocumentsTask,
   InMemoryModelRepository,
   ImageToTextTask,
   ImageToTextOutputSchema,
@@ -7640,6 +7772,7 @@ export {
   DocumentUpsertTask,
   DocumentEnricherTask,
   DirectExecutionStrategy,
+  DEFAULT_AI_PROVIDER_WORKER_IDLE_TIMEOUT_MS,
   CountTokensTask,
   CountTokensOutputSchema,
   CountTokensInputSchema,
@@ -7663,4 +7796,4 @@ export {
   AgentInputSchema
 };
 
-//# debugId=745FC34B8779988C64756E2164756E21
+//# debugId=FF0FA51260752A4C64756E2164756E21