📮 fix: Count Undelivered Background Results as Outstanding

api/server/services/Endpoints/agents/backgroundCompletion.js

@@ -1,9 +1,14 @@
 const {
   createBackgroundToolCompletionWakeupHandler,
   createBackgroundToolDeadClaimRecovery,
+  createPendingBackgroundCompletions,
   createBackgroundToolResultHandler,
   claimBackgroundToolResult: claimResult,
 } = require('@librechat/api');
+const {
+  listPendingAgentBackgroundToolCompletions,
+  listUndeliveredAgentTriggerTaskIds,
+} = require('~/models');
 const {
   enqueueAgentTrigger,
   persistAgentBackgroundToolResult,
@@ -23,6 +28,12 @@ const preregisterBackgroundToolCompletion = createBackgroundToolCompletionWakeup
   (deliveryKey) => expediteCompletionWakeups({ deliveryKeys: [deliveryKey] }),
 );
 
+const pendingBackgroundToolCompletions = createPendingBackgroundCompletions({
+  list: listPendingAgentBackgroundToolCompletions,
+  listTaskIds: listUndeliveredAgentTriggerTaskIds,
+  retire: retireAgentTrigger,
+});
+
 function createBackgroundToolResultPersistence({ req, updateToolCallResult }) {
   return createBackgroundToolResultHandler({ req, updateToolCallResult });
 }
@@ -46,6 +57,7 @@ function createDeadBackgroundToolClaimRecovery(
 
 module.exports = {
   preregisterBackgroundToolCompletion,
+  pendingBackgroundToolCompletions,
   createBackgroundToolResultPersistence,
   claimBackgroundToolResult,
   createDeadBackgroundToolClaimRecovery,

api/server/services/Endpoints/agents/initialize.js

@@ -94,6 +94,7 @@ const { processAddedConvo } = require('./addedConvo');
 const subagentThreadTaskStore = require('./subagentThreadStore');
 const {
   preregisterBackgroundToolCompletion,
+  pendingBackgroundToolCompletions,
   createBackgroundToolResultPersistence,
   claimBackgroundToolResult,
   createDeadBackgroundToolClaimRecovery,
@@ -515,6 +516,8 @@ const initializeClientWithProvider = async ({
     }),
     backgroundToolCompletion: {
       ...(completionWakeupsEnabled ? { preregister: preregisterBackgroundToolCompletion } : {}),
+      /** Deliveries admitted before wake-ups were disabled still drain and still count. */
+      pending: pendingBackgroundToolCompletions,
       persist: createBackgroundToolResultPersistence({
         req,
         updateToolCallResult: db.updateToolCallResult,