@orion-agents/orion-code 0.1.7 → 0.1.9

package/dist/runtime/chat-controller.js

@@ -20,10 +20,12 @@ const tool_artifacts_1 = require("../core/tool-artifacts");
 const skills_1 = require("../skills");
 const subagents_1 = require("./subagents");
 const file_context_1 = require("../services/file-context");
-const prompt_context_1 = require("../services/prompt-context");
+const prompt_context_1 = require("../memory/prompt-context");
+const prompt_context_2 = require("../services/prompt-context");
 const workspace_state_1 = require("../services/workspace-state");
 const verification_profile_1 = require("../services/verification-profile");
 const ui_events_1 = require("./ui-events");
+const ui_view_model_1 = require("./ui-view-model");
 const agent_status_1 = require("./agent-status");
 const loop_budget_1 = require("./loop-budget");
 const observability_1 = require("../utils/observability");
@@ -32,7 +34,9 @@ const chat_checkpoint_1 = require("./chat-checkpoint");
 const chat_workspace_1 = require("./chat-workspace");
 const chat_presentation_1 = require("./chat-presentation");
 const chat_effort_1 = require("./chat-effort");
+const chat_compact_1 = require("./chat-compact");
 const agent_mode_1 = require("../framework/agent-mode");
+const chat_command_scope_1 = require("./chat-command-scope");
 var chat_presentation_2 = require("./chat-presentation");
 Object.defineProperty(exports, "createAssistantStreamPresenter", { enumerable: true, get: function () { return chat_presentation_2.createAssistantStreamPresenter; } });
 Object.defineProperty(exports, "createToolEventPresenter", { enumerable: true, get: function () { return chat_presentation_2.createToolEventPresenter; } });
@@ -42,9 +46,11 @@ class AgentChatController {
         this.goalCoordinator = coord;
     }
     resolveToolAllowlist() {
-        // Reload on every scheduling boundary so a project/global approval selected
-        // by the interactive prompt affects the very next tool call in this process.
-        return (0, tool_allowlist_1.resolveProjectToolAllowlist)(this.runtime.cwd).evaluator;
+        // Resolve on every evaluation, including execution. A model response can
+        // contain several serial calls prepared before the first permission prompt;
+        // once the user chooses project/global scope, the next call in that same
+        // response must observe the persisted grant instead of prompting again.
+        return (toolName, args) => (0, tool_allowlist_1.resolveProjectToolAllowlist)(this.runtime.cwd).evaluator?.(toolName, args);
     }
     constructor(runtime, events, controllerOptions = {}) {
         this.runtime = runtime;
@@ -79,21 +85,6 @@ class AgentChatController {
             await this.runChat(text, options);
             return;
         }
-        if (parsed.name === 'clear') {
-            if (this.events.clearView) {
-                this.events.clearView();
-            }
-            else {
-                this.events.clearTranscript();
-            }
-            this.events.setStatus('View cleared. Conversation context is preserved.');
-            return;
-        }
-        if (parsed.name === 'exit' || parsed.name === 'quit' || parsed.name === 'q') {
-            this.events.shutdownRequested?.('user request');
-            await this.runtime.shutdown();
-            return;
-        }
         const command = (0, commands_1.findCommand)(parsed.name);
         if (!command) {
             if ((0, skills_1.hasMatchingSkill)(text, this.runtime.cwd)) {
@@ -111,6 +102,24 @@ class AgentChatController {
             });
             return;
         }
+        const activeRenderer = this.controllerOptions.uiRenderer ?? this.runtime.config.ui?.renderer ?? 'terminal';
+        if ((0, chat_command_scope_1.rejectUnsupportedRenderer)(this.events, command, activeRenderer))
+            return;
+        if (parsed.name === 'clear') {
+            if (this.events.clearView) {
+                this.events.clearView();
+            }
+            else {
+                this.events.clearTranscript();
+            }
+            this.events.setStatus('View cleared. Conversation context is preserved.');
+            return;
+        }
+        if (parsed.name === 'exit' || parsed.name === 'quit' || parsed.name === 'q') {
+            this.events.shutdownRequested?.('user request');
+            await this.runtime.shutdown();
+            return;
+        }
         const compatibilityAlias = command.compatibilityAliases?.find(alias => alias.name.toLowerCase() === parsed.name.toLowerCase());
         const deprecation = compatibilityAlias?.lifecycle ??
             (command.lifecycle.status === 'deprecated' ? command.lifecycle : undefined);
@@ -342,12 +351,7 @@ class AgentChatController {
                 allowlist: toolAllowlist?.(action.tool, action.args),
                 toolConfirmation: this.runtime.config.toolConfirmation,
             });
-            // Local exec is a hidden shortcut rather than a plan artifact. Even a
-            // read-only command must not execute through that shortcut in plan mode.
-            if (effectivePermissionMode === 'plan' && action.tool === 'exec_command') {
-                throw new chat_presentation_1.LocalFastPathBlockedError('Local run-test fast path is blocked in plan mode; switch mode before executing commands.');
-            }
-            if (effective.outcome === 'deny' || effective.outcome === 'block') {
+            if (effective.outcome === 'deny') {
                 throw new chat_presentation_1.LocalFastPathBlockedError(effective.reason ||
                     `Local fast path blocked by ${effective.source ?? 'permission policy'}.`);
             }
@@ -445,6 +449,7 @@ class AgentChatController {
                     turnId,
                     type: 'complete',
                     finishReason: stats.finishReason,
+                    stopDecision: stats.stopDecision,
                     llmRequests: stats.llmRequests,
                     toolCalls: stats.toolCalls,
                     readOnlyToolCalls: stats.readOnlyToolCalls,
@@ -456,10 +461,6 @@ class AgentChatController {
                     content: assistantContent,
                     timestamp: Date.now(),
                 });
-                const recordedMessages = (0, session_storage_1.readSessionMessages)(sessionId);
-                if (recordedMessages.length > 0) {
-                    (0, session_storage_1.updateSessionSummary)(sessionId, recordedMessages);
-                }
             }
             this.events.setStatus(envelope.success ? `Completed local ${action.label}` : `Failed local ${action.label}`);
         }
@@ -487,6 +488,7 @@ class AgentChatController {
                     turnId,
                     type: 'complete',
                     finishReason: stats.finishReason,
+                    stopDecision: stats.stopDecision,
                     llmRequests: stats.llmRequests,
                     toolCalls: stats.toolCalls,
                     localFastPathUsed: true,
@@ -507,6 +509,7 @@ class AgentChatController {
             store: this.runtime.store,
             llm: this.runtime.llm,
             compactCoordinator: this.runtime.compactCoordinator,
+            modelCoordinator: this.runtime.modelCoordinator,
             runtime: this.runtime.runtime,
             sessionId: this.runtime.getSession()?.id,
             turnId,
@@ -523,6 +526,7 @@ class AgentChatController {
                 this.events.sessionRestored?.(event);
             },
             getSession: this.runtime.getSession,
+            getActiveGoal: () => this.goalCoordinator?.goal ?? null,
             abortSignal,
             writeOutput: text => {
                 if (text.trim()) {
@@ -687,7 +691,7 @@ class AgentChatController {
         if (options.persistAsUserMessage !== false) {
             this.runtime.store.addMessage({ role: 'user', content: input });
         }
-        (0, prompt_context_1.refreshProjectInstructions)(this.runtime.store, this.runtime.cwd);
+        (0, prompt_context_2.refreshProjectInstructions)(this.runtime.store, this.runtime.cwd);
         const snapshot = this.runtime.store.getSnapshot();
         const effectivePermissionMode = this.runtime.store.getEffectivePermissionMode();
         const onPlanModeChange = (0, agent_mode_1.createPlanModeChangeHandler)(this.runtime.store, this.controllerOptions.agentModeLifecycle);
@@ -832,12 +836,31 @@ class AgentChatController {
             : null;
         const subtaskTool = subagentBundle?.tool ?? null;
         const turnTools = subtaskTool ? [...skillResolution.tools, subtaskTool] : skillResolution.tools;
+        const memoryPrompt = (0, prompt_context_1.buildMemoryPromptContext)(input, this.runtime.cwd);
+        if (sessionId) {
+            const selected = memoryPrompt.manifest.selected.map(item => `memory:${item.name}`);
+            const omitted = memoryPrompt.manifest.omitted.map(item => `memory:${item.name}`);
+            (0, chat_trace_1.recordTraceEvent)(this.events, sessionId, {
+                turnId: String(turnId),
+                type: 'message',
+                name: 'memory_context',
+                argsSummary: JSON.stringify({
+                    budgetChars: memoryPrompt.manifest.budgetChars,
+                    usedChars: memoryPrompt.manifest.usedChars,
+                    candidateCount: memoryPrompt.manifest.candidateCount,
+                    entrypointTruncated: memoryPrompt.manifest.entrypoint.truncated,
+                    selected: selected.slice(0, 10),
+                    omitted: omitted.slice(0, 10),
+                }),
+                contentBytes: Buffer.byteLength(memoryPrompt.content, 'utf8'),
+            });
+        }
         const promptCtx = {
             cwd: this.runtime.cwd,
             platform: process.platform,
             nodeVersion: process.version,
             tools: turnTools,
-            memoryContent: snapshot.memoryContent,
+            memoryContent: memoryPrompt.content,
             skillsContent: snapshot.skillsContent,
             projectInstructionsContent: snapshot.projectInstructionsContent,
             activeSkillsContent: skillResolution.promptInjection,
@@ -915,10 +938,9 @@ class AgentChatController {
                 onPlanModeChange,
             });
         };
-        const resolvedLoopBudget = (0, loop_budget_1.resolveRuntimeLoopBudget)(input, this.runtime.config, harness.toJSON());
         const loopBudget = options.inputKind === 'goal_continuation'
-            ? (0, loop_budget_1.capAutonomousGoalLoopBudget)(resolvedLoopBudget)
-            : resolvedLoopBudget;
+            ? (0, loop_budget_1.resolveAutonomousGoalLoopBudget)(this.runtime.config)
+            : (0, loop_budget_1.resolveRuntimeLoopBudget)(input, this.runtime.config, harness.toJSON());
         let observedTurnsStarted = 0;
         let observedLlmRequests = 0;
         let observedToolCalls = 0;
@@ -980,6 +1002,9 @@ class AgentChatController {
                     this.runtime.store.setContextUsage(usage);
                 },
                 compactCoordinator: this.runtime.compactCoordinator,
+                onCompactPrepare: sessionId
+                    ? () => (0, session_storage_1.prepareSessionCompactSourceReceipt)(sessionId)
+                    : undefined,
             })) {
                 switch (event.type) {
                     case 'request_start':
@@ -1012,6 +1037,10 @@ class AgentChatController {
                                 promptOmittedEvidence: event.omittedEvidence,
                                 promptIncludedEvidenceCount: event.includedEvidenceCount,
                                 promptOmittedEvidenceCount: event.omittedEvidenceCount,
+                                promptSectionManifest: event.sectionManifest,
+                                promptOverBudget: event.overBudget,
+                                promptCapabilityProfileVersion: event.capabilityProfileVersion,
+                                promptCapabilityProfileFingerprint: event.capabilityProfileFingerprint,
                             });
                         }
                         break;
@@ -1094,6 +1123,7 @@ class AgentChatController {
                         }
                         break;
                     case 'permission_decision':
+                        toolEvents.permission(event);
                         if (sessionId) {
                             (0, chat_trace_1.recordTraceEvent)(this.events, sessionId, {
                                 turnId,
@@ -1220,6 +1250,7 @@ class AgentChatController {
                                 model: event.model,
                                 contentBytes: (0, chat_trace_1.byteLength)(event.content || ''),
                                 finishReason: event.stats.finishReason,
+                                stopDecision: event.stats.stopDecision,
                                 llmRequests: event.stats.llmRequests,
                                 toolCalls: event.stats.toolCalls,
                                 readOnlyToolCalls: event.stats.readOnlyToolCalls,
@@ -1234,6 +1265,9 @@ class AgentChatController {
                                 budgetExceededReason: event.stats.budgetExceededReason,
                                 continuationActions: event.stats.continuationActions,
                                 continuationHint: event.stats.continuationHint,
+                                lastToolName: event.stats.lastToolName,
+                                lastToolSummary: event.stats.lastToolSummary,
+                                lastToolSuccess: event.stats.lastToolSuccess,
                                 localFastPathUsed: event.stats.localFastPathUsed,
                             };
                         }
@@ -1325,10 +1359,10 @@ class AgentChatController {
                     }
                     const stats = pendingCompleteStats ?? this.runtime.store.getSnapshot().lastLoopStats;
                     if (stats) {
-                        pendingCompleteStats = {
-                            ...(0, chat_workspace_1.withVerificationLoopStats)(stats, summary),
-                            finishReason: 'completion_gate',
-                        };
+                        pendingCompleteStats = (0, framework_1.withLoopFinishReason)((0, chat_workspace_1.withVerificationLoopStats)(stats, summary), 'completion_gate');
+                        if (pendingCompleteTrace) {
+                            pendingCompleteTrace.stopDecision = pendingCompleteStats.stopDecision;
+                        }
                     }
                 }
                 else if (profile.changedFiles.length > 0 && profile.required) {
@@ -1337,35 +1371,20 @@ class AgentChatController {
             }
             if (pendingCompleteStats?.finishReason === 'budget_exceeded') {
                 const stats = pendingCompleteStats;
-                const lines = ['Loop budget reached — stopping this turn.'];
-                if (stats.budgetExceededReason) {
-                    lines.push(`Reason: ${stats.budgetExceededReason}`);
-                }
-                const progressParts = [];
-                if (typeof stats.loopBudgetMaxLlmRequests === 'number') {
-                    progressParts.push(`${stats.llmRequests ?? 0}/${stats.loopBudgetMaxLlmRequests} LLM requests`);
-                }
-                if (typeof stats.loopBudgetMaxToolCalls === 'number') {
-                    progressParts.push(`${stats.toolCalls ?? 0}/${stats.loopBudgetMaxToolCalls} tool calls`);
-                }
-                if (progressParts.length) {
-                    lines.push(`Progress: ${progressParts.join(', ')}`);
-                }
-                if (stats.continuationActions?.length) {
-                    lines.push(`Next: ${stats.continuationActions.join('; ')}`);
-                }
-                else if (stats.continuationHint) {
-                    lines.push(`Next: ${stats.continuationHint}`);
-                }
-                const notice = lines.join('\n');
+                const budgetStop = (0, ui_view_model_1.createLoopBudgetStopView)(stats);
+                const notice = (0, ui_view_model_1.formatLoopBudgetStopView)(budgetStop, (0, ui_view_model_1.budgetStopLocaleForInput)(input));
                 this.events.append({
                     role: 'status',
                     title: 'budget',
                     statusTone: 'warning',
                     content: notice,
+                    budgetStop,
                 });
                 finalContent = finalContent ? `${finalContent}\n\n${notice}` : notice;
                 (0, chat_trace_1.appendAssistantNotice)(sessionMessagesToRecord, notice);
+                if (pendingCompleteTrace) {
+                    pendingCompleteTrace.contentBytes = (0, chat_trace_1.byteLength)(finalContent);
+                }
             }
             if (pendingCompleteStats) {
                 this.setLoopStats(this.foldSubagentUsage(pendingCompleteStats, subagentBundle));
@@ -1376,64 +1395,30 @@ class AgentChatController {
             if (sessionId && sessionMessagesToRecord.length > 0) {
                 (0, session_storage_1.appendSessionMessages)(sessionId, sessionMessagesToRecord);
             }
+            harness.ingestTurn({
+                userInput: input,
+                assistantContent: finalContent,
+                sessionMessages: sessionMessagesToRecord,
+                intent,
+            });
+            const harnessState = harness.toJSON();
+            this.runtime.store.setState({ harnessState });
+            (0, chat_trace_1.emitHarnessDiagnostics)(this.events, harnessState);
+            if (sessionId) {
+                (0, session_storage_1.updateSessionHarnessState)(sessionId, harnessState);
+            }
             if (pendingCompact) {
-                try {
-                    let committedCheckpointId;
-                    if (sessionId) {
-                        const sourceMessageCount = (0, session_storage_1.readSessionMessages)(sessionId).length;
-                        const checkpoint = (0, session_storage_1.commitSessionCompactCheckpoint)({
-                            sessionId,
-                            mode: pendingCompact.mode,
-                            modelId: finalModel || this.runtime.llm.getModel(),
-                            sourceMessageCount,
-                            transcriptStartMessageIndex: Math.max(0, sourceMessageCount - 20),
-                            modelHistory: pendingCompact.modelHistory,
-                            summary: pendingCompact.summary,
-                            beforeUsage: pendingCompact.before,
-                            afterUsage: pendingCompact.after,
-                        });
-                        committedCheckpointId = checkpoint.checkpointId;
-                        this.runtime.store.setState({ conversationHistory: checkpoint.modelHistory });
-                    }
-                    else {
-                        this.runtime.store.setState({
-                            conversationHistory: pendingCompact.modelHistory.filter(message => message.role !== 'system'),
-                        });
-                    }
-                    this.runtime.store.setContextUsage(pendingCompact.after);
-                    if (sessionId) {
-                        (0, chat_trace_1.recordTraceEvent)(this.events, sessionId, {
-                            turnId,
-                            type: 'compact_completed',
-                            checkpointId: committedCheckpointId,
-                            model: finalModel || this.runtime.llm.getModel(),
-                            note: pendingCompact.mode,
-                        });
-                    }
-                    this.events.append({
-                        role: 'status',
-                        title: 'auto-compact',
-                        statusTone: 'neutral',
-                        content: `Context reached ${pendingCompact.before.percent}% of the safe input budget. Agent core committed a ${pendingCompact.mode} compact checkpoint; current context is ${pendingCompact.after.percent}%.`,
-                    });
-                }
-                catch (error) {
-                    if (sessionId) {
-                        (0, chat_trace_1.recordTraceEvent)(this.events, sessionId, {
-                            turnId,
-                            type: 'compact_failed',
-                            model: finalModel || this.runtime.llm.getModel(),
-                            error: (0, chat_trace_1.compactTraceError)(error),
-                            note: pendingCompact.mode,
-                        });
-                    }
-                    this.events.append({
-                        role: 'error',
-                        title: 'compact-failed',
-                        content: `Compact checkpoint failed; the previous model context remains active. ${error instanceof Error ? error.message : String(error)}`,
-                        errorLayer: 'runtime',
-                    });
-                }
+                (0, chat_compact_1.commitChatCompact)({
+                    pendingCompact,
+                    sessionId,
+                    turnId,
+                    modelId: finalModel || this.runtime.llm.getModel(),
+                    sessionMessages: sessionMessagesToRecord,
+                    harnessState,
+                    activeGoal: this.goalCoordinator?.goal,
+                    store: this.runtime.store,
+                    events: this.events,
+                });
             }
             if (finalUsage) {
                 // Fold subagent token usage into /cost accounting.
@@ -1446,25 +1431,11 @@ class AgentChatController {
                 }
                 this.runtime.store.setTokenUsage(finalUsage);
             }
-            harness.ingestTurn({
-                userInput: input,
-                assistantContent: finalContent,
-                sessionMessages: sessionMessagesToRecord,
-                intent,
-            });
-            const harnessState = harness.toJSON();
-            this.runtime.store.setState({ harnessState });
-            (0, chat_trace_1.emitHarnessDiagnostics)(this.events, harnessState);
             if (sessionId) {
                 if (pendingCompleteTrace) {
                     (0, chat_trace_1.recordTraceEvent)(this.events, sessionId, pendingCompleteTrace);
                 }
                 (0, session_storage_1.updateSessionSkills)(sessionId, appliedSkillNames);
-                (0, session_storage_1.updateSessionHarnessState)(sessionId, harnessState);
-                const recordedMessages = (0, session_storage_1.readSessionMessages)(sessionId);
-                if (recordedMessages.length > 0) {
-                    (0, session_storage_1.updateSessionSummary)(sessionId, recordedMessages);
-                }
             }
             this.events.setStatus(finalModel ? `Completed with ${finalModel}` : 'Completed');
         }
@@ -1594,6 +1565,7 @@ class AgentChatController {
                     model: failedStats.providerFinalModel ?? this.runtime.llm.getModel(),
                     contentBytes: 0,
                     finishReason: failedStats.finishReason,
+                    stopDecision: failedStats.stopDecision,
                     llmRequests: failedStats.llmRequests,
                     toolCalls: failedStats.toolCalls,
                     readOnlyToolCalls: failedStats.readOnlyToolCalls,