@resolveio/server-lib 22.1.31 → 22.2.0

package/methods/ai-terminal.js

@@ -162,7 +162,6 @@ var ai_terminal_message_collection_1 = require("../collections/ai-terminal-messa
 var ai_terminal_issue_report_collection_1 = require("../collections/ai-terminal-issue-report.collection");
 var common_1 = require("../util/common");
 var tokenizer_1 = require("../util/tokenizer");
-var openai_client_1 = require("../services/openai-client");
 var codex_client_1 = require("../services/codex-client");
 var openai_usage_ledger_manager_1 = require("../managers/openai-usage-ledger.manager");
 var DEFAULT_MAX_FILE_MB = 50;
@@ -172,6 +171,8 @@ var DEFAULT_MAX_TOTAL_ATTACHMENT_CHARS = 40000;
 var DEFAULT_CODEX_MODEL = 'gpt-5.3-codex';
 var DEFAULT_CODEX_FALLBACK_MODEL = 'gpt-5.2-codex';
 var DEFAULT_CODEX_TIMEOUT_MS = 180000;
+var DEFAULT_AI_ASSISTANT_RUN_BUDGET_MS = 60000;
+var DEFAULT_AI_ASSISTANT_MONGO_MAX_TIME_MS = 8000;
 var AI_ASSISTANT_MONGO_DEFAULT_LIMIT = 20;
 var AI_ASSISTANT_MONGO_MAX_LIMIT = 200;
 var AI_ASSISTANT_MONGO_EXPORT_MAX_LIMIT = 5000;
@@ -199,6 +200,8 @@ var AI_ASSISTANT_CURRENCY_CODE = 'USD';
 var AI_ASSISTANT_USD_CURRENCY_TEXT_PATTERN = /\b(?:USD|US\$)(?:\s| | |[\u00A0\u202F\u2007])*\$?\s*([-+]?[0-9][0-9,]*(?:\.[0-9]+)?)/gi;
 var AI_ASSISTANT_REVENUE_VERIFY_DIFF_WARN_THRESHOLD = 0.15;
 var AI_ASSISTANT_REVENUE_VERIFY_PAID_GRAND_GAP_WARN_THRESHOLD = 0.10;
+var AI_ASSISTANT_CODEX_MIN_STAGE_BUDGET_MS = 1500;
+var AI_ASSISTANT_MONGO_MIN_STAGE_BUDGET_MS = 80;
 var AI_ASSISTANT_DISPLAY_PRIORITY_FIELDS = [
     'name',
     'title',
@@ -552,7 +555,7 @@ var AI_ASSISTANT_SYSTEM_PROMPT = [
     '- Do not modify files, run destructive commands, or access databases directly.',
     '- Read-only data access is allowed only via the REPORT_BUILDER_READ/REPORT_BUILDER_AGG directives (see below).',
     '- Do not access secrets, credentials, or user data.',
-    '- If asked about the underlying AI/model/provider/runtime (including Codex/OpenAI/local-vs-hosted/version details), refuse. This information is not available to users, including admins and super admins.',
+    '- If asked about the underlying AI/model/provider/runtime (including local-vs-hosted/version details), refuse. This information is not available to users, including admins and super admins.',
     '- If the user has a customer portal scope (other.id_customer), only discuss that customer\'s data and what is visible in their customer portal. Never reference other customers or internal/admin-only data. If asked for anything outside the portal, say it isn\'t available.',
     '- Do not assist with hacking, bypassing security, or abuse.',
     'Accuracy & tools:',
@@ -1306,7 +1309,7 @@ function loadAiTerminalMethods(methodManager) {
 }
 function executeAiTerminalRun(payload, context) {
     return __awaiter(this, void 0, void 0, function () {
-        var input, message, requestId, identityGuardrail, conversation_1, now_1, userMsg, assistantMsg, isSuperAdmin, guardrailsEnabled, guardrail, conversation_2, now_2, userMsg, assistantMsg, conversation, now, attachments, attachmentData, config, systemPrompt, userPromptTemplate, userPrompt, historyLimit, history, _a, messages, openaiSettings, client, response, usage, idClient, userDoc, assistantDoc, insertResult;
+        var input, message, requestId, identityGuardrail, conversation_1, now_1, userMsg, assistantMsg, isSuperAdmin, guardrailsEnabled, guardrail, conversation_2, now_2, userMsg, assistantMsg, conversation, now, attachments, attachmentData, config, systemPrompt, userPromptTemplate, userPrompt, historyLimit, history, _a, messages, codexModel, codexFallbackModels, codexSettings, client, responseText, usage, idClient, userDoc, assistantDoc, insertResult;
         return __generator(this, function (_b) {
             switch (_b.label) {
                 case 0:
@@ -1321,7 +1324,7 @@ function executeAiTerminalRun(payload, context) {
                     requestId = normalizeOptionalString(input.request_id);
                     identityGuardrail = evaluateAssistantIdentityDisclosureGuardrail(message);
                     if (!(identityGuardrail === null || identityGuardrail === void 0 ? void 0 : identityGuardrail.blocked)) return [3 /*break*/, 5];
-                    return [4 /*yield*/, ensureConversation(input, 'openai', context === null || context === void 0 ? void 0 : context.id_user)];
+                    return [4 /*yield*/, ensureConversation(input, 'codex', context === null || context === void 0 ? void 0 : context.id_user)];
                 case 1:
                     conversation_1 = _b.sent();
                     now_1 = new Date();
@@ -1362,7 +1365,7 @@ function executeAiTerminalRun(payload, context) {
                     if (!guardrailsEnabled) return [3 /*break*/, 11];
                     guardrail = evaluateGuardrails(message);
                     if (!(guardrail === null || guardrail === void 0 ? void 0 : guardrail.blocked)) return [3 /*break*/, 11];
-                    return [4 /*yield*/, ensureConversation(input, 'openai', context === null || context === void 0 ? void 0 : context.id_user)];
+                    return [4 /*yield*/, ensureConversation(input, 'codex', context === null || context === void 0 ? void 0 : context.id_user)];
                 case 7:
                     conversation_2 = _b.sent();
                     now_2 = new Date();
@@ -1396,7 +1399,7 @@ function executeAiTerminalRun(payload, context) {
                             message: assistantMsg,
                             guardrails: { blocked: true, reason: guardrail.reason }
                         }];
-                case 11: return [4 /*yield*/, ensureConversation(input, 'openai', context === null || context === void 0 ? void 0 : context.id_user)];
+                case 11: return [4 /*yield*/, ensureConversation(input, 'codex', context === null || context === void 0 ? void 0 : context.id_user)];
                 case 12:
                     conversation = _b.sent();
                     now = new Date();
@@ -1433,19 +1436,32 @@ function executeAiTerminalRun(payload, context) {
                     if (userPrompt) {
                         messages.push({ role: 'user', content: userPrompt });
                     }
-                    openaiSettings = resolveOpenAISettings(config);
-                    client = new openai_client_1.OpenAIClient(openaiSettings);
-                    return [4 /*yield*/, client.chat(messages, { timeoutMs: 60000, responseFormat: config.response_format })];
+                    codexModel = resolveCodexModel(config);
+                    codexFallbackModels = resolveCodexFallbackModels(config, codexModel);
+                    codexSettings = resolveCodexSettings({ model: codexModel, fallbackModels: codexFallbackModels });
+                    client = getAssistantCodexClient(codexSettings);
+                    return [4 /*yield*/, client.run(buildCodexChatPrompt(messages), {
+                            timeoutMs: 60000,
+                            threadOptions: {
+                                model: codexModel,
+                                sandboxMode: 'read-only',
+                                skipGitRepoCheck: true,
+                                networkAccessEnabled: false,
+                                webSearchMode: 'disabled',
+                                webSearchEnabled: false,
+                                approvalPolicy: 'never'
+                            }
+                        })];
                 case 17:
-                    response = _b.sent();
-                    usage = response.usage || estimateUsage(messages, response.content, openaiSettings.model);
+                    responseText = _b.sent();
+                    usage = estimateUsage(messages, responseText, codexModel);
                     return [4 /*yield*/, resolveClientId(conversation, input.id_client, context === null || context === void 0 ? void 0 : context.id_user)];
                 case 18:
                     idClient = _b.sent();
                     if (!idClient) return [3 /*break*/, 20];
                     return [4 /*yield*/, (0, openai_usage_ledger_manager_1.recordOpenAIUsage)({
                             id_client: idClient,
-                            model: response.model || openaiSettings.model || 'unknown',
+                            model: codexModel || 'unknown',
                             input_tokens: usage.inputTokens,
                             output_tokens: usage.outputTokens,
                             total_tokens: usage.totalTokens,
@@ -1468,10 +1484,10 @@ function executeAiTerminalRun(payload, context) {
                     assistantDoc = {
                         id_conversation: conversation._id,
                         role: 'assistant',
-                        content: response.content,
+                        content: responseText,
                         metadata: input.request_id ? { request_id: input.request_id } : undefined,
                         usage: {
-                            model: response.model || openaiSettings.model,
+                            model: codexModel,
                             input_tokens: usage.inputTokens,
                             output_tokens: usage.outputTokens,
                             total_tokens: usage.totalTokens
@@ -1504,7 +1520,7 @@ function executeAiTerminalRun(payload, context) {
 }
 function executeAiFormPatch(payload, context) {
     return __awaiter(this, void 0, void 0, function () {
-        var input, message, allowedFields, guardrail, systemPrompt, userPrompt, messages, openaiSettings, client, response, usage, idClient, parsed;
+        var input, message, allowedFields, guardrail, systemPrompt, userPrompt, messages, config, codexModel, codexFallbackModels, codexSettings, client, responseText, usage, idClient, parsed;
         var _a;
         return __generator(this, function (_b) {
             switch (_b.label) {
@@ -1531,19 +1547,33 @@ function executeAiFormPatch(payload, context) {
                         { role: 'system', content: systemPrompt },
                         { role: 'user', content: userPrompt }
                     ];
-                    openaiSettings = resolveOpenAISettings(input.config || {});
-                    client = new openai_client_1.OpenAIClient(openaiSettings);
-                    return [4 /*yield*/, client.chat(messages, { timeoutMs: 60000, responseFormat: 'json_object' })];
+                    config = sanitizeConfig(input.config || {});
+                    codexModel = resolveCodexModel(config);
+                    codexFallbackModels = resolveCodexFallbackModels(config, codexModel);
+                    codexSettings = resolveCodexSettings({ model: codexModel, fallbackModels: codexFallbackModels });
+                    client = getAssistantCodexClient(codexSettings);
+                    return [4 /*yield*/, client.run(buildCodexChatPrompt(messages), {
+                            timeoutMs: 60000,
+                            threadOptions: {
+                                model: codexModel,
+                                sandboxMode: 'read-only',
+                                skipGitRepoCheck: true,
+                                networkAccessEnabled: false,
+                                webSearchMode: 'disabled',
+                                webSearchEnabled: false,
+                                approvalPolicy: 'never'
+                            }
+                        })];
                 case 1:
-                    response = _b.sent();
-                    usage = response.usage || estimateUsage(messages, response.content, response.model || openaiSettings.model);
+                    responseText = _b.sent();
+                    usage = estimateUsage(messages, responseText, codexModel);
                     return [4 /*yield*/, resolveClientIdFromConfig(input.id_client, context === null || context === void 0 ? void 0 : context.id_user)];
                 case 2:
                     idClient = _b.sent();
                     if (!idClient) return [3 /*break*/, 4];
                     return [4 /*yield*/, (0, openai_usage_ledger_manager_1.recordOpenAIUsage)({
                             id_client: idClient,
-                            model: response.model || openaiSettings.model || 'unknown',
+                            model: codexModel || 'unknown',
                             input_tokens: usage.inputTokens,
                             output_tokens: usage.outputTokens,
                             total_tokens: usage.totalTokens,
@@ -1553,7 +1583,7 @@ function executeAiFormPatch(payload, context) {
                     _b.sent();
                     _b.label = 4;
                 case 4:
-                    parsed = parseJsonObject(response.content);
+                    parsed = parseJsonObject(responseText);
                     if (!parsed || typeof parsed !== 'object') {
                         throw new Error('AI form patch response was not valid JSON.');
                     }
@@ -1561,7 +1591,7 @@ function executeAiFormPatch(payload, context) {
                             patch: (_a = parsed.patch) !== null && _a !== void 0 ? _a : parsed,
                             notes: normalizeOptionalString(parsed.notes) || undefined,
                             usage: {
-                                model: response.model || openaiSettings.model,
+                                model: codexModel,
                                 input_tokens: usage.inputTokens,
                                 output_tokens: usage.outputTokens,
                                 total_tokens: usage.totalTokens
@@ -1781,18 +1811,23 @@ function executeAiAssistantCodexRun(payload, context) {
                     insertResult = _d.sent();
                     assistantMessageId = (insertResult === null || insertResult === void 0 ? void 0 : insertResult._id) || (insertResult === null || insertResult === void 0 ? void 0 : insertResult.insertedId);
                     enqueueAssistantCodexRun(function () { return __awaiter(_this, void 0, void 0, function () {
-                        var runStart, steps, recordStep, progressTracker, streamProgress, assistantContent, toolResult, assistantDebug, directiveSource, requestClassification, dataQuestion, lastDirective, heuristicDirectivePrecomputed, usedDeterministicHeuristicFastPath, requestedTimeGrain, requestedBreakdownDimensions, enforceDatedDirective, enforceGroupedDirective, datedDirectiveRetryUsed, datedDirectiveResolved, toolResponseDebug, toolError, termHints, collectionHints, fieldHints, methodHints, publicationHints, collectionTokenization, collectionRanking, collectionSelection, collectionOverride, collectionNames, plannerEnabled, plannerUsed, plannerSkipReason, plannerOutput, plannerRaw, timingBreakdown, codexUsage, accumulateCodexUsage, contextRoute, contextMode, hintSeed, termExpansion, hintText, baseTokens, expandedTokens, baseWeights, expandedWeights, dbName, db, surfaceHints, _a, allowedRoutes, rankedAllowedRoutes, routeHints, rankedCollectionHints, rankedCollections, hintCollections, assistantContext, hasDeterministicHeuristicFastPath, prompt_1, workspaceRoot, codexConfig, runOptions, plannerRunOptions, shouldRunPlanner, plannerPrompt, plannerStart, _b, preferListDirective, directiveStyleHint, directivePromptMode, responseText, directiveText, directive, heuristicDirectiveFastPath, directivePrompt, directiveStart, forcedDirective, _c, initialStart, extractedDirective, error_2, directivePrompt, forcedStart, forcedDirective, _d, strictDirectivePrompt, strictStart, strictDirectiveText, strictDirective, strictDirectiveIsDated, shouldUseStrictDirective, _e, guardDirectivePrompt, guardStart, guardDirectiveText, guardDirective, _f, groupedDirectivePrompt, groupedStart, groupedDirectiveText, groupedDirective, _g, heuristicDirective, requestedCollection, allowCollectionOverride, cleanedResponseText, deniedModuleByIntent, permissionLabel, effectiveDirective, toolRequest, toolStart, toolResponse, _h, toolPayload, skipFollowupCodex, followupPrompt, followupStart, followupText, _j, error_3, error_4, finishedAt, finalNow, finishedAt, codexMs, draftingMs, finalMetadata, finalUsage, usageClientId, usageError_1, finalAssistantDoc, setPayload, finalizeError_1, failedAt, fallbackContent, failureMetadata, persistError_1, _k;
+                        var runStart, fastModeEnabled, runBudgetMs, runDeadlineMs, steps, recordStep, progressTracker, streamProgress, remainingRunBudgetMs, hasRunBudget, applyCodexStageBudget, assistantContent, toolResult, assistantDebug, directiveSource, requestClassification, dataQuestion, lastDirective, heuristicDirectivePrecomputed, usedDeterministicHeuristicFastPath, requestedTimeGrain, requestedBreakdownDimensions, enforceDatedDirective, enforceGroupedDirective, datedDirectiveRetryUsed, datedDirectiveResolved, toolResponseDebug, toolError, termHints, collectionHints, fieldHints, methodHints, publicationHints, collectionTokenization, collectionRanking, collectionSelection, collectionOverride, collectionNames, plannerEnabled, plannerUsed, plannerSkipReason, plannerOutput, plannerRaw, timingBreakdown, codexUsage, accumulateCodexUsage, contextRoute, contextMode, hintSeed, termExpansion, hintText, baseTokens, expandedTokens, baseWeights, expandedWeights, dbName, db, surfaceHints, _a, allowedRoutes, rankedAllowedRoutes, routeHints, rankedCollectionHints, rankedCollections, hintCollections, assistantContext, hasDeterministicHeuristicFastPath, prompt_1, workspaceRoot, codexConfig, runOptions, plannerRunOptions, shouldRunPlanner, plannerBudgetAvailable, plannerPrompt, plannerStart, _b, preferListDirective, directiveStyleHint, directivePromptMode, responseText, directiveText, directive, heuristicDirectiveFastPath, directivePrompt, directiveStart, forcedDirective, _c, initialStart, extractedDirective, error_2, directivePrompt, forcedStart, forcedDirective, _d, strictDirectivePrompt, strictStart, strictDirectiveText, strictDirective, strictDirectiveIsDated, shouldUseStrictDirective, _e, guardDirectivePrompt, guardStart, guardDirectiveText, guardDirective, _f, groupedDirectivePrompt, groupedStart, groupedDirectiveText, groupedDirective, _g, heuristicDirective, requestedCollection, allowCollectionOverride, cleanedResponseText, deniedModuleByIntent, permissionLabel, effectiveDirective, toolRequest, toolStart, toolResponse, _h, toolPayload, zeroRowResult, skipFollowupCodex, followupPrompt, followupStart, followupText, _j, error_3, error_4, runBudgetExceeded, finishedAt, finalNow, finishedAt, codexMs, draftingMs, finalMetadata, finalUsage, usageClientId, usageError_1, finalAssistantDoc, setPayload, finalizeError_1, failedAt, fallbackContent, failureMetadata, persistError_1, _k;
                         var _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13;
                         return __generator(this, function (_14) {
                             switch (_14.label) {
                                 case 0:
                                     runStart = Date.now();
+                                    fastModeEnabled = resolveAssistantFastModeEnabled();
+                                    runBudgetMs = resolveAssistantRunBudgetMs();
+                                    runDeadlineMs = fastModeEnabled ? runStart + runBudgetMs : null;
                                     if (aiWorkerDebug) {
                                         console.log(new Date(), '[AI Worker Debug] codex run start', {
                                             requestId: requestId || null,
                                             id_user: (context === null || context === void 0 ? void 0 : context.id_user) || null,
                                             id_conversation: (conversation === null || conversation === void 0 ? void 0 : conversation._id) || null,
                                             id_message: assistantMessageId ? String(assistantMessageId) : null,
+                                            fastModeEnabled: fastModeEnabled,
+                                            runBudgetMs: runDeadlineMs ? runBudgetMs : null,
                                             isWorkerInstance: process.env.IS_WORKER_INSTANCE || null,
                                             workerIndex: process.env.WORKER_INDEX || null,
                                             workerInstance: process.env.NODE_APP_INSTANCE || null
@@ -1809,6 +1844,36 @@ function executeAiAssistantCodexRun(payload, context) {
                                     };
                                     progressTracker = createAssistantProgressTracker(assistantMessageId, initialProgress);
                                     streamProgress = createAssistantStreamProgressHandler(progressTracker);
+                                    remainingRunBudgetMs = function () {
+                                        if (!runDeadlineMs) {
+                                            return null;
+                                        }
+                                        return Math.max(runDeadlineMs - Date.now(), 0);
+                                    };
+                                    hasRunBudget = function (phase, minMs) {
+                                        if (minMs === void 0) { minMs = AI_ASSISTANT_CODEX_MIN_STAGE_BUDGET_MS; }
+                                        var remaining = remainingRunBudgetMs();
+                                        if (remaining === null || remaining >= minMs) {
+                                            return true;
+                                        }
+                                        recordStep("".concat(phase, ": skipped"), {
+                                            reason: 'run_budget_exhausted',
+                                            remainingMs: remaining,
+                                            budgetMs: runBudgetMs
+                                        });
+                                        return false;
+                                    };
+                                    applyCodexStageBudget = function (base) {
+                                        var remaining = remainingRunBudgetMs();
+                                        if (remaining === null) {
+                                            return base;
+                                        }
+                                        var baseTimeout = typeof base.timeoutMs === 'number' && base.timeoutMs > 0
+                                            ? base.timeoutMs
+                                            : resolveCodexTimeoutMs();
+                                        var timeoutMs = Math.max(1000, Math.min(baseTimeout, remaining));
+                                        return __assign(__assign({}, base), { timeoutMs: timeoutMs });
+                                    };
                                     assistantContent = '';
                                     toolResult = null;
                                     assistantDebug = null;
@@ -1879,7 +1944,7 @@ function executeAiAssistantCodexRun(payload, context) {
                                     });
                                     _14.label = 1;
                                 case 1:
-                                    _14.trys.push([1, 53, 54, 55]);
+                                    _14.trys.push([1, 56, 57, 58]);
                                     hintSeed = [message, contextRoute].filter(Boolean).join(' ');
                                     termExpansion = expandAssistantTermSynonyms(hintSeed);
                                     hintText = termExpansion.expanded || hintSeed;
@@ -2022,7 +2087,8 @@ function executeAiAssistantCodexRun(payload, context) {
                                         requestClassification: requestClassification,
                                         hasDeterministicHeuristicFastPath: hasDeterministicHeuristicFastPath
                                     });
-                                    if (!shouldRunPlanner) return [3 /*break*/, 12];
+                                    plannerBudgetAvailable = !shouldRunPlanner || hasRunBudget('Planning: planner');
+                                    if (!(shouldRunPlanner && plannerBudgetAvailable)) return [3 /*break*/, 12];
                                     plannerUsed = true;
                                     recordStep('Planning: planner prompt');
                                     plannerPrompt = buildAssistantPlannerPrompt({
@@ -2041,7 +2107,7 @@ function executeAiAssistantCodexRun(payload, context) {
                                 case 8:
                                     _14.trys.push([8, 10, , 11]);
                                     plannerStart = Date.now();
-                                    return [4 /*yield*/, runCodexInWorkerThread(plannerPrompt, plannerRunOptions, codexConfig, streamProgress)];
+                                    return [4 /*yield*/, runCodexInWorkerThread(plannerPrompt, applyCodexStageBudget(plannerRunOptions), codexConfig, streamProgress)];
                                 case 9:
                                     plannerRaw = _14.sent();
                                     accumulateCodexUsage(plannerPrompt, plannerRaw);
@@ -2062,6 +2128,9 @@ function executeAiAssistantCodexRun(payload, context) {
                                         else if (!requestClassification.dataQuestion) {
                                             plannerSkipReason = "non_data_".concat(requestClassification.type || 'unknown');
                                         }
+                                        else if (!plannerBudgetAvailable) {
+                                            plannerSkipReason = 'run_budget_exhausted';
+                                        }
                                         else {
                                             plannerSkipReason = 'policy';
                                         }
@@ -2122,11 +2191,12 @@ function executeAiAssistantCodexRun(payload, context) {
                                     if (!(dataQuestion && !directive)) return [3 /*break*/, 17];
                                     recordStep('Directive: determine tool', { type: 'data-question' });
                                     directivePrompt = buildAssistantCodexDirectivePrompt(message, attachmentData.promptText, historyLines.join('\n'), assistantContext, directiveStyleHint, directivePromptMode);
+                                    if (!hasRunBudget('Directive: determine tool')) return [3 /*break*/, 17];
                                     _14.label = 14;
                                 case 14:
                                     _14.trys.push([14, 16, , 17]);
                                     directiveStart = Date.now();
-                                    return [4 /*yield*/, runCodexInWorkerThread(directivePrompt, runOptions, codexConfig, streamProgress)];
+                                    return [4 /*yield*/, runCodexInWorkerThread(directivePrompt, applyCodexStageBudget(runOptions), codexConfig, streamProgress)];
                                 case 15:
                                     directiveText = _14.sent();
                                     accumulateCodexUsage(directivePrompt, directiveText);
@@ -2151,11 +2221,12 @@ function executeAiAssistantCodexRun(payload, context) {
                                 case 17:
                                     if (!(!directive && !dataQuestion)) return [3 /*break*/, 21];
                                     recordStep('Response: draft initial answer', { mode: 'full' });
+                                    if (!hasRunBudget('Response: initial draft')) return [3 /*break*/, 21];
                                     _14.label = 18;
                                 case 18:
                                     _14.trys.push([18, 20, , 21]);
                                     initialStart = Date.now();
-                                    return [4 /*yield*/, runCodexInWorkerThread(prompt_1, runOptions, codexConfig, streamProgress)];
+                                    return [4 /*yield*/, runCodexInWorkerThread(prompt_1, applyCodexStageBudget(runOptions), codexConfig, streamProgress)];
                                 case 19:
                                     responseText = _14.sent();
                                     accumulateCodexUsage(prompt_1, responseText);
@@ -2184,11 +2255,12 @@ function executeAiAssistantCodexRun(payload, context) {
                                     if (!(!directive && dataQuestion)) return [3 /*break*/, 25];
                                     recordStep('Directive: forced retry', { mode: 'directive-only' });
                                     directivePrompt = buildAssistantCodexDirectivePrompt(message, attachmentData.promptText, historyLines.join('\n'), assistantContext, directiveStyleHint, directivePromptMode);
+                                    if (!hasRunBudget('Directive: forced retry')) return [3 /*break*/, 25];
                                     _14.label = 22;
                                 case 22:
                                     _14.trys.push([22, 24, , 25]);
                                     forcedStart = Date.now();
-                                    return [4 /*yield*/, runCodexInWorkerThread(directivePrompt, runOptions, codexConfig, streamProgress)];
+                                    return [4 /*yield*/, runCodexInWorkerThread(directivePrompt, applyCodexStageBudget(runOptions), codexConfig, streamProgress)];
                                 case 23:
                                     directiveText = _14.sent();
                                     accumulateCodexUsage(directivePrompt, directiveText);
@@ -2226,11 +2298,12 @@ function executeAiAssistantCodexRun(payload, context) {
                                         requestedBreakdowns: requestedBreakdownDimensions,
                                         strict: true
                                     }), 'aggregate_only');
+                                    if (!hasRunBudget('Directive: dated retry')) return [3 /*break*/, 29];
                                     _14.label = 26;
                                 case 26:
                                     _14.trys.push([26, 28, , 29]);
                                     strictStart = Date.now();
-                                    return [4 /*yield*/, runCodexInWorkerThread(strictDirectivePrompt, runOptions, codexConfig, streamProgress)];
+                                    return [4 /*yield*/, runCodexInWorkerThread(strictDirectivePrompt, applyCodexStageBudget(runOptions), codexConfig, streamProgress)];
                                 case 27:
                                     strictDirectiveText = _14.sent();
                                     accumulateCodexUsage(strictDirectivePrompt, strictDirectiveText);
@@ -2273,11 +2346,12 @@ function executeAiAssistantCodexRun(payload, context) {
                                         requestedBreakdowns: requestedBreakdownDimensions,
                                         strict: true
                                     }), 'aggregate_only');
+                                    if (!hasRunBudget('Directive: dated guard retry')) return [3 /*break*/, 33];
                                     _14.label = 30;
                                 case 30:
                                     _14.trys.push([30, 32, , 33]);
                                     guardStart = Date.now();
-                                    return [4 /*yield*/, runCodexInWorkerThread(guardDirectivePrompt, runOptions, codexConfig, streamProgress)];
+                                    return [4 /*yield*/, runCodexInWorkerThread(guardDirectivePrompt, applyCodexStageBudget(runOptions), codexConfig, streamProgress)];
                                 case 31:
                                     guardDirectiveText = _14.sent();
                                     accumulateCodexUsage(guardDirectivePrompt, guardDirectiveText);
@@ -2318,11 +2392,12 @@ function executeAiAssistantCodexRun(payload, context) {
                                         requestedBreakdowns: requestedBreakdownDimensions,
                                         strict: true
                                     }), 'aggregate_only');
+                                    if (!hasRunBudget('Directive: grouped guard retry')) return [3 /*break*/, 37];
                                     _14.label = 34;
                                 case 34:
                                     _14.trys.push([34, 36, , 37]);
                                     groupedStart = Date.now();
-                                    return [4 /*yield*/, runCodexInWorkerThread(groupedDirectivePrompt, runOptions, codexConfig, streamProgress)];
+                                    return [4 /*yield*/, runCodexInWorkerThread(groupedDirectivePrompt, applyCodexStageBudget(runOptions), codexConfig, streamProgress)];
                                 case 35:
                                     groupedDirectiveText = _14.sent();
                                     accumulateCodexUsage(groupedDirectivePrompt, groupedDirectiveText);
@@ -2405,12 +2480,17 @@ function executeAiAssistantCodexRun(payload, context) {
                                     });
                                     assistantContent = "I couldn't run that request because this account does not have ".concat(permissionLabel, " permission.");
                                     toolError = new Error('AI assistant report builder bridge: Access denied.');
-                                    return [3 /*break*/, 52];
+                                    return [3 /*break*/, 55];
                                 case 38:
-                                    if (!((directive === null || directive === void 0 ? void 0 : directive.payload) && AI_ASSISTANT_TOOL_MAX_STEPS > 0)) return [3 /*break*/, 51];
+                                    if (!((directive === null || directive === void 0 ? void 0 : directive.payload) && AI_ASSISTANT_TOOL_MAX_STEPS > 0)) return [3 /*break*/, 54];
                                     effectiveDirective = collectionOverride
                                         ? __assign(__assign({}, directive), { payload: __assign(__assign({}, (directive.payload || {})), { collection: collectionOverride.to }) }) : directive;
                                     toolRequest = buildAssistantToolRequest(effectiveDirective, input);
+                                    toolRequest.__assistantRuntime = {
+                                        fastMode: fastModeEnabled,
+                                        deadlineMs: runDeadlineMs || undefined,
+                                        maxTimeMS: resolveAssistantMongoDefaultMaxTimeMs()
+                                    };
                                     progressTracker.push('Grabbing Data');
                                     recordStep('Grabbing Data: start', {
                                         type: effectiveDirective.type,
@@ -2419,18 +2499,22 @@ function executeAiAssistantCodexRun(payload, context) {
                                     });
                                     _14.label = 39;
                                 case 39:
-                                    _14.trys.push([39, 49, , 50]);
+                                    _14.trys.push([39, 52, , 53]);
+                                    if (!!hasRunBudget('Grabbing Data', AI_ASSISTANT_MONGO_MIN_STAGE_BUDGET_MS)) return [3 /*break*/, 40];
+                                    assistantContent = buildAssistantRunBudgetExceededMessage();
+                                    return [3 /*break*/, 51];
+                                case 40:
                                     toolStart = Date.now();
-                                    if (!(effectiveDirective.type === 'aggregate')) return [3 /*break*/, 41];
+                                    if (!(effectiveDirective.type === 'aggregate')) return [3 /*break*/, 42];
                                     return [4 /*yield*/, executeAiAssistantReportBuilderAggregate(toolRequest, context)];
-                                case 40:
-                                    _h = _14.sent();
-                                    return [3 /*break*/, 43];
-                                case 41: return [4 /*yield*/, executeAiAssistantReportBuilderRead(toolRequest, context)];
-                                case 42:
+                                case 41:
                                     _h = _14.sent();
-                                    _14.label = 43;
+                                    return [3 /*break*/, 44];
+                                case 42: return [4 /*yield*/, executeAiAssistantReportBuilderRead(toolRequest, context)];
                                 case 43:
+                                    _h = _14.sent();
+                                    _14.label = 44;
+                                case 44:
                                     toolResponse = _h;
                                     timingBreakdown.toolMs = Date.now() - toolStart;
                                     toolResponseDebug = (toolResponse === null || toolResponse === void 0 ? void 0 : toolResponse.debug) && typeof toolResponse.debug === 'object'
@@ -2444,58 +2528,71 @@ function executeAiAssistantCodexRun(payload, context) {
                                         columns: toolPayload.result.output.columns
                                     });
                                     progressTracker.push('Drafting response');
+                                    zeroRowResult = toolPayload.result.output.rowCount <= 0;
                                     skipFollowupCodex = usedDeterministicHeuristicFastPath
-                                        || isAssistantDeterministicHeuristicDirective(effectiveDirective);
-                                    if (!skipFollowupCodex) return [3 /*break*/, 44];
+                                        || isAssistantDeterministicHeuristicDirective(effectiveDirective)
+                                        || zeroRowResult;
+                                    if (!skipFollowupCodex) return [3 /*break*/, 45];
                                     recordStep('Drafting response: deterministic summary', {
-                                        reason: normalizeOptionalString(effectiveDirective.rawLine) || 'deterministic_heuristic'
+                                        reason: zeroRowResult
+                                            ? 'zero_rows'
+                                            : (normalizeOptionalString(effectiveDirective.rawLine) || 'deterministic_heuristic')
                                     });
                                     assistantContent = buildAssistantToolFallbackResponse(toolPayload.result);
                                     assistantContent = applyAssistantDisplayTableToResponse(assistantContent, toolPayload.result.output.display);
-                                    return [3 /*break*/, 48];
-                                case 44:
+                                    return [3 /*break*/, 51];
+                                case 45:
                                     recordStep('Drafting response');
                                     followupPrompt = buildAssistantCodexToolFollowupPrompt(message, attachmentData.promptText, historyLines.join('\n'), assistantContext, toolPayload.prompt);
-                                    _14.label = 45;
-                                case 45:
-                                    _14.trys.push([45, 47, , 48]);
-                                    followupStart = Date.now();
-                                    return [4 /*yield*/, runCodexInWorkerThread(followupPrompt, runOptions, codexConfig, streamProgress)];
+                                    if (!hasRunBudget('Drafting response: followup')) return [3 /*break*/, 50];
+                                    _14.label = 46;
                                 case 46:
+                                    _14.trys.push([46, 48, , 49]);
+                                    followupStart = Date.now();
+                                    return [4 /*yield*/, runCodexInWorkerThread(followupPrompt, applyCodexStageBudget(runOptions), codexConfig, streamProgress)];
+                                case 47:
                                     followupText = _14.sent();
                                     accumulateCodexUsage(followupPrompt, followupText);
                                     timingBreakdown.followupMs = Date.now() - followupStart;
                                     assistantContent = sanitizeAssistantResponse(followupText);
                                     assistantContent = applyAssistantDisplayTableToResponse(assistantContent, toolPayload.result.output.display);
-                                    return [3 /*break*/, 48];
-                                case 47:
+                                    return [3 /*break*/, 49];
+                                case 48:
                                     _j = _14.sent();
                                     assistantContent = buildAssistantToolFallbackResponse(toolPayload.result);
                                     assistantContent = applyAssistantDisplayTableToResponse(assistantContent, toolPayload.result.output.display);
-                                    return [3 /*break*/, 48];
-                                case 48: return [3 /*break*/, 50];
-                                case 49:
+                                    return [3 /*break*/, 49];
+                                case 49: return [3 /*break*/, 51];
+                                case 50:
+                                    assistantContent = buildAssistantToolFallbackResponse(toolPayload.result);
+                                    assistantContent = applyAssistantDisplayTableToResponse(assistantContent, toolPayload.result.output.display);
+                                    _14.label = 51;
+                                case 51: return [3 /*break*/, 53];
+                                case 52:
                                     error_3 = _14.sent();
                                     assistantContent = buildAssistantToolErrorMessage(error_3);
                                     toolError = error_3;
-                                    return [3 /*break*/, 50];
-                                case 50: return [3 /*break*/, 52];
-                                case 51:
+                                    return [3 /*break*/, 53];
+                                case 53: return [3 /*break*/, 55];
+                                case 54:
                                     progressTracker.push('Drafting response');
                                     recordStep('Drafting response');
-                                    _14.label = 52;
-                                case 52: return [3 /*break*/, 55];
-                                case 53:
+                                    _14.label = 55;
+                                case 55: return [3 /*break*/, 58];
+                                case 56:
                                     error_4 = _14.sent();
                                     assistantContent = buildAssistantCodexErrorMessage(error_4);
                                     recordStep('Error', { message: normalizeOptionalString(error_4 === null || error_4 === void 0 ? void 0 : error_4.message) || 'Unknown error' });
-                                    return [3 /*break*/, 55];
-                                case 54:
+                                    return [3 /*break*/, 58];
+                                case 57:
                                     progressTracker.stop();
                                     return [7 /*endfinally*/];
-                                case 55:
+                                case 58:
+                                    runBudgetExceeded = runDeadlineMs ? Date.now() > runDeadlineMs : false;
                                     if (!assistantContent) {
-                                        assistantContent = buildAssistantCodexErrorMessage(null);
+                                        assistantContent = runBudgetExceeded
+                                            ? buildAssistantRunBudgetExceededMessage()
+                                            : buildAssistantCodexErrorMessage(null);
                                     }
                                     assistantContent = applyAssistantVerificationNotes(assistantContent, toolResult);
                                     assistantContent = applyAssistantDatedReportWindow(assistantContent, toolResult);
@@ -2511,15 +2608,17 @@ function executeAiAssistantCodexRun(payload, context) {
                                             contentChars: assistantContent.length,
                                             totalMs: finishedAt - runStart,
                                             timingBreakdown: timingBreakdown,
+                                            runBudgetMs: runDeadlineMs ? runBudgetMs : null,
+                                            runBudgetRemainingMs: remainingRunBudgetMs(),
                                             isWorkerInstance: process.env.IS_WORKER_INSTANCE || null,
                                             workerIndex: process.env.WORKER_INDEX || null,
                                             workerInstance: process.env.NODE_APP_INSTANCE || null
                                         });
                                     }
                                     finalNow = new Date();
-                                    _14.label = 56;
-                                case 56:
-                                    _14.trys.push([56, 67, , 78]);
+                                    _14.label = 59;
+                                case 59:
+                                    _14.trys.push([59, 70, , 81]);
                                     if (canViewDebug) {
                                         finishedAt = Date.now();
                                         codexMs = timingBreakdown.directiveMs
@@ -2587,7 +2686,11 @@ function executeAiAssistantCodexRun(payload, context) {
                                             }
                                         });
                                     }
-                                    finalMetadata = __assign(__assign(__assign(__assign(__assign(__assign({ model: codexModel }, (codexFallbackModels.length ? { model_fallbacks: codexFallbackModels } : {})), (requestId ? { request_id: requestId } : {})), { pending: false }), (changeHistoryFastPathBypassedReason ? {
+                                    finalMetadata = __assign(__assign(__assign(__assign(__assign(__assign(__assign({ model: codexModel }, (codexFallbackModels.length ? { model_fallbacks: codexFallbackModels } : {})), (requestId ? { request_id: requestId } : {})), { pending: false }), (fastModeEnabled ? {
+                                        fast_mode: true,
+                                        run_budget_ms: runBudgetMs,
+                                        run_budget_exhausted: runBudgetExceeded
+                                    } : {})), (changeHistoryFastPathBypassedReason ? {
                                         fast_path_change_history_bypassed_reason: changeHistoryFastPathBypassedReason
                                     } : {})), (toolResult ? { tool_result: toolResult } : {})), (assistantDebug ? { debug: assistantDebug } : {}));
                                     finalUsage = codexUsage.total_tokens > 0 ? {
@@ -2596,14 +2699,14 @@ function executeAiAssistantCodexRun(payload, context) {
                                         output_tokens: codexUsage.output_tokens,
                                         total_tokens: codexUsage.total_tokens
                                     } : null;
-                                    if (!finalUsage) return [3 /*break*/, 61];
+                                    if (!finalUsage) return [3 /*break*/, 64];
                                     return [4 /*yield*/, resolveClientId(conversation, input.id_client, context === null || context === void 0 ? void 0 : context.id_user)];
-                                case 57:
+                                case 60:
                                     usageClientId = _14.sent();
-                                    if (!usageClientId) return [3 /*break*/, 61];
-                                    _14.label = 58;
-                                case 58:
-                                    _14.trys.push([58, 60, , 61]);
+                                    if (!usageClientId) return [3 /*break*/, 64];
+                                    _14.label = 61;
+                                case 61:
+                                    _14.trys.push([61, 63, , 64]);
                                     return [4 /*yield*/, (0, openai_usage_ledger_manager_1.recordOpenAIUsage)({
                                             id_client: usageClientId,
                                             model: finalUsage.model,
@@ -2614,16 +2717,16 @@ function executeAiAssistantCodexRun(payload, context) {
                                             id_request: requestId || undefined,
                                             id_conversation: conversation._id
                                         })];
-                                case 59:
+                                case 62:
                                     _14.sent();
-                                    return [3 /*break*/, 61];
-                                case 60:
+                                    return [3 /*break*/, 64];
+                                case 63:
                                     usageError_1 = _14.sent();
-                                    console.error(new Date(), 'Failed to record codex usage', usageError_1);
-                                    return [3 /*break*/, 61];
-                                case 61:
+                                    console.error(new Date(), 'Failed to record AI usage', usageError_1);
+                                    return [3 /*break*/, 64];
+                                case 64:
                                     finalAssistantDoc = __assign(__assign(__assign(__assign({}, assistantDoc), { _id: assistantMessageId, content: assistantContent, metadata: finalMetadata }), (finalUsage ? { usage: finalUsage } : {})), { updatedAt: finalNow });
-                                    if (!assistantMessageId) return [3 /*break*/, 63];
+                                    if (!assistantMessageId) return [3 /*break*/, 66];
                                     setPayload = {
                                         content: assistantContent,
                                         metadata: finalMetadata,
@@ -2633,59 +2736,59 @@ function executeAiAssistantCodexRun(payload, context) {
                                         setPayload.usage = finalUsage;
                                     }
                                     return [4 /*yield*/, updateAssistantMessageWithFallback(assistantMessageId, setPayload)];
-                                case 62:
-                                    _14.sent();
-                                    _14.label = 63;
-                                case 63: return [4 /*yield*/, touchConversation(conversation._id, finalNow, assistantMessageId ? String(assistantMessageId) : undefined)];
-                                case 64:
-                                    _14.sent();
-                                    if (!(input.delete_files_after_run !== false)) return [3 /*break*/, 66];
-                                    return [4 /*yield*/, cleanupAttachments(attachmentData.attachments)];
                                 case 65:
                                     _14.sent();
                                     _14.label = 66;
-                                case 66: return [2 /*return*/, finalAssistantDoc];
+                                case 66: return [4 /*yield*/, touchConversation(conversation._id, finalNow, assistantMessageId ? String(assistantMessageId) : undefined)];
                                 case 67:
+                                    _14.sent();
+                                    if (!(input.delete_files_after_run !== false)) return [3 /*break*/, 69];
+                                    return [4 /*yield*/, cleanupAttachments(attachmentData.attachments)];
+                                case 68:
+                                    _14.sent();
+                                    _14.label = 69;
+                                case 69: return [2 /*return*/, finalAssistantDoc];
+                                case 70:
                                     finalizeError_1 = _14.sent();
                                     failedAt = new Date();
                                     fallbackContent = assistantContent || buildAssistantCodexErrorMessage(finalizeError_1);
                                     failureMetadata = __assign(__assign(__assign({ model: codexModel }, (codexFallbackModels.length ? { model_fallbacks: codexFallbackModels } : {})), (requestId ? { request_id: requestId } : {})), { pending: false, failed: true, error_message: normalizeOptionalString(finalizeError_1 === null || finalizeError_1 === void 0 ? void 0 : finalizeError_1.message) || 'Unknown error' });
-                                    _14.label = 68;
-                                case 68:
-                                    _14.trys.push([68, 72, , 73]);
-                                    if (!assistantMessageId) return [3 /*break*/, 70];
+                                    _14.label = 71;
+                                case 71:
+                                    _14.trys.push([71, 75, , 76]);
+                                    if (!assistantMessageId) return [3 /*break*/, 73];
                                     return [4 /*yield*/, updateAssistantMessageWithFallback(assistantMessageId, {
                                             content: fallbackContent,
                                             metadata: failureMetadata,
                                             updatedAt: failedAt
                                         })];
-                                case 69:
+                                case 72:
                                     _14.sent();
-                                    _14.label = 70;
-                                case 70: return [4 /*yield*/, touchConversation(conversation._id, failedAt, assistantMessageId ? String(assistantMessageId) : undefined)];
-                                case 71:
+                                    _14.label = 73;
+                                case 73: return [4 /*yield*/, touchConversation(conversation._id, failedAt, assistantMessageId ? String(assistantMessageId) : undefined)];
+                                case 74:
                                     _14.sent();
-                                    return [3 /*break*/, 73];
-                                case 72:
+                                    return [3 /*break*/, 76];
+                                case 75:
                                     persistError_1 = _14.sent();
                                     console.error(new Date(), 'AI assistant finalize fallback update failed', persistError_1);
-                                    return [3 /*break*/, 73];
-                                case 73:
-                                    if (!(input.delete_files_after_run !== false)) return [3 /*break*/, 77];
-                                    _14.label = 74;
-                                case 74:
-                                    _14.trys.push([74, 76, , 77]);
+                                    return [3 /*break*/, 76];
+                                case 76:
+                                    if (!(input.delete_files_after_run !== false)) return [3 /*break*/, 80];
+                                    _14.label = 77;
+                                case 77:
+                                    _14.trys.push([77, 79, , 80]);
                                     return [4 /*yield*/, cleanupAttachments(attachmentData.attachments)];
-                                case 75:
+                                case 78:
                                     _14.sent();
-                                    return [3 /*break*/, 77];
-                                case 76:
+                                    return [3 /*break*/, 80];
+                                case 79:
                                     _k = _14.sent();
-                                    return [3 /*break*/, 77];
-                                case 77:
+                                    return [3 /*break*/, 80];
+                                case 80:
                                     console.error(new Date(), 'AI assistant run finalize failed:', finalizeError_1);
                                     return [2 /*return*/, __assign(__assign({}, assistantDoc), { _id: assistantMessageId, content: fallbackContent, metadata: failureMetadata, updatedAt: failedAt })];
-                                case 78: return [2 /*return*/];
+                                case 81: return [2 /*return*/];
                             }
                         });
                     }); });
@@ -2928,7 +3031,7 @@ function executeAiAssistantReportIssue(payload, context) {
 }
 function executeAiAssistantMongoRead(payload, context) {
     return __awaiter(this, void 0, void 0, function () {
-        var input, rawCollection, retryState, triedCollections, crossCollectionRetryEnabled, dbName, db, collectionResolution, collection, bridgeCollection, retryRootCollection, schemaFields, effectivePermissionView, _a, user, isSuperAdmin, canViewDebug, customerId, fallbackMeta, baseQuery, stripped, userId, normalizedClient, shouldScopeByClient, _b, strippedClient, clientScopedQuery, scopedQuery, normalized, findOptions, runFindWithRepair, initialRead, documents, executedQuery, probeDocs, dateFallback, fallbackQuery, fallbackRead, fallbackDocs, expanded, fallbackRead, fallbackDocs, nameFallback, fallbackRead, fallbackDocs, _c, chemicalLookup, fallbackRead, fallbackDocs, queryFields, _d, aliases, rewrittenQuery, fallbackRead, fallbackDocs, _e, activeFallback, fallbackRead, fallbackDocs, _f, idLookup, fallbackRead, fallbackDocs, baseCollection, fallbackPayload, fallbackResult, existingFallbacks, hasActiveStatusFilters, crossCollectionCandidates, crossCollectionCandidates_1, crossCollectionCandidates_1_1, candidateCollection, fallbackPayload, fallbackResult, fallbackDocs, existingFallbacks, rejectedQuality, error_5, existingErrors, e_1_1, nameFields, dateFields, diagnostics, queryNoName, _g, queryNoDate, _h, _j, _k, _l, allCollections, base, alt, altCount, _m, total, sanitizedDocuments, requestedFields, missingFields, _o, projectionAliases, expandedProjection, refreshedDocs, includeIds, fieldAliases, displayDocs, idLookupDisplay, priorityFields, displayMaxRows, display;
+        var input, rawCollection, retryState, triedCollections, crossCollectionRetryEnabled, dbName, db, collectionResolution, collection, bridgeCollection, retryRootCollection, schemaFields, effectivePermissionView, _a, user, isSuperAdmin, canViewDebug, customerId, fallbackMeta, runtimeControl, runtimeFastMode, activeRuntimeControl, hasReadRuntimeBudget, resolveReadMaxTimeMS, baseQuery, stripped, userId, normalizedClient, shouldScopeByClient, _b, strippedClient, clientScopedQuery, scopedQuery, normalized, findOptions, buildFindOptions, buildCountOptions, runFindWithRepair, initialRead, documents, executedQuery, probeDocs, dateFallback, fallbackQuery, fallbackRead, fallbackDocs, expanded, fallbackRead, fallbackDocs, nameFallback, fallbackRead, fallbackDocs, _c, chemicalLookup, fallbackRead, fallbackDocs, queryFields, _d, aliases, rewrittenQuery, fallbackRead, fallbackDocs, _e, activeFallback, fallbackRead, fallbackDocs, _f, idLookup, fallbackRead, fallbackDocs, baseCollection, fallbackPayload, fallbackResult, existingFallbacks, hasActiveStatusFilters, crossCollectionCandidates, crossCollectionCandidates_1, crossCollectionCandidates_1_1, candidateCollection, fallbackPayload, fallbackResult, fallbackDocs, existingFallbacks, rejectedQuality, error_5, existingErrors, e_1_1, nameFields, dateFields, diagnostics, countOptions, queryNoName, _g, queryNoDate, _h, _j, _k, _l, allCollections, base, alt, altCount, _m, total, sanitizedDocuments, requestedFields, missingFields, _o, projectionAliases, expandedProjection, refreshedDocs, includeIds, fieldAliases, displayDocs, idLookupDisplay, priorityFields, displayMaxRows, display;
         var e_1, _p;
         var _this = this;
         var _q, _r;
@@ -2972,6 +3075,26 @@ function executeAiAssistantMongoRead(payload, context) {
                     }
                     customerId = normalizeOptionalString((_q = user === null || user === void 0 ? void 0 : user.other) === null || _q === void 0 ? void 0 : _q.id_customer);
                     fallbackMeta = {};
+                    runtimeControl = normalizeAssistantRuntimeControl(input.__assistantRuntime);
+                    runtimeFastMode = (runtimeControl === null || runtimeControl === void 0 ? void 0 : runtimeControl.fastMode) === undefined
+                        ? resolveAssistantFastModeEnabled()
+                        : runtimeControl.fastMode === true;
+                    activeRuntimeControl = runtimeFastMode ? runtimeControl : null;
+                    hasReadRuntimeBudget = function (stage, minimumMs) {
+                        if (minimumMs === void 0) { minimumMs = AI_ASSISTANT_MONGO_MIN_STAGE_BUDGET_MS; }
+                        if (!hasAssistantRuntimeBudget(activeRuntimeControl, minimumMs)) {
+                            fallbackMeta.runtimeBudget = {
+                                stage: stage,
+                                exhausted: true,
+                                deadlineMs: (activeRuntimeControl === null || activeRuntimeControl === void 0 ? void 0 : activeRuntimeControl.deadlineMs) || undefined
+                            };
+                            return false;
+                        }
+                        return true;
+                    };
+                    resolveReadMaxTimeMS = function () { return resolveAssistantEffectiveMongoMaxTimeMs({
+                        runtime: activeRuntimeControl
+                    }); };
                     if (bridgeCollection.fallbackFrom) {
                         fallbackMeta.reportBuilderBridge = {
                             from: bridgeCollection.fallbackFrom,
@@ -3029,17 +3152,33 @@ function executeAiAssistantMongoRead(payload, context) {
                     scopedQuery = applyCustomerScopeFilter(clientScopedQuery, collection, customerId, isSuperAdmin);
                     normalized = normalizeAssistantFindOptions(input.options);
                     findOptions = __assign(__assign({}, normalized.findOptions), { readPreference: AI_ASSISTANT_READ_PREFERENCE });
+                    buildFindOptions = function () {
+                        var maxTimeMS = resolveReadMaxTimeMS();
+                        return maxTimeMS ? __assign(__assign({}, findOptions), { maxTimeMS: maxTimeMS }) : findOptions;
+                    };
+                    buildCountOptions = function () {
+                        var maxTimeMS = resolveReadMaxTimeMS();
+                        return maxTimeMS
+                            ? { readPreference: AI_ASSISTANT_READ_PREFERENCE, maxTimeMS: maxTimeMS }
+                            : { readPreference: AI_ASSISTANT_READ_PREFERENCE };
+                    };
                     runFindWithRepair = function (query, stage) { return __awaiter(_this, void 0, void 0, function () {
-                        var docs, error_6, repaired, docs;
+                        var stageFindOptions, docs, error_6, repaired, docs;
                         return __generator(this, function (_a) {
                             switch (_a.label) {
                                 case 0:
-                                    _a.trys.push([0, 2, , 4]);
-                                    return [4 /*yield*/, db.collection(collection).find(query, findOptions).toArray()];
+                                    if (!hasReadRuntimeBudget(stage)) {
+                                        return [2 /*return*/, { documents: [], query: query }];
+                                    }
+                                    stageFindOptions = buildFindOptions();
+                                    _a.label = 1;
                                 case 1:
+                                    _a.trys.push([1, 3, , 5]);
+                                    return [4 /*yield*/, db.collection(collection).find(query, stageFindOptions).toArray()];
+                                case 2:
                                     docs = _a.sent();
                                     return [2 /*return*/, { documents: docs, query: query }];
-                                case 2:
+                                case 3:
                                     error_6 = _a.sent();
                                     if (!isAssistantDateArithmeticArgumentError(error_6)) {
                                         throw error_6;
@@ -3048,8 +3187,8 @@ function executeAiAssistantMongoRead(payload, context) {
                                     if (!repaired.changed || containsForbiddenMongoOperators(repaired.query)) {
                                         throw error_6;
                                     }
-                                    return [4 /*yield*/, db.collection(collection).find(repaired.query, findOptions).toArray()];
-                                case 3:
+                                    return [4 /*yield*/, db.collection(collection).find(repaired.query, stageFindOptions).toArray()];
+                                case 4:
                                     docs = _a.sent();
                                     fallbackMeta.dateArithmetic = {
                                         attempted: true,
@@ -3061,7 +3200,7 @@ function executeAiAssistantMongoRead(payload, context) {
                                             documents: docs,
                                             query: repaired.query
                                         }];
-                                case 4: return [2 /*return*/];
+                                case 5: return [2 /*return*/];
                             }
                         });
                     }); };
@@ -3135,7 +3274,8 @@ function executeAiAssistantMongoRead(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveReadMaxTimeMS()
                         })];
                 case 14:
                     _c = (_s.sent());
@@ -3177,7 +3317,8 @@ function executeAiAssistantMongoRead(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveReadMaxTimeMS()
                         })];
                 case 19:
                     _d = (_s.sent());
@@ -3217,7 +3358,8 @@ function executeAiAssistantMongoRead(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveReadMaxTimeMS()
                         })];
                 case 23:
                     _e = (_s.sent());
@@ -3252,7 +3394,8 @@ function executeAiAssistantMongoRead(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveReadMaxTimeMS()
                         })];
                 case 27:
                     _f = (_s.sent());
@@ -3283,7 +3426,7 @@ function executeAiAssistantMongoRead(payload, context) {
                     }
                     _s.label = 31;
                 case 31:
-                    if (!!documents.length) return [3 /*break*/, 34];
+                    if (!(!documents.length && hasReadRuntimeBudget('report_fallback'))) return [3 /*break*/, 34];
                     return [4 /*yield*/, resolveBaseCollectionFromReport(db, dbName, collection)];
                 case 32:
                     baseCollection = _s.sent();
@@ -3314,7 +3457,7 @@ function executeAiAssistantMongoRead(payload, context) {
                     _s.label = 34;
                 case 34:
                     hasActiveStatusFilters = collectAssistantActiveMatchFields(executedQuery).length > 0;
-                    if (!(!documents.length && crossCollectionRetryEnabled && !hasActiveStatusFilters)) return [3 /*break*/, 46];
+                    if (!(!documents.length && crossCollectionRetryEnabled && !hasActiveStatusFilters && hasReadRuntimeBudget('collection_retry'))) return [3 /*break*/, 46];
                     return [4 /*yield*/, resolveAssistantAvailableCrossCollectionFallbacks(db, dbName, collection, triedCollections, {
                             fieldHints: extractQueryFieldPaths(executedQuery || {})
                         })];
@@ -3414,7 +3557,7 @@ function executeAiAssistantMongoRead(payload, context) {
                     }
                     _s.label = 47;
                 case 47:
-                    if (!(!documents.length && canViewDebug)) return [3 /*break*/, 64];
+                    if (!(!documents.length && canViewDebug && hasReadRuntimeBudget('zero_diagnostics', 20))) return [3 /*break*/, 64];
                     nameFields = collectMatchFieldsByCondition(executedQuery, function (field, condition) { return isRegexMatchCondition(condition)
                         || (typeof condition === 'string' && shouldApplyAssistantNameRegex(field)); });
                     dateFields = collectMatchFieldsByCondition(executedQuery, function (_field, condition) { return isDateCondition(condition); });
@@ -3425,12 +3568,11 @@ function executeAiAssistantMongoRead(payload, context) {
                     _s.label = 48;
                 case 48:
                     _s.trys.push([48, 53, , 54]);
+                    countOptions = buildCountOptions();
                     if (!nameFields.length) return [3 /*break*/, 50];
                     queryNoName = stripMatchFields(executedQuery, nameFields);
                     _g = diagnostics;
-                    return [4 /*yield*/, db.collection(collection).countDocuments(queryNoName, {
-                            readPreference: AI_ASSISTANT_READ_PREFERENCE
-                        })];
+                    return [4 /*yield*/, db.collection(collection).countDocuments(queryNoName, countOptions)];
                 case 49:
                     _g.recentCount = _s.sent();
                     _s.label = 50;
@@ -3438,9 +3580,7 @@ function executeAiAssistantMongoRead(payload, context) {
                     if (!dateFields.length) return [3 /*break*/, 52];
                     queryNoDate = stripMatchFields(executedQuery, dateFields);
                     _h = diagnostics;
-                    return [4 /*yield*/, db.collection(collection).countDocuments(queryNoDate, {
-                            readPreference: AI_ASSISTANT_READ_PREFERENCE
-                        })];
+                    return [4 /*yield*/, db.collection(collection).countDocuments(queryNoDate, countOptions)];
                 case 51:
                     _h.nameMatchCount = _s.sent();
                     _s.label = 52;
@@ -3458,7 +3598,8 @@ function executeAiAssistantMongoRead(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveReadMaxTimeMS()
                         })];
                 case 55:
                     _k = (_s.sent());
@@ -3480,7 +3621,7 @@ function executeAiAssistantMongoRead(payload, context) {
                     base = stripVersionSuffix(collection.startsWith('report-') ? collection.slice('report-'.length) : collection);
                     alt = collection.startsWith('report-') ? base : "report-".concat(base);
                     if (!(alt && alt !== collection && allCollections.includes(alt))) return [3 /*break*/, 61];
-                    return [4 /*yield*/, db.collection(alt).countDocuments({}, { readPreference: AI_ASSISTANT_READ_PREFERENCE })];
+                    return [4 /*yield*/, db.collection(alt).countDocuments({}, buildCountOptions())];
                 case 60:
                     altCount = _s.sent();
                     diagnostics.alternateCollection = alt;
@@ -3495,8 +3636,8 @@ function executeAiAssistantMongoRead(payload, context) {
                     _s.label = 64;
                 case 64:
                     total = null;
-                    if (!normalized.includeTotal) return [3 /*break*/, 66];
-                    return [4 /*yield*/, db.collection(collection).countDocuments(executedQuery, { readPreference: AI_ASSISTANT_READ_PREFERENCE })];
+                    if (!(normalized.includeTotal && hasReadRuntimeBudget('total_count', 20))) return [3 /*break*/, 66];
+                    return [4 /*yield*/, db.collection(collection).countDocuments(executedQuery, buildCountOptions())];
                 case 65:
                     total = _s.sent();
                     _s.label = 66;
@@ -3516,7 +3657,8 @@ function executeAiAssistantMongoRead(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveReadMaxTimeMS()
                         })];
                 case 67:
                     _o = (_s.sent());
@@ -3535,7 +3677,7 @@ function executeAiAssistantMongoRead(payload, context) {
                     if (!(expandedProjection && expandedProjection !== findOptions.projection)) return [3 /*break*/, 70];
                     findOptions = __assign(__assign({}, findOptions), { projection: expandedProjection });
                     normalized.findOptions.projection = expandedProjection;
-                    return [4 /*yield*/, db.collection(collection).find(executedQuery, findOptions).toArray()];
+                    return [4 /*yield*/, db.collection(collection).find(executedQuery, buildFindOptions()).toArray()];
                 case 69:
                     refreshedDocs = _s.sent();
                     if (refreshedDocs.length) {
@@ -3607,12 +3749,12 @@ function executeAiAssistantMongoRead(payload, context) {
 }
 function executeAiAssistantMongoAggregate(payload, context) {
     return __awaiter(this, void 0, void 0, function () {
-        var input, rawCollection, retryState, triedCollections, crossCollectionRetryEnabled, dbName, db, collectionResolution, collection, bridgeCollection, retryRootCollection, schemaFields, effectivePermissionView, _a, user, isSuperAdmin, canViewDebug, customerId, fallbackMeta, baseQuery, stripped, userId, normalizedClient, shouldScopeByClient, _b, clientScopedQuery, scopedQuery, normalizedPipeline, sanitizedPipeline, strippedPipeline, pipelineWithScope, normalizedOptions, limitedPipeline, executedPipeline, dateField, aggregateOptions, runAggregateWithRepair, initialAggregate, documents, probeDocs, fallback, fallbackPipeline, fallbackAggregate, fallbackDocs, createdFallback, createdPipeline, createdAggregate, createdDocs, expanded, expandedAggregate, expandedDocs, completionFallback, fallbackPipeline, fallbackAggregate, fallbackDocs, completionExprFallback, fallbackPipeline, fallbackAggregate, fallbackDocs, unwindFallback, shouldUnwind, _c, _d, fallbackPipeline, fallbackAggregate, fallbackDocs, nameFallback, fallbackPipeline, fallbackAggregate, fallbackDocs, _e, _loop_1, i, state_1, matchFields_1, _f, aliases, rewrittenPipeline, fallbackAggregate, fallbackDocs, _loop_2, i, state_2, nestedFallbacks, nestedFallbacks_1, nestedFallbacks_1_1, nestedFallback, fallbackPayload, fallbackResult, existingFallbacks, error_7, errorMessage, existingErrors, e_2_1, baseCollection, fallbackPayload, fallbackResult, existingFallbacks, fallbackAcceptanceOptions, crossCollectionCandidates, crossCollectionCandidates_2, crossCollectionCandidates_2_1, candidateCollection, fallbackPayload, fallbackResult, fallbackDocs, fallbackQuality, existingFallbacks, rejectedQuality, error_8, existingErrors, e_3_1, matchStages, diagnostics, combinedMatch, nameFields, dateFields, queryNoName, _g, queryNoDate, _h, _j, _k, _l, allCollections, base, alt, altCount, _m, verification, sanitizedDocuments, includeIds, displayDocs, idLookupDisplay, display;
-        var e_2, _o, e_3, _p;
+        var input, rawCollection, retryState, triedCollections, crossCollectionRetryEnabled, dbName, db, collectionResolution, collection, bridgeCollection, retryRootCollection, schemaFields, effectivePermissionView, _a, user, isSuperAdmin, canViewDebug, customerId, fallbackMeta, runtimeControl, runtimeFastMode, activeRuntimeControl, hasAggregateRuntimeBudget, resolveAggregateMaxTimeMS, baseQuery, stripped, userId, normalizedClient, shouldScopeByClient, _b, clientScopedQuery, scopedQuery, normalizedPipeline, sanitizedPipeline, strippedPipeline, pipelineWithScope, normalizedOptions, limitedPipeline, executedPipeline, dateField, buildAggregateOptions, buildAggregateCountOptions, runAggregateWithRepair, initialAggregate, documents, probeDocs, fallback, fallbackPipeline, fallbackAggregate, fallbackDocs, createdFallback, createdPipeline, createdAggregate, createdDocs, expanded, expandedAggregate, expandedDocs, completionFallback, fallbackPipeline, fallbackAggregate, fallbackDocs, completionExprFallback, fallbackPipeline, fallbackAggregate, fallbackDocs, unwindFallback, shouldUnwind, _c, _d, fallbackPipeline, fallbackAggregate, fallbackDocs, nameFallback, fallbackPipeline, fallbackAggregate, fallbackDocs, _e, _loop_1, i, state_1, matchFields_1, _f, aliases, rewrittenPipeline, fallbackAggregate, fallbackDocs, _loop_2, i, state_2, nestedFallbacks, nestedFallbacks_1, nestedFallbacks_1_1, nestedFallback, fallbackPayload, fallbackResult, existingFallbacks, error_7, errorMessage, existingErrors, e_2_1, baseCollection, fallbackPayload, fallbackResult, existingFallbacks, fallbackAcceptanceOptions, crossCollectionCandidates, crossCollectionCandidates_2, crossCollectionCandidates_2_1, candidateCollection, fallbackPayload, fallbackResult, fallbackDocs, fallbackQuality, existingFallbacks, rejectedQuality, error_8, existingErrors, e_3_1, matchStages, diagnostics, combinedMatch, nameFields, dateFields, countOptions, queryNoName, _g, queryNoDate, _h, _j, _k, _l, allCollections, base, alt, altCount, _m, verification, _o, sanitizedDocuments, includeIds, displayDocs, idLookupDisplay, display;
+        var e_2, _p, e_3, _q;
         var _this = this;
-        var _q, _r;
-        return __generator(this, function (_s) {
-            switch (_s.label) {
+        var _r, _s;
+        return __generator(this, function (_t) {
+            switch (_t.label) {
                 case 0:
                     input = payload || {};
                     rawCollection = normalizeOptionalString(input.collection);
@@ -3629,11 +3771,11 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     db = resolveio_server_app_1.ResolveIOServer.getMongoConnection().db(dbName);
                     return [4 /*yield*/, resolveAssistantCollectionName(db, dbName, rawCollection)];
                 case 1:
-                    collectionResolution = _s.sent();
+                    collectionResolution = _t.sent();
                     collection = collectionResolution.name || rawCollection;
                     return [4 /*yield*/, resolveAssistantReportBuilderBridgeCollection(collection, db, dbName)];
                 case 2:
-                    bridgeCollection = _s.sent();
+                    bridgeCollection = _t.sent();
                     if (bridgeCollection.fallbackFrom) {
                         collection = bridgeCollection.collection;
                     }
@@ -3645,12 +3787,33 @@ function executeAiAssistantMongoAggregate(payload, context) {
                         : input.permissionView;
                     return [4 /*yield*/, ensureAssistantReadAccess(context, effectivePermissionView, collection)];
                 case 3:
-                    _a = _s.sent(), user = _a.user, isSuperAdmin = _a.isSuperAdmin, canViewDebug = _a.canViewDebug;
+                    _a = _t.sent(), user = _a.user, isSuperAdmin = _a.isSuperAdmin, canViewDebug = _a.canViewDebug;
                     if (!isSuperAdmin && AI_ASSISTANT_BLOCKED_COLLECTIONS.has(collection)) {
                         throw new Error('AI assistant report builder bridge: Access denied.');
                     }
-                    customerId = normalizeOptionalString((_q = user === null || user === void 0 ? void 0 : user.other) === null || _q === void 0 ? void 0 : _q.id_customer);
+                    customerId = normalizeOptionalString((_r = user === null || user === void 0 ? void 0 : user.other) === null || _r === void 0 ? void 0 : _r.id_customer);
                     fallbackMeta = {};
+                    runtimeControl = normalizeAssistantRuntimeControl(input.__assistantRuntime);
+                    runtimeFastMode = (runtimeControl === null || runtimeControl === void 0 ? void 0 : runtimeControl.fastMode) === undefined
+                        ? resolveAssistantFastModeEnabled()
+                        : runtimeControl.fastMode === true;
+                    activeRuntimeControl = runtimeFastMode ? runtimeControl : null;
+                    hasAggregateRuntimeBudget = function (stage, minimumMs) {
+                        if (minimumMs === void 0) { minimumMs = AI_ASSISTANT_MONGO_MIN_STAGE_BUDGET_MS; }
+                        if (!hasAssistantRuntimeBudget(activeRuntimeControl, minimumMs)) {
+                            fallbackMeta.runtimeBudget = {
+                                stage: stage,
+                                exhausted: true,
+                                deadlineMs: (activeRuntimeControl === null || activeRuntimeControl === void 0 ? void 0 : activeRuntimeControl.deadlineMs) || undefined
+                            };
+                            return false;
+                        }
+                        return true;
+                    };
+                    resolveAggregateMaxTimeMS = function (requestedMaxTimeMS) { return resolveAssistantEffectiveMongoMaxTimeMs({
+                        requestedMaxTimeMS: requestedMaxTimeMS,
+                        runtime: activeRuntimeControl
+                    }); };
                     if (bridgeCollection.fallbackFrom) {
                         fallbackMeta.reportBuilderBridge = {
                             from: bridgeCollection.fallbackFrom,
@@ -3686,11 +3849,11 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (!(!isSuperAdmin && normalizedClient)) return [3 /*break*/, 5];
                     return [4 /*yield*/, collectionHasClientIndex(db, dbName, collection)];
                 case 4:
-                    _b = _s.sent();
+                    _b = _t.sent();
                     return [3 /*break*/, 6];
                 case 5:
                     _b = false;
-                    _s.label = 6;
+                    _t.label = 6;
                 case 6:
                     shouldScopeByClient = _b;
                     clientScopedQuery = shouldScopeByClient
@@ -3716,20 +3879,35 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (containsForbiddenMongoOperators(executedPipeline)) {
                         throw new Error('AI assistant report builder bridge: Pipeline contains restricted operators.');
                     }
-                    aggregateOptions = __assign(__assign({}, normalizedOptions.aggregateOptions), { readPreference: AI_ASSISTANT_READ_PREFERENCE });
+                    buildAggregateOptions = function () {
+                        var maxTimeMS = resolveAggregateMaxTimeMS(normalizedOptions.aggregateOptions.maxTimeMS);
+                        return __assign(__assign(__assign({}, normalizedOptions.aggregateOptions), (maxTimeMS ? { maxTimeMS: maxTimeMS } : {})), { readPreference: AI_ASSISTANT_READ_PREFERENCE });
+                    };
+                    buildAggregateCountOptions = function () {
+                        var maxTimeMS = resolveAggregateMaxTimeMS(normalizedOptions.aggregateOptions.maxTimeMS);
+                        return maxTimeMS
+                            ? { readPreference: AI_ASSISTANT_READ_PREFERENCE, maxTimeMS: maxTimeMS }
+                            : { readPreference: AI_ASSISTANT_READ_PREFERENCE };
+                    };
                     runAggregateWithRepair = function (pipeline, stage) { return __awaiter(_this, void 0, void 0, function () {
-                        var docs, error_9, isDateRepair, isRankSortByRepair, isDottedOutputFieldRepair, repaired, dottedFieldPath, dottedFieldReferenceRepairUsed, pathReferenceRepair, docs;
+                        var aggregateOptions, docs, error_9, isDateRepair, isRankSortByRepair, isDottedOutputFieldRepair, repaired, dottedFieldPath, dottedFieldReferenceRepairUsed, pathReferenceRepair, docs;
                         return __generator(this, function (_a) {
                             switch (_a.label) {
                                 case 0:
-                                    _a.trys.push([0, 2, , 4]);
+                                    if (!hasAggregateRuntimeBudget(stage)) {
+                                        return [2 /*return*/, { documents: [], pipeline: pipeline }];
+                                    }
+                                    aggregateOptions = buildAggregateOptions();
+                                    _a.label = 1;
+                                case 1:
+                                    _a.trys.push([1, 3, , 5]);
                                     return [4 /*yield*/, db.collection(collection)
                                             .aggregate(pipeline, aggregateOptions)
                                             .toArray()];
-                                case 1:
+                                case 2:
                                     docs = _a.sent();
                                     return [2 /*return*/, { documents: docs, pipeline: pipeline }];
-                                case 2:
+                                case 3:
                                     error_9 = _a.sent();
                                     isDateRepair = isAssistantDateArithmeticArgumentError(error_9);
                                     isRankSortByRepair = isAssistantRankSortByError(error_9);
@@ -3763,7 +3941,7 @@ function executeAiAssistantMongoAggregate(payload, context) {
                                     return [4 /*yield*/, db.collection(collection)
                                             .aggregate(repaired.pipeline, aggregateOptions)
                                             .toArray()];
-                                case 3:
+                                case 4:
                                     docs = _a.sent();
                                     if (isDateRepair) {
                                         fallbackMeta.dateArithmetic = {
@@ -3793,13 +3971,13 @@ function executeAiAssistantMongoAggregate(payload, context) {
                                         };
                                     }
                                     return [2 /*return*/, { documents: docs, pipeline: repaired.pipeline }];
-                                case 4: return [2 /*return*/];
+                                case 5: return [2 /*return*/];
                             }
                         });
                     }); };
                     return [4 /*yield*/, runAggregateWithRepair(executedPipeline, 'initial')];
                 case 7:
-                    initialAggregate = _s.sent();
+                    initialAggregate = _t.sent();
                     documents = initialAggregate.documents;
                     executedPipeline = initialAggregate.pipeline;
                     dateField = findAggregateDateField(executedPipeline);
@@ -3812,14 +3990,14 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (!!containsForbiddenMongoOperators(fallbackPipeline)) return [3 /*break*/, 9];
                     return [4 /*yield*/, runAggregateWithRepair(fallbackPipeline, 'date_field_fallback')];
                 case 8:
-                    fallbackAggregate = _s.sent();
+                    fallbackAggregate = _t.sent();
                     fallbackDocs = fallbackAggregate.documents;
                     if (fallbackDocs.length) {
                         documents = fallbackDocs;
                         executedPipeline = fallbackAggregate.pipeline;
                         fallbackMeta.dateField.used = true;
                     }
-                    _s.label = 9;
+                    _t.label = 9;
                 case 9:
                     if (!(!documents.length && dateField !== 'createdAt')) return [3 /*break*/, 11];
                     createdFallback = { from: dateField, to: 'createdAt', attempted: true, used: false };
@@ -3828,14 +4006,14 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (!!containsForbiddenMongoOperators(createdPipeline)) return [3 /*break*/, 11];
                     return [4 /*yield*/, runAggregateWithRepair(createdPipeline, 'date_field_created_at')];
                 case 10:
-                    createdAggregate = _s.sent();
+                    createdAggregate = _t.sent();
                     createdDocs = createdAggregate.documents;
                     if (createdDocs.length) {
                         documents = createdDocs;
                         executedPipeline = createdAggregate.pipeline;
                         fallbackMeta.dateFieldCreatedAt.used = true;
                     }
-                    _s.label = 11;
+                    _t.label = 11;
                 case 11:
                     if (!!documents.length) return [3 /*break*/, 13];
                     expanded = expandAggregateDateMatchFallback(executedPipeline, schemaFields);
@@ -3848,14 +4026,14 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (!!containsForbiddenMongoOperators(expanded.pipeline)) return [3 /*break*/, 13];
                     return [4 /*yield*/, runAggregateWithRepair(expanded.pipeline, 'date_fields_expanded')];
                 case 12:
-                    expandedAggregate = _s.sent();
+                    expandedAggregate = _t.sent();
                     expandedDocs = expandedAggregate.documents;
                     if (expandedDocs.length) {
                         documents = expandedDocs;
                         executedPipeline = expandedAggregate.pipeline;
                         fallbackMeta.dateFieldsExpanded.used = true;
                     }
-                    _s.label = 13;
+                    _t.label = 13;
                 case 13:
                     if (!!documents.length) return [3 /*break*/, 17];
                     completionFallback = resolveAggregateCompletionFallback(executedPipeline);
@@ -3872,14 +4050,14 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (!(fallbackPipeline.length && !containsForbiddenMongoOperators(fallbackPipeline))) return [3 /*break*/, 15];
                     return [4 /*yield*/, runAggregateWithRepair(fallbackPipeline, 'completion_add_fields')];
                 case 14:
-                    fallbackAggregate = _s.sent();
+                    fallbackAggregate = _t.sent();
                     fallbackDocs = fallbackAggregate.documents;
                     if (fallbackDocs.length) {
                         documents = fallbackDocs;
                         executedPipeline = fallbackAggregate.pipeline;
                         fallbackMeta.completion.used = true;
                     }
-                    _s.label = 15;
+                    _t.label = 15;
                 case 15:
                     if (!!documents.length) return [3 /*break*/, 17];
                     completionExprFallback = resolveAggregateCompletionExprFallback(executedPipeline);
@@ -3896,23 +4074,23 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (!(fallbackPipeline.length && !containsForbiddenMongoOperators(fallbackPipeline))) return [3 /*break*/, 17];
                     return [4 /*yield*/, runAggregateWithRepair(fallbackPipeline, 'completion_expr')];
                 case 16:
-                    fallbackAggregate = _s.sent();
+                    fallbackAggregate = _t.sent();
                     fallbackDocs = fallbackAggregate.documents;
                     if (fallbackDocs.length) {
                         documents = fallbackDocs;
                         executedPipeline = fallbackAggregate.pipeline;
                         fallbackMeta.completion.used = true;
                     }
-                    _s.label = 17;
+                    _t.label = 17;
                 case 17:
                     if (!(documents.length <= 1)) return [3 /*break*/, 26];
                     unwindFallback = resolveAggregateUnwindFallback(executedPipeline);
                     if (!unwindFallback) return [3 /*break*/, 26];
                     fallbackMeta.unwind = { path: unwindFallback.path, attempted: true, used: false };
                     shouldUnwind = false;
-                    _s.label = 18;
+                    _t.label = 18;
                 case 18:
-                    _s.trys.push([18, 21, , 22]);
+                    _t.trys.push([18, 21, , 22]);
                     _c = probeDocs;
                     if (_c) return [3 /*break*/, 20];
                     return [4 /*yield*/, fetchAssistantProbeDocs({
@@ -3921,17 +4099,18 @@ function executeAiAssistantMongoAggregate(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveAggregateMaxTimeMS()
                         })];
                 case 19:
-                    _c = (_s.sent());
-                    _s.label = 20;
+                    _c = (_t.sent());
+                    _t.label = 20;
                 case 20:
                     probeDocs = _c;
                     shouldUnwind = probeDocs.length ? hasArrayValueAtPath(probeDocs, unwindFallback.path) : false;
                     return [3 /*break*/, 22];
                 case 21:
-                    _d = _s.sent();
+                    _d = _t.sent();
                     shouldUnwind = false;
                     return [3 /*break*/, 22];
                 case 22:
@@ -3940,18 +4119,18 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (!!containsForbiddenMongoOperators(fallbackPipeline)) return [3 /*break*/, 24];
                     return [4 /*yield*/, runAggregateWithRepair(fallbackPipeline, 'unwind')];
                 case 23:
-                    fallbackAggregate = _s.sent();
+                    fallbackAggregate = _t.sent();
                     fallbackDocs = fallbackAggregate.documents;
                     if (fallbackDocs.length > documents.length) {
                         documents = fallbackDocs;
                         executedPipeline = fallbackAggregate.pipeline;
                         fallbackMeta.unwind.used = true;
                     }
-                    _s.label = 24;
+                    _t.label = 24;
                 case 24: return [3 /*break*/, 26];
                 case 25:
                     fallbackMeta.unwind.skipped = 'not_array';
-                    _s.label = 26;
+                    _t.label = 26;
                 case 26:
                     if (!!documents.length) return [3 /*break*/, 28];
                     nameFallback = resolveAggregateNameMatchFallback(executedPipeline);
@@ -3966,14 +4145,14 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (!(fallbackPipeline.length && !containsForbiddenMongoOperators(fallbackPipeline))) return [3 /*break*/, 28];
                     return [4 /*yield*/, runAggregateWithRepair(fallbackPipeline, 'name_match')];
                 case 27:
-                    fallbackAggregate = _s.sent();
+                    fallbackAggregate = _t.sent();
                     fallbackDocs = fallbackAggregate.documents;
                     if (fallbackDocs.length) {
                         documents = fallbackDocs;
                         executedPipeline = fallbackAggregate.pipeline;
                         fallbackMeta.nameMatch.used = true;
                     }
-                    _s.label = 28;
+                    _t.label = 28;
                 case 28:
                     if (!!documents.length) return [3 /*break*/, 34];
                     _e = probeDocs;
@@ -3984,17 +4163,18 @@ function executeAiAssistantMongoAggregate(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveAggregateMaxTimeMS()
                         })];
                 case 29:
-                    _e = (_s.sent());
-                    _s.label = 30;
+                    _e = (_t.sent());
+                    _t.label = 30;
                 case 30:
                     probeDocs = _e;
                     _loop_1 = function (i) {
                         var stage, chemicalLookup, rewrittenPipeline, fallbackAggregate, fallbackDocs;
-                        return __generator(this, function (_t) {
-                            switch (_t.label) {
+                        return __generator(this, function (_u) {
+                            switch (_u.label) {
                                 case 0:
                                     stage = executedPipeline[i];
                                     if (!stage || typeof stage !== 'object' || !stage.$match || typeof stage.$match !== 'object') {
@@ -4010,7 +4190,7 @@ function executeAiAssistantMongoAggregate(payload, context) {
                                             probeDocs: probeDocs || undefined
                                         })];
                                 case 1:
-                                    chemicalLookup = _t.sent();
+                                    chemicalLookup = _u.sent();
                                     if (!chemicalLookup) {
                                         return [2 /*return*/, "continue"];
                                     }
@@ -4019,7 +4199,7 @@ function executeAiAssistantMongoAggregate(payload, context) {
                                     if (!!containsForbiddenMongoOperators(rewrittenPipeline)) return [3 /*break*/, 3];
                                     return [4 /*yield*/, runAggregateWithRepair(rewrittenPipeline, 'chemical_lookup')];
                                 case 2:
-                                    fallbackAggregate = _t.sent();
+                                    fallbackAggregate = _u.sent();
                                     fallbackDocs = fallbackAggregate.documents;
                                     if (fallbackDocs.length) {
                                         documents = fallbackDocs;
@@ -4027,21 +4207,21 @@ function executeAiAssistantMongoAggregate(payload, context) {
                                         fallbackMeta.chemicalLookup.used = true;
                                         return [2 /*return*/, "break"];
                                     }
-                                    _t.label = 3;
+                                    _u.label = 3;
                                 case 3: return [2 /*return*/];
                             }
                         });
                     };
                     i = 0;
-                    _s.label = 31;
+                    _t.label = 31;
                 case 31:
                     if (!(i < (executedPipeline || []).length)) return [3 /*break*/, 34];
                     return [5 /*yield**/, _loop_1(i)];
                 case 32:
-                    state_1 = _s.sent();
+                    state_1 = _t.sent();
                     if (state_1 === "break")
                         return [3 /*break*/, 34];
-                    _s.label = 33;
+                    _t.label = 33;
                 case 33:
                     i += 1;
                     return [3 /*break*/, 31];
@@ -4064,11 +4244,12 @@ function executeAiAssistantMongoAggregate(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveAggregateMaxTimeMS()
                         })];
                 case 35:
-                    _f = (_s.sent());
-                    _s.label = 36;
+                    _f = (_t.sent());
+                    _t.label = 36;
                 case 36:
                     probeDocs = _f;
                     if (!probeDocs.length) return [3 /*break*/, 38];
@@ -4087,40 +4268,41 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (!!containsForbiddenMongoOperators(rewrittenPipeline)) return [3 /*break*/, 38];
                     return [4 /*yield*/, runAggregateWithRepair(rewrittenPipeline, 'query_field_aliases')];
                 case 37:
-                    fallbackAggregate = _s.sent();
+                    fallbackAggregate = _t.sent();
                     fallbackDocs = fallbackAggregate.documents;
                     if (fallbackDocs.length) {
                         documents = fallbackDocs;
                         executedPipeline = fallbackAggregate.pipeline;
                         fallbackMeta.queryFieldAliases.used = true;
                     }
-                    _s.label = 38;
+                    _t.label = 38;
                 case 38:
                     if (!!documents.length) return [3 /*break*/, 42];
                     _loop_2 = function (i) {
-                        var stage, _u, idLookup, rewrittenPipeline, fallbackAggregate, fallbackDocs;
-                        return __generator(this, function (_v) {
-                            switch (_v.label) {
+                        var stage, _v, idLookup, rewrittenPipeline, fallbackAggregate, fallbackDocs;
+                        return __generator(this, function (_w) {
+                            switch (_w.label) {
                                 case 0:
                                     stage = executedPipeline[i];
                                     if (!stage || typeof stage !== 'object' || !stage.$match || typeof stage.$match !== 'object') {
                                         return [2 /*return*/, "continue"];
                                     }
-                                    _u = probeDocs;
-                                    if (_u) return [3 /*break*/, 2];
+                                    _v = probeDocs;
+                                    if (_v) return [3 /*break*/, 2];
                                     return [4 /*yield*/, fetchAssistantProbeDocs({
                                             db: db,
                                             collection: collection,
                                             idClient: normalizedClient,
                                             idCustomer: customerId,
                                             isSuperAdmin: isSuperAdmin,
-                                            includeClientScope: shouldScopeByClient
+                                            includeClientScope: shouldScopeByClient,
+                                            maxTimeMS: resolveAggregateMaxTimeMS()
                                         })];
                                 case 1:
-                                    _u = (_v.sent());
-                                    _v.label = 2;
+                                    _v = (_w.sent());
+                                    _w.label = 2;
                                 case 2:
-                                    probeDocs = _u;
+                                    probeDocs = _v;
                                     return [4 /*yield*/, applyIdLookupFallbackToQuery({
                                             query: stage.$match,
                                             db: db,
@@ -4131,7 +4313,7 @@ function executeAiAssistantMongoAggregate(payload, context) {
                                             probeDocs: probeDocs || undefined
                                         })];
                                 case 3:
-                                    idLookup = _v.sent();
+                                    idLookup = _w.sent();
                                     if (!idLookup) {
                                         return [2 /*return*/, "continue"];
                                     }
@@ -4140,7 +4322,7 @@ function executeAiAssistantMongoAggregate(payload, context) {
                                     if (!!containsForbiddenMongoOperators(rewrittenPipeline)) return [3 /*break*/, 5];
                                     return [4 /*yield*/, runAggregateWithRepair(rewrittenPipeline, 'id_lookup')];
                                 case 4:
-                                    fallbackAggregate = _v.sent();
+                                    fallbackAggregate = _w.sent();
                                     fallbackDocs = fallbackAggregate.documents;
                                     if (fallbackDocs.length) {
                                         documents = fallbackDocs;
@@ -4148,26 +4330,26 @@ function executeAiAssistantMongoAggregate(payload, context) {
                                         fallbackMeta.idLookup.used = true;
                                         return [2 /*return*/, "break"];
                                     }
-                                    _v.label = 5;
+                                    _w.label = 5;
                                 case 5: return [2 /*return*/];
                             }
                         });
                     };
                     i = 0;
-                    _s.label = 39;
+                    _t.label = 39;
                 case 39:
                     if (!(i < (executedPipeline || []).length)) return [3 /*break*/, 42];
                     return [5 /*yield**/, _loop_2(i)];
                 case 40:
-                    state_2 = _s.sent();
+                    state_2 = _t.sent();
                     if (state_2 === "break")
                         return [3 /*break*/, 42];
-                    _s.label = 41;
+                    _t.label = 41;
                 case 41:
                     i += 1;
                     return [3 /*break*/, 39];
                 case 42:
-                    if (!!documents.length) return [3 /*break*/, 53];
+                    if (!(!documents.length && hasAggregateRuntimeBudget('relationship_retry'))) return [3 /*break*/, 53];
                     return [4 /*yield*/, resolveAssistantNestedAggregateFallbacks({
                             db: db,
                             dbName: dbName,
@@ -4176,7 +4358,7 @@ function executeAiAssistantMongoAggregate(payload, context) {
                             triedCollections: triedCollections
                         })];
                 case 43:
-                    nestedFallbacks = _s.sent();
+                    nestedFallbacks = _t.sent();
                     if (!nestedFallbacks.length) return [3 /*break*/, 53];
                     fallbackMeta.relationshipRetry = {
                         from: collection,
@@ -4189,11 +4371,11 @@ function executeAiAssistantMongoAggregate(payload, context) {
                         attempted: true,
                         used: false
                     };
-                    _s.label = 44;
+                    _t.label = 44;
                 case 44:
-                    _s.trys.push([44, 51, 52, 53]);
+                    _t.trys.push([44, 51, 52, 53]);
                     nestedFallbacks_1 = __values(nestedFallbacks), nestedFallbacks_1_1 = nestedFallbacks_1.next();
-                    _s.label = 45;
+                    _t.label = 45;
                 case 45:
                     if (!!nestedFallbacks_1_1.done) return [3 /*break*/, 50];
                     nestedFallback = nestedFallbacks_1_1.value;
@@ -4202,12 +4384,12 @@ function executeAiAssistantMongoAggregate(payload, context) {
                             rootCollection: retryRootCollection,
                             disableCrossCollectionRetry: true
                         }) });
-                    _s.label = 46;
+                    _t.label = 46;
                 case 46:
-                    _s.trys.push([46, 48, , 49]);
+                    _t.trys.push([46, 48, , 49]);
                     return [4 /*yield*/, executeAiAssistantMongoAggregate(fallbackPayload, context)];
                 case 47:
-                    fallbackResult = _s.sent();
+                    fallbackResult = _t.sent();
                     if (Array.isArray(fallbackResult === null || fallbackResult === void 0 ? void 0 : fallbackResult.documents) && fallbackResult.documents.length) {
                         if (canViewDebug && (fallbackResult === null || fallbackResult === void 0 ? void 0 : fallbackResult.debug) && typeof fallbackResult.debug === 'object') {
                             existingFallbacks = fallbackResult.debug.fallbacks && typeof fallbackResult.debug.fallbacks === 'object'
@@ -4226,7 +4408,7 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     }
                     return [3 /*break*/, 49];
                 case 48:
-                    error_7 = _s.sent();
+                    error_7 = _t.sent();
                     errorMessage = normalizeOptionalString(error_7 === null || error_7 === void 0 ? void 0 : error_7.message) || 'Unknown error';
                     existingErrors = Array.isArray(fallbackMeta.relationshipRetry.errors)
                         ? fallbackMeta.relationshipRetry.errors
@@ -4242,20 +4424,20 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     return [3 /*break*/, 45];
                 case 50: return [3 /*break*/, 53];
                 case 51:
-                    e_2_1 = _s.sent();
+                    e_2_1 = _t.sent();
                     e_2 = { error: e_2_1 };
                     return [3 /*break*/, 53];
                 case 52:
                     try {
-                        if (nestedFallbacks_1_1 && !nestedFallbacks_1_1.done && (_o = nestedFallbacks_1.return)) _o.call(nestedFallbacks_1);
+                        if (nestedFallbacks_1_1 && !nestedFallbacks_1_1.done && (_p = nestedFallbacks_1.return)) _p.call(nestedFallbacks_1);
                     }
                     finally { if (e_2) throw e_2.error; }
                     return [7 /*endfinally*/];
                 case 53:
-                    if (!!documents.length) return [3 /*break*/, 56];
+                    if (!(!documents.length && hasAggregateRuntimeBudget('report_fallback'))) return [3 /*break*/, 56];
                     return [4 /*yield*/, resolveBaseCollectionFromReport(db, dbName, collection)];
                 case 54:
-                    baseCollection = _s.sent();
+                    baseCollection = _t.sent();
                     if (!(baseCollection && baseCollection !== collection)) return [3 /*break*/, 56];
                     addAssistantCollectionToTriedSet(triedCollections, baseCollection);
                     fallbackPayload = __assign(__assign({}, input), { collection: baseCollection, __assistantRetryState: buildAssistantRetryState(retryState, triedCollections, {
@@ -4263,7 +4445,7 @@ function executeAiAssistantMongoAggregate(payload, context) {
                         }) });
                     return [4 /*yield*/, executeAiAssistantMongoAggregate(fallbackPayload, context)];
                 case 55:
-                    fallbackResult = _s.sent();
+                    fallbackResult = _t.sent();
                     if (Array.isArray(fallbackResult === null || fallbackResult === void 0 ? void 0 : fallbackResult.documents) && fallbackResult.documents.length) {
                         if (canViewDebug && (fallbackResult === null || fallbackResult === void 0 ? void 0 : fallbackResult.debug) && typeof fallbackResult.debug === 'object') {
                             existingFallbacks = fallbackResult.debug.fallbacks && typeof fallbackResult.debug.fallbacks === 'object'
@@ -4280,15 +4462,15 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     if (canViewDebug) {
                         fallbackMeta.reportFallback = { from: collection, to: baseCollection, attempted: true, used: false };
                     }
-                    _s.label = 56;
+                    _t.label = 56;
                 case 56:
-                    if (!(!documents.length && crossCollectionRetryEnabled)) return [3 /*break*/, 67];
+                    if (!(!documents.length && crossCollectionRetryEnabled && hasAggregateRuntimeBudget('collection_retry'))) return [3 /*break*/, 67];
                     fallbackAcceptanceOptions = buildAssistantAggregateFallbackAcceptanceOptions(executedPipeline || []);
                     return [4 /*yield*/, resolveAssistantAvailableCrossCollectionFallbacks(db, dbName, collection, triedCollections, {
                             fieldHints: collectAggregateReferencedFieldPaths(executedPipeline || [])
                         })];
                 case 57:
-                    crossCollectionCandidates = _s.sent();
+                    crossCollectionCandidates = _t.sent();
                     if (!crossCollectionCandidates.length) return [3 /*break*/, 67];
                     fallbackMeta.collectionRetry = {
                         from: collection,
@@ -4297,11 +4479,11 @@ function executeAiAssistantMongoAggregate(payload, context) {
                         attempted: true,
                         used: false
                     };
-                    _s.label = 58;
+                    _t.label = 58;
                 case 58:
-                    _s.trys.push([58, 65, 66, 67]);
+                    _t.trys.push([58, 65, 66, 67]);
                     crossCollectionCandidates_2 = __values(crossCollectionCandidates), crossCollectionCandidates_2_1 = crossCollectionCandidates_2.next();
-                    _s.label = 59;
+                    _t.label = 59;
                 case 59:
                     if (!!crossCollectionCandidates_2_1.done) return [3 /*break*/, 64];
                     candidateCollection = crossCollectionCandidates_2_1.value;
@@ -4310,12 +4492,12 @@ function executeAiAssistantMongoAggregate(payload, context) {
                             rootCollection: retryRootCollection,
                             disableCrossCollectionRetry: true
                         }) });
-                    _s.label = 60;
+                    _t.label = 60;
                 case 60:
-                    _s.trys.push([60, 62, , 63]);
+                    _t.trys.push([60, 62, , 63]);
                     return [4 /*yield*/, executeAiAssistantMongoAggregate(fallbackPayload, context)];
                 case 61:
-                    fallbackResult = _s.sent();
+                    fallbackResult = _t.sent();
                     fallbackDocs = Array.isArray(fallbackResult === null || fallbackResult === void 0 ? void 0 : fallbackResult.documents) ? fallbackResult.documents : [];
                     fallbackQuality = evaluateAssistantFallbackDocumentsQuality(fallbackDocs, fallbackAcceptanceOptions);
                     if (fallbackDocs.length && fallbackQuality.accepted) {
@@ -4350,7 +4532,7 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     }
                     return [3 /*break*/, 63];
                 case 62:
-                    error_8 = _s.sent();
+                    error_8 = _t.sent();
                     if (fallbackMeta.collectionRetry && typeof fallbackMeta.collectionRetry === 'object') {
                         existingErrors = Array.isArray(fallbackMeta.collectionRetry.errors)
                             ? fallbackMeta.collectionRetry.errors
@@ -4367,17 +4549,17 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     return [3 /*break*/, 59];
                 case 64: return [3 /*break*/, 67];
                 case 65:
-                    e_3_1 = _s.sent();
+                    e_3_1 = _t.sent();
                     e_3 = { error: e_3_1 };
                     return [3 /*break*/, 67];
                 case 66:
                     try {
-                        if (crossCollectionCandidates_2_1 && !crossCollectionCandidates_2_1.done && (_p = crossCollectionCandidates_2.return)) _p.call(crossCollectionCandidates_2);
+                        if (crossCollectionCandidates_2_1 && !crossCollectionCandidates_2_1.done && (_q = crossCollectionCandidates_2.return)) _q.call(crossCollectionCandidates_2);
                     }
                     finally { if (e_3) throw e_3.error; }
                     return [7 /*endfinally*/];
                 case 67:
-                    if (!(!documents.length && canViewDebug)) return [3 /*break*/, 84];
+                    if (!(!documents.length && canViewDebug && hasAggregateRuntimeBudget('zero_diagnostics', 20))) return [3 /*break*/, 84];
                     matchStages = (executedPipeline || []).filter(function (stage) { return stage && typeof stage === 'object' && stage.$match && typeof stage.$match === 'object'; });
                     diagnostics = {};
                     if (!matchStages.length) return [3 /*break*/, 84];
@@ -4387,34 +4569,31 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     dateFields = collectMatchFieldsByCondition(combinedMatch, function (_field, condition) { return isDateCondition(condition); });
                     diagnostics.nameFields = nameFields.length ? nameFields : undefined;
                     diagnostics.dateFields = dateFields.length ? dateFields : undefined;
-                    _s.label = 68;
+                    _t.label = 68;
                 case 68:
-                    _s.trys.push([68, 73, , 74]);
+                    _t.trys.push([68, 73, , 74]);
+                    countOptions = buildAggregateCountOptions();
                     if (!nameFields.length) return [3 /*break*/, 70];
                     queryNoName = stripMatchFields(combinedMatch, nameFields);
                     _g = diagnostics;
-                    return [4 /*yield*/, db.collection(collection).countDocuments(queryNoName, {
-                            readPreference: AI_ASSISTANT_READ_PREFERENCE
-                        })];
+                    return [4 /*yield*/, db.collection(collection).countDocuments(queryNoName, countOptions)];
                 case 69:
-                    _g.recentCount = _s.sent();
-                    _s.label = 70;
+                    _g.recentCount = _t.sent();
+                    _t.label = 70;
                 case 70:
                     if (!dateFields.length) return [3 /*break*/, 72];
                     queryNoDate = stripMatchFields(combinedMatch, dateFields);
                     _h = diagnostics;
-                    return [4 /*yield*/, db.collection(collection).countDocuments(queryNoDate, {
-                            readPreference: AI_ASSISTANT_READ_PREFERENCE
-                        })];
+                    return [4 /*yield*/, db.collection(collection).countDocuments(queryNoDate, countOptions)];
                 case 71:
-                    _h.nameMatchCount = _s.sent();
-                    _s.label = 72;
+                    _h.nameMatchCount = _t.sent();
+                    _t.label = 72;
                 case 72: return [3 /*break*/, 74];
                 case 73:
-                    _j = _s.sent();
+                    _j = _t.sent();
                     return [3 /*break*/, 74];
                 case 74:
-                    _s.trys.push([74, 77, , 78]);
+                    _t.trys.push([74, 77, , 78]);
                     _k = probeDocs;
                     if (_k) return [3 /*break*/, 76];
                     return [4 /*yield*/, fetchAssistantProbeDocs({
@@ -4423,11 +4602,12 @@ function executeAiAssistantMongoAggregate(payload, context) {
                             idClient: normalizedClient,
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin,
-                            includeClientScope: shouldScopeByClient
+                            includeClientScope: shouldScopeByClient,
+                            maxTimeMS: resolveAggregateMaxTimeMS()
                         })];
                 case 75:
-                    _k = (_s.sent());
-                    _s.label = 76;
+                    _k = (_t.sent());
+                    _t.label = 76;
                 case 76:
                     probeDocs = _k;
                     if (probeDocs.length && nameFields.length) {
@@ -4435,42 +4615,50 @@ function executeAiAssistantMongoAggregate(payload, context) {
                     }
                     return [3 /*break*/, 78];
                 case 77:
-                    _l = _s.sent();
+                    _l = _t.sent();
                     return [3 /*break*/, 78];
                 case 78:
-                    _s.trys.push([78, 82, , 83]);
+                    _t.trys.push([78, 82, , 83]);
                     return [4 /*yield*/, listAssistantCollections(db, dbName)];
                 case 79:
-                    allCollections = _s.sent();
+                    allCollections = _t.sent();
                     base = stripVersionSuffix(collection.startsWith('report-') ? collection.slice('report-'.length) : collection);
                     alt = collection.startsWith('report-') ? base : "report-".concat(base);
                     if (!(alt && alt !== collection && allCollections.includes(alt))) return [3 /*break*/, 81];
-                    return [4 /*yield*/, db.collection(alt).countDocuments({}, { readPreference: AI_ASSISTANT_READ_PREFERENCE })];
+                    return [4 /*yield*/, db.collection(alt).countDocuments({}, buildAggregateCountOptions())];
                 case 80:
-                    altCount = _s.sent();
+                    altCount = _t.sent();
                     diagnostics.alternateCollection = alt;
                     diagnostics.alternateCollectionCount = altCount;
-                    _s.label = 81;
+                    _t.label = 81;
                 case 81: return [3 /*break*/, 83];
                 case 82:
-                    _m = _s.sent();
+                    _m = _t.sent();
                     return [3 /*break*/, 83];
                 case 83:
                     fallbackMeta.zeroDiagnostics = diagnostics;
-                    _s.label = 84;
-                case 84: return [4 /*yield*/, verifyAssistantAggregateReliability({
-                        db: db,
-                        collection: collection,
-                        pipeline: executedPipeline,
-                        documents: documents,
-                        aggregateOptions: aggregateOptions
-                    })];
+                    _t.label = 84;
+                case 84:
+                    if (!hasAggregateRuntimeBudget('verification', 40)) return [3 /*break*/, 86];
+                    return [4 /*yield*/, verifyAssistantAggregateReliability({
+                            db: db,
+                            collection: collection,
+                            pipeline: executedPipeline,
+                            documents: documents,
+                            aggregateOptions: buildAggregateOptions()
+                        })];
                 case 85:
-                    verification = _s.sent();
+                    _o = _t.sent();
+                    return [3 /*break*/, 87];
+                case 86:
+                    _o = null;
+                    _t.label = 87;
+                case 87:
+                    verification = _o;
                     sanitizedDocuments = isSuperAdmin
                         ? documents
                         : documents.map(function (doc) { return redactSensitiveFields((0, common_1.deepCopy)(doc)); });
-                    includeIds = ((_r = input.options) === null || _r === void 0 ? void 0 : _r.includeIds) === true;
+                    includeIds = ((_s = input.options) === null || _s === void 0 ? void 0 : _s.includeIds) === true;
                     displayDocs = sanitizedDocuments.map(function (doc) { return flattenForTable(doc, {
                         includeGroupFromId: true,
                         includeIds: true
@@ -4484,8 +4672,8 @@ function executeAiAssistantMongoAggregate(payload, context) {
                             idCustomer: customerId,
                             isSuperAdmin: isSuperAdmin
                         })];
-                case 86:
-                    idLookupDisplay = _s.sent();
+                case 88:
+                    idLookupDisplay = _t.sent();
                     if (idLookupDisplay === null || idLookupDisplay === void 0 ? void 0 : idLookupDisplay.docs) {
                         displayDocs = idLookupDisplay.docs;
                     }
@@ -7930,6 +8118,9 @@ function buildAssistantCodexErrorMessage(error) {
     }
     return 'I ran into an internal error while preparing that response. Please try again.';
 }
+function buildAssistantRunBudgetExceededMessage() {
+    return 'I could not finish this request within the assistant time budget. Please narrow the date range or ask for a smaller slice of data and retry.';
+}
 function isAssistantIdField(key) {
     var normalized = String(key || '').trim().toLowerCase();
     if (!normalized) {
@@ -16503,13 +16694,13 @@ function buildAssistantProbeQuery(collection, idClient, idCustomer, isSuperAdmin
     return applyCustomerScopeFilter(query, collection, idCustomer, isSuperAdmin);
 }
 function fetchAssistantProbeDocs(params) {
-    var db = params.db, collection = params.collection, idClient = params.idClient, idCustomer = params.idCustomer, isSuperAdmin = params.isSuperAdmin, includeClientScope = params.includeClientScope;
+    var db = params.db, collection = params.collection, idClient = params.idClient, idCustomer = params.idCustomer, isSuperAdmin = params.isSuperAdmin, includeClientScope = params.includeClientScope, maxTimeMS = params.maxTimeMS;
     var probeQuery = buildAssistantProbeQuery(collection, idClient, idCustomer, isSuperAdmin, includeClientScope);
+    var resolvedMaxTimeMS = resolveAssistantEffectiveMongoMaxTimeMs({
+        requestedMaxTimeMS: maxTimeMS
+    });
     return db.collection(collection)
-        .find(probeQuery, {
-        limit: AI_ASSISTANT_PROBE_LIMIT,
-        readPreference: AI_ASSISTANT_READ_PREFERENCE
-    })
+        .find(probeQuery, __assign({ limit: AI_ASSISTANT_PROBE_LIMIT, readPreference: AI_ASSISTANT_READ_PREFERENCE }, (resolvedMaxTimeMS ? { maxTimeMS: resolvedMaxTimeMS } : {})))
         .toArray();
 }
 var AI_ASSISTANT_PERMISSION_ACTION_TOKENS = new Set([
@@ -16778,6 +16969,84 @@ function resolveCodexTimeoutMs() {
     }
     return DEFAULT_CODEX_TIMEOUT_MS;
 }
+function resolveAssistantFastModeEnabled() {
+    var config = resolveio_server_app_1.ResolveIOServer.getServerConfig() || {};
+    var raw = normalizeOptionalBoolean(config['AI_ASSISTANT_FAST_MODE_ENABLED']
+        || process.env.AI_ASSISTANT_FAST_MODE_ENABLED);
+    return raw === undefined ? true : raw === true;
+}
+function resolveAssistantRunBudgetMs() {
+    var config = resolveio_server_app_1.ResolveIOServer.getServerConfig() || {};
+    var raw = normalizeOptionalNumber(config['AI_ASSISTANT_RUN_BUDGET_MS']
+        || process.env.AI_ASSISTANT_RUN_BUDGET_MS);
+    if (raw && raw > 0) {
+        return Math.max((0, common_1.round)(raw), 5000);
+    }
+    return DEFAULT_AI_ASSISTANT_RUN_BUDGET_MS;
+}
+function resolveAssistantMongoDefaultMaxTimeMs() {
+    var config = resolveio_server_app_1.ResolveIOServer.getServerConfig() || {};
+    var raw = normalizeOptionalNumber(config['AI_ASSISTANT_MONGO_MAX_TIME_MS']
+        || process.env.AI_ASSISTANT_MONGO_MAX_TIME_MS);
+    if (raw && raw > 0) {
+        return Math.max((0, common_1.round)(raw), 250);
+    }
+    return DEFAULT_AI_ASSISTANT_MONGO_MAX_TIME_MS;
+}
+function normalizeAssistantRuntimeControl(value) {
+    if (!value || typeof value !== 'object') {
+        return null;
+    }
+    var deadlineRaw = normalizeOptionalNumber(value.deadlineMs);
+    var maxTimeRaw = normalizeOptionalNumber(value.maxTimeMS);
+    var fastModeRaw = normalizeOptionalBoolean(value.fastMode);
+    var next = {};
+    if (deadlineRaw && deadlineRaw > 0) {
+        next.deadlineMs = (0, common_1.round)(deadlineRaw);
+    }
+    if (maxTimeRaw && maxTimeRaw > 0) {
+        next.maxTimeMS = (0, common_1.round)(maxTimeRaw);
+    }
+    if (typeof fastModeRaw === 'boolean') {
+        next.fastMode = fastModeRaw;
+    }
+    return Object.keys(next).length ? next : null;
+}
+function resolveAssistantRuntimeRemainingMs(runtime) {
+    var deadline = normalizeOptionalNumber(runtime === null || runtime === void 0 ? void 0 : runtime.deadlineMs);
+    if (!deadline || deadline <= 0) {
+        return null;
+    }
+    return Math.max((0, common_1.round)(deadline) - Date.now(), 0);
+}
+function hasAssistantRuntimeBudget(runtime, minimumMs) {
+    if (minimumMs === void 0) { minimumMs = AI_ASSISTANT_MONGO_MIN_STAGE_BUDGET_MS; }
+    var remaining = resolveAssistantRuntimeRemainingMs(runtime);
+    if (remaining === null) {
+        return true;
+    }
+    return remaining >= Math.max((0, common_1.round)(minimumMs || 0), 1);
+}
+function resolveAssistantEffectiveMongoMaxTimeMs(params) {
+    var _a;
+    var requested = normalizeOptionalNumber(params === null || params === void 0 ? void 0 : params.requestedMaxTimeMS);
+    var runtimeMax = normalizeOptionalNumber((_a = params === null || params === void 0 ? void 0 : params.runtime) === null || _a === void 0 ? void 0 : _a.maxTimeMS);
+    var configured = resolveAssistantMongoDefaultMaxTimeMs();
+    var resolved = requested && requested > 0
+        ? (0, common_1.round)(requested)
+        : configured;
+    if (runtimeMax && runtimeMax > 0) {
+        resolved = Math.min(resolved, (0, common_1.round)(runtimeMax));
+    }
+    var remaining = resolveAssistantRuntimeRemainingMs((params === null || params === void 0 ? void 0 : params.runtime) || null);
+    if (remaining !== null) {
+        if (remaining < AI_ASSISTANT_MONGO_MIN_STAGE_BUDGET_MS) {
+            return undefined;
+        }
+        resolved = Math.min(resolved, remaining);
+    }
+    return resolved > 0 ? Math.max((0, common_1.round)(resolved), 50) : undefined;
+}
 function resolveCodexWorkerThreadEnabled() {
     var config = resolveio_server_app_1.ResolveIOServer.getServerConfig() || {};
     var raw = normalizeOptionalString(config['AI_ASSISTANT_CODEX_USE_WORKER_THREAD']
@@ -16892,7 +17161,7 @@ function resolveCodexSettings(options) {
     var serverConfig = resolveio_server_app_1.ResolveIOServer.getServerConfig() || {};
     var apiKey = (serverConfig['OPENAI_API_KEY'] || process.env.OPENAI_API_KEY || '').trim();
     if (!apiKey) {
-        throw new Error('OpenAI API key missing. Add OPENAI_API_KEY to server config.');
+        throw new Error('AI API key missing. Add an AI API key to server config.');
     }
     var model = normalizeOptionalString(options === null || options === void 0 ? void 0 : options.model) || resolveCodexModel();
     var fallbackModels = resolveCodexFallbackModels({ fallbackModels: options === null || options === void 0 ? void 0 : options.fallbackModels }, model);
@@ -17001,7 +17270,7 @@ function waitForCodexWorkerMessage(worker, streamStatusHandler) {
                     if (e_44) throw e_44.error;
                     return [7 /*endfinally*/];
                 case 10: return [7 /*endfinally*/];
-                case 11: throw new CodexWorkerBootstrapError('Codex worker exited before completing.');
+                case 11: throw new CodexWorkerBootstrapError('AI worker exited before completing.');
             }
         });
     });
@@ -17040,7 +17309,7 @@ function runCodexInWorkerThread(prompt, runOptions, config, streamStatusHandler)
                     if (!(error_10 instanceof CodexWorkerBootstrapError)) {
                         throw error_10;
                     }
-                    console.error('Codex worker bootstrap failed, falling back to in-process run.', error_10);
+                    console.error('AI worker bootstrap failed, falling back to in-process run.', error_10);
                     codexClient = getAssistantCodexClient(config);
                     return [4 /*yield*/, codexClient.run(prompt, streamedOptions)];
                 case 10: return [2 /*return*/, _a.sent()];
@@ -17069,7 +17338,7 @@ function runCodexInWorkerThreadInternal(workerPath, prompt, runOptions, config,
                         });
                     }
                     catch (error) {
-                        wrapped = new CodexWorkerBootstrapError('Failed to start codex worker.');
+                        wrapped = new CodexWorkerBootstrapError('Failed to start AI worker.');
                         if (error instanceof Error && error.stack) {
                             wrapped.stack = error.stack;
                         }
@@ -17150,28 +17419,28 @@ function runCodexInWorkerThreadInternal(workerPath, prompt, runOptions, config,
                     if (outcome.message && outcome.message.ok) {
                         return [2 /*return*/, outcome.message.result || ''];
                     }
-                    throw new Error(((_a = outcome.message) === null || _a === void 0 ? void 0 : _a.error) || 'Codex worker failed.');
+                    throw new Error(((_a = outcome.message) === null || _a === void 0 ? void 0 : _a.error) || 'AI worker failed.');
                 case 6:
                     if (!(outcome.type === 'timeout')) return [3 /*break*/, 8];
                     return [4 /*yield*/, terminateWorker(worker)];
                 case 7:
                     _c.sent();
-                    throw new CodexWorkerBootstrapError('Codex worker timed out.');
+                    throw new CodexWorkerBootstrapError('AI worker timed out.');
                 case 8:
                     if (!(outcome.type === 'error')) return [3 /*break*/, 10];
                     return [4 /*yield*/, terminateWorker(worker)];
                 case 9:
                     _c.sent();
-                    wrapped = new CodexWorkerBootstrapError('Codex worker errored before completing.');
+                    wrapped = new CodexWorkerBootstrapError('AI worker errored before completing.');
                     if ((_b = outcome.error) === null || _b === void 0 ? void 0 : _b.stack) {
                         wrapped.stack = outcome.error.stack;
                     }
                     throw wrapped;
                 case 10:
                     if (outcome.type === 'exit') {
-                        throw new CodexWorkerBootstrapError("Codex worker exited before completing (code ".concat(outcome.code, ")."));
+                        throw new CodexWorkerBootstrapError("AI worker exited before completing (code ".concat(outcome.code, ")."));
                     }
-                    throw new CodexWorkerBootstrapError('Codex worker aborted before completing.');
+                    throw new CodexWorkerBootstrapError('AI worker aborted before completing.');
             }
         });
     });
@@ -19959,22 +20228,6 @@ function evaluateAssistantGuardrails(message) {
     }
     return null;
 }
-function resolveOpenAISettings(config) {
-    var serverConfig = resolveio_server_app_1.ResolveIOServer.getServerConfig() || {};
-    var apiKey = (serverConfig['OPENAI_API_KEY'] || process.env.OPENAI_API_KEY || '').trim();
-    if (!apiKey) {
-        throw new Error('OpenAI API key missing. Add OPENAI_API_KEY to server config.');
-    }
-    return {
-        apiKey: apiKey,
-        baseUrl: (serverConfig['OPENAI_BASE_URL'] || process.env.OPENAI_BASE_URL || '').trim() || undefined,
-        model: normalizeOptionalString(config === null || config === void 0 ? void 0 : config.model) || (serverConfig['OPENAI_MODEL'] || process.env.OPENAI_MODEL || 'gpt-5.1').trim(),
-        temperature: normalizeOptionalNumber(config === null || config === void 0 ? void 0 : config.temperature),
-        maxTokens: normalizeOptionalNumber(config === null || config === void 0 ? void 0 : config.max_tokens),
-        maxRetries: normalizeOptionalNumber(serverConfig['OPENAI_MAX_RETRIES'] || process.env.OPENAI_MAX_RETRIES),
-        retryDelayMs: normalizeOptionalNumber(serverConfig['OPENAI_RETRY_DELAY_MS'] || process.env.OPENAI_RETRY_DELAY_MS)
-    };
-}
 function buildAiFormSystemPrompt() {
     return AI_FORM_PATCH_SYSTEM_PROMPT;
 }
@@ -20481,6 +20734,29 @@ function buildUserPrompt(template, inputText, attachmentText) {
     }
     return "".concat(inputText || '').concat(attachmentText).trim();
 }
+function buildCodexChatPrompt(messages) {
+    var systemParts = [];
+    var dialogParts = [];
+    (messages || []).forEach(function (message) {
+        var content = normalizeOptionalString(message === null || message === void 0 ? void 0 : message.content);
+        if (!content) {
+            return;
+        }
+        if (message.role === 'system') {
+            systemParts.push(content);
+            return;
+        }
+        var label = message.role === 'assistant' ? 'Assistant' : 'User';
+        dialogParts.push("".concat(label, ":\n").concat(content));
+    });
+    var systemPrompt = systemParts.length
+        ? systemParts.join('\n\n')
+        : 'You are a helpful AI assistant.';
+    var dialogPrompt = dialogParts.length
+        ? dialogParts.join('\n\n')
+        : 'User:\nContinue.';
+    return "System:\n".concat(systemPrompt, "\n\n").concat(dialogPrompt).trim();
+}
 function applyTemplate(template, data) {
     var output = template;
     Object.keys(data).forEach(function (key) {
@@ -20491,9 +20767,17 @@ function applyTemplate(template, data) {
     return output;
 }
 function sanitizeConfig(source) {
+    var fallbackModels = Array.isArray(source === null || source === void 0 ? void 0 : source.fallback_models)
+        ? source.fallback_models
+        : (Array.isArray(source === null || source === void 0 ? void 0 : source.fallbackModels) ? source.fallbackModels : []);
     return {
         mode: normalizeOptionalString(source === null || source === void 0 ? void 0 : source.mode),
         model: normalizeOptionalString(source === null || source === void 0 ? void 0 : source.model),
+        codex_model: normalizeOptionalString(source === null || source === void 0 ? void 0 : source.codex_model),
+        fallback_model: normalizeOptionalString((source === null || source === void 0 ? void 0 : source.fallback_model) || (source === null || source === void 0 ? void 0 : source.fallbackModel)),
+        fallback_models: fallbackModels
+            .map(function (entry) { return normalizeOptionalString(entry); })
+            .filter(Boolean),
         system_prompt: normalizeOptionalString(source === null || source === void 0 ? void 0 : source.system_prompt),
         user_prompt_template: normalizeOptionalString(source === null || source === void 0 ? void 0 : source.user_prompt_template),
         response_format: normalizeOptionalString(source === null || source === void 0 ? void 0 : source.response_format),