@workglow/ai 0.0.122 → 0.0.124

package/dist/bun.js

@@ -24,6 +24,15 @@ class AiProviderRegistry {
   getProviders() {
     return new Map(this.providers);
   }
+  getInstalledProviderIds() {
+    return [...this.providers.keys()].sort();
+  }
+  getProviderIdsForTask(taskType) {
+    const taskMap = this.runFnRegistry.get(taskType);
+    if (!taskMap)
+      return [];
+    return [...taskMap.keys()].sort();
+  }
   registerRunFn(modelProvider, taskType, runFn) {
     if (!this.runFnRegistry.has(taskType)) {
       this.runFnRegistry.set(taskType, new Map);
@@ -470,7 +479,7 @@ import {
 import { getLogger as getLogger3 } from "@workglow/util";
 
 // src/task/AgentTypes.ts
-import { parseDataUri } from "@workglow/util";
+import { parseDataUri } from "@workglow/util/media";
 function imageBlock(mimeType, data) {
   return { type: "image", mimeType, data };
 }
@@ -1463,7 +1472,7 @@ Workflow2.prototype.agent = CreateWorkflow2(AgentTask);
 import { CreateWorkflow as CreateWorkflow3, Workflow as Workflow3 } from "@workglow/task-graph";
 
 // src/task/base/AiVisionTask.ts
-import { convertImageDataToUseableForm } from "@workglow/util";
+import { convertImageDataToUseableForm } from "@workglow/util/media";
 class AiVisionTask extends AiTask {
   static type = "AiVisionTask";
   async getJobInput(input) {
@@ -1538,13 +1547,13 @@ import {
 import {
   isTypedArray,
   TypedArraySchema as TypedArraySchema2
-} from "@workglow/util";
+} from "@workglow/util/schema";
 
 // src/task/TextEmbeddingTask.ts
 import { CreateWorkflow as CreateWorkflow4, Workflow as Workflow4 } from "@workglow/task-graph";
 import {
   TypedArraySchema
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var modelSchema4 = TypeModel("model:TextEmbeddingTask");
 var TextEmbeddingInputSchema = {
   type: "object",
@@ -1789,7 +1798,7 @@ import {
 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema3
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema2 = {
   type: "object",
   properties: {
@@ -1923,7 +1932,7 @@ import {
 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema4
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema3 = {
   type: "object",
   properties: {
@@ -2097,7 +2106,7 @@ import {
 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema5
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema4 = {
   type: "object",
   properties: {
@@ -2220,7 +2229,7 @@ import {
 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema6
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema5 = {
   type: "object",
   properties: {
@@ -4216,7 +4225,7 @@ Workflow22.prototype.imageClassification = CreateWorkflow22(ImageClassificationT
 import { CreateWorkflow as CreateWorkflow23, Workflow as Workflow23 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema7
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var modelSchema16 = TypeModel("model:ImageEmbeddingTask");
 var embeddingSchema = TypedArraySchema7({
   title: "Embedding",
@@ -4482,7 +4491,7 @@ var ModelSearchInputSchema = {
     provider: {
       type: "string",
       title: "Provider",
-      description: "The model provider to search (e.g. ANTHROPIC, OPENAI, HF_TRANSFORMERS_ONNX)"
+      description: "Registered AI provider id to use for model search. At runtime the workflow UI lists only providers that support model search."
     },
     query: {
       type: "string",
@@ -4493,6 +4502,30 @@ var ModelSearchInputSchema = {
   required: ["provider", "query"],
   additionalProperties: false
 };
+function buildModelSearchInputSchemaDynamic() {
+  const registry = getAiProviderRegistry();
+  const ids = registry.getProviderIdsForTask("ModelSearchTask");
+  const enumLabels = {};
+  for (const id of ids) {
+    enumLabels[id] = registry.getProvider(id)?.displayName ?? id;
+  }
+  const providerProp = {
+    ...ModelSearchInputSchema.properties.provider
+  };
+  if (ids.length > 0) {
+    providerProp.enum = ids;
+    providerProp["x-ui-enum-labels"] = enumLabels;
+  }
+  return {
+    type: "object",
+    properties: {
+      provider: providerProp,
+      query: ModelSearchInputSchema.properties.query
+    },
+    required: [...ModelSearchInputSchema.required],
+    additionalProperties: ModelSearchInputSchema.additionalProperties
+  };
+}
 var ModelSearchOutputSchema = {
   type: "object",
   properties: {
@@ -4522,12 +4555,16 @@ class ModelSearchTask extends Task11 {
   static title = "Model Search";
   static description = "Search for models using provider-specific search functions";
   static cacheable = false;
+  static hasDynamicSchemas = true;
   static inputSchema() {
     return ModelSearchInputSchema;
   }
   static outputSchema() {
     return ModelSearchOutputSchema;
   }
+  inputSchema() {
+    return buildModelSearchInputSchemaDynamic();
+  }
   async execute(input, context) {
     const registry = getAiProviderRegistry();
     const noop = () => {};
@@ -6481,7 +6518,7 @@ import {
   normalizeNumberArray,
   TensorType,
   TypedArraySchema as TypedArraySchema8
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema15 = {
   type: "object",
   properties: {
@@ -6665,7 +6702,7 @@ import {
   hammingSimilarity,
   jaccardSimilarity,
   TypedArraySchema as TypedArraySchema9
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var SimilarityFn = {
   COSINE: "cosine",
   JACCARD: "jaccard",
@@ -7209,4 +7246,4 @@ export {
   AgentInputSchema
 };
 
-//# debugId=01FBA636739047B364756E2164756E21
+//# debugId=10C25A453F3375E964756E2164756E21