diff --git a/src/store/execution/generateAudioExecutor.ts b/src/store/execution/generateAudioExecutor.ts index 47745f5b..8b76024c 100644 --- a/src/store/execution/generateAudioExecutor.ts +++ b/src/store/execution/generateAudioExecutor.ts @@ -141,7 +141,7 @@ export async function executeGenerateAudio( }); // Track cost - if (nodeData.selectedModel?.pricing) { + if (nodeData.selectedModel?.provider === "fal" && nodeData.selectedModel?.pricing) { addIncurredCost(nodeData.selectedModel.pricing.amount); } diff --git a/src/store/execution/generateVideoExecutor.ts b/src/store/execution/generateVideoExecutor.ts index 0c1952b2..4bb47222 100644 --- a/src/store/execution/generateVideoExecutor.ts +++ b/src/store/execution/generateVideoExecutor.ts @@ -148,7 +148,7 @@ export async function executeGenerateVideo( }); // Track cost - if (nodeData.selectedModel?.pricing) { + if (nodeData.selectedModel?.provider === "fal" && nodeData.selectedModel?.pricing) { addIncurredCost(nodeData.selectedModel.pricing.amount); } diff --git a/src/utils/costCalculator.ts b/src/utils/costCalculator.ts index 320faf21..019d381e 100644 --- a/src/utils/costCalculator.ts +++ b/src/utils/costCalculator.ts @@ -258,6 +258,9 @@ export function hasNonGeminiProviders(nodes: WorkflowNode[]): boolean { const data = node.data as Generate3DNodeData; return data.selectedModel?.provider !== undefined && data.selectedModel.provider !== "gemini"; } + if (node.type === "generateAudio") { + return true; // Audio nodes are always non-Gemini + } return false; }); }