@bratsos/workflow-engine 0.5.1 → 0.7.0

package/dist/{chunk-ZYMT2PAO.js → chunk-WWK2SPN7.js}

@@ -278,34 +278,6 @@ var ModelStatsTracker = class _ModelStatsTracker {
     tracker.modelConfig = void 0;
     return tracker;
   }
-  /**
-   * Get the model ID for use with AI SDK
-   * @deprecated Use getModelById(modelKey).id instead for parallel execution
-   */
-  getModelId() {
-    if (!this.modelConfig) {
-      throw new Error("Model not set for this tracker");
-    }
-    return this.modelConfig.id;
-  }
-  /**
-   * Get the model configuration
-   * @deprecated Use getModelById(modelKey) instead for parallel execution
-   */
-  getModelConfig() {
-    if (!this.modelConfig) {
-      throw new Error("Model not set for this tracker");
-    }
-    return this.modelConfig;
-  }
-  /**
-   * Switch model (useful for sequential model switching)
-   * @deprecated For parallel execution, pass model key to recordCall() instead
-   */
-  switchModel(modelKey) {
-    this.modelKey = modelKey;
-    this.modelConfig = getModel(modelKey);
-  }
   /**
    * Get a model helper with bound recordCall for parallel execution
    * Perfect for running multiple AI calls in parallel with different models
@@ -1405,7 +1377,8 @@ var AIHelperImpl = class _AIHelperImpl {
   aiCallLogger;
   logContext;
   batchLogFn;
-  constructor(topic, aiCallLogger, logContext) {
+  providerResolver;
+  constructor(topic, aiCallLogger, logContext, providerResolver) {
     if (!aiCallLogger) {
       throw new Error(
         "AIHelperImpl requires a logger. Create one using createPrismaAICallLogger(prisma)."
@@ -1414,6 +1387,7 @@ var AIHelperImpl = class _AIHelperImpl {
     this.topic = topic;
     this.aiCallLogger = aiCallLogger;
     this.logContext = logContext;
+    this.providerResolver = providerResolver;
     if (logContext) {
       this.batchLogFn = (level, message, meta) => {
         logContext.createLog({
@@ -1429,7 +1403,7 @@ var AIHelperImpl = class _AIHelperImpl {
   }
   async generateText(modelKey, prompt, options = {}) {
     const modelConfig = getModel(modelKey);
-    const model = getModelProvider(modelConfig);
+    const model = this.providerResolver?.(modelConfig) ?? getModelProvider(modelConfig);
     const startTime = Date.now();
     const isMultimodal = Array.isArray(prompt);
     const hasTools = options.tools !== void 0;
@@ -1606,7 +1580,7 @@ var AIHelperImpl = class _AIHelperImpl {
   }
   async generateObject(modelKey, prompt, schema, options = {}) {
     const modelConfig = getModel(modelKey);
-    const model = getModelProvider(modelConfig);
+    const model = this.providerResolver?.(modelConfig) ?? getModelProvider(modelConfig);
     const startTime = Date.now();
     const isMultimodal = Array.isArray(prompt);
     const hasTools = options.tools !== void 0;
@@ -1840,7 +1814,7 @@ var AIHelperImpl = class _AIHelperImpl {
   }
   streamText(modelKey, input, options = {}) {
     const modelConfig = getModel(modelKey);
-    const model = getModelProvider(modelConfig);
+    const model = this.providerResolver?.(modelConfig) ?? getModelProvider(modelConfig);
     const startTime = Date.now();
     const hasTools = options.tools !== void 0;
     const promptForLog = "prompt" in input && input.prompt ? input.prompt : JSON.stringify(input.messages);
@@ -1994,7 +1968,12 @@ var AIHelperImpl = class _AIHelperImpl {
   }
   createChild(segment, id) {
     const newTopic = id ? `${this.topic}.${segment}.${id}` : `${this.topic}.${segment}`;
-    return new _AIHelperImpl(newTopic, this.aiCallLogger, this.logContext);
+    return new _AIHelperImpl(
+      newTopic,
+      this.aiCallLogger,
+      this.logContext,
+      this.providerResolver
+    );
   }
   /** @internal Get the logger for batch operations */
   getLogger() {
@@ -2308,10 +2287,10 @@ var AIBatchImpl = class {
     );
   }
 };
-function createAIHelper(topic, logger2, logContext) {
-  return new AIHelperImpl(topic, logger2, logContext);
+function createAIHelper(topic, logger2, logContext, providerResolver) {
+  return new AIHelperImpl(topic, logger2, logContext, providerResolver);
 }
 
 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 };
-//# sourceMappingURL=chunk-ZYMT2PAO.js.map
-//# sourceMappingURL=chunk-ZYMT2PAO.js.map
+//# sourceMappingURL=chunk-WWK2SPN7.js.map
+//# sourceMappingURL=chunk-WWK2SPN7.js.map