diff --git a/src/store/execution/generateAudioExecutor.ts b/src/store/execution/generateAudioExecutor.ts index 8b76024c..47745f5b 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?.provider === "fal" && nodeData.selectedModel?.pricing) { + if (nodeData.selectedModel?.pricing) { addIncurredCost(nodeData.selectedModel.pricing.amount); } diff --git a/src/store/execution/generateVideoExecutor.ts b/src/store/execution/generateVideoExecutor.ts index 4bb47222..0c1952b2 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?.provider === "fal" && nodeData.selectedModel?.pricing) { + if (nodeData.selectedModel?.pricing) { addIncurredCost(nodeData.selectedModel.pricing.amount); }