CERBERUS/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2026-07-11 08:43:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor(agents): streamline bulk write logic in recordCollectedUsage function

Browse source 
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.
This commit is contained in:
Danny Avila 2026-03-11 22:32:12 -04:00
parent 4ca86fae3d
commit 908a08fd99
No known key found for this signature in database
GPG key ID: BF31EEB2C5CA0956
1 changed files with 4 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

15
packages/api/src/agents/usage.ts
View file

@ -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);
}
}