@workglow/ai 0.0.120 → 0.0.122

package/dist/bun.js

@@ -8,7 +8,7 @@ import {
 } from "@workglow/job-queue";
 
 // src/provider/AiProviderRegistry.ts
-import { globalServiceRegistry, WORKER_MANAGER } from "@workglow/util";
+import { globalServiceRegistry, WORKER_MANAGER } from "@workglow/util/worker";
 
 class AiProviderRegistry {
   runFnRegistry = new Map;
@@ -146,6 +146,34 @@ class AiJob extends Job {
     yield* streamFn(input.taskInput, model, context.signal, input.outputSchema);
   }
 }
+
+// src/task/ToolCallingUtils.ts
+import { getLogger } from "@workglow/util/worker";
+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) {
+  return toolCalls.filter((tc) => {
+    if (tc.name && isAllowedToolName(tc.name, allowedTools)) {
+      return true;
+    }
+    getLogger().warn(`Filtered out tool call with unknown name "${tc.name ?? "(missing)"}"`, {
+      callId: tc.id,
+      toolName: tc.name
+    });
+    return false;
+  });
+}
+
 // src/model/InMemoryModelRepository.ts
 import { InMemoryTabularStorage } from "@workglow/storage";
 
@@ -284,6 +312,7 @@ class InMemoryModelRepository extends ModelRepository {
     super(new InMemoryTabularStorage(ModelRecordSchema, ModelPrimaryKeyNames));
   }
 }
+
 // src/model/ModelRegistry.ts
 import {
   createServiceToken,
@@ -309,31 +338,12 @@ async function resolveModelFromRegistry(id, format, registry) {
   return model;
 }
 registerInputResolver("model", resolveModelFromRegistry);
-// src/provider/AiProvider.ts
-import { globalServiceRegistry as globalServiceRegistry3, WORKER_MANAGER as WORKER_MANAGER2 } from "@workglow/util";
-
-// src/queue/createDefaultQueue.ts
-import { ConcurrencyLimiter, JobQueueClient, JobQueueServer } from "@workglow/job-queue";
-import { InMemoryQueueStorage } from "@workglow/storage";
-import { getTaskQueueRegistry } from "@workglow/task-graph";
-async function createDefaultQueue(providerName, concurrency) {
-  const storage = new InMemoryQueueStorage(providerName);
-  await storage.setupDatabase();
-  const server = new JobQueueServer(AiJob, {
-    storage,
-    queueName: providerName,
-    limiter: new ConcurrencyLimiter(concurrency)
-  });
-  const client = new JobQueueClient({
-    storage,
-    queueName: providerName
-  });
-  client.attach(server);
-  getTaskQueueRegistry().registerQueue({ server, client, storage });
-  await server.start();
-}
 
 // src/provider/AiProvider.ts
+import {
+  globalServiceRegistry as globalServiceRegistry3,
+  WORKER_MANAGER as WORKER_MANAGER2
+} from "@workglow/util/worker";
 class AiProvider {
   tasks;
   streamTasks;
@@ -355,19 +365,21 @@ class AiProvider {
   getReactiveRunFn(taskType) {
     return this.reactiveTasks?.[taskType];
   }
-  async register(options = { mode: "inline" }) {
-    await this.onInitialize(options);
-    if (options.mode === "worker") {
-      if (!options.worker) {
-        throw new Error(`AiProvider "${this.name}": worker is required when mode is "worker". ` + `Pass a Web Worker instance, e.g. register({ mode: "worker", worker: new Worker(...) }).`);
+  async register(options = {}) {
+    const isInline = !!this.tasks;
+    const context = { ...options, isInline };
+    await this.onInitialize(context);
+    if (isInline) {
+      if (!this.tasks) {
+        throw new Error(`AiProvider "${this.name}": tasks must be provided via the constructor for inline registration. ` + `Pass the tasks record when constructing the provider, e.g. new MyProvider(MY_TASKS).`);
       }
     } else {
-      if (!this.tasks) {
-        throw new Error(`AiProvider "${this.name}": tasks must be provided via the constructor for inline mode. ` + `Pass the tasks record when constructing the provider, e.g. new MyProvider(MY_TASKS).`);
+      if (!options.worker) {
+        throw new Error(`AiProvider "${this.name}": worker is required when no tasks are provided (worker-backed registration). ` + `Pass worker: new Worker(...) or worker: () => new Worker(...).`);
       }
     }
     const registry = getAiProviderRegistry();
-    if (options.mode === "worker" && options.worker) {
+    if (!isInline && options.worker) {
       const workerManager = globalServiceRegistry3.get(WORKER_MANAGER2);
       workerManager.registerWorker(this.name, options.worker);
       for (const taskType of this.taskTypes) {
@@ -391,9 +403,7 @@ class AiProvider {
       }
     }
     registry.registerProvider(this);
-    if (options.queue?.autoCreate !== false) {
-      await this.createQueue(options.queue?.concurrency ?? 1);
-    }
+    await this.afterRegister(options);
   }
   registerOnWorkerServer(workerServer) {
     if (!this.tasks) {
@@ -415,10 +425,39 @@ class AiProvider {
   }
   async onInitialize(_options) {}
   async dispose() {}
-  async createQueue(concurrency) {
-    await createDefaultQueue(this.name, concurrency);
+  async afterRegister(_options) {}
+}
+
+// src/queue/createDefaultQueue.ts
+import { ConcurrencyLimiter, JobQueueClient, JobQueueServer } from "@workglow/job-queue";
+import { InMemoryQueueStorage } from "@workglow/storage";
+import { getTaskQueueRegistry } from "@workglow/task-graph";
+async function createDefaultQueue(providerName, concurrency) {
+  const storage = new InMemoryQueueStorage(providerName);
+  await storage.setupDatabase();
+  const server = new JobQueueServer(AiJob, {
+    storage,
+    queueName: providerName,
+    limiter: new ConcurrencyLimiter(concurrency)
+  });
+  const client = new JobQueueClient({
+    storage,
+    queueName: providerName
+  });
+  client.attach(server);
+  getTaskQueueRegistry().registerQueue({ server, client, storage });
+  await server.start();
+}
+
+// src/provider/QueuedAiProvider.ts
+class QueuedAiProvider extends AiProvider {
+  async afterRegister(options) {
+    if (options.queue?.autoCreate !== false) {
+      await createDefaultQueue(this.name, options.queue?.concurrency ?? 1);
+    }
   }
 }
+
 // src/task/index.ts
 import { TaskRegistry } from "@workglow/task-graph";
 
@@ -498,7 +537,6 @@ import {
   getTaskConstructors,
   Workflow
 } from "@workglow/task-graph";
-import { getLogger } from "@workglow/util";
 
 // src/task/base/AiTaskSchemas.ts
 var TypeLanguage = (annotations = {}) => ({
@@ -818,30 +856,6 @@ class StreamingAiTask extends AiTask {
 }
 
 // src/task/ToolCallingTask.ts
-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) {
-  return toolCalls.filter((tc) => {
-    if (tc.name && isAllowedToolName(tc.name, allowedTools)) {
-      return true;
-    }
-    getLogger().warn(`Filtered out tool call with unknown name "${tc.name ?? "(missing)"}"`, {
-      callId: tc.id,
-      toolName: tc.name
-    });
-    return false;
-  });
-}
 function taskTypesToTools(taskNames, registry) {
   const constructors = getTaskConstructors(registry);
   return taskNames.map((name) => {
@@ -1458,9 +1472,9 @@ class AiVisionTask extends AiTask {
     const queueName = registeredQueue?.server.queueName;
     const supports = ["Blob"];
     if (input.image) {
-      if (queueName === "TENSORFLOW_MEDIAPIPE" && "ImageBitmap" in globalThis) {
+      if (typeof queueName === "string" && queueName.startsWith("TENSORFLOW_MEDIAPIPE") && "ImageBitmap" in globalThis) {
         supports.push("ImageBitmap");
-      } else if (queueName === "TENSORFLOW_MEDIAPIPE" && "VideoFrame" in globalThis) {
+      } else if (typeof queueName === "string" && queueName.startsWith("TENSORFLOW_MEDIAPIPE") && "VideoFrame" in globalThis) {
         supports.push("VideoFrame");
       }
       const image = await convertImageDataToUseableForm(input.image, supports);
@@ -3063,7 +3077,7 @@ var DownloadModelOutputSchema = {
 
 class DownloadModelTask extends AiTask {
   static type = "DownloadModelTask";
-  static category = "AI Text Model";
+  static category = "AI Model";
   static title = "Download Model";
   static description = "Downloads and caches AI models locally with progress tracking";
   static inputSchema() {
@@ -3082,7 +3096,18 @@ class DownloadModelTask extends AiTask {
     });
   }
   processProgress(progress, message = "", details) {
-    if (details?.file) {
+    if (details?.files && typeof details.files === "object") {
+      const entries = Object.entries(details.files);
+      if (entries.length > 0) {
+        this.files = entries.map(([file, info]) => ({
+          file,
+          progress: info.total > 0 ? info.loaded / info.total * 100 : 0
+        })).sort((a, b) => a.file.localeCompare(b.file));
+        this.progress = progress;
+        return;
+      }
+    }
+    if (details?.file !== undefined && details.progress !== undefined) {
       const file = this.files.find((f) => f.file === details.file);
       if (file) {
         file.progress = details.progress;
@@ -4376,7 +4401,10 @@ var imageToText = (input, config) => {
 Workflow25.prototype.imageToText = CreateWorkflow25(ImageToTextTask);
 
 // src/task/ModelInfoTask.ts
-import { CreateWorkflow as CreateWorkflow26, Workflow as Workflow26 } from "@workglow/task-graph";
+import {
+  CreateWorkflow as CreateWorkflow26,
+  Workflow as Workflow26
+} from "@workglow/task-graph";
 var modelSchema19 = TypeModel("model");
 var ModelInfoInputSchema = {
   type: "object",
@@ -4401,7 +4429,12 @@ var ModelInfoOutputSchema = {
     supports_node: { type: "boolean" },
     is_cached: { type: "boolean" },
     is_loaded: { type: "boolean" },
-    file_sizes: {}
+    file_sizes: {},
+    quantizations: {
+      type: "array",
+      items: { type: "string" },
+      description: "Available quantization variants (e.g. fp32, fp16, q8). Only present for models with quantization options."
+    }
   },
   required: [
     "model",
@@ -4418,7 +4451,8 @@ var ModelInfoOutputSchema = {
 
 class ModelInfoTask extends AiTask {
   static type = "ModelInfoTask";
-  static category = "Hidden";
+  static category = "AI Model";
+  static cacheable = false;
   static title = "Model Info";
   static description = "Returns runtime information about a model including locality, cache status, and file sizes";
   static inputSchema() {
@@ -4427,15 +4461,87 @@ class ModelInfoTask extends AiTask {
   static outputSchema() {
     return ModelInfoOutputSchema;
   }
-  static cacheable = false;
+  async execute(input, context) {
+    const model = input.model;
+    const registry = getAiProviderRegistry();
+    const noop = () => {};
+    const runFn = registry.getDirectRunFn(model.provider, "ModelInfoTask");
+    return runFn(input, model, noop, context.signal);
+  }
 }
 var modelInfo = (input, config) => {
   return new ModelInfoTask({}, config).run(input);
 };
 Workflow26.prototype.modelInfo = CreateWorkflow26(ModelInfoTask);
 
+// src/task/ModelSearchTask.ts
+import { CreateWorkflow as CreateWorkflow27, Task as Task11, Workflow as Workflow27 } from "@workglow/task-graph";
+var ModelSearchInputSchema = {
+  type: "object",
+  properties: {
+    provider: {
+      type: "string",
+      title: "Provider",
+      description: "The model provider to search (e.g. ANTHROPIC, OPENAI, HF_TRANSFORMERS_ONNX)"
+    },
+    query: {
+      type: "string",
+      title: "Query",
+      description: "Search query string"
+    }
+  },
+  required: ["provider", "query"],
+  additionalProperties: false
+};
+var ModelSearchOutputSchema = {
+  type: "object",
+  properties: {
+    results: {
+      type: "array",
+      items: {
+        type: "object",
+        properties: {
+          id: { type: "string" },
+          label: { type: "string" },
+          description: { type: "string" },
+          record: { type: "object", additionalProperties: true },
+          raw: {}
+        },
+        required: ["id", "label", "description", "record"],
+        additionalProperties: false
+      }
+    }
+  },
+  required: ["results"],
+  additionalProperties: false
+};
+
+class ModelSearchTask extends Task11 {
+  static type = "ModelSearchTask";
+  static category = "AI Model";
+  static title = "Model Search";
+  static description = "Search for models using provider-specific search functions";
+  static cacheable = false;
+  static inputSchema() {
+    return ModelSearchInputSchema;
+  }
+  static outputSchema() {
+    return ModelSearchOutputSchema;
+  }
+  async execute(input, context) {
+    const registry = getAiProviderRegistry();
+    const noop = () => {};
+    const runFn = registry.getDirectRunFn(input.provider, "ModelSearchTask");
+    return runFn(input, undefined, noop, context.signal);
+  }
+}
+var modelSearch = (input, config) => {
+  return new ModelSearchTask({}, config).run(input);
+};
+Workflow27.prototype.modelSearch = CreateWorkflow27(ModelSearchTask);
+
 // src/task/ObjectDetectionTask.ts
-import { CreateWorkflow as CreateWorkflow27, Workflow as Workflow27 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow28, Workflow as Workflow28 } from "@workglow/task-graph";
 var modelSchema20 = TypeModel("model:ObjectDetectionTask");
 var detectionSchema = {
   type: "object",
@@ -4515,10 +4621,10 @@ class ObjectDetectionTask extends AiVisionTask {
 var objectDetection = (input, config) => {
   return new ObjectDetectionTask({}, config).run(input);
 };
-Workflow27.prototype.objectDetection = CreateWorkflow27(ObjectDetectionTask);
+Workflow28.prototype.objectDetection = CreateWorkflow28(ObjectDetectionTask);
 
 // src/task/PoseLandmarkerTask.ts
-import { CreateWorkflow as CreateWorkflow28, Workflow as Workflow28 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow29, Workflow as Workflow29 } from "@workglow/task-graph";
 var modelSchema21 = TypeModel("model:PoseLandmarkerTask");
 var TypePoseLandmark = {
   type: "object",
@@ -4677,13 +4783,13 @@ class PoseLandmarkerTask extends AiVisionTask {
 var poseLandmarker = (input, config) => {
   return new PoseLandmarkerTask({}, config).run(input);
 };
-Workflow28.prototype.poseLandmarker = CreateWorkflow28(PoseLandmarkerTask);
+Workflow29.prototype.poseLandmarker = CreateWorkflow29(PoseLandmarkerTask);
 
 // src/task/QueryExpanderTask.ts
 import {
-  CreateWorkflow as CreateWorkflow29,
-  Task as Task11,
-  Workflow as Workflow29
+  CreateWorkflow as CreateWorkflow30,
+  Task as Task12,
+  Workflow as Workflow30
 } from "@workglow/task-graph";
 var QueryExpansionMethod = {
   MULTI_QUERY: "multi-query",
@@ -4752,7 +4858,7 @@ var outputSchema10 = {
   additionalProperties: false
 };
 
-class QueryExpanderTask extends Task11 {
+class QueryExpanderTask extends Task12 {
   static type = "QueryExpanderTask";
   static category = "RAG";
   static title = "Query Expander";
@@ -4894,17 +5000,17 @@ class QueryExpanderTask extends Task11 {
 var queryExpander = (input, config) => {
   return new QueryExpanderTask({}, config).run(input);
 };
-Workflow29.prototype.queryExpander = CreateWorkflow29(QueryExpanderTask);
+Workflow30.prototype.queryExpander = CreateWorkflow30(QueryExpanderTask);
 
 // src/task/RerankerTask.ts
 import {
-  CreateWorkflow as CreateWorkflow31,
-  Task as Task12,
-  Workflow as Workflow31
+  CreateWorkflow as CreateWorkflow32,
+  Task as Task13,
+  Workflow as Workflow32
 } from "@workglow/task-graph";
 
 // src/task/TextClassificationTask.ts
-import { CreateWorkflow as CreateWorkflow30, Workflow as Workflow30 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow31, Workflow as Workflow31 } from "@workglow/task-graph";
 var modelSchema22 = TypeModel("model:TextClassificationTask");
 var TextClassificationInputSchema = {
   type: "object",
@@ -4982,7 +5088,7 @@ class TextClassificationTask extends AiTask {
 var textClassification = (input, config) => {
   return new TextClassificationTask({}, config).run(input);
 };
-Workflow30.prototype.textClassification = CreateWorkflow30(TextClassificationTask);
+Workflow31.prototype.textClassification = CreateWorkflow31(TextClassificationTask);
 
 // src/task/RerankerTask.ts
 var inputSchema11 = {
@@ -5077,7 +5183,7 @@ var outputSchema11 = {
   additionalProperties: false
 };
 
-class RerankerTask extends Task12 {
+class RerankerTask extends Task13 {
   static type = "RerankerTask";
   static category = "RAG";
   static title = "Reranker";
@@ -5206,14 +5312,14 @@ class RerankerTask extends Task12 {
 var reranker = (input, config) => {
   return new RerankerTask({}, config).run(input);
 };
-Workflow31.prototype.reranker = CreateWorkflow31(RerankerTask);
+Workflow32.prototype.reranker = CreateWorkflow32(RerankerTask);
 
 // src/task/StructuralParserTask.ts
 import { StructuralParser } from "@workglow/knowledge-base";
 import {
-  CreateWorkflow as CreateWorkflow32,
-  Task as Task13,
-  Workflow as Workflow32
+  CreateWorkflow as CreateWorkflow33,
+  Task as Task14,
+  Workflow as Workflow33
 } from "@workglow/task-graph";
 import { uuid4 as uuid42 } from "@workglow/util";
 var inputSchema12 = {
@@ -5272,7 +5378,7 @@ var outputSchema12 = {
   additionalProperties: false
 };
 
-class StructuralParserTask extends Task13 {
+class StructuralParserTask extends Task14 {
   static type = "StructuralParserTask";
   static category = "Document";
   static title = "Structural Parser";
@@ -5315,10 +5421,10 @@ class StructuralParserTask extends Task13 {
 var structuralParser = (input, config) => {
   return new StructuralParserTask({}, config).run(input);
 };
-Workflow32.prototype.structuralParser = CreateWorkflow32(StructuralParserTask);
+Workflow33.prototype.structuralParser = CreateWorkflow33(StructuralParserTask);
 
 // src/task/StructuredGenerationTask.ts
-import { CreateWorkflow as CreateWorkflow33, Workflow as Workflow33 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow34, Workflow as Workflow34 } from "@workglow/task-graph";
 var modelSchema23 = TypeModel("model:StructuredGenerationTask");
 var StructuredGenerationInputSchema = {
   type: "object",
@@ -5386,13 +5492,13 @@ class StructuredGenerationTask extends StreamingAiTask {
 var structuredGeneration = (input, config) => {
   return new StructuredGenerationTask({}, config).run(input);
 };
-Workflow33.prototype.structuredGeneration = CreateWorkflow33(StructuredGenerationTask);
+Workflow34.prototype.structuredGeneration = CreateWorkflow34(StructuredGenerationTask);
 
 // src/task/TextChunkerTask.ts
 import {
-  CreateWorkflow as CreateWorkflow34,
-  Task as Task14,
-  Workflow as Workflow34
+  CreateWorkflow as CreateWorkflow35,
+  Task as Task15,
+  Workflow as Workflow35
 } from "@workglow/task-graph";
 var ChunkingStrategy = {
   FIXED: "fixed",
@@ -5462,7 +5568,7 @@ var outputSchema13 = {
   additionalProperties: false
 };
 
-class TextChunkerTask extends Task14 {
+class TextChunkerTask extends Task15 {
   static type = "TextChunkerTask";
   static category = "Document";
   static title = "Text Chunker";
@@ -5642,10 +5748,10 @@ class TextChunkerTask extends Task14 {
 var textChunker = (input, config) => {
   return new TextChunkerTask({}, config).run(input);
 };
-Workflow34.prototype.textChunker = CreateWorkflow34(TextChunkerTask);
+Workflow35.prototype.textChunker = CreateWorkflow35(TextChunkerTask);
 
 // src/task/TextFillMaskTask.ts
-import { CreateWorkflow as CreateWorkflow35, Workflow as Workflow35 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow36, Workflow as Workflow36 } from "@workglow/task-graph";
 var modelSchema24 = TypeModel("model:TextFillMaskTask");
 var TextFillMaskInputSchema = {
   type: "object",
@@ -5710,10 +5816,10 @@ class TextFillMaskTask extends AiTask {
 var textFillMask = (input, config) => {
   return new TextFillMaskTask({}, config).run(input);
 };
-Workflow35.prototype.textFillMask = CreateWorkflow35(TextFillMaskTask);
+Workflow36.prototype.textFillMask = CreateWorkflow36(TextFillMaskTask);
 
 // src/task/TextGenerationTask.ts
-import { CreateWorkflow as CreateWorkflow36, Workflow as Workflow36 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow37, Workflow as Workflow37 } from "@workglow/task-graph";
 var generatedTextSchema2 = TypeSingleOrArray({
   type: "string",
   title: "Text",
@@ -5798,10 +5904,10 @@ class TextGenerationTask extends StreamingAiTask {
 var textGeneration = (input, config) => {
   return new TextGenerationTask({}, config).run(input);
 };
-Workflow36.prototype.textGeneration = CreateWorkflow36(TextGenerationTask);
+Workflow37.prototype.textGeneration = CreateWorkflow37(TextGenerationTask);
 
 // src/task/TextLanguageDetectionTask.ts
-import { CreateWorkflow as CreateWorkflow37, Workflow as Workflow37 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow38, Workflow as Workflow38 } from "@workglow/task-graph";
 var modelSchema26 = TypeModel("model:TextLanguageDetectionTask");
 var TextLanguageDetectionInputSchema = {
   type: "object",
@@ -5869,10 +5975,10 @@ class TextLanguageDetectionTask extends AiTask {
 var textLanguageDetection = (input, config) => {
   return new TextLanguageDetectionTask({}, config).run(input);
 };
-Workflow37.prototype.textLanguageDetection = CreateWorkflow37(TextLanguageDetectionTask);
+Workflow38.prototype.textLanguageDetection = CreateWorkflow38(TextLanguageDetectionTask);
 
 // src/task/TextQuestionAnswerTask.ts
-import { CreateWorkflow as CreateWorkflow38, Workflow as Workflow38 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow39, Workflow as Workflow39 } from "@workglow/task-graph";
 var contextSchema = TypeSingleOrArray({
   type: "string",
   title: "Context",
@@ -5924,10 +6030,10 @@ class TextQuestionAnswerTask extends StreamingAiTask {
 var textQuestionAnswer = (input, config) => {
   return new TextQuestionAnswerTask({}, config).run(input);
 };
-Workflow38.prototype.textQuestionAnswer = CreateWorkflow38(TextQuestionAnswerTask);
+Workflow39.prototype.textQuestionAnswer = CreateWorkflow39(TextQuestionAnswerTask);
 
 // src/task/TextRewriterTask.ts
-import { CreateWorkflow as CreateWorkflow39, Workflow as Workflow39 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow40, Workflow as Workflow40 } from "@workglow/task-graph";
 var modelSchema28 = TypeModel("model:TextRewriterTask");
 var TextRewriterInputSchema = {
   type: "object",
@@ -5976,10 +6082,10 @@ class TextRewriterTask extends StreamingAiTask {
 var textRewriter = (input, config) => {
   return new TextRewriterTask({}, config).run(input);
 };
-Workflow39.prototype.textRewriter = CreateWorkflow39(TextRewriterTask);
+Workflow40.prototype.textRewriter = CreateWorkflow40(TextRewriterTask);
 
 // src/task/TextTranslationTask.ts
-import { CreateWorkflow as CreateWorkflow40, Workflow as Workflow40 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow41, Workflow as Workflow41 } from "@workglow/task-graph";
 var modelSchema29 = TypeModel("model:TextTranslationTask");
 var translationTextSchema = TypeSingleOrArray({
   type: "string",
@@ -6042,13 +6148,13 @@ class TextTranslationTask extends StreamingAiTask {
 var textTranslation = (input, config) => {
   return new TextTranslationTask({}, config).run(input);
 };
-Workflow40.prototype.textTranslation = CreateWorkflow40(TextTranslationTask);
+Workflow41.prototype.textTranslation = CreateWorkflow41(TextTranslationTask);
 
 // src/task/TopicSegmenterTask.ts
 import {
-  CreateWorkflow as CreateWorkflow41,
-  Task as Task15,
-  Workflow as Workflow41
+  CreateWorkflow as CreateWorkflow42,
+  Task as Task16,
+  Workflow as Workflow42
 } from "@workglow/task-graph";
 var SegmentationMethod = {
   HEURISTIC: "heuristic",
@@ -6124,7 +6230,7 @@ var outputSchema14 = {
   additionalProperties: false
 };
 
-class TopicSegmenterTask extends Task15 {
+class TopicSegmenterTask extends Task16 {
   static type = "TopicSegmenterTask";
   static category = "Document";
   static title = "Topic Segmenter";
@@ -6329,10 +6435,10 @@ class TopicSegmenterTask extends Task15 {
 var topicSegmenter = (input, config) => {
   return new TopicSegmenterTask({}, config).run(input);
 };
-Workflow41.prototype.topicSegmenter = CreateWorkflow41(TopicSegmenterTask);
+Workflow42.prototype.topicSegmenter = CreateWorkflow42(TopicSegmenterTask);
 
 // src/task/UnloadModelTask.ts
-import { CreateWorkflow as CreateWorkflow42, Workflow as Workflow42 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow43, Workflow as Workflow43 } from "@workglow/task-graph";
 var modelSchema30 = TypeModel("model");
 var UnloadModelInputSchema = {
   type: "object",
@@ -6353,7 +6459,7 @@ var UnloadModelOutputSchema = {
 
 class UnloadModelTask extends AiTask {
   static type = "UnloadModelTask";
-  static category = "Hidden";
+  static category = "AI Model";
   static title = "Unload Model";
   static description = "Unloads and clears cached AI models from memory and storage";
   static inputSchema() {
@@ -6367,10 +6473,10 @@ class UnloadModelTask extends AiTask {
 var unloadModel = (input, config) => {
   return new UnloadModelTask({}, config).run(input);
 };
-Workflow42.prototype.unloadModel = CreateWorkflow42(UnloadModelTask);
+Workflow43.prototype.unloadModel = CreateWorkflow43(UnloadModelTask);
 
 // src/task/VectorQuantizeTask.ts
-import { CreateWorkflow as CreateWorkflow43, Task as Task16, Workflow as Workflow43 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow44, Task as Task17, Workflow as Workflow44 } from "@workglow/task-graph";
 import {
   normalizeNumberArray,
   TensorType,
@@ -6450,7 +6556,7 @@ var outputSchema15 = {
   additionalProperties: false
 };
 
-class VectorQuantizeTask extends Task16 {
+class VectorQuantizeTask extends Task17 {
   static type = "VectorQuantizeTask";
   static category = "Vector";
   static title = "Quantize";
@@ -6550,10 +6656,10 @@ class VectorQuantizeTask extends Task16 {
 var vectorQuantize = (input, config) => {
   return new VectorQuantizeTask({}, config).run(input);
 };
-Workflow43.prototype.vectorQuantize = CreateWorkflow43(VectorQuantizeTask);
+Workflow44.prototype.vectorQuantize = CreateWorkflow44(VectorQuantizeTask);
 
 // src/task/VectorSimilarityTask.ts
-import { CreateWorkflow as CreateWorkflow44, GraphAsTask, Workflow as Workflow44 } from "@workglow/task-graph";
+import { CreateWorkflow as CreateWorkflow45, GraphAsTask, Workflow as Workflow45 } from "@workglow/task-graph";
 import {
   cosineSimilarity,
   hammingSimilarity,
@@ -6659,7 +6765,7 @@ class VectorSimilarityTask extends GraphAsTask {
 var similarity = (input, config) => {
   return new VectorSimilarityTask({}, config).run(input);
 };
-Workflow44.prototype.similarity = CreateWorkflow44(VectorSimilarityTask);
+Workflow45.prototype.similarity = CreateWorkflow45(VectorSimilarityTask);
 // src/task/MessageConversion.ts
 function getInputMessages(input) {
   const messages = input.messages;
@@ -6882,6 +6988,7 @@ var registerAiTasks = () => {
     ImageSegmentationTask,
     ImageToTextTask,
     ModelInfoTask,
+    ModelSearchTask,
     ObjectDetectionTask,
     PoseLandmarkerTask,
     QueryExpanderTask,
@@ -6941,6 +7048,7 @@ export {
   queryExpander,
   poseLandmarker,
   objectDetection,
+  modelSearch,
   modelInfo,
   isAllowedToolName,
   imageToText,
@@ -7033,6 +7141,7 @@ export {
   SimilarityFn,
   SegmentationMethod,
   RerankerTask,
+  QueuedAiProvider,
   QueryExpansionMethod,
   QueryExpanderTask,
   PoseLandmarkerTask,
@@ -7041,6 +7150,7 @@ export {
   ObjectDetectionTask,
   ObjectDetectionOutputSchema,
   ObjectDetectionInputSchema,
+  ModelSearchTask,
   ModelRepository,
   ModelRecordSchema,
   ModelPrimaryKeyNames,
@@ -7099,4 +7209,4 @@ export {
   AgentInputSchema
 };
 
-//# debugId=470E906FF8218D7164756E2164756E21
+//# debugId=01FBA636739047B364756E2164756E21