Update extraction-service.ts

mendableai · Jan 19, 2025 · 64607f3 · 64607f3
1 parent b8a30a5
commit 64607f3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/apps/api/src/lib/extract/extraction-service.ts b/apps/api/src/lib/extract/extraction-service.ts
@@ -675,12 +675,11 @@ export async function performExtraction(
 
   const totalTokensUsed = tokenUsage.reduce((a, b) => a + b.totalTokens, 0);
   const llmUsage = estimateTotalCost(tokenUsage);
-  const tokensToBill = calculateFinalResultCost(finalResult);
+  let tokensToBill = calculateFinalResultCost(finalResult);
 
-  let linksBilled = links.length * 5;
 
   if (CUSTOM_U_TEAMS.includes(teamId)) {
-    linksBilled = 1;
+    tokensToBill = 1;
   }
   // Bill team for usage
   billTeam(teamId, subId, tokensToBill, logger, true).catch((error) => {