@bratsos/workflow-engine 0.3.0 → 0.4.0

package/dist/client.d.ts

@@ -1,4 +1,5 @@
-export { g as AIHelper, k as AVAILABLE_MODELS, l as AsyncBatchStageDefinition, D as DEFAULT_MODEL_KEY, m as EnhancedStageContext, I as InferInput, V as LogPayload, n as ModelConfig, X as ModelFilter, M as ModelKey, o as ModelRegistry, N as NoInputSchema, S as SimpleStageResult, Y as StageCompletedPayload, Z as StageFailedPayload, _ as StageStartedPayload, t as SyncStageDefinition, $ as WorkflowCompletedPayload, W as WorkflowEventType, a0 as WorkflowFailedPayload, u as WorkflowSSEEvent, a1 as WorkflowStartedPayload, a2 as WorkflowSuspendedPayload, v as calculateCost, x as defineAsyncBatchStage, y as defineStage, H as listModels, K as modelSupportsBatch, Q as registerModels } from './client-D4PoxADF.js';
+export { g as AIHelper, k as AVAILABLE_MODELS, l as AsyncBatchStageDefinition, D as DEFAULT_MODEL_KEY, m as EnhancedStageContext, I as InferInput, X as LogPayload, n as ModelConfig, Y as ModelFilter, M as ModelKey, o as ModelRegistry, N as NoInputSchema, S as SimpleStageResult, Z as StageCompletedPayload, _ as StageFailedPayload, $ as StageStartedPayload, t as SyncStageDefinition, a0 as WorkflowCompletedPayload, W as WorkflowEventType, a1 as WorkflowFailedPayload, u as WorkflowSSEEvent, a2 as WorkflowStartedPayload, a3 as WorkflowSuspendedPayload, v as calculateCost, x as defineAsyncBatchStage, y as defineStage, H as listModels, K as modelSupportsBatch, U as registerModels } from './client-DxNS_NoL.js';
+import '@ai-sdk/provider';
 import 'ai';
 import 'zod';
 import './interface-MMqhfQQK.js';

package/dist/client.js

@@ -1,4 +1,4 @@
-export { AVAILABLE_MODELS, DEFAULT_MODEL_KEY, ModelKey, NoInputSchema, calculateCost, defineAsyncBatchStage, defineStage, listModels, modelSupportsBatch, registerModels } from './chunk-QBYB2PJJ.js';
+export { AVAILABLE_MODELS, DEFAULT_MODEL_KEY, ModelKey, NoInputSchema, calculateCost, defineAsyncBatchStage, defineStage, listModels, modelSupportsBatch, registerModels } from './chunk-XS3ZX4KW.js';
 import './chunk-MUWP5SF2.js';
 //# sourceMappingURL=client.js.map
 //# sourceMappingURL=client.js.map

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { M as ModelKey } from './client-D4PoxADF.js';
-export { A as AIBatch, a as AIBatchHandle, b as AIBatchProvider, c as AIBatchRequest, d as AIBatchResult, e as AICallType, f as AIEmbedResult, g as AIHelper, h as AIObjectResult, i as AIStreamResult, j as AITextResult, k as AVAILABLE_MODELS, l as AsyncBatchStageDefinition, B as BatchLogFn, D as DEFAULT_MODEL_KEY, E as EmbedOptions, m as EnhancedStageContext, I as InferInput, L as LogContext, n as ModelConfig, o as ModelRegistry, p as ModelStats, q as ModelStatsTracker, r as ModelSyncConfig, N as NoInputSchema, O as ObjectOptions, R as RecordCallParams, S as SimpleStageResult, s as StreamOptions, t as SyncStageDefinition, T as TextOptions, W as WorkflowEventType, u as WorkflowSSEEvent, v as calculateCost, w as createAIHelper, x as defineAsyncBatchStage, y as defineStage, z as getDefaultModel, C as getModel, F as getModelById, G as getRegisteredModel, H as listModels, J as listRegisteredModels, K as modelSupportsBatch, P as printAvailableModels, Q as registerModels, U as requireStageOutput } from './client-D4PoxADF.js';
+import { M as ModelKey } from './client-DxNS_NoL.js';
+export { A as AIBatch, a as AIBatchHandle, b as AIBatchProvider, c as AIBatchRequest, d as AIBatchResult, e as AICallType, f as AIEmbedResult, g as AIHelper, h as AIObjectResult, i as AIStreamResult, j as AITextResult, k as AVAILABLE_MODELS, l as AsyncBatchStageDefinition, B as BatchLogFn, D as DEFAULT_MODEL_KEY, E as EmbedOptions, m as EnhancedStageContext, I as InferInput, L as LogContext, n as ModelConfig, o as ModelRegistry, p as ModelStats, q as ModelStatsTracker, r as ModelSyncConfig, N as NoInputSchema, O as ObjectOptions, R as RecordCallParams, S as SimpleStageResult, s as StreamOptions, t as SyncStageDefinition, T as TextOptions, W as WorkflowEventType, u as WorkflowSSEEvent, v as calculateCost, w as createAIHelper, x as defineAsyncBatchStage, y as defineStage, z as getDefaultModel, C as getModel, F as getModelById, G as getRegisteredModel, H as listModels, J as listRegisteredModels, K as modelSupportsBatch, P as printAvailableModels, Q as registerEmbeddingProvider, U as registerModels, V as requireStageOutput } from './client-DxNS_NoL.js';
 import { L as LogLevel } from './stage-BPw7m9Wx.js';
 export { S as Stage, a as StageResult } from './stage-BPw7m9Wx.js';
 export { C as CommandResult, I as IdempotencyInProgressError, y as InferWorkflowStageIds, J as JobExecuteCommand, a as JobExecuteResult, b as Kernel, c as KernelCommand, d as KernelCommandType, e as KernelConfig, W as KernelWorkflowRegistry, L as LeaseReapStaleCommand, f as LeaseReapStaleResult, O as OutboxFlushCommand, g as OutboxFlushResult, P as PluginDefinition, h as PluginReplayDLQCommand, i as PluginReplayDLQResult, j as PluginRunner, k as PluginRunnerConfig, R as RunCancelCommand, l as RunCancelResult, m as RunClaimPendingCommand, n as RunClaimPendingResult, o as RunCreateCommand, p as RunCreateResult, q as RunRerunFromCommand, r as RunRerunFromResult, s as RunTransitionCommand, t as RunTransitionResult, S as StagePollSuspendedCommand, u as StagePollSuspendedResult, z as Workflow, A as WorkflowBuilder, v as createKernel, w as createPluginRunner, x as definePlugin } from './plugins-CPC-X0rR.js';
@@ -8,6 +8,7 @@ export { A as AICallLogger, i as AICallRecord, h as AIHelperStats, r as Artifact
 export { P as PrismaAICallLogger, a as PrismaJobQueue, c as PrismaWorkflowPersistence, e as createPrismaAICallLogger, f as createPrismaJobQueue, g as createPrismaWorkflowPersistence } from './index-DAzCfO1R.js';
 import { ToolSet } from 'ai';
 export { B as BlobStore, C as Clock, E as EventSink, J as JobTransport, K as KernelEvent, a as KernelEventType, P as Persistence, S as Scheduler } from './ports-tU3rzPXJ.js';
+import '@ai-sdk/provider';
 
 /**
  * Config Presets - Common configuration patterns for workflow stages

package/dist/index.js

@@ -1,5 +1,5 @@
-import { ModelKey } from './chunk-QBYB2PJJ.js';
-export { AVAILABLE_MODELS, AnthropicBatchProvider, DEFAULT_MODEL_KEY, GoogleBatchProvider, ModelKey, ModelStatsTracker, NoInputSchema, OpenAIBatchProvider, calculateCost, createAIHelper, defineAsyncBatchStage, defineStage, getBestProviderForModel, getDefaultModel, getModel, getModelById, getRegisteredModel, listModels, listRegisteredModels, modelSupportsBatch, printAvailableModels, registerModels, requireStageOutput, resolveModelForProvider } from './chunk-QBYB2PJJ.js';
+import { ModelKey } from './chunk-XS3ZX4KW.js';
+export { AVAILABLE_MODELS, AnthropicBatchProvider, DEFAULT_MODEL_KEY, GoogleBatchProvider, ModelKey, ModelStatsTracker, NoInputSchema, OpenAIBatchProvider, calculateCost, createAIHelper, defineAsyncBatchStage, defineStage, getBestProviderForModel, getDefaultModel, getModel, getModelById, getRegisteredModel, listModels, listRegisteredModels, modelSupportsBatch, printAvailableModels, registerEmbeddingProvider, registerModels, requireStageOutput, resolveModelForProvider } from './chunk-XS3ZX4KW.js';
 import './chunk-D7RVRRM2.js';
 export { PrismaAICallLogger, PrismaJobQueue, PrismaWorkflowPersistence, createPrismaAICallLogger, createPrismaJobQueue, createPrismaWorkflowPersistence } from './chunk-NYKMT46J.js';
 import './chunk-MUWP5SF2.js';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bratsos/workflow-engine",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "Type-safe, distributed workflow engine for AI-orchestrated processes with suspend/resume, parallel execution, and cost tracking",
   "type": "module",
   "license": "MIT",
@@ -78,6 +78,7 @@
   ],
   "dependencies": {
     "@ai-sdk/google": "^3.0.1",
+    "@ai-sdk/provider": "^3.0.0",
     "@openrouter/ai-sdk-provider": "^2.1.1",
     "ai": "^6.0.3",
     "js-tiktoken": "^1.0.21",

package/skills/workflow-engine/SKILL.md

@@ -108,6 +108,7 @@ await kernel.dispatch({
 | `createNodeHost` | Function | `@bratsos/workflow-engine-host-node` | Create Node.js host |
 | `createServerlessHost` | Function | `@bratsos/workflow-engine-host-serverless` | Create serverless host |
 | `createAIHelper` | Function | `@bratsos/workflow-engine` | AI operations (text, object, embed, batch) |
+| `registerEmbeddingProvider` | Function | `@bratsos/workflow-engine` | Register custom embedding providers (Voyage, Cohere, etc.) |
 | `definePlugin` | Function | `@bratsos/workflow-engine/kernel` | Define kernel plugins |
 | `createPluginRunner` | Function | `@bratsos/workflow-engine/kernel` | Create plugin event processor |
 
@@ -339,6 +340,12 @@ const { object } = await ai.generateObject("gemini-2.5-flash", prompt, schema);
 const { embedding } = await ai.embed("text-embedding-004", ["text1"], { dimensions: 768 });
 // OpenRouter embedding models (OpenAI, Cohere, etc.)
 const { embedding } = await ai.embed("openai/text-embedding-3-small", ["text1"]);
+
+// Custom embedding providers (Voyage, Cohere, Jina, etc.)
+import { registerEmbeddingProvider } from "@bratsos/workflow-engine";
+import { voyage } from "voyage-ai-provider";
+registerEmbeddingProvider("voyage", (modelId) => voyage.embeddingModel(modelId));
+// Then register models with provider: "voyage" and use ai.embed() as usual
 ```
 
 ## Persistence Setup

package/skills/workflow-engine/references/04-ai-integration.md

@@ -248,7 +248,7 @@ const result = await ai.generateObject(
 
 ## embed
 
-Generate embeddings for text. Supports Google and OpenRouter embedding providers.
+Generate embeddings for text. Supports Google and OpenRouter as built-in providers, plus any custom provider registered via `registerEmbeddingProvider()`.
 
 ```typescript
 // Google embedding model (with Google-specific options)
@@ -287,6 +287,39 @@ console.log(result.embedding);     // First embedding (convenience)
 > OpenRouter embedding models work without provider-specific options. The provider is determined
 > by the `provider` field in the model's `ModelConfig`.
 
+### Custom Embedding Providers
+
+Use `registerEmbeddingProvider()` to add support for any AI SDK community embedding provider (Voyage, Cohere, Jina, etc.) without modifying the library. Call this once at application startup, before any `embed()` calls.
+
+```typescript
+import { registerEmbeddingProvider, registerModels } from "@bratsos/workflow-engine";
+import { voyage } from "voyage-ai-provider";
+
+// 1. Register the provider factory (once at startup)
+registerEmbeddingProvider("voyage", (modelId) => voyage.embeddingModel(modelId));
+
+// 2. Register models that use the provider
+registerModels({
+  "voyage-4-large": {
+    id: "voyage-4-large",
+    name: "Voyage 4 Large",
+    provider: "voyage",           // Must match the name in registerEmbeddingProvider()
+    inputCostPerMillion: 0.06,
+    outputCostPerMillion: 0,
+    isEmbeddingModel: true,
+  },
+});
+
+// 3. Use it like any other embedding model
+const { embedding } = await ai.embed("voyage-4-large", "Hello world");
+```
+
+**How it works:**
+- The factory receives the model's `id` from `ModelConfig` and must return an `EmbeddingModelV3` instance (from `@ai-sdk/provider`)
+- Custom providers are checked **before** built-in providers, so you can even override `"openrouter"` or `"google"` if needed
+- The workflow engine stays provider-agnostic — install your chosen provider package as your app's dependency, not the library's
+- Provider-specific options (like Google's `taskType`) are handled by each provider through the AI SDK's standard mechanism
+
 ## streamText
 
 Stream text generation.