@bratsos/workflow-engine 0.2.1 → 0.3.0

package/dist/index.js

@@ -1,5 +1,5 @@
-import { ModelKey } from './chunk-P4KMGCT3.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-P4KMGCT3.js';
+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 './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.2.1",
+  "version": "0.3.0",
   "description": "Type-safe, distributed workflow engine for AI-orchestrated processes with suspend/resume, parallel execution, and cost tracking",
   "type": "module",
   "license": "MIT",
@@ -76,12 +76,6 @@
     "README.md",
     "LICENSE"
   ],
-  "scripts": {
-    "typecheck": "tsc --noEmit",
-    "build": "tsup",
-    "test": "vitest run",
-    "sync-models": "tsx src/cli/sync-models.ts"
-  },
   "dependencies": {
     "@ai-sdk/google": "^3.0.1",
     "@openrouter/ai-sdk-provider": "^2.1.1",
@@ -118,5 +112,11 @@
   },
   "engines": {
     "node": ">=22.11.0"
+  },
+  "scripts": {
+    "typecheck": "tsc --noEmit",
+    "build": "tsup",
+    "test": "vitest run",
+    "sync-models": "tsx src/cli/sync-models.ts"
   }
-}
+}

package/skills/workflow-engine/SKILL.md

@@ -337,6 +337,8 @@ const ai = createAIHelper(
 const { text, cost } = await ai.generateText("gemini-2.5-flash", prompt);
 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"]);
 ```
 
 ## Persistence Setup

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

@@ -248,16 +248,16 @@ const result = await ai.generateObject(
 
 ## embed
 
-Generate embeddings for text.
+Generate embeddings for text. Supports Google and OpenRouter embedding providers.
 
 ```typescript
-// Single text
+// Google embedding model (with Google-specific options)
 const result = await ai.embed(
   "text-embedding-004",
   "The quick brown fox",
   {
     dimensions: 768,  // Output dimensions (default: 768)
-    taskType: "RETRIEVAL_DOCUMENT",  // or "RETRIEVAL_QUERY", "SEMANTIC_SIMILARITY"
+    taskType: "RETRIEVAL_DOCUMENT",  // Google-only: "RETRIEVAL_QUERY", "SEMANTIC_SIMILARITY"
   }
 );
 
@@ -266,6 +266,12 @@ console.log(result.dimensions);    // 768
 console.log(result.inputTokens);
 console.log(result.cost);
 
+// OpenRouter embedding model (OpenAI, Cohere, etc.)
+const result = await ai.embed(
+  "openai/text-embedding-3-small",
+  "The quick brown fox",
+);
+
 // Multiple texts (batch)
 const result = await ai.embed("text-embedding-004", [
   "First document",
@@ -277,6 +283,10 @@ console.log(result.embeddings);    // number[][] (3 embeddings)
 console.log(result.embedding);     // First embedding (convenience)
 ```
 
+> **Note:** `taskType` and `outputDimensionality` options only apply to Google embedding models.
+> OpenRouter embedding models work without provider-specific options. The provider is determined
+> by the `provider` field in the model's `ModelConfig`.
+
 ## streamText
 
 Stream text generation.
@@ -465,6 +475,26 @@ registerModels({
 const result = await ai.generateText("my-custom-model", prompt);
 ```
 
+### Register Custom Embedding Models
+
+```typescript
+import { registerModels } from "@bratsos/workflow-engine";
+
+registerModels({
+  "openai/text-embedding-3-small": {
+    id: "openai/text-embedding-3-small",
+    name: "OpenAI text-embedding-3-small",
+    provider: "openrouter",
+    inputCostPerMillion: 0.02,
+    outputCostPerMillion: 0,
+    isEmbeddingModel: true,
+  },
+});
+
+// Now usable
+const { embedding } = await ai.embed("openai/text-embedding-3-small", "text");
+```
+
 ### Cost Calculation
 
 ```typescript