@resolveio/server-lib 22.1.28 → 22.1.29

package/methods/ai-terminal.js

@@ -1614,6 +1614,15 @@ function executeAiAssistantCodexRun(payload, context) {
                     user = _d.sent();
                     isSuperAdmin = !!((_a = user === null || user === void 0 ? void 0 : user.roles) === null || _a === void 0 ? void 0 : _a.super_admin);
                     canViewDebug = isAssistantDebugUser(user);
+                    if (shouldLogAssistantFinalizeDebug()) {
+                        console.log(new Date(), '[AI-Dashboard][AI-Assistant] debug-visibility', {
+                            requestId: requestId || null,
+                            id_user: (context === null || context === void 0 ? void 0 : context.id_user) || null,
+                            username: normalizeOptionalString(user === null || user === void 0 ? void 0 : user.username) || null,
+                            isSuperAdmin: isSuperAdmin,
+                            canViewDebug: canViewDebug
+                        });
+                    }
                     hasInvoiceAccess = userHasInvoiceAccess(user);
                     customerId = normalizeOptionalString((_b = user === null || user === void 0 ? void 0 : user.other) === null || _b === void 0 ? void 0 : _b.id_customer);
                     return [4 /*yield*/, ensureConversation(input, 'codex', context === null || context === void 0 ? void 0 : context.id_user)];
@@ -20485,54 +20494,110 @@ function resolveAssistantDatabaseNameForSystemCollections() {
 }
 function updateAssistantMessageWithFallback(messageId, setPayload) {
     return __awaiter(this, void 0, void 0, function () {
-        var error_12, dbName, _a;
-        return __generator(this, function (_b) {
-            switch (_b.label) {
+        var safeSetPayload, fallbackId, debugEnabled, result, error_12, dbName, messagesCollection, result, _a, retryResult;
+        var _b, _c, _d, _e, _f, _g;
+        return __generator(this, function (_h) {
+            switch (_h.label) {
                 case 0:
                     if (!messageId || !setPayload || typeof setPayload !== 'object') {
                         return [2 /*return*/];
                     }
-                    _b.label = 1;
+                    safeSetPayload = sanitizeAssistantMessageSetPayload(setPayload);
+                    fallbackId = normalizeOptionalString(typeof messageId === 'string' ? messageId : String(messageId));
+                    debugEnabled = shouldLogAssistantFinalizeDebug();
+                    _h.label = 1;
                 case 1:
-                    _b.trys.push([1, 3, , 10]);
-                    return [4 /*yield*/, ai_terminal_message_collection_1.AiTerminalMessages.updateOne({ _id: messageId }, { $set: setPayload }, undefined, false, false, true)];
+                    _h.trys.push([1, 5, , 17]);
+                    return [4 /*yield*/, ai_terminal_message_collection_1.AiTerminalMessages.updateOne({ _id: messageId }, { $set: safeSetPayload }, undefined, false, false, true)];
                 case 2:
-                    _b.sent();
-                    return [2 /*return*/];
+                    result = _h.sent();
+                    if (!((result === null || result === void 0 ? void 0 : result.matchedCount) === 0 && fallbackId && fallbackId !== messageId)) return [3 /*break*/, 4];
+                    return [4 /*yield*/, ai_terminal_message_collection_1.AiTerminalMessages.updateOne({ _id: fallbackId }, { $set: safeSetPayload }, undefined, false, false, true)];
                 case 3:
-                    error_12 = _b.sent();
+                    result = _h.sent();
+                    _h.label = 4;
+                case 4:
+                    if (debugEnabled) {
+                        console.log(new Date(), '[AI-Dashboard][AI-Assistant] message finalize update', {
+                            messageId: fallbackId || messageId,
+                            matchedCount: (_b = result === null || result === void 0 ? void 0 : result.matchedCount) !== null && _b !== void 0 ? _b : 0,
+                            modifiedCount: (_c = result === null || result === void 0 ? void 0 : result.modifiedCount) !== null && _c !== void 0 ? _c : 0,
+                            via: 'collection_manager'
+                        });
+                    }
+                    return [2 /*return*/];
+                case 5:
+                    error_12 = _h.sent();
                     if (!isAssistantTransactionCommittedError(error_12)) {
                         throw error_12;
                     }
+                    if (debugEnabled) {
+                        console.log(new Date(), '[AI-Dashboard][AI-Assistant] message finalize update fallback triggered', {
+                            messageId: fallbackId || messageId,
+                            reason: normalizeOptionalString(error_12 === null || error_12 === void 0 ? void 0 : error_12.message) || 'transaction_committed'
+                        });
+                    }
                     dbName = resolveAssistantDatabaseNameForSystemCollections();
-                    _b.label = 4;
-                case 4:
-                    _b.trys.push([4, 6, , 7]);
-                    return [4 /*yield*/, resolveio_server_app_1.ResolveIOServer.getMongoConnection()
-                            .db(dbName)
-                            .collection('ai-terminal-messages')
-                            .updateOne({ _id: messageId }, { $set: setPayload })];
-                case 5:
-                    _b.sent();
-                    return [2 /*return*/];
+                    _h.label = 6;
                 case 6:
-                    _a = _b.sent();
-                    return [3 /*break*/, 7];
-                case 7: return [4 /*yield*/, (0, promises_1.setTimeout)(15)];
+                    _h.trys.push([6, 11, , 12]);
+                    messagesCollection = resolveio_server_app_1.ResolveIOServer.getMongoConnection()
+                        .db(dbName)
+                        .collection('ai-terminal-messages');
+                    return [4 /*yield*/, messagesCollection
+                            .updateOne({ _id: String(messageId) }, { $set: safeSetPayload })];
+                case 7:
+                    result = _h.sent();
+                    if (!((result === null || result === void 0 ? void 0 : result.matchedCount) === 0 && fallbackId && fallbackId !== messageId)) return [3 /*break*/, 9];
+                    return [4 /*yield*/, messagesCollection
+                            .updateOne({ _id: fallbackId }, { $set: safeSetPayload })];
                 case 8:
-                    _b.sent();
-                    return [4 /*yield*/, ai_terminal_message_collection_1.AiTerminalMessages.updateOne({ _id: messageId }, { $set: setPayload }, undefined, false, false, true)];
-                case 9:
-                    _b.sent();
-                    return [3 /*break*/, 10];
-                case 10: return [2 /*return*/];
+                    result = _h.sent();
+                    _h.label = 9;
+                case 9: return [4 /*yield*/, notifyAiTerminalCollectionUpdate('ai-terminal-messages', fallbackId || messageId)];
+                case 10:
+                    _h.sent();
+                    if (debugEnabled) {
+                        console.log(new Date(), '[AI-Dashboard][AI-Assistant] message finalize update', {
+                            messageId: fallbackId || messageId,
+                            matchedCount: (_d = result === null || result === void 0 ? void 0 : result.matchedCount) !== null && _d !== void 0 ? _d : 0,
+                            modifiedCount: (_e = result === null || result === void 0 ? void 0 : result.modifiedCount) !== null && _e !== void 0 ? _e : 0,
+                            via: 'direct_mongo_fallback'
+                        });
+                    }
+                    return [2 /*return*/];
+                case 11:
+                    _a = _h.sent();
+                    return [3 /*break*/, 12];
+                case 12: return [4 /*yield*/, (0, promises_1.setTimeout)(15)];
+                case 13:
+                    _h.sent();
+                    return [4 /*yield*/, ai_terminal_message_collection_1.AiTerminalMessages.updateOne({ _id: messageId }, { $set: safeSetPayload }, undefined, false, false, true)];
+                case 14:
+                    retryResult = _h.sent();
+                    if (!((retryResult === null || retryResult === void 0 ? void 0 : retryResult.matchedCount) === 0 && fallbackId && fallbackId !== messageId)) return [3 /*break*/, 16];
+                    return [4 /*yield*/, ai_terminal_message_collection_1.AiTerminalMessages.updateOne({ _id: fallbackId }, { $set: safeSetPayload }, undefined, false, false, true)];
+                case 15:
+                    retryResult = _h.sent();
+                    _h.label = 16;
+                case 16:
+                    if (debugEnabled) {
+                        console.log(new Date(), '[AI-Dashboard][AI-Assistant] message finalize update', {
+                            messageId: fallbackId || messageId,
+                            matchedCount: (_f = retryResult === null || retryResult === void 0 ? void 0 : retryResult.matchedCount) !== null && _f !== void 0 ? _f : 0,
+                            modifiedCount: (_g = retryResult === null || retryResult === void 0 ? void 0 : retryResult.modifiedCount) !== null && _g !== void 0 ? _g : 0,
+                            via: 'collection_manager_retry'
+                        });
+                    }
+                    return [3 /*break*/, 17];
+                case 17: return [2 /*return*/];
             }
         });
     });
 }
 function touchConversation(idConversation, timestamp, lastMessageId) {
     return __awaiter(this, void 0, void 0, function () {
-        var update, error_13, dbName;
+        var update, error_13, dbName, conversationsCollection;
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -20545,7 +20610,7 @@ function touchConversation(idConversation, timestamp, lastMessageId) {
                     }
                     _a.label = 1;
                 case 1:
-                    _a.trys.push([1, 3, , 5]);
+                    _a.trys.push([1, 3, , 6]);
                     return [4 /*yield*/, ai_terminal_conversation_collection_1.AiTerminalConversations.updateOne({ _id: idConversation }, { $set: update }, undefined, false, false, true)];
                 case 2:
                     _a.sent();
@@ -20556,18 +20621,106 @@ function touchConversation(idConversation, timestamp, lastMessageId) {
                         throw error_13;
                     }
                     dbName = resolveAssistantDatabaseNameForSystemCollections();
-                    return [4 /*yield*/, resolveio_server_app_1.ResolveIOServer.getMongoConnection()
-                            .db(dbName)
-                            .collection('ai-terminal-conversations')
-                            .updateOne({ _id: idConversation }, { $set: update })];
+                    conversationsCollection = resolveio_server_app_1.ResolveIOServer.getMongoConnection()
+                        .db(dbName)
+                        .collection('ai-terminal-conversations');
+                    return [4 /*yield*/, conversationsCollection.updateOne({ _id: idConversation }, { $set: update })];
                 case 4:
                     _a.sent();
+                    return [4 /*yield*/, notifyAiTerminalCollectionUpdate('ai-terminal-conversations', idConversation)];
+                case 5:
+                    _a.sent();
+                    return [3 /*break*/, 6];
+                case 6: return [2 /*return*/];
+            }
+        });
+    });
+}
+function shouldLogAssistantFinalizeDebug() {
+    return parseDebugFlag(process.env.AI_DASHBOARD_DEBUG)
+        || parseDebugFlag(process.env.AI_ASSISTANT_WORKER_DEBUG);
+}
+function notifyAiTerminalCollectionUpdate(collection, documentId) {
+    return __awaiter(this, void 0, void 0, function () {
+        var subManager, _a;
+        var _b, _c;
+        return __generator(this, function (_d) {
+            switch (_d.label) {
+                case 0:
+                    try {
+                        resolveio_server_app_1.ResolveIOServer.getMongoManager().invalidateQueryCache(collection, true);
+                    }
+                    catch (_e) { }
+                    _d.label = 1;
+                case 1:
+                    _d.trys.push([1, 4, , 5]);
+                    subManager = (_c = (_b = resolveio_server_app_1.ResolveIOServer.getMainServer()) === null || _b === void 0 ? void 0 : _b.getSubscriptionManager) === null || _c === void 0 ? void 0 : _c.call(_b);
+                    if (!(subManager && typeof subManager.invalidatePubsCache === 'function')) return [3 /*break*/, 3];
+                    return [4 /*yield*/, subManager.invalidatePubsCache(collection, 'update', documentId)];
+                case 2:
+                    _d.sent();
+                    _d.label = 3;
+                case 3: return [3 /*break*/, 5];
+                case 4:
+                    _a = _d.sent();
                     return [3 /*break*/, 5];
                 case 5: return [2 /*return*/];
             }
         });
     });
 }
+function sanitizeAssistantMessageSetPayload(setPayload) {
+    var next = __assign({}, setPayload);
+    if (next.metadata && typeof next.metadata === 'object') {
+        next.metadata = sanitizeMongoSafeObject(next.metadata);
+    }
+    return next;
+}
+function sanitizeMongoSafeObject(value) {
+    var e_57, _a;
+    if (value === null || value === undefined) {
+        return value;
+    }
+    if (Array.isArray(value)) {
+        return value.map(function (entry) { return sanitizeMongoSafeObject(entry); });
+    }
+    if (value instanceof Date) {
+        return value;
+    }
+    if (isMongoObjectId(value)) {
+        return value;
+    }
+    if (!isPlainObject(value)) {
+        return value;
+    }
+    var out = {};
+    try {
+        for (var _b = __values(Object.entries(value)), _c = _b.next(); !_c.done; _c = _b.next()) {
+            var _d = __read(_c.value, 2), rawKey = _d[0], rawValue = _d[1];
+            var key = sanitizeMongoKey(rawKey);
+            out[key] = sanitizeMongoSafeObject(rawValue);
+        }
+    }
+    catch (e_57_1) { e_57 = { error: e_57_1 }; }
+    finally {
+        try {
+            if (_c && !_c.done && (_a = _b.return)) _a.call(_b);
+        }
+        finally { if (e_57) throw e_57.error; }
+    }
+    return out;
+}
+function sanitizeMongoKey(rawKey) {
+    var key = String(rawKey || '');
+    if (!key) {
+        return '_';
+    }
+    var sanitized = key.replace(/\./g, '__dot__');
+    if (sanitized.startsWith('$')) {
+        sanitized = "_dollar_".concat(sanitized.slice(1));
+    }
+    return sanitized || '_';
+}
 function normalizeOptionalString(value) {
     var raw = typeof value === 'string' ? value.trim() : '';
     return raw || '';
@@ -20654,7 +20807,7 @@ function estimateUsage(messages, responseText, model) {
     };
 }
 function evaluateGuardrails(message) {
-    var e_57, _a;
+    var e_58, _a;
     var normalized = String(message || '').toLowerCase();
     var identityGuardrail = evaluateAssistantIdentityDisclosureGuardrail(normalized);
     if (identityGuardrail === null || identityGuardrail === void 0 ? void 0 : identityGuardrail.blocked) {
@@ -20680,12 +20833,12 @@ function evaluateGuardrails(message) {
             }
         }
     }
-    catch (e_57_1) { e_57 = { error: e_57_1 }; }
+    catch (e_58_1) { e_58 = { error: e_58_1 }; }
     finally {
         try {
             if (patterns_3_1 && !patterns_3_1.done && (_a = patterns_3.return)) _a.call(patterns_3);
         }
-        finally { if (e_57) throw e_57.error; }
+        finally { if (e_58) throw e_58.error; }
     }
     return null;
 }