@workglow/ai 0.0.122 → 0.0.124

package/dist/job/AiJob.d.ts

@@ -5,7 +5,7 @@
  */
 import { IJobExecuteContext, Job } from "@workglow/job-queue";
 import { TaskInput, TaskOutput, type StreamEvent } from "@workglow/task-graph";
-import type { JsonSchema } from "@workglow/util";
+import type { JsonSchema } from "@workglow/util/schema";
 import type { ModelConfig } from "../model/ModelSchema";
 /**
  * Input data for the AiJob

package/dist/job/AiJob.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AiJob.d.ts","sourceRoot":"","sources":["../../src/job/AiJob.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAEL,kBAAkB,EAClB,GAAG,EAGJ,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,KAAK,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC/E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAGxD;;GAEG;AACH,MAAM,WAAW,UAAU,CAAC,KAAK,SAAS,SAAS,GAAG,SAAS;IAC7D,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,KAAK,GAAG;QAAE,KAAK,EAAE,WAAW,CAAA;KAAE,CAAC;IAC1C,qFAAqF;IACrF,YAAY,CAAC,EAAE,UAAU,CAAC;CAC3B;AAED;;;GAGG;AACH,qBAAa,KAAK,CAChB,KAAK,SAAS,UAAU,CAAC,SAAS,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC,EAC3D,MAAM,SAAS,UAAU,GAAG,UAAU,CACtC,SAAQ,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC;IAC1B;;OAEG;IACG,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,MAAM,CAAC,CAgDxE;IAED;;;OAGG;IACI,aAAa,CAClB,KAAK,EAAE,KAAK,EACZ,OAAO,EAAE,kBAAkB,GAC1B,aAAa,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAkBpC;CACF"}
+{"version":3,"file":"AiJob.d.ts","sourceRoot":"","sources":["../../src/job/AiJob.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAEL,kBAAkB,EAClB,GAAG,EAGJ,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,KAAK,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAC/E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAGxD;;GAEG;AACH,MAAM,WAAW,UAAU,CAAC,KAAK,SAAS,SAAS,GAAG,SAAS;IAC7D,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,KAAK,GAAG;QAAE,KAAK,EAAE,WAAW,CAAA;KAAE,CAAC;IAC1C,qFAAqF;IACrF,YAAY,CAAC,EAAE,UAAU,CAAC;CAC3B;AAED;;;GAGG;AACH,qBAAa,KAAK,CAChB,KAAK,SAAS,UAAU,CAAC,SAAS,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC,EAC3D,MAAM,SAAS,UAAU,GAAG,UAAU,CACtC,SAAQ,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC;IAC1B;;OAEG;IACG,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,MAAM,CAAC,CAgDxE;IAED;;;OAGG;IACI,aAAa,CAClB,KAAK,EAAE,KAAK,EACZ,OAAO,EAAE,kBAAkB,GAC1B,aAAa,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAkBpC;CACF"}

package/dist/node.js

@@ -23,6 +23,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);
@@ -469,7 +478,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 };
 }
@@ -1462,7 +1471,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) {
@@ -1537,13 +1546,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",
@@ -1788,7 +1797,7 @@ import {
 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema3
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema2 = {
   type: "object",
   properties: {
@@ -1922,7 +1931,7 @@ import {
 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema4
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema3 = {
   type: "object",
   properties: {
@@ -2096,7 +2105,7 @@ import {
 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema5
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema4 = {
   type: "object",
   properties: {
@@ -2219,7 +2228,7 @@ import {
 } from "@workglow/task-graph";
 import {
   TypedArraySchema as TypedArraySchema6
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema5 = {
   type: "object",
   properties: {
@@ -4215,7 +4224,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",
@@ -4481,7 +4490,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",
@@ -4492,6 +4501,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: {
@@ -4521,12 +4554,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 = () => {};
@@ -6480,7 +6517,7 @@ import {
   normalizeNumberArray,
   TensorType,
   TypedArraySchema as TypedArraySchema8
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var inputSchema15 = {
   type: "object",
   properties: {
@@ -6664,7 +6701,7 @@ import {
   hammingSimilarity,
   jaccardSimilarity,
   TypedArraySchema as TypedArraySchema9
-} from "@workglow/util";
+} from "@workglow/util/schema";
 var SimilarityFn = {
   COSINE: "cosine",
   JACCARD: "jaccard",
@@ -7208,4 +7245,4 @@ export {
   AgentInputSchema
 };
 
-//# debugId=3E409008C441159964756E2164756E21
+//# debugId=D07E930D64256EAD64756E2164756E21