@workglow/ai 0.0.123 → 0.0.125

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);
@@ -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 = () => {};
@@ -7209,4 +7246,4 @@ export {
   AgentInputSchema
 };
 
-//# debugId=D89F77C8F1A116D164756E2164756E21
+//# debugId=10C25A453F3375E964756E2164756E21