From 908a08fd9913b0f17dd416fb7784c8e8e526da7c Mon Sep 17 00:00:00 2001 From: Danny Avila Date: Wed, 11 Mar 2026 22:32:12 -0400 Subject: [PATCH] refactor(agents): streamline bulk write logic in recordCollectedUsage function Removed redundant bulk write operations and consolidated document handling in the recordCollectedUsage function. The logic now combines all documents into a single bulk write operation, improving efficiency and reducing error handling complexity. Updated logging to provide consistent error messages for bulk write failures. --- packages/api/src/agents/usage.ts | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/packages/api/src/agents/usage.ts b/packages/api/src/agents/usage.ts index bdb701a0cb..c34eb1570a 100644 --- a/packages/api/src/agents/usage.ts +++ b/packages/api/src/agents/usage.ts @@ -182,14 +182,6 @@ export async function recordCollectedUsage( }); } - if (useBulk && allDocs.length > 0) { - try { - await bulkWriteTransactions({ user, docs: allDocs }, bulkWriteOps); - } catch (err) { - logger.error('[packages/api #recordCollectedUsage] Error in bulk write', err); - } - } - const summarizationDocs: PreparedEntry[] = []; for (const usage of summarizationUsages) { @@ -273,11 +265,12 @@ export async function recordCollectedUsage( }); } - if (useBulk && summarizationDocs.length > 0) { + const combinedDocs = [...allDocs, ...summarizationDocs]; + if (useBulk && combinedDocs.length > 0) { try { - await bulkWriteTransactions({ user, docs: summarizationDocs }, bulkWriteOps); + await bulkWriteTransactions({ user, docs: combinedDocs }, bulkWriteOps); } catch (err) { - logger.error('[packages/api #recordCollectedUsage] Error in summarization bulk write', err); + logger.error('[packages/api #recordCollectedUsage] Error in bulk write', err); } }