6354 lines
228 KiB
TypeScript
6354 lines
228 KiB
TypeScript
// src/cli/App.tsx
|
||
|
||
import { existsSync, readFileSync, writeFileSync } from "node:fs";
|
||
import { APIUserAbortError } from "@letta-ai/letta-client/core/error";
|
||
import type {
|
||
AgentState,
|
||
MessageCreate,
|
||
} from "@letta-ai/letta-client/resources/agents/agents";
|
||
import type {
|
||
ApprovalCreate,
|
||
Message,
|
||
} from "@letta-ai/letta-client/resources/agents/messages";
|
||
import type { LlmConfig } from "@letta-ai/letta-client/resources/models/models";
|
||
import type { StopReasonType } from "@letta-ai/letta-client/resources/runs/runs";
|
||
import { Box, Static, Text } from "ink";
|
||
import { useCallback, useEffect, useMemo, useRef, useState } from "react";
|
||
import {
|
||
type ApprovalResult,
|
||
executeAutoAllowedTools,
|
||
} from "../agent/approval-execution";
|
||
import {
|
||
buildApprovalRecoveryMessage,
|
||
fetchRunErrorDetail,
|
||
isApprovalStateDesyncError,
|
||
} from "../agent/approval-recovery";
|
||
import { prefetchAvailableModelHandles } from "../agent/available-models";
|
||
import { getResumeData } from "../agent/check-approval";
|
||
import { getClient } from "../agent/client";
|
||
import { getCurrentAgentId, setCurrentAgentId } from "../agent/context";
|
||
import { type AgentProvenance, createAgent } from "../agent/create";
|
||
import { sendMessageStream } from "../agent/message";
|
||
import { getModelDisplayName, getModelInfo } from "../agent/model";
|
||
import { SessionStats } from "../agent/stats";
|
||
import type { ApprovalContext } from "../permissions/analyzer";
|
||
import { type PermissionMode, permissionMode } from "../permissions/mode";
|
||
import { updateProjectSettings } from "../settings";
|
||
import { settingsManager } from "../settings-manager";
|
||
import { telemetry } from "../telemetry";
|
||
import {
|
||
analyzeToolApproval,
|
||
checkToolPermission,
|
||
executeTool,
|
||
isGeminiModel,
|
||
isOpenAIModel,
|
||
savePermissionRule,
|
||
type ToolExecutionResult,
|
||
} from "../tools/manager";
|
||
import {
|
||
handleMcpAdd,
|
||
handleMcpUsage,
|
||
type McpCommandContext,
|
||
} from "./commands/mcp";
|
||
import {
|
||
addCommandResult,
|
||
handlePin,
|
||
handleProfileDelete,
|
||
handleProfileSave,
|
||
handleProfileUsage,
|
||
handleUnpin,
|
||
type ProfileCommandContext,
|
||
validateProfileLoad,
|
||
} from "./commands/profile";
|
||
import { AgentSelector } from "./components/AgentSelector";
|
||
// ApprovalDialog removed - all approvals now render inline
|
||
import { ApprovalPreview } from "./components/ApprovalPreview";
|
||
import { AssistantMessage } from "./components/AssistantMessageRich";
|
||
import { BashCommandMessage } from "./components/BashCommandMessage";
|
||
import { CommandMessage } from "./components/CommandMessage";
|
||
import { colors } from "./components/colors";
|
||
// EnterPlanModeDialog removed - now using InlineEnterPlanModeApproval
|
||
import { ErrorMessage } from "./components/ErrorMessageRich";
|
||
import { FeedbackDialog } from "./components/FeedbackDialog";
|
||
import { HelpDialog } from "./components/HelpDialog";
|
||
import { InlineBashApproval } from "./components/InlineBashApproval";
|
||
import { InlineEnterPlanModeApproval } from "./components/InlineEnterPlanModeApproval";
|
||
import { InlineFileEditApproval } from "./components/InlineFileEditApproval";
|
||
import { InlineGenericApproval } from "./components/InlineGenericApproval";
|
||
import { InlineQuestionApproval } from "./components/InlineQuestionApproval";
|
||
import { InlineTaskApproval } from "./components/InlineTaskApproval";
|
||
import { Input } from "./components/InputRich";
|
||
import { McpSelector } from "./components/McpSelector";
|
||
import { MemoryViewer } from "./components/MemoryViewer";
|
||
import { MessageSearch } from "./components/MessageSearch";
|
||
import { ModelSelector } from "./components/ModelSelector";
|
||
import { NewAgentDialog } from "./components/NewAgentDialog";
|
||
import { OAuthCodeDialog } from "./components/OAuthCodeDialog";
|
||
import { PendingApprovalStub } from "./components/PendingApprovalStub";
|
||
import { PinDialog, validateAgentName } from "./components/PinDialog";
|
||
// QuestionDialog removed - now using InlineQuestionApproval
|
||
import { ReasoningMessage } from "./components/ReasoningMessageRich";
|
||
import { ResumeSelector } from "./components/ResumeSelector";
|
||
import { formatUsageStats } from "./components/SessionStats";
|
||
// InlinePlanApproval kept for easy rollback if needed
|
||
// import { InlinePlanApproval } from "./components/InlinePlanApproval";
|
||
import { StaticPlanApproval } from "./components/StaticPlanApproval";
|
||
import { StatusMessage } from "./components/StatusMessage";
|
||
import { SubagentGroupDisplay } from "./components/SubagentGroupDisplay";
|
||
import { SubagentGroupStatic } from "./components/SubagentGroupStatic";
|
||
import { SubagentManager } from "./components/SubagentManager";
|
||
import { SystemPromptSelector } from "./components/SystemPromptSelector";
|
||
import { ToolCallMessage } from "./components/ToolCallMessageRich";
|
||
import { ToolsetSelector } from "./components/ToolsetSelector";
|
||
import { UserMessage } from "./components/UserMessageRich";
|
||
import { WelcomeScreen } from "./components/WelcomeScreen";
|
||
import {
|
||
type Buffers,
|
||
createBuffers,
|
||
type Line,
|
||
markIncompleteToolsAsCancelled,
|
||
onChunk,
|
||
toLines,
|
||
} from "./helpers/accumulator";
|
||
import { backfillBuffers } from "./helpers/backfill";
|
||
import {
|
||
type AdvancedDiffSuccess,
|
||
computeAdvancedDiff,
|
||
parsePatchToAdvancedDiff,
|
||
} from "./helpers/diff";
|
||
import { formatErrorDetails } from "./helpers/errorFormatter";
|
||
import { parsePatchOperations } from "./helpers/formatArgsDisplay";
|
||
import {
|
||
buildMemoryReminder,
|
||
parseMemoryPreference,
|
||
} from "./helpers/memoryReminder";
|
||
import {
|
||
buildMessageContentFromDisplay,
|
||
clearPlaceholdersInText,
|
||
resolvePlaceholders,
|
||
} from "./helpers/pasteRegistry";
|
||
import { generatePlanFilePath } from "./helpers/planName";
|
||
import { safeJsonParseOr } from "./helpers/safeJsonParse";
|
||
import { type ApprovalRequest, drainStreamWithResume } from "./helpers/stream";
|
||
import {
|
||
collectFinishedTaskToolCalls,
|
||
createSubagentGroupItem,
|
||
hasInProgressTaskToolCalls,
|
||
} from "./helpers/subagentAggregation";
|
||
import {
|
||
clearCompletedSubagents,
|
||
clearSubagentsByIds,
|
||
} from "./helpers/subagentState";
|
||
import { getRandomThinkingVerb } from "./helpers/thinkingMessages";
|
||
import {
|
||
isFileEditTool,
|
||
isFileWriteTool,
|
||
isPatchTool,
|
||
isShellTool,
|
||
} from "./helpers/toolNameMapping";
|
||
import {
|
||
alwaysRequiresUserInput,
|
||
isTaskTool,
|
||
} from "./helpers/toolNameMapping.js";
|
||
import { useSuspend } from "./hooks/useSuspend/useSuspend.ts";
|
||
import { useSyncedState } from "./hooks/useSyncedState";
|
||
import { useTerminalWidth } from "./hooks/useTerminalWidth";
|
||
|
||
// Used only for terminal resize, not for dialog dismissal (see PR for details)
|
||
const CLEAR_SCREEN_AND_HOME = "\u001B[2J\u001B[H";
|
||
|
||
// Feature flag: Check for pending approvals before sending messages
|
||
// This prevents infinite thinking state when there's an orphaned approval
|
||
// Can be disabled if the latency check adds too much overhead
|
||
const CHECK_PENDING_APPROVALS_BEFORE_SEND = true;
|
||
|
||
// Feature flag: Eagerly cancel streams client-side when user presses ESC
|
||
// When true (default), immediately abort the stream after calling .cancel()
|
||
// This provides instant feedback to the user without waiting for backend acknowledgment
|
||
// When false, wait for backend to send "cancelled" stop_reason (useful for testing backend behavior)
|
||
const EAGER_CANCEL = true;
|
||
|
||
// Maximum retries for transient LLM API errors (matches headless.ts)
|
||
const LLM_API_ERROR_MAX_RETRIES = 3;
|
||
|
||
// tiny helper for unique ids (avoid overwriting prior user lines)
|
||
function uid(prefix: string) {
|
||
return `${prefix}-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
|
||
}
|
||
|
||
// Send desktop notification via terminal bell
|
||
// Modern terminals (iTerm2, Ghostty, WezTerm, Kitty) convert this to a desktop
|
||
// notification when the terminal is not focused
|
||
function sendDesktopNotification() {
|
||
process.stdout.write("\x07");
|
||
}
|
||
|
||
// Check if error is retriable based on stop reason and run metadata
|
||
async function isRetriableError(
|
||
stopReason: StopReasonType,
|
||
lastRunId: string | null | undefined,
|
||
): Promise<boolean> {
|
||
// Primary check: backend sets stop_reason=llm_api_error for LLMError exceptions
|
||
if (stopReason === "llm_api_error") return true;
|
||
|
||
// Fallback check: in case stop_reason is "error" but metadata indicates LLM error
|
||
// This could happen if there's a backend edge case where LLMError is raised but
|
||
// stop_reason isn't set correctly. The metadata.error is a LettaErrorMessage with
|
||
// error_type="llm_error" for LLM errors (see streaming_service.py:402-411)
|
||
if (stopReason === "error" && lastRunId) {
|
||
try {
|
||
const client = await getClient();
|
||
const run = await client.runs.retrieve(lastRunId);
|
||
const metaError = run.metadata?.error as
|
||
| {
|
||
error_type?: string;
|
||
detail?: string;
|
||
// Handle nested error structure (error.error) that can occur in some edge cases
|
||
error?: { error_type?: string; detail?: string };
|
||
}
|
||
| undefined;
|
||
|
||
// Check for llm_error at top level or nested (handles error.error nesting)
|
||
const errorType = metaError?.error_type ?? metaError?.error?.error_type;
|
||
if (errorType === "llm_error") return true;
|
||
|
||
// Fallback: detect LLM provider errors from detail even if misclassified as internal_error
|
||
// This handles edge cases where streaming errors weren't properly converted to LLMError
|
||
// Patterns are derived from handle_llm_error() message formats in the backend
|
||
const detail = metaError?.detail ?? metaError?.error?.detail ?? "";
|
||
const llmProviderPatterns = [
|
||
"Anthropic API error", // anthropic_client.py:759
|
||
"OpenAI API error", // openai_client.py:1034
|
||
"Google Vertex API error", // google_vertex_client.py:848
|
||
"overloaded", // anthropic_client.py:753 - used for LLMProviderOverloaded
|
||
"api_error", // Anthropic SDK error type field
|
||
];
|
||
if (
|
||
errorType === "internal_error" &&
|
||
llmProviderPatterns.some((pattern) => detail.includes(pattern))
|
||
) {
|
||
return true;
|
||
}
|
||
|
||
return false;
|
||
} catch {
|
||
return false;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
// Save current agent as lastAgent before exiting
|
||
// This ensures subagent overwrites during the session don't persist
|
||
function saveLastAgentBeforeExit() {
|
||
try {
|
||
const currentAgentId = getCurrentAgentId();
|
||
settingsManager.updateLocalProjectSettings({ lastAgent: currentAgentId });
|
||
settingsManager.updateSettings({ lastAgent: currentAgentId });
|
||
} catch {
|
||
// Ignore if no agent context set
|
||
}
|
||
}
|
||
|
||
// Get plan mode system reminder if in plan mode
|
||
function getPlanModeReminder(): string {
|
||
if (permissionMode.getMode() !== "plan") {
|
||
return "";
|
||
}
|
||
|
||
const planFilePath = permissionMode.getPlanFilePath();
|
||
|
||
// Generate dynamic reminder with plan file path
|
||
return `<system-reminder>
|
||
Plan mode is active. The user indicated that they do not want you to execute yet -- you MUST NOT make any edits (with the exception of the plan file mentioned below), run any non-readonly tools (including changing configs or making commits), or otherwise make any changes to the system. This supersedes any other instructions you have received.
|
||
|
||
## Plan File Info:
|
||
${planFilePath ? `No plan file exists yet. You should create your plan at ${planFilePath} using a write tool (e.g. Write, ApplyPatch, etc. depending on your toolset).` : "No plan file path assigned."}
|
||
|
||
You should build your plan incrementally by writing to or editing this file. NOTE that this is the only file you are allowed to edit - other than this you are only allowed to take READ-ONLY actions.
|
||
|
||
**Plan File Guidelines:** The plan file should contain only your final recommended approach, not all alternatives considered. Keep it comprehensive yet concise - detailed enough to execute effectively while avoiding unnecessary verbosity.
|
||
|
||
## Enhanced Planning Workflow
|
||
|
||
### Phase 1: Initial Understanding
|
||
Goal: Gain a comprehensive understanding of the user's request by reading through code and asking them questions.
|
||
|
||
1. Understand the user's request thoroughly
|
||
2. Explore the codebase to understand existing patterns and relevant code
|
||
3. Use AskUserQuestion tool to clarify ambiguities in the user request up front.
|
||
|
||
### Phase 2: Planning
|
||
Goal: Come up with an approach to solve the problem identified in phase 1.
|
||
|
||
- Provide any background context that may help with the task without prescribing the exact design itself
|
||
- Create a detailed plan
|
||
|
||
### Phase 3: Synthesis
|
||
Goal: Synthesize the perspectives from Phase 2, and ensure that it aligns with the user's intentions by asking them questions.
|
||
|
||
1. Collect all findings from exploration
|
||
2. Keep track of critical files that should be read before implementing the plan
|
||
3. Use AskUserQuestion to ask the user questions about trade offs.
|
||
|
||
### Phase 4: Final Plan
|
||
Once you have all the information you need, ensure that the plan file has been updated with your synthesized recommendation including:
|
||
|
||
- Recommended approach with rationale
|
||
- Key insights from different perspectives
|
||
- Critical files that need modification
|
||
|
||
### Phase 5: Call ExitPlanMode
|
||
At the very end of your turn, once you have asked the user questions and are happy with your final plan file - you should always call ExitPlanMode to indicate to the user that you are done planning.
|
||
|
||
This is critical - your turn should only end with either asking the user a question or calling ExitPlanMode. Do not stop unless it's for these 2 reasons.
|
||
|
||
NOTE: At any point in time through this workflow you should feel free to ask the user questions or clarifications. Don't make large assumptions about user intent. The goal is to present a well researched plan to the user, and tie any loose ends before implementation begins.
|
||
</system-reminder>
|
||
`;
|
||
}
|
||
|
||
// Check if plan file exists
|
||
function planFileExists(): boolean {
|
||
const planFilePath = permissionMode.getPlanFilePath();
|
||
return !!planFilePath && existsSync(planFilePath);
|
||
}
|
||
|
||
// Read plan content from the plan file
|
||
function _readPlanFile(): string {
|
||
const planFilePath = permissionMode.getPlanFilePath();
|
||
if (!planFilePath) {
|
||
return "No plan file path set.";
|
||
}
|
||
if (!existsSync(planFilePath)) {
|
||
return `Plan file not found at ${planFilePath}`;
|
||
}
|
||
try {
|
||
return readFileSync(planFilePath, "utf-8");
|
||
} catch {
|
||
return `Failed to read plan file at ${planFilePath}`;
|
||
}
|
||
}
|
||
|
||
// Extract questions from AskUserQuestion tool args
|
||
function getQuestionsFromApproval(approval: ApprovalRequest) {
|
||
const parsed = safeJsonParseOr<Record<string, unknown>>(
|
||
approval.toolArgs,
|
||
{},
|
||
);
|
||
return (
|
||
(parsed.questions as Array<{
|
||
question: string;
|
||
header: string;
|
||
options: Array<{ label: string; description: string }>;
|
||
multiSelect: boolean;
|
||
}>) || []
|
||
);
|
||
}
|
||
|
||
// Get skill unload reminder if skills are loaded (using cached flag)
|
||
function getSkillUnloadReminder(): string {
|
||
const { hasLoadedSkills } = require("../agent/context");
|
||
if (hasLoadedSkills()) {
|
||
const { SKILL_UNLOAD_REMINDER } = require("../agent/promptAssets");
|
||
return SKILL_UNLOAD_REMINDER;
|
||
}
|
||
return "";
|
||
}
|
||
|
||
// Items that have finished rendering and no longer change
|
||
type StaticItem =
|
||
| {
|
||
kind: "welcome";
|
||
id: string;
|
||
snapshot: {
|
||
continueSession: boolean;
|
||
agentState?: AgentState | null;
|
||
agentProvenance?: AgentProvenance | null;
|
||
terminalWidth: number;
|
||
};
|
||
}
|
||
| {
|
||
kind: "subagent_group";
|
||
id: string;
|
||
agents: Array<{
|
||
id: string;
|
||
type: string;
|
||
description: string;
|
||
status: "completed" | "error";
|
||
toolCount: number;
|
||
totalTokens: number;
|
||
agentURL: string | null;
|
||
error?: string;
|
||
}>;
|
||
}
|
||
| {
|
||
// Preview content committed early during approval to enable flicker-free UI
|
||
// When an approval's content is tall enough to overflow the viewport,
|
||
// we commit the preview to static and only show small approval options in dynamic
|
||
kind: "approval_preview";
|
||
id: string;
|
||
toolCallId: string;
|
||
toolName: string;
|
||
toolArgs: string;
|
||
// Optional precomputed/cached data for rendering
|
||
precomputedDiff?: AdvancedDiffSuccess;
|
||
planContent?: string; // For ExitPlanMode
|
||
planFilePath?: string; // For ExitPlanMode
|
||
}
|
||
| Line;
|
||
|
||
export default function App({
|
||
agentId: initialAgentId,
|
||
agentState: initialAgentState,
|
||
loadingState = "ready",
|
||
continueSession = false,
|
||
startupApproval = null,
|
||
startupApprovals = [],
|
||
messageHistory = [],
|
||
tokenStreaming = false,
|
||
agentProvenance = null,
|
||
}: {
|
||
agentId: string;
|
||
agentState?: AgentState | null;
|
||
loadingState?:
|
||
| "assembling"
|
||
| "importing"
|
||
| "initializing"
|
||
| "checking"
|
||
| "ready";
|
||
continueSession?: boolean;
|
||
startupApproval?: ApprovalRequest | null; // Deprecated: use startupApprovals
|
||
startupApprovals?: ApprovalRequest[];
|
||
messageHistory?: Message[];
|
||
tokenStreaming?: boolean;
|
||
agentProvenance?: AgentProvenance | null;
|
||
}) {
|
||
// Warm the model-access cache in the background so /model is fast on first open.
|
||
useEffect(() => {
|
||
prefetchAvailableModelHandles();
|
||
}, []);
|
||
|
||
// Track current agent (can change when swapping)
|
||
const [agentId, setAgentId] = useState(initialAgentId);
|
||
const [agentState, setAgentState] = useState(initialAgentState);
|
||
|
||
// Keep a ref to the current agentId for use in callbacks that need the latest value
|
||
const agentIdRef = useRef(agentId);
|
||
useEffect(() => {
|
||
agentIdRef.current = agentId;
|
||
telemetry.setCurrentAgentId(agentId);
|
||
}, [agentId]);
|
||
|
||
const resumeKey = useSuspend();
|
||
|
||
// Track previous prop values to detect actual prop changes (not internal state changes)
|
||
const prevInitialAgentIdRef = useRef(initialAgentId);
|
||
const prevInitialAgentStateRef = useRef(initialAgentState);
|
||
|
||
// Sync with prop changes (e.g., when parent updates from "loading" to actual ID)
|
||
// Only sync when the PROP actually changes, not when internal state changes
|
||
useEffect(() => {
|
||
if (initialAgentId !== prevInitialAgentIdRef.current) {
|
||
prevInitialAgentIdRef.current = initialAgentId;
|
||
agentIdRef.current = initialAgentId;
|
||
setAgentId(initialAgentId);
|
||
}
|
||
}, [initialAgentId]);
|
||
|
||
useEffect(() => {
|
||
if (initialAgentState !== prevInitialAgentStateRef.current) {
|
||
prevInitialAgentStateRef.current = initialAgentState;
|
||
setAgentState(initialAgentState);
|
||
}
|
||
}, [initialAgentState]);
|
||
|
||
// Set agent context for tools (especially Task tool)
|
||
useEffect(() => {
|
||
if (agentId) {
|
||
setCurrentAgentId(agentId);
|
||
}
|
||
}, [agentId]);
|
||
|
||
// Whether a stream is in flight (disables input)
|
||
// Uses synced state to keep ref in sync for reliable async checks
|
||
const [streaming, setStreaming, streamingRef] = useSyncedState(false);
|
||
|
||
// Guard ref for preventing concurrent processConversation calls
|
||
// Separate from streaming state which may be set early for UI responsiveness
|
||
// Tracks depth to allow intentional reentry while blocking parallel calls
|
||
const processingConversationRef = useRef(0);
|
||
|
||
// Whether an interrupt has been requested for the current stream
|
||
const [interruptRequested, setInterruptRequested] = useState(false);
|
||
|
||
// Whether a command is running (disables input but no streaming UI)
|
||
// Uses synced state to keep ref in sync for reliable async checks
|
||
const [commandRunning, setCommandRunning, commandRunningRef] =
|
||
useSyncedState(false);
|
||
|
||
// Profile load confirmation - when loading a profile and current agent is unsaved
|
||
const [profileConfirmPending, setProfileConfirmPending] = useState<{
|
||
name: string;
|
||
agentId: string;
|
||
cmdId: string;
|
||
} | null>(null);
|
||
|
||
// If we have approval requests, we should show the approval dialog instead of the input area
|
||
const [pendingApprovals, setPendingApprovals] = useState<ApprovalRequest[]>(
|
||
[],
|
||
);
|
||
const [approvalContexts, setApprovalContexts] = useState<ApprovalContext[]>(
|
||
[],
|
||
);
|
||
|
||
// Sequential approval: track results as user reviews each approval
|
||
const [approvalResults, setApprovalResults] = useState<
|
||
Array<
|
||
| { type: "approve"; approval: ApprovalRequest }
|
||
| { type: "deny"; approval: ApprovalRequest; reason: string }
|
||
>
|
||
>([]);
|
||
const [isExecutingTool, setIsExecutingTool] = useState(false);
|
||
const [queuedApprovalResults, setQueuedApprovalResults] = useState<
|
||
ApprovalResult[] | null
|
||
>(null);
|
||
const toolAbortControllerRef = useRef<AbortController | null>(null);
|
||
|
||
// Track auto-handled results to combine with user decisions
|
||
const [autoHandledResults, setAutoHandledResults] = useState<
|
||
Array<{
|
||
toolCallId: string;
|
||
result: ToolExecutionResult;
|
||
}>
|
||
>([]);
|
||
const [autoDeniedApprovals, setAutoDeniedApprovals] = useState<
|
||
Array<{
|
||
approval: ApprovalRequest;
|
||
reason: string;
|
||
}>
|
||
>([]);
|
||
|
||
// Bash mode: cache bash commands to prefix next user message
|
||
// Use ref instead of state to avoid stale closure issues in onSubmit
|
||
const bashCommandCacheRef = useRef<Array<{ input: string; output: string }>>(
|
||
[],
|
||
);
|
||
|
||
// Derive current approval from pending approvals and results
|
||
// This is the approval currently being shown to the user
|
||
const currentApproval = pendingApprovals[approvalResults.length];
|
||
const currentApprovalContext = approvalContexts[approvalResults.length];
|
||
const activeApprovalId = currentApproval?.toolCallId ?? null;
|
||
|
||
// Build Sets/Maps for three approval states (excluding the active one):
|
||
// - pendingIds: undecided approvals (index > approvalResults.length)
|
||
// - queuedIds: decided but not yet executed (index < approvalResults.length)
|
||
// Used to render appropriate stubs while one approval is active
|
||
const {
|
||
pendingIds,
|
||
queuedIds,
|
||
approvalMap,
|
||
stubDescriptions,
|
||
queuedDecisions,
|
||
} = useMemo(() => {
|
||
const pending = new Set<string>();
|
||
const queued = new Set<string>();
|
||
const map = new Map<string, ApprovalRequest>();
|
||
const descriptions = new Map<string, string>();
|
||
const decisions = new Map<
|
||
string,
|
||
{ type: "approve" | "deny"; reason?: string }
|
||
>();
|
||
|
||
// Helper to compute stub description - called once per approval during memo
|
||
const computeStubDescription = (
|
||
approval: ApprovalRequest,
|
||
): string | undefined => {
|
||
try {
|
||
const args = JSON.parse(approval.toolArgs || "{}");
|
||
|
||
if (
|
||
isFileEditTool(approval.toolName) ||
|
||
isFileWriteTool(approval.toolName)
|
||
) {
|
||
return args.file_path || undefined;
|
||
}
|
||
if (isShellTool(approval.toolName)) {
|
||
const cmd =
|
||
typeof args.command === "string"
|
||
? args.command
|
||
: Array.isArray(args.command)
|
||
? args.command.join(" ")
|
||
: "";
|
||
return cmd.length > 50 ? `${cmd.slice(0, 50)}...` : cmd || undefined;
|
||
}
|
||
if (isPatchTool(approval.toolName)) {
|
||
return "patch operation";
|
||
}
|
||
return undefined;
|
||
} catch {
|
||
return undefined;
|
||
}
|
||
};
|
||
|
||
const activeIndex = approvalResults.length;
|
||
|
||
for (let i = 0; i < pendingApprovals.length; i++) {
|
||
const approval = pendingApprovals[i];
|
||
if (!approval?.toolCallId || approval.toolCallId === activeApprovalId) {
|
||
continue;
|
||
}
|
||
|
||
const id = approval.toolCallId;
|
||
map.set(id, approval);
|
||
|
||
const desc = computeStubDescription(approval);
|
||
if (desc) {
|
||
descriptions.set(id, desc);
|
||
}
|
||
|
||
if (i < activeIndex) {
|
||
// Decided but not yet executed
|
||
queued.add(id);
|
||
const result = approvalResults[i];
|
||
if (result) {
|
||
decisions.set(id, {
|
||
type: result.type,
|
||
reason: result.type === "deny" ? result.reason : undefined,
|
||
});
|
||
}
|
||
} else {
|
||
// Undecided (waiting in queue)
|
||
pending.add(id);
|
||
}
|
||
}
|
||
|
||
return {
|
||
pendingIds: pending,
|
||
queuedIds: queued,
|
||
approvalMap: map,
|
||
stubDescriptions: descriptions,
|
||
queuedDecisions: decisions,
|
||
};
|
||
}, [pendingApprovals, approvalResults, activeApprovalId]);
|
||
|
||
// Overlay/selector state - only one can be open at a time
|
||
type ActiveOverlay =
|
||
| "model"
|
||
| "toolset"
|
||
| "system"
|
||
| "agent"
|
||
| "resume"
|
||
| "search"
|
||
| "subagent"
|
||
| "feedback"
|
||
| "memory"
|
||
| "pin"
|
||
| "new"
|
||
| "mcp"
|
||
| "help"
|
||
| "oauth"
|
||
| null;
|
||
const [activeOverlay, setActiveOverlay] = useState<ActiveOverlay>(null);
|
||
const [feedbackPrefill, setFeedbackPrefill] = useState("");
|
||
const closeOverlay = useCallback(() => {
|
||
setActiveOverlay(null);
|
||
setFeedbackPrefill("");
|
||
}, []);
|
||
|
||
// Pin dialog state
|
||
const [pinDialogLocal, setPinDialogLocal] = useState(false);
|
||
|
||
// Derived: check if any selector/overlay is open (blocks queue processing and hides input)
|
||
const anySelectorOpen = activeOverlay !== null;
|
||
|
||
// Other model/agent state
|
||
const [currentSystemPromptId, setCurrentSystemPromptId] = useState<
|
||
string | null
|
||
>("default");
|
||
const [currentToolset, setCurrentToolset] = useState<
|
||
| "codex"
|
||
| "codex_snake"
|
||
| "default"
|
||
| "gemini"
|
||
| "gemini_snake"
|
||
| "none"
|
||
| null
|
||
>(null);
|
||
const [llmConfig, setLlmConfig] = useState<LlmConfig | null>(null);
|
||
const llmConfigRef = useRef(llmConfig);
|
||
useEffect(() => {
|
||
llmConfigRef.current = llmConfig;
|
||
}, [llmConfig]);
|
||
const [currentModelId, setCurrentModelId] = useState<string | null>(null);
|
||
const [agentName, setAgentName] = useState<string | null>(null);
|
||
const [agentDescription, setAgentDescription] = useState<string | null>(null);
|
||
const [agentLastRunAt, setAgentLastRunAt] = useState<string | null>(null);
|
||
const currentModelLabel =
|
||
llmConfig?.model_endpoint_type && llmConfig?.model
|
||
? `${llmConfig.model_endpoint_type}/${llmConfig.model}`
|
||
: (llmConfig?.model ?? null);
|
||
const currentModelDisplay = currentModelLabel
|
||
? (getModelDisplayName(currentModelLabel) ??
|
||
currentModelLabel.split("/").pop())
|
||
: null;
|
||
const currentModelProvider = llmConfig?.provider_name ?? null;
|
||
|
||
// Token streaming preference (can be toggled at runtime)
|
||
const [tokenStreamingEnabled, setTokenStreamingEnabled] =
|
||
useState(tokenStreaming);
|
||
|
||
// Live, approximate token counter (resets each turn)
|
||
const [tokenCount, setTokenCount] = useState(0);
|
||
|
||
// Current thinking message (rotates each turn)
|
||
const [thinkingMessage, setThinkingMessage] = useState(
|
||
getRandomThinkingVerb(),
|
||
);
|
||
|
||
// Session stats tracking
|
||
const sessionStatsRef = useRef(new SessionStats());
|
||
|
||
// Wire up session stats to telemetry for safety net handlers
|
||
useEffect(() => {
|
||
telemetry.setSessionStatsGetter(() =>
|
||
sessionStatsRef.current.getSnapshot(),
|
||
);
|
||
|
||
// Cleanup on unmount (defensive, prevents potential memory leak)
|
||
return () => {
|
||
telemetry.setSessionStatsGetter(undefined);
|
||
};
|
||
}, []);
|
||
|
||
// Show exit stats on exit (double Ctrl+C)
|
||
const [showExitStats, setShowExitStats] = useState(false);
|
||
|
||
// Track if we've sent the session context for this CLI session
|
||
const hasSentSessionContextRef = useRef(false);
|
||
|
||
// Track conversation turn count for periodic memory reminders
|
||
const turnCountRef = useRef(0);
|
||
|
||
// Static items (things that are done rendering and can be frozen)
|
||
const [staticItems, setStaticItems] = useState<StaticItem[]>([]);
|
||
|
||
// Track committed ids to avoid duplicates
|
||
const emittedIdsRef = useRef<Set<string>>(new Set());
|
||
|
||
// Guard to append welcome snapshot only once
|
||
const welcomeCommittedRef = useRef(false);
|
||
|
||
// AbortController for stream cancellation
|
||
const abortControllerRef = useRef<AbortController | null>(null);
|
||
|
||
// Track if user wants to cancel (persists across state updates)
|
||
const userCancelledRef = useRef(false);
|
||
|
||
// Retry counter for transient LLM API errors (ref for synchronous access in loop)
|
||
const llmApiErrorRetriesRef = useRef(0);
|
||
|
||
// Message queue state for queueing messages during streaming
|
||
const [messageQueue, setMessageQueue] = useState<string[]>([]);
|
||
|
||
// Queue cancellation: when any message is queued, we send cancel and wait for stream to end
|
||
const waitingForQueueCancelRef = useRef(false);
|
||
const queueSnapshotRef = useRef<string[]>([]);
|
||
const [restoreQueueOnCancel, setRestoreQueueOnCancel] = useState(false);
|
||
const restoreQueueOnCancelRef = useRef(restoreQueueOnCancel);
|
||
useEffect(() => {
|
||
restoreQueueOnCancelRef.current = restoreQueueOnCancel;
|
||
}, [restoreQueueOnCancel]);
|
||
|
||
// Helper to check if agent is busy (streaming, executing tool, or running command)
|
||
// Uses refs for synchronous access outside React's closure system
|
||
// biome-ignore lint/correctness/useExhaustiveDependencies: refs are stable objects, .current is read dynamically
|
||
const isAgentBusy = useCallback(() => {
|
||
return (
|
||
streamingRef.current ||
|
||
isExecutingTool ||
|
||
commandRunningRef.current ||
|
||
abortControllerRef.current !== null
|
||
);
|
||
}, [isExecutingTool]);
|
||
|
||
// Helper to wrap async handlers that need to close overlay and lock input
|
||
// Closes overlay and sets commandRunning before executing, releases lock in finally
|
||
const withCommandLock = useCallback(
|
||
async (asyncFn: () => Promise<void>) => {
|
||
setActiveOverlay(null);
|
||
setCommandRunning(true);
|
||
try {
|
||
await asyncFn();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
},
|
||
[setCommandRunning],
|
||
);
|
||
|
||
// Track terminal shrink events to refresh static output (prevents wrapped leftovers)
|
||
const columns = useTerminalWidth();
|
||
const prevColumnsRef = useRef(columns);
|
||
const [staticRenderEpoch, setStaticRenderEpoch] = useState(0);
|
||
useEffect(() => {
|
||
const prev = prevColumnsRef.current;
|
||
if (columns === prev) return;
|
||
|
||
if (
|
||
columns < prev &&
|
||
typeof process !== "undefined" &&
|
||
process.stdout &&
|
||
"write" in process.stdout &&
|
||
process.stdout.isTTY
|
||
) {
|
||
process.stdout.write(CLEAR_SCREEN_AND_HOME);
|
||
}
|
||
|
||
setStaticRenderEpoch((epoch) => epoch + 1);
|
||
prevColumnsRef.current = columns;
|
||
}, [columns]);
|
||
|
||
// Commit immutable/finished lines into the historical log
|
||
const commitEligibleLines = useCallback((b: Buffers) => {
|
||
const newlyCommitted: StaticItem[] = [];
|
||
let firstTaskIndex = -1;
|
||
|
||
// Check if there are any in-progress Task tool_calls
|
||
const hasInProgress = hasInProgressTaskToolCalls(
|
||
b.order,
|
||
b.byId,
|
||
emittedIdsRef.current,
|
||
);
|
||
|
||
// Collect finished Task tool_calls for grouping
|
||
const finishedTaskToolCalls = collectFinishedTaskToolCalls(
|
||
b.order,
|
||
b.byId,
|
||
emittedIdsRef.current,
|
||
hasInProgress,
|
||
);
|
||
|
||
// Commit regular lines (non-Task tools)
|
||
for (const id of b.order) {
|
||
if (emittedIdsRef.current.has(id)) continue;
|
||
const ln = b.byId.get(id);
|
||
if (!ln) continue;
|
||
if (ln.kind === "user" || ln.kind === "error" || ln.kind === "status") {
|
||
emittedIdsRef.current.add(id);
|
||
newlyCommitted.push({ ...ln });
|
||
continue;
|
||
}
|
||
// Commands with phase should only commit when finished
|
||
if (ln.kind === "command" || ln.kind === "bash_command") {
|
||
if (!ln.phase || ln.phase === "finished") {
|
||
emittedIdsRef.current.add(id);
|
||
newlyCommitted.push({ ...ln });
|
||
}
|
||
continue;
|
||
}
|
||
// Handle Task tool_calls specially - track position but don't add individually
|
||
// (unless there's no subagent data, in which case commit as regular tool call)
|
||
if (ln.kind === "tool_call" && ln.name && isTaskTool(ln.name)) {
|
||
// Check if this specific Task tool has subagent data (will be grouped)
|
||
const hasSubagentData = finishedTaskToolCalls.some(
|
||
(tc) => tc.lineId === id,
|
||
);
|
||
if (hasSubagentData) {
|
||
// Has subagent data - will be grouped later
|
||
if (firstTaskIndex === -1) {
|
||
firstTaskIndex = newlyCommitted.length;
|
||
}
|
||
continue;
|
||
}
|
||
// No subagent data (e.g., backfilled from history) - commit as regular tool call
|
||
if (ln.phase === "finished") {
|
||
emittedIdsRef.current.add(id);
|
||
newlyCommitted.push({ ...ln });
|
||
}
|
||
continue;
|
||
}
|
||
if ("phase" in ln && ln.phase === "finished") {
|
||
emittedIdsRef.current.add(id);
|
||
newlyCommitted.push({ ...ln });
|
||
// Note: We intentionally don't cleanup precomputedDiffs here because
|
||
// the Static area renders AFTER this function returns (on next React tick),
|
||
// and the diff needs to be available for ToolCallMessage to render.
|
||
// The diffs will be cleaned up when the session ends or on next session start.
|
||
}
|
||
}
|
||
|
||
// If we collected Task tool_calls (all are finished), create a subagent_group
|
||
if (finishedTaskToolCalls.length > 0) {
|
||
// Mark all as emitted
|
||
for (const tc of finishedTaskToolCalls) {
|
||
emittedIdsRef.current.add(tc.lineId);
|
||
}
|
||
|
||
const groupItem = createSubagentGroupItem(finishedTaskToolCalls);
|
||
|
||
// Insert at the position of the first Task tool_call
|
||
newlyCommitted.splice(
|
||
firstTaskIndex >= 0 ? firstTaskIndex : newlyCommitted.length,
|
||
0,
|
||
groupItem,
|
||
);
|
||
|
||
// Clear these agents from the subagent store
|
||
clearSubagentsByIds(groupItem.agents.map((a) => a.id));
|
||
}
|
||
|
||
if (newlyCommitted.length > 0) {
|
||
setStaticItems((prev) => [...prev, ...newlyCommitted]);
|
||
}
|
||
}, []);
|
||
|
||
// Render-ready transcript
|
||
const [lines, setLines] = useState<Line[]>([]);
|
||
|
||
// Canonical buffers stored in a ref (mutated by onChunk), PERSISTED for session
|
||
const buffersRef = useRef(createBuffers());
|
||
|
||
// Track whether we've already backfilled history (should only happen once)
|
||
const hasBackfilledRef = useRef(false);
|
||
|
||
// Cache precomputed diffs from approval dialogs for tool return rendering
|
||
// Key: toolCallId or "toolCallId:filePath" for Patch operations
|
||
const precomputedDiffsRef = useRef<Map<string, AdvancedDiffSuccess>>(
|
||
new Map(),
|
||
);
|
||
|
||
// Store the last plan file path for post-approval rendering
|
||
// (needed because plan mode is exited before rendering the result)
|
||
const lastPlanFilePathRef = useRef<string | null>(null);
|
||
|
||
// Track which approval tool call IDs have had their previews eagerly committed
|
||
// This prevents double-committing when the approval changes
|
||
const eagerCommittedPreviewsRef = useRef<Set<string>>(new Set());
|
||
|
||
// Recompute UI state from buffers after each streaming chunk
|
||
const refreshDerived = useCallback(() => {
|
||
const b = buffersRef.current;
|
||
setTokenCount(b.tokenCount);
|
||
const newLines = toLines(b);
|
||
setLines(newLines);
|
||
commitEligibleLines(b);
|
||
}, [commitEligibleLines]);
|
||
|
||
// Throttled version for streaming updates (~60fps max)
|
||
const refreshDerivedThrottled = useCallback(() => {
|
||
// Use a ref to track pending refresh
|
||
if (!buffersRef.current.pendingRefresh) {
|
||
buffersRef.current.pendingRefresh = true;
|
||
// Capture the current generation to detect if resume invalidates this refresh
|
||
const capturedGeneration = buffersRef.current.commitGeneration || 0;
|
||
setTimeout(() => {
|
||
buffersRef.current.pendingRefresh = false;
|
||
// Skip refresh if stream was interrupted - prevents stale updates appearing
|
||
// after user cancels. Normal stream completion still renders (interrupted=false).
|
||
// Also skip if commitGeneration changed - this means a resume is in progress and
|
||
// committing now would lock in the stale "Interrupted by user" state.
|
||
if (
|
||
!buffersRef.current.interrupted &&
|
||
(buffersRef.current.commitGeneration || 0) === capturedGeneration
|
||
) {
|
||
refreshDerived();
|
||
}
|
||
}, 16); // ~60fps
|
||
}
|
||
}, [refreshDerived]);
|
||
|
||
// Restore pending approval from startup when ready
|
||
// All approvals (including fancy UI tools) go through pendingApprovals
|
||
// The render logic determines which UI to show based on tool name
|
||
useEffect(() => {
|
||
// Use new plural field if available, otherwise wrap singular in array for backward compat
|
||
const approvals =
|
||
startupApprovals?.length > 0
|
||
? startupApprovals
|
||
: startupApproval
|
||
? [startupApproval]
|
||
: [];
|
||
|
||
if (loadingState === "ready" && approvals.length > 0) {
|
||
// All approvals go through the same flow - UI rendering decides which dialog to show
|
||
setPendingApprovals(approvals);
|
||
|
||
// Analyze approval contexts for all restored approvals
|
||
const analyzeStartupApprovals = async () => {
|
||
try {
|
||
const contexts = await Promise.all(
|
||
approvals.map(async (approval) => {
|
||
const parsedArgs = safeJsonParseOr<Record<string, unknown>>(
|
||
approval.toolArgs,
|
||
{},
|
||
);
|
||
return await analyzeToolApproval(approval.toolName, parsedArgs);
|
||
}),
|
||
);
|
||
setApprovalContexts(contexts);
|
||
} catch (error) {
|
||
// If analysis fails, leave context as null (will show basic options)
|
||
console.error("Failed to analyze startup approvals:", error);
|
||
}
|
||
};
|
||
|
||
analyzeStartupApprovals();
|
||
}
|
||
}, [loadingState, startupApproval, startupApprovals]);
|
||
|
||
// Eager commit for ExitPlanMode: Always commit plan preview to staticItems
|
||
// This keeps the dynamic area small (just approval options) to avoid flicker
|
||
useEffect(() => {
|
||
if (!currentApproval) return;
|
||
if (currentApproval.toolName !== "ExitPlanMode") return;
|
||
|
||
const toolCallId = currentApproval.toolCallId;
|
||
if (!toolCallId) return;
|
||
|
||
// Already committed preview for this approval?
|
||
if (eagerCommittedPreviewsRef.current.has(toolCallId)) return;
|
||
|
||
const planFilePath = permissionMode.getPlanFilePath();
|
||
if (!planFilePath) return;
|
||
|
||
try {
|
||
const { readFileSync, existsSync } = require("node:fs");
|
||
if (!existsSync(planFilePath)) return;
|
||
|
||
const planContent = readFileSync(planFilePath, "utf-8");
|
||
|
||
// Commit preview to static area
|
||
const previewItem: StaticItem = {
|
||
kind: "approval_preview",
|
||
id: `approval-preview-${toolCallId}`,
|
||
toolCallId,
|
||
toolName: currentApproval.toolName,
|
||
toolArgs: currentApproval.toolArgs || "{}",
|
||
planContent,
|
||
planFilePath,
|
||
};
|
||
|
||
setStaticItems((prev) => [...prev, previewItem]);
|
||
eagerCommittedPreviewsRef.current.add(toolCallId);
|
||
|
||
// Also capture plan file path for post-approval rendering
|
||
lastPlanFilePathRef.current = planFilePath;
|
||
} catch {
|
||
// Failed to read plan, don't commit preview
|
||
}
|
||
}, [currentApproval]);
|
||
|
||
// Backfill message history when resuming (only once)
|
||
useEffect(() => {
|
||
if (
|
||
loadingState === "ready" &&
|
||
messageHistory.length > 0 &&
|
||
!hasBackfilledRef.current
|
||
) {
|
||
// Set flag FIRST to prevent double-execution in strict mode
|
||
hasBackfilledRef.current = true;
|
||
// Append welcome snapshot FIRST so it appears above history
|
||
if (!welcomeCommittedRef.current) {
|
||
welcomeCommittedRef.current = true;
|
||
setStaticItems((prev) => [
|
||
...prev,
|
||
{
|
||
kind: "welcome",
|
||
id: `welcome-${Date.now().toString(36)}`,
|
||
snapshot: {
|
||
continueSession,
|
||
agentState,
|
||
agentProvenance,
|
||
terminalWidth: columns,
|
||
},
|
||
},
|
||
]);
|
||
}
|
||
// Use backfillBuffers to properly populate the transcript from history
|
||
backfillBuffers(buffersRef.current, messageHistory);
|
||
|
||
// Add combined status at the END so user sees it without scrolling
|
||
const statusId = `status-resumed-${Date.now().toString(36)}`;
|
||
const cwd = process.cwd();
|
||
const shortCwd = cwd.startsWith(process.env.HOME || "")
|
||
? `~${cwd.slice((process.env.HOME || "").length)}`
|
||
: cwd;
|
||
|
||
// Check if agent is pinned (locally or globally)
|
||
const isPinned = agentState?.id
|
||
? settingsManager.getLocalPinnedAgents().includes(agentState.id) ||
|
||
settingsManager.getGlobalPinnedAgents().includes(agentState.id)
|
||
: false;
|
||
|
||
// Build status message
|
||
const agentName = agentState?.name || "Unnamed Agent";
|
||
const headerMessage = `Connecting to **${agentName}** (last used in ${shortCwd})`;
|
||
|
||
// Command hints - for pinned agents show /memory, for unpinned show /pin
|
||
const commandHints = isPinned
|
||
? [
|
||
"→ **/memory** view your agent's memory blocks",
|
||
"→ **/init** initialize your agent's memory",
|
||
"→ **/remember** teach your agent",
|
||
"→ **/agents** list agents",
|
||
"→ **/ade** open in the browser (web UI)",
|
||
]
|
||
: [
|
||
"→ **/pin** save + name your agent",
|
||
"→ **/init** initialize your agent's memory",
|
||
"→ **/remember** teach your agent",
|
||
"→ **/agents** list agents",
|
||
"→ **/ade** open in the browser (web UI)",
|
||
];
|
||
|
||
const statusLines = [headerMessage, ...commandHints];
|
||
buffersRef.current.byId.set(statusId, {
|
||
kind: "status",
|
||
id: statusId,
|
||
lines: statusLines,
|
||
});
|
||
buffersRef.current.order.push(statusId);
|
||
|
||
refreshDerived();
|
||
commitEligibleLines(buffersRef.current);
|
||
}
|
||
}, [
|
||
loadingState,
|
||
messageHistory,
|
||
refreshDerived,
|
||
commitEligibleLines,
|
||
continueSession,
|
||
columns,
|
||
agentState,
|
||
agentProvenance,
|
||
]);
|
||
|
||
// Fetch llmConfig when agent is ready
|
||
useEffect(() => {
|
||
if (loadingState === "ready" && agentId && agentId !== "loading") {
|
||
const fetchConfig = async () => {
|
||
try {
|
||
const { getClient } = await import("../agent/client");
|
||
const client = await getClient();
|
||
const agent = await client.agents.retrieve(agentId);
|
||
setLlmConfig(agent.llm_config);
|
||
setAgentName(agent.name);
|
||
setAgentDescription(agent.description ?? null);
|
||
// Get last message timestamp from agent state if available
|
||
const lastRunCompletion = (agent as { last_run_completion?: string })
|
||
.last_run_completion;
|
||
setAgentLastRunAt(lastRunCompletion ?? null);
|
||
|
||
// Derive model ID from llm_config for ModelSelector
|
||
const agentModelHandle =
|
||
agent.llm_config.model_endpoint_type && agent.llm_config.model
|
||
? `${agent.llm_config.model_endpoint_type}/${agent.llm_config.model}`
|
||
: agent.llm_config.model;
|
||
const modelInfo = getModelInfo(agentModelHandle || "");
|
||
if (modelInfo) {
|
||
setCurrentModelId(modelInfo.id);
|
||
} else {
|
||
setCurrentModelId(agentModelHandle || null);
|
||
}
|
||
|
||
// Derive toolset from agent's model (not persisted, computed on resume)
|
||
if (agentModelHandle) {
|
||
const derivedToolset = isOpenAIModel(agentModelHandle)
|
||
? "codex"
|
||
: isGeminiModel(agentModelHandle)
|
||
? "gemini"
|
||
: "default";
|
||
setCurrentToolset(derivedToolset);
|
||
}
|
||
} catch (error) {
|
||
console.error("Error fetching agent config:", error);
|
||
}
|
||
};
|
||
fetchConfig();
|
||
}
|
||
}, [loadingState, agentId]);
|
||
|
||
// Helper to append an error to the transcript
|
||
// Also tracks the error in telemetry so we know an error was shown
|
||
const appendError = useCallback(
|
||
(message: string, skipTelemetry = false) => {
|
||
// Defensive: ensure message is always a string (guards against [object Object])
|
||
const text =
|
||
typeof message === "string"
|
||
? message
|
||
: message != null
|
||
? JSON.stringify(message)
|
||
: "[Unknown error]";
|
||
|
||
const id = uid("err");
|
||
buffersRef.current.byId.set(id, {
|
||
kind: "error",
|
||
id,
|
||
text,
|
||
});
|
||
buffersRef.current.order.push(id);
|
||
refreshDerived();
|
||
|
||
// Track error in telemetry (unless explicitly skipped for user-initiated actions)
|
||
if (!skipTelemetry) {
|
||
telemetry.trackError("ui_error", text, "error_display", {
|
||
modelId: currentModelId || undefined,
|
||
});
|
||
}
|
||
},
|
||
[refreshDerived, currentModelId],
|
||
);
|
||
|
||
// Core streaming function - iterative loop that processes conversation turns
|
||
const processConversation = useCallback(
|
||
async (
|
||
initialInput: Array<MessageCreate | ApprovalCreate>,
|
||
options?: { allowReentry?: boolean },
|
||
): Promise<void> => {
|
||
// Copy so we can safely mutate for retry recovery flows
|
||
const currentInput = [...initialInput];
|
||
const allowReentry = options?.allowReentry ?? false;
|
||
|
||
// Guard against concurrent processConversation calls
|
||
// This can happen if user submits two messages in quick succession
|
||
// Uses dedicated ref (not streamingRef) since streaming may be set early for UI responsiveness
|
||
if (processingConversationRef.current > 0 && !allowReentry) {
|
||
return;
|
||
}
|
||
processingConversationRef.current += 1;
|
||
|
||
// Reset retry counter for new conversation turns (fresh budget per user message)
|
||
if (!allowReentry) {
|
||
llmApiErrorRetriesRef.current = 0;
|
||
}
|
||
|
||
// Track last run ID for error reporting (accessible in catch block)
|
||
let currentRunId: string | undefined;
|
||
|
||
try {
|
||
// Check if user hit escape before we started
|
||
if (userCancelledRef.current) {
|
||
userCancelledRef.current = false; // Reset for next time
|
||
return;
|
||
}
|
||
|
||
setStreaming(true);
|
||
abortControllerRef.current = new AbortController();
|
||
|
||
// Clear any stale pending tool calls from previous turns
|
||
// If we're sending a new message, old pending state is no longer relevant
|
||
markIncompleteToolsAsCancelled(buffersRef.current);
|
||
// Reset interrupted flag since we're starting a fresh stream
|
||
buffersRef.current.interrupted = false;
|
||
|
||
// Clear completed subagents from the UI when starting a new turn
|
||
clearCompletedSubagents();
|
||
|
||
while (true) {
|
||
// Capture the signal BEFORE any async operations
|
||
// This prevents a race where handleInterrupt nulls the ref during await
|
||
const signal = abortControllerRef.current?.signal;
|
||
|
||
// Check if cancelled before starting new stream
|
||
if (signal?.aborted) {
|
||
setStreaming(false);
|
||
return;
|
||
}
|
||
|
||
// Stream one turn - use ref to always get the latest agentId
|
||
const stream = await sendMessageStream(
|
||
agentIdRef.current,
|
||
currentInput,
|
||
);
|
||
|
||
// Check again after network call - user may have pressed Escape during sendMessageStream
|
||
if (signal?.aborted) {
|
||
setStreaming(false);
|
||
return;
|
||
}
|
||
|
||
// Define callback to sync agent state on first message chunk
|
||
// This ensures the UI shows the correct model as early as possible
|
||
const syncAgentState = async () => {
|
||
try {
|
||
const client = await getClient();
|
||
const agent = await client.agents.retrieve(agentIdRef.current);
|
||
|
||
// Check if the model has changed by comparing llm_config
|
||
const currentModel = llmConfigRef.current?.model;
|
||
const currentEndpoint = llmConfigRef.current?.model_endpoint_type;
|
||
const agentModel = agent.llm_config.model;
|
||
const agentEndpoint = agent.llm_config.model_endpoint_type;
|
||
|
||
if (
|
||
currentModel !== agentModel ||
|
||
currentEndpoint !== agentEndpoint
|
||
) {
|
||
// Model has changed - update local state
|
||
setLlmConfig(agent.llm_config);
|
||
|
||
// Derive model ID from llm_config for ModelSelector
|
||
// Try to find matching model by handle in models.json
|
||
const { getModelInfo } = await import("../agent/model");
|
||
const agentModelHandle =
|
||
agent.llm_config.model_endpoint_type && agent.llm_config.model
|
||
? `${agent.llm_config.model_endpoint_type}/${agent.llm_config.model}`
|
||
: agent.llm_config.model;
|
||
|
||
const modelInfo = getModelInfo(agentModelHandle || "");
|
||
if (modelInfo) {
|
||
setCurrentModelId(modelInfo.id);
|
||
} else {
|
||
// Model not in models.json (e.g., BYOK model) - use handle as ID
|
||
setCurrentModelId(agentModelHandle || null);
|
||
}
|
||
|
||
// Also update agent state if other fields changed
|
||
setAgentName(agent.name);
|
||
setAgentDescription(agent.description ?? null);
|
||
const lastRunCompletion = (
|
||
agent as { last_run_completion?: string }
|
||
).last_run_completion;
|
||
setAgentLastRunAt(lastRunCompletion ?? null);
|
||
}
|
||
} catch (error) {
|
||
// Silently fail - don't interrupt the conversation flow
|
||
console.error("Failed to sync agent state:", error);
|
||
}
|
||
};
|
||
|
||
const {
|
||
stopReason,
|
||
approval,
|
||
approvals,
|
||
apiDurationMs,
|
||
lastRunId,
|
||
fallbackError,
|
||
} = await drainStreamWithResume(
|
||
stream,
|
||
buffersRef.current,
|
||
refreshDerivedThrottled,
|
||
signal, // Use captured signal, not ref (which may be nulled by handleInterrupt)
|
||
syncAgentState,
|
||
);
|
||
|
||
// Update currentRunId for error reporting in catch block
|
||
currentRunId = lastRunId ?? undefined;
|
||
|
||
// Track API duration
|
||
sessionStatsRef.current.endTurn(apiDurationMs);
|
||
sessionStatsRef.current.updateUsageFromBuffers(buffersRef.current);
|
||
|
||
const wasInterrupted = !!buffersRef.current.interrupted;
|
||
const wasAborted = !!signal?.aborted;
|
||
let stopReasonToHandle = wasAborted ? "cancelled" : stopReason;
|
||
|
||
// Immediate refresh after stream completes to show final state unless
|
||
// the user already cancelled (handleInterrupt rendered the UI).
|
||
if (!wasInterrupted) {
|
||
refreshDerived();
|
||
}
|
||
|
||
// If the turn was interrupted client-side but the backend had already emitted
|
||
// requires_approval, treat it as a cancel. This avoids re-entering approval flow
|
||
// and keeps queue-cancel flags consistent with the normal cancel branch below.
|
||
if (wasInterrupted && stopReasonToHandle === "requires_approval") {
|
||
stopReasonToHandle = "cancelled";
|
||
}
|
||
|
||
// Case 1: Turn ended normally
|
||
if (stopReasonToHandle === "end_turn") {
|
||
setStreaming(false);
|
||
llmApiErrorRetriesRef.current = 0; // Reset retry counter on success
|
||
|
||
// Send desktop notification when turn completes
|
||
// and we're not about to auto-send another queued message
|
||
if (!waitingForQueueCancelRef.current) {
|
||
sendDesktopNotification();
|
||
}
|
||
|
||
// Check if we were waiting for cancel but stream finished naturally
|
||
if (waitingForQueueCancelRef.current) {
|
||
if (restoreQueueOnCancelRef.current) {
|
||
// User hit ESC during queue cancel - abort the auto-send
|
||
setRestoreQueueOnCancel(false);
|
||
// Don't clear queue, don't send - let dequeue effect handle them one by one
|
||
} else {
|
||
// Auto-send concatenated message
|
||
// Clear the queue
|
||
setMessageQueue([]);
|
||
|
||
// Concatenate the snapshot
|
||
const concatenatedMessage = queueSnapshotRef.current.join("\n");
|
||
|
||
if (concatenatedMessage.trim()) {
|
||
onSubmitRef.current(concatenatedMessage);
|
||
}
|
||
}
|
||
|
||
// Reset flags
|
||
waitingForQueueCancelRef.current = false;
|
||
queueSnapshotRef.current = [];
|
||
}
|
||
|
||
return;
|
||
}
|
||
|
||
// Case 1.5: Stream was cancelled by user
|
||
if (stopReasonToHandle === "cancelled") {
|
||
setStreaming(false);
|
||
|
||
// Check if this cancel was triggered by queue threshold
|
||
if (waitingForQueueCancelRef.current) {
|
||
if (restoreQueueOnCancelRef.current) {
|
||
// User hit ESC during queue cancel - abort the auto-send
|
||
setRestoreQueueOnCancel(false);
|
||
// Don't clear queue, don't send - let dequeue effect handle them one by one
|
||
} else {
|
||
// Auto-send concatenated message
|
||
// Clear the queue
|
||
setMessageQueue([]);
|
||
|
||
// Concatenate the snapshot
|
||
const concatenatedMessage = queueSnapshotRef.current.join("\n");
|
||
|
||
if (concatenatedMessage.trim()) {
|
||
onSubmitRef.current(concatenatedMessage);
|
||
}
|
||
}
|
||
|
||
// Reset flags
|
||
waitingForQueueCancelRef.current = false;
|
||
queueSnapshotRef.current = [];
|
||
} else {
|
||
// Regular user cancellation - show error
|
||
if (!EAGER_CANCEL) {
|
||
appendError("Stream interrupted by user", true);
|
||
}
|
||
}
|
||
|
||
return;
|
||
}
|
||
|
||
// Case 2: Requires approval
|
||
if (stopReasonToHandle === "requires_approval") {
|
||
// Clear stale state immediately to prevent ID mismatch bugs
|
||
setAutoHandledResults([]);
|
||
setAutoDeniedApprovals([]);
|
||
|
||
// Use new approvals array, fallback to legacy approval for backward compat
|
||
const approvalsToProcess =
|
||
approvals && approvals.length > 0
|
||
? approvals
|
||
: approval
|
||
? [approval]
|
||
: [];
|
||
|
||
if (approvalsToProcess.length === 0) {
|
||
appendError(
|
||
`Unexpected empty approvals with stop reason: ${stopReason}`,
|
||
);
|
||
setStreaming(false);
|
||
return;
|
||
}
|
||
|
||
// If in quietCancel mode (user queued messages), auto-reject all approvals
|
||
// and send denials + queued messages together
|
||
if (waitingForQueueCancelRef.current) {
|
||
if (restoreQueueOnCancelRef.current) {
|
||
// User hit ESC during queue cancel - abort the auto-send
|
||
setRestoreQueueOnCancel(false);
|
||
// Don't clear queue, don't send - let dequeue effect handle them one by one
|
||
} else {
|
||
// Create denial results for all approvals
|
||
const denialResults = approvalsToProcess.map(
|
||
(approvalItem) => ({
|
||
type: "approval" as const,
|
||
tool_call_id: approvalItem.toolCallId,
|
||
approve: false,
|
||
reason: "User cancelled - new message queued",
|
||
}),
|
||
);
|
||
|
||
// Update buffers to show tools as cancelled
|
||
for (const approvalItem of approvalsToProcess) {
|
||
onChunk(buffersRef.current, {
|
||
message_type: "tool_return_message",
|
||
id: "dummy",
|
||
date: new Date().toISOString(),
|
||
tool_call_id: approvalItem.toolCallId,
|
||
tool_return: "Cancelled - user sent new message",
|
||
status: "error",
|
||
});
|
||
}
|
||
refreshDerived();
|
||
|
||
// Queue denial results to be sent with the queued message
|
||
setQueuedApprovalResults(denialResults);
|
||
|
||
// Get queued messages and clear queue
|
||
const concatenatedMessage = queueSnapshotRef.current.join("\n");
|
||
setMessageQueue([]);
|
||
|
||
// Send via onSubmit which will combine queuedApprovalResults + message
|
||
if (concatenatedMessage.trim()) {
|
||
onSubmitRef.current(concatenatedMessage);
|
||
}
|
||
}
|
||
|
||
// Reset flags
|
||
waitingForQueueCancelRef.current = false;
|
||
queueSnapshotRef.current = [];
|
||
setStreaming(false);
|
||
return;
|
||
}
|
||
|
||
// Check if user cancelled before starting permission checks
|
||
if (
|
||
userCancelledRef.current ||
|
||
abortControllerRef.current?.signal.aborted
|
||
) {
|
||
setStreaming(false);
|
||
markIncompleteToolsAsCancelled(buffersRef.current);
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
|
||
// Check permissions for all approvals (including fancy UI tools)
|
||
const approvalResults = await Promise.all(
|
||
approvalsToProcess.map(async (approvalItem) => {
|
||
// Check if approval is incomplete (missing name)
|
||
// Note: toolArgs can be empty string for tools with no arguments (e.g., EnterPlanMode)
|
||
if (!approvalItem.toolName) {
|
||
return {
|
||
approval: approvalItem,
|
||
permission: {
|
||
decision: "deny" as const,
|
||
reason:
|
||
"Tool call incomplete - missing name or arguments",
|
||
},
|
||
context: null,
|
||
};
|
||
}
|
||
|
||
const parsedArgs = safeJsonParseOr<Record<string, unknown>>(
|
||
approvalItem.toolArgs,
|
||
{},
|
||
);
|
||
const permission = await checkToolPermission(
|
||
approvalItem.toolName,
|
||
parsedArgs,
|
||
);
|
||
const context = await analyzeToolApproval(
|
||
approvalItem.toolName,
|
||
parsedArgs,
|
||
);
|
||
return { approval: approvalItem, permission, context };
|
||
}),
|
||
);
|
||
|
||
// Categorize approvals by permission decision
|
||
// Fancy UI tools should always go through their dialog, even if auto-allowed
|
||
const needsUserInput: typeof approvalResults = [];
|
||
const autoDenied: typeof approvalResults = [];
|
||
const autoAllowed: typeof approvalResults = [];
|
||
|
||
for (const ac of approvalResults) {
|
||
const { approval, permission } = ac;
|
||
let decision = permission.decision;
|
||
|
||
// Some tools always need user input regardless of yolo mode
|
||
if (
|
||
alwaysRequiresUserInput(approval.toolName) &&
|
||
decision === "allow"
|
||
) {
|
||
decision = "ask";
|
||
}
|
||
|
||
if (decision === "ask") {
|
||
needsUserInput.push(ac);
|
||
} else if (decision === "deny") {
|
||
autoDenied.push(ac);
|
||
} else {
|
||
// decision === "allow"
|
||
autoAllowed.push(ac);
|
||
}
|
||
}
|
||
|
||
// Precompute diffs for file edit tools before execution (both auto-allowed and needs-user-input)
|
||
// This is needed for inline approval UI to show diffs, and for post-approval rendering
|
||
for (const ac of [...autoAllowed, ...needsUserInput]) {
|
||
const toolName = ac.approval.toolName;
|
||
const toolCallId = ac.approval.toolCallId;
|
||
try {
|
||
const args = JSON.parse(ac.approval.toolArgs || "{}");
|
||
|
||
if (isFileWriteTool(toolName)) {
|
||
const filePath = args.file_path as string | undefined;
|
||
if (filePath) {
|
||
const result = computeAdvancedDiff({
|
||
kind: "write",
|
||
filePath,
|
||
content: (args.content as string) || "",
|
||
});
|
||
if (result.mode === "advanced") {
|
||
precomputedDiffsRef.current.set(toolCallId, result);
|
||
}
|
||
}
|
||
} else if (isFileEditTool(toolName)) {
|
||
const filePath = args.file_path as string | undefined;
|
||
if (filePath) {
|
||
// Check if it's a multi-edit (has edits array) or single edit
|
||
if (args.edits && Array.isArray(args.edits)) {
|
||
const result = computeAdvancedDiff({
|
||
kind: "multi_edit",
|
||
filePath,
|
||
edits: args.edits as Array<{
|
||
old_string: string;
|
||
new_string: string;
|
||
replace_all?: boolean;
|
||
}>,
|
||
});
|
||
if (result.mode === "advanced") {
|
||
precomputedDiffsRef.current.set(toolCallId, result);
|
||
}
|
||
} else {
|
||
const result = computeAdvancedDiff({
|
||
kind: "edit",
|
||
filePath,
|
||
oldString: (args.old_string as string) || "",
|
||
newString: (args.new_string as string) || "",
|
||
replaceAll: args.replace_all as boolean | undefined,
|
||
});
|
||
if (result.mode === "advanced") {
|
||
precomputedDiffsRef.current.set(toolCallId, result);
|
||
}
|
||
}
|
||
}
|
||
} else if (isPatchTool(toolName) && args.input) {
|
||
// Patch tools - parse hunks directly (patches ARE diffs)
|
||
const operations = parsePatchOperations(args.input as string);
|
||
for (const op of operations) {
|
||
const key = `${toolCallId}:${op.path}`;
|
||
if (op.kind === "add" || op.kind === "update") {
|
||
const result = parsePatchToAdvancedDiff(
|
||
op.patchLines,
|
||
op.path,
|
||
);
|
||
if (result) {
|
||
precomputedDiffsRef.current.set(key, result);
|
||
}
|
||
}
|
||
// Delete operations don't need diffs
|
||
}
|
||
}
|
||
} catch {
|
||
// Ignore errors in diff computation for auto-allowed tools
|
||
}
|
||
}
|
||
|
||
// Execute auto-allowed tools (sequential for writes, parallel for reads)
|
||
const autoAllowedResults = await executeAutoAllowedTools(
|
||
autoAllowed,
|
||
(chunk) => onChunk(buffersRef.current, chunk),
|
||
);
|
||
|
||
// Create denial results for auto-denied tools and update buffers
|
||
const autoDeniedResults = autoDenied.map((ac) => {
|
||
// Prefer the detailed reason over the short matchedRule name
|
||
// (e.g., reason contains plan file path info, matchedRule is just "plan mode")
|
||
const reason = ac.permission.reason
|
||
? `Permission denied: ${ac.permission.reason}`
|
||
: "matchedRule" in ac.permission && ac.permission.matchedRule
|
||
? `Permission denied by rule: ${ac.permission.matchedRule}`
|
||
: "Permission denied: Unknown reason";
|
||
|
||
// Update buffers with tool rejection for UI
|
||
onChunk(buffersRef.current, {
|
||
message_type: "tool_return_message",
|
||
id: "dummy",
|
||
date: new Date().toISOString(),
|
||
tool_call_id: ac.approval.toolCallId,
|
||
tool_return: `Error: request to call tool denied. User reason: ${reason}`,
|
||
status: "error",
|
||
stdout: null,
|
||
stderr: null,
|
||
});
|
||
|
||
return {
|
||
approval: ac.approval,
|
||
reason,
|
||
};
|
||
});
|
||
|
||
// If all are auto-handled, continue immediately without showing dialog
|
||
if (needsUserInput.length === 0) {
|
||
// Check if user cancelled before continuing
|
||
if (
|
||
userCancelledRef.current ||
|
||
abortControllerRef.current?.signal.aborted
|
||
) {
|
||
setStreaming(false);
|
||
markIncompleteToolsAsCancelled(buffersRef.current);
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
|
||
// Combine auto-allowed results + auto-denied responses
|
||
const allResults = [
|
||
...autoAllowedResults.map((ar) => ({
|
||
type: "tool" as const,
|
||
tool_call_id: ar.toolCallId,
|
||
tool_return: ar.result.toolReturn,
|
||
status: ar.result.status,
|
||
stdout: ar.result.stdout,
|
||
stderr: ar.result.stderr,
|
||
})),
|
||
...autoDeniedResults.map((ad) => ({
|
||
type: "approval" as const,
|
||
tool_call_id: ad.approval.toolCallId,
|
||
approve: false,
|
||
reason: ad.reason,
|
||
})),
|
||
];
|
||
|
||
// Check if user queued messages during auto-allowed tool execution
|
||
if (waitingForQueueCancelRef.current) {
|
||
if (restoreQueueOnCancelRef.current) {
|
||
// User hit ESC during queue cancel - abort the auto-send
|
||
setRestoreQueueOnCancel(false);
|
||
} else {
|
||
// Queue results to be sent with the queued message
|
||
setQueuedApprovalResults(allResults);
|
||
|
||
// Get queued messages and clear queue
|
||
const concatenatedMessage =
|
||
queueSnapshotRef.current.join("\n");
|
||
setMessageQueue([]);
|
||
|
||
// Send via onSubmit
|
||
if (concatenatedMessage.trim()) {
|
||
onSubmitRef.current(concatenatedMessage);
|
||
}
|
||
}
|
||
|
||
// Reset flags
|
||
waitingForQueueCancelRef.current = false;
|
||
queueSnapshotRef.current = [];
|
||
setStreaming(false);
|
||
return;
|
||
}
|
||
|
||
// Rotate to a new thinking message
|
||
setThinkingMessage(getRandomThinkingVerb());
|
||
refreshDerived();
|
||
|
||
await processConversation(
|
||
[
|
||
{
|
||
type: "approval",
|
||
approvals: allResults,
|
||
},
|
||
],
|
||
{ allowReentry: true },
|
||
);
|
||
return;
|
||
}
|
||
|
||
// Check again if user queued messages during auto-allowed tool execution
|
||
if (waitingForQueueCancelRef.current) {
|
||
if (restoreQueueOnCancelRef.current) {
|
||
// User hit ESC during queue cancel - abort the auto-send
|
||
setRestoreQueueOnCancel(false);
|
||
} else {
|
||
// Create denial results for tools that need user input
|
||
const denialResults = needsUserInput.map((ac) => ({
|
||
type: "approval" as const,
|
||
tool_call_id: ac.approval.toolCallId,
|
||
approve: false,
|
||
reason: "User cancelled - new message queued",
|
||
}));
|
||
|
||
// Update buffers to show tools as cancelled
|
||
for (const ac of needsUserInput) {
|
||
onChunk(buffersRef.current, {
|
||
message_type: "tool_return_message",
|
||
id: "dummy",
|
||
date: new Date().toISOString(),
|
||
tool_call_id: ac.approval.toolCallId,
|
||
tool_return: "Cancelled - user sent new message",
|
||
status: "error",
|
||
});
|
||
}
|
||
refreshDerived();
|
||
|
||
// Combine with auto-handled results and queue for sending
|
||
const allResults = [
|
||
...autoAllowedResults.map((ar) => ({
|
||
type: "tool" as const,
|
||
tool_call_id: ar.toolCallId,
|
||
tool_return: ar.result.toolReturn,
|
||
status: ar.result.status,
|
||
})),
|
||
...autoDeniedResults.map((ad) => ({
|
||
type: "approval" as const,
|
||
tool_call_id: ad.approval.toolCallId,
|
||
approve: false,
|
||
reason: ad.reason,
|
||
})),
|
||
...denialResults,
|
||
];
|
||
setQueuedApprovalResults(allResults);
|
||
|
||
// Get queued messages and clear queue
|
||
const concatenatedMessage = queueSnapshotRef.current.join("\n");
|
||
setMessageQueue([]);
|
||
|
||
// Send via onSubmit
|
||
if (concatenatedMessage.trim()) {
|
||
onSubmitRef.current(concatenatedMessage);
|
||
}
|
||
}
|
||
|
||
// Reset flags
|
||
waitingForQueueCancelRef.current = false;
|
||
queueSnapshotRef.current = [];
|
||
setStreaming(false);
|
||
return;
|
||
}
|
||
|
||
// Check if user cancelled before showing dialog
|
||
if (
|
||
userCancelledRef.current ||
|
||
abortControllerRef.current?.signal.aborted
|
||
) {
|
||
setStreaming(false);
|
||
markIncompleteToolsAsCancelled(buffersRef.current);
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
|
||
// Show approval dialog for tools that need user input
|
||
setPendingApprovals(needsUserInput.map((ac) => ac.approval));
|
||
setApprovalContexts(
|
||
needsUserInput
|
||
.map((ac) => ac.context)
|
||
.filter((ctx): ctx is ApprovalContext => ctx !== null),
|
||
);
|
||
setAutoHandledResults(autoAllowedResults);
|
||
setAutoDeniedApprovals(autoDeniedResults);
|
||
setStreaming(false);
|
||
// Notify user that approval is needed
|
||
sendDesktopNotification();
|
||
return;
|
||
}
|
||
|
||
// Unexpected stop reason (error, llm_api_error, etc.)
|
||
// Cache desync detection and last failure for consistent handling
|
||
const isApprovalPayload =
|
||
currentInput.length === 1 && currentInput[0]?.type === "approval";
|
||
|
||
// Capture the most recent error text in this turn (if any)
|
||
let latestErrorText: string | null = null;
|
||
for (let i = buffersRef.current.order.length - 1; i >= 0; i -= 1) {
|
||
const id = buffersRef.current.order[i];
|
||
if (!id) continue;
|
||
const entry = buffersRef.current.byId.get(id);
|
||
if (entry?.kind === "error" && typeof entry.text === "string") {
|
||
latestErrorText = entry.text;
|
||
break;
|
||
}
|
||
}
|
||
|
||
// Detect approval desync once per turn
|
||
const detailFromRun = await fetchRunErrorDetail(lastRunId);
|
||
const desyncDetected =
|
||
isApprovalStateDesyncError(detailFromRun) ||
|
||
isApprovalStateDesyncError(latestErrorText);
|
||
|
||
// Track last failure info so we can emit it if retries stop
|
||
const lastFailureMessage = latestErrorText || detailFromRun || null;
|
||
|
||
// Check for approval desync errors even if stop_reason isn't llm_api_error.
|
||
if (isApprovalPayload && desyncDetected) {
|
||
if (llmApiErrorRetriesRef.current < LLM_API_ERROR_MAX_RETRIES) {
|
||
llmApiErrorRetriesRef.current += 1;
|
||
const statusId = uid("status");
|
||
buffersRef.current.byId.set(statusId, {
|
||
kind: "status",
|
||
id: statusId,
|
||
lines: [
|
||
"Approval state desynced; resending keep-alive recovery prompt...",
|
||
],
|
||
});
|
||
buffersRef.current.order.push(statusId);
|
||
refreshDerived();
|
||
|
||
currentInput.splice(
|
||
0,
|
||
currentInput.length,
|
||
buildApprovalRecoveryMessage(),
|
||
);
|
||
|
||
// Remove the transient status before retrying
|
||
buffersRef.current.byId.delete(statusId);
|
||
buffersRef.current.order = buffersRef.current.order.filter(
|
||
(id) => id !== statusId,
|
||
);
|
||
refreshDerived();
|
||
continue;
|
||
}
|
||
|
||
// No retries left: emit the failure and exit
|
||
const errorToShow =
|
||
lastFailureMessage ||
|
||
`An error occurred during agent execution\n(run_id: ${lastRunId ?? "unknown"}, stop_reason: ${stopReasonToHandle})`;
|
||
appendError(errorToShow, true);
|
||
setStreaming(false);
|
||
sendDesktopNotification();
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
|
||
// Check if this is a retriable error (transient LLM API error)
|
||
const retriable = await isRetriableError(
|
||
stopReasonToHandle,
|
||
lastRunId,
|
||
);
|
||
|
||
if (
|
||
retriable &&
|
||
llmApiErrorRetriesRef.current < LLM_API_ERROR_MAX_RETRIES
|
||
) {
|
||
llmApiErrorRetriesRef.current += 1;
|
||
const attempt = llmApiErrorRetriesRef.current;
|
||
const delayMs = 1000 * 2 ** (attempt - 1); // 1s, 2s, 4s
|
||
|
||
// Show subtle grey status message
|
||
const statusId = uid("status");
|
||
const statusLines = [
|
||
"Unexpected downstream LLM API error, retrying...",
|
||
];
|
||
buffersRef.current.byId.set(statusId, {
|
||
kind: "status",
|
||
id: statusId,
|
||
lines: statusLines,
|
||
});
|
||
buffersRef.current.order.push(statusId);
|
||
refreshDerived();
|
||
|
||
// Wait before retry (check abort signal periodically for ESC cancellation)
|
||
let cancelled = false;
|
||
const startTime = Date.now();
|
||
while (Date.now() - startTime < delayMs) {
|
||
if (
|
||
abortControllerRef.current?.signal.aborted ||
|
||
userCancelledRef.current
|
||
) {
|
||
cancelled = true;
|
||
break;
|
||
}
|
||
await new Promise((resolve) => setTimeout(resolve, 100)); // Check every 100ms
|
||
}
|
||
|
||
// Remove status message
|
||
buffersRef.current.byId.delete(statusId);
|
||
buffersRef.current.order = buffersRef.current.order.filter(
|
||
(id) => id !== statusId,
|
||
);
|
||
refreshDerived();
|
||
|
||
if (!cancelled) {
|
||
// Retry by continuing the while loop (same currentInput)
|
||
continue;
|
||
}
|
||
// User pressed ESC - fall through to error handling
|
||
}
|
||
|
||
// Reset retry counter on non-retriable error (or max retries exceeded)
|
||
llmApiErrorRetriesRef.current = 0;
|
||
|
||
// Mark incomplete tool calls as finished to prevent stuck blinking UI
|
||
markIncompleteToolsAsCancelled(buffersRef.current);
|
||
|
||
// Track the error in telemetry
|
||
telemetry.trackError(
|
||
fallbackError
|
||
? "FallbackError"
|
||
: stopReasonToHandle || "unknown_stop_reason",
|
||
fallbackError ||
|
||
`Stream stopped with reason: ${stopReasonToHandle}`,
|
||
"message_stream",
|
||
{
|
||
modelId: currentModelId || undefined,
|
||
runId: lastRunId ?? undefined,
|
||
},
|
||
);
|
||
|
||
// If we have a client-side stream error (e.g., JSON parse error), show it directly
|
||
// Fallback error: no run_id available, show whatever error message we have
|
||
if (fallbackError) {
|
||
const errorMsg = lastRunId
|
||
? `Stream error: ${fallbackError}\n(run_id: ${lastRunId})`
|
||
: `Stream error: ${fallbackError}`;
|
||
appendError(errorMsg, true); // Skip telemetry - already tracked above
|
||
setStreaming(false);
|
||
sendDesktopNotification(); // Notify user of error
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
|
||
// Fetch error details from the run if available (server-side errors)
|
||
if (lastRunId) {
|
||
try {
|
||
const client = await getClient();
|
||
const run = await client.runs.retrieve(lastRunId);
|
||
|
||
// Check if run has error information in metadata
|
||
if (run.metadata?.error) {
|
||
const errorData = run.metadata.error as {
|
||
type?: string;
|
||
message?: string;
|
||
detail?: string;
|
||
};
|
||
|
||
// Pass structured error data to our formatter
|
||
const errorObject = {
|
||
error: {
|
||
error: errorData,
|
||
run_id: lastRunId,
|
||
},
|
||
};
|
||
const errorDetails = formatErrorDetails(
|
||
errorObject,
|
||
agentIdRef.current,
|
||
);
|
||
appendError(errorDetails, true); // Skip telemetry - already tracked above
|
||
} else {
|
||
// No error metadata, show generic error with run info
|
||
appendError(
|
||
`An error occurred during agent execution\n(run_id: ${lastRunId}, stop_reason: ${stopReason})`,
|
||
true, // Skip telemetry - already tracked above
|
||
);
|
||
}
|
||
} catch (_e) {
|
||
// If we can't fetch error details, show generic error
|
||
appendError(
|
||
`An error occurred during agent execution\n(run_id: ${lastRunId}, stop_reason: ${stopReason})\n(Unable to fetch additional error details from server)`,
|
||
true, // Skip telemetry - already tracked above
|
||
);
|
||
return;
|
||
}
|
||
} else {
|
||
// No run_id available - but this is unusual since errors should have run_ids
|
||
appendError(
|
||
`An error occurred during agent execution\n(stop_reason: ${stopReason})`,
|
||
true, // Skip telemetry - already tracked above
|
||
);
|
||
}
|
||
|
||
setStreaming(false);
|
||
sendDesktopNotification(); // Notify user of error
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
} catch (e) {
|
||
// Mark incomplete tool calls as cancelled to prevent stuck blinking UI
|
||
markIncompleteToolsAsCancelled(buffersRef.current);
|
||
|
||
// If using eager cancel and this is an abort error, silently ignore it
|
||
// The user already got "Stream interrupted by user" feedback from handleInterrupt
|
||
if (EAGER_CANCEL && e instanceof APIUserAbortError) {
|
||
setStreaming(false);
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
|
||
// Track error with enhanced context
|
||
const errorType =
|
||
e instanceof Error ? e.constructor.name : "UnknownError";
|
||
const errorMessage = e instanceof Error ? e.message : String(e);
|
||
|
||
// Extract HTTP status code if available (API errors often have this)
|
||
const httpStatus =
|
||
e &&
|
||
typeof e === "object" &&
|
||
"status" in e &&
|
||
typeof e.status === "number"
|
||
? e.status
|
||
: undefined;
|
||
|
||
telemetry.trackError(errorType, errorMessage, "message_stream", {
|
||
httpStatus,
|
||
modelId: currentModelId || undefined,
|
||
runId: currentRunId,
|
||
});
|
||
|
||
// Use comprehensive error formatting
|
||
const errorDetails = formatErrorDetails(e, agentIdRef.current);
|
||
appendError(errorDetails, true); // Skip telemetry - already tracked above with more context
|
||
setStreaming(false);
|
||
sendDesktopNotification(); // Notify user of error
|
||
refreshDerived();
|
||
} finally {
|
||
abortControllerRef.current = null;
|
||
processingConversationRef.current = Math.max(
|
||
0,
|
||
processingConversationRef.current - 1,
|
||
);
|
||
}
|
||
},
|
||
[
|
||
appendError,
|
||
refreshDerived,
|
||
refreshDerivedThrottled,
|
||
setStreaming,
|
||
currentModelId,
|
||
],
|
||
);
|
||
|
||
const handleExit = useCallback(async () => {
|
||
saveLastAgentBeforeExit();
|
||
|
||
// Track session end explicitly (before exit) with stats
|
||
const stats = sessionStatsRef.current.getSnapshot();
|
||
telemetry.trackSessionEnd(stats, "exit_command");
|
||
|
||
// Flush telemetry before exit
|
||
await telemetry.flush();
|
||
|
||
setShowExitStats(true);
|
||
// Give React time to render the stats, then exit
|
||
setTimeout(() => {
|
||
process.exit(0);
|
||
}, 100);
|
||
}, []);
|
||
|
||
// Handler when user presses UP/ESC to load queue into input for editing
|
||
const handleEnterQueueEditMode = useCallback(() => {
|
||
setMessageQueue([]);
|
||
}, []);
|
||
|
||
const handleInterrupt = useCallback(async () => {
|
||
// If we're executing client-side tools, abort them AND the main stream
|
||
if (isExecutingTool && toolAbortControllerRef.current) {
|
||
toolAbortControllerRef.current.abort();
|
||
|
||
// ALSO abort the main stream - don't leave it running
|
||
buffersRef.current.abortGeneration =
|
||
(buffersRef.current.abortGeneration || 0) + 1;
|
||
const toolsCancelled = markIncompleteToolsAsCancelled(buffersRef.current);
|
||
|
||
// Show interrupt feedback (yellow message if no tools were cancelled)
|
||
if (!toolsCancelled) {
|
||
appendError("Stream interrupted by user", true);
|
||
}
|
||
|
||
if (abortControllerRef.current) {
|
||
abortControllerRef.current.abort();
|
||
abortControllerRef.current = null;
|
||
}
|
||
|
||
userCancelledRef.current = true; // Prevent dequeue
|
||
setStreaming(false);
|
||
setIsExecutingTool(false);
|
||
refreshDerived();
|
||
|
||
// Delay flag reset to ensure React has flushed state updates before dequeue can fire.
|
||
// Use setTimeout(50) instead of setTimeout(0) - the longer delay ensures React's
|
||
// batched state updates have been fully processed before we allow the dequeue effect.
|
||
setTimeout(() => {
|
||
userCancelledRef.current = false;
|
||
}, 50);
|
||
|
||
return;
|
||
}
|
||
|
||
if (!streaming || interruptRequested) return;
|
||
|
||
// If we're in the middle of queue cancel, set flag to restore instead of auto-send
|
||
if (waitingForQueueCancelRef.current) {
|
||
setRestoreQueueOnCancel(true);
|
||
// Don't reset flags - let the cancel complete naturally
|
||
}
|
||
|
||
// If EAGER_CANCEL is enabled, immediately stop everything client-side first
|
||
if (EAGER_CANCEL) {
|
||
// Prevent multiple handleInterrupt calls while state updates are pending
|
||
setInterruptRequested(true);
|
||
|
||
// Set interrupted flag FIRST, before abort() triggers any async work.
|
||
// This ensures onChunk and other guards see interrupted=true immediately.
|
||
buffersRef.current.abortGeneration =
|
||
(buffersRef.current.abortGeneration || 0) + 1;
|
||
const toolsCancelled = markIncompleteToolsAsCancelled(buffersRef.current);
|
||
|
||
// NOW abort the stream - interrupted flag is already set
|
||
if (abortControllerRef.current) {
|
||
abortControllerRef.current.abort();
|
||
abortControllerRef.current = null; // Clear ref so isAgentBusy() returns false
|
||
}
|
||
|
||
// Set cancellation flag to prevent processConversation from starting
|
||
userCancelledRef.current = true;
|
||
|
||
// Stop streaming and show error message (unless tool calls were cancelled,
|
||
// since the tool result will show "Interrupted by user")
|
||
setStreaming(false);
|
||
if (!toolsCancelled) {
|
||
appendError("Stream interrupted by user", true);
|
||
}
|
||
refreshDerived();
|
||
|
||
// Clear any pending approvals since we're cancelling
|
||
setPendingApprovals([]);
|
||
setApprovalContexts([]);
|
||
setApprovalResults([]);
|
||
setAutoHandledResults([]);
|
||
setAutoDeniedApprovals([]);
|
||
|
||
// Send cancel request to backend asynchronously (fire-and-forget)
|
||
// Don't wait for it or show errors since user already got feedback
|
||
getClient()
|
||
.then((client) => client.agents.messages.cancel(agentId))
|
||
.catch(() => {
|
||
// Silently ignore - cancellation already happened client-side
|
||
});
|
||
|
||
// Reset cancellation flags after cleanup is complete.
|
||
// Use setTimeout(50) instead of setTimeout(0) to ensure React has fully processed
|
||
// the streaming=false state before we allow the dequeue effect to start a new conversation.
|
||
// This prevents the "Maximum update depth exceeded" infinite render loop.
|
||
setTimeout(() => {
|
||
userCancelledRef.current = false;
|
||
setInterruptRequested(false);
|
||
}, 50);
|
||
|
||
return;
|
||
} else {
|
||
setInterruptRequested(true);
|
||
try {
|
||
const client = await getClient();
|
||
await client.agents.messages.cancel(agentId);
|
||
|
||
if (abortControllerRef.current) {
|
||
abortControllerRef.current.abort();
|
||
}
|
||
} catch (e) {
|
||
const errorDetails = formatErrorDetails(e, agentId);
|
||
appendError(`Failed to interrupt stream: ${errorDetails}`);
|
||
setInterruptRequested(false);
|
||
}
|
||
}
|
||
}, [
|
||
agentId,
|
||
streaming,
|
||
interruptRequested,
|
||
appendError,
|
||
isExecutingTool,
|
||
refreshDerived,
|
||
setStreaming,
|
||
]);
|
||
|
||
// Keep ref to latest processConversation to avoid circular deps in useEffect
|
||
const processConversationRef = useRef(processConversation);
|
||
useEffect(() => {
|
||
processConversationRef.current = processConversation;
|
||
}, [processConversation]);
|
||
|
||
const handleAgentSelect = useCallback(
|
||
async (targetAgentId: string, _opts?: { profileName?: string }) => {
|
||
// Close selector immediately
|
||
setActiveOverlay(null);
|
||
|
||
// Skip if already on this agent (no async work needed, queue can proceed)
|
||
if (targetAgentId === agentId) {
|
||
const label = agentName || targetAgentId.slice(0, 12);
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/pinned",
|
||
output: `Already on "${label}"`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
|
||
// Lock input for async operation (set before any await to prevent queue processing)
|
||
setCommandRunning(true);
|
||
|
||
const inputCmd = "/pinned";
|
||
const cmdId = uid("cmd");
|
||
|
||
// Show loading indicator while switching
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: inputCmd,
|
||
output: "Switching agent...",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
try {
|
||
const client = await getClient();
|
||
// Fetch new agent
|
||
const agent = await client.agents.retrieve(targetAgentId);
|
||
|
||
// Fetch agent's message history
|
||
const messagesPage = await client.agents.messages.list(targetAgentId);
|
||
const messages = messagesPage.items;
|
||
|
||
// Update project settings with new agent
|
||
await updateProjectSettings({ lastAgent: targetAgentId });
|
||
|
||
// Clear current transcript and static items
|
||
buffersRef.current.byId.clear();
|
||
buffersRef.current.order = [];
|
||
buffersRef.current.tokenCount = 0;
|
||
emittedIdsRef.current.clear();
|
||
setStaticItems([]);
|
||
setStaticRenderEpoch((e) => e + 1);
|
||
|
||
// Reset turn counter for memory reminders when switching agents
|
||
turnCountRef.current = 0;
|
||
|
||
// Update agent state - also update ref immediately for any code that runs before re-render
|
||
agentIdRef.current = targetAgentId;
|
||
setAgentId(targetAgentId);
|
||
setAgentState(agent);
|
||
setAgentName(agent.name);
|
||
setLlmConfig(agent.llm_config);
|
||
|
||
// Build success command
|
||
const agentUrl = `https://app.letta.com/projects/default-project/agents/${targetAgentId}`;
|
||
const successOutput = `Resumed "${agent.name || targetAgentId}"\n⎿ ${agentUrl}`;
|
||
const successItem: StaticItem = {
|
||
kind: "command",
|
||
id: uid("cmd"),
|
||
input: inputCmd,
|
||
output: successOutput,
|
||
phase: "finished",
|
||
success: true,
|
||
};
|
||
|
||
// Backfill message history with visual separator, then success command at end
|
||
if (messages.length > 0) {
|
||
hasBackfilledRef.current = false;
|
||
backfillBuffers(buffersRef.current, messages);
|
||
// Collect backfilled items
|
||
const backfilledItems: StaticItem[] = [];
|
||
for (const id of buffersRef.current.order) {
|
||
const ln = buffersRef.current.byId.get(id);
|
||
if (!ln) continue;
|
||
emittedIdsRef.current.add(id);
|
||
backfilledItems.push({ ...ln } as StaticItem);
|
||
}
|
||
// Add separator before backfilled messages, then success at end
|
||
const separator = {
|
||
kind: "separator" as const,
|
||
id: uid("sep"),
|
||
};
|
||
setStaticItems([separator, ...backfilledItems, successItem]);
|
||
setLines(toLines(buffersRef.current));
|
||
hasBackfilledRef.current = true;
|
||
} else {
|
||
setStaticItems([successItem]);
|
||
setLines(toLines(buffersRef.current));
|
||
}
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
const errorCmdId = uid("cmd");
|
||
buffersRef.current.byId.set(errorCmdId, {
|
||
kind: "command",
|
||
id: errorCmdId,
|
||
input: inputCmd,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
buffersRef.current.order.push(errorCmdId);
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
},
|
||
[refreshDerived, agentId, agentName, setCommandRunning],
|
||
);
|
||
|
||
// Handle creating a new agent and switching to it
|
||
const handleCreateNewAgent = useCallback(
|
||
async (name: string) => {
|
||
// Close dialog immediately
|
||
setActiveOverlay(null);
|
||
|
||
// Lock input for async operation
|
||
setCommandRunning(true);
|
||
|
||
const inputCmd = "/new";
|
||
const cmdId = uid("cmd");
|
||
|
||
// Show "Creating..." status while we wait
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: inputCmd,
|
||
output: `Creating agent "${name}"...`,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
try {
|
||
// Create the new agent
|
||
const { agent } = await createAgent(name);
|
||
|
||
// Update project settings with new agent
|
||
await updateProjectSettings({ lastAgent: agent.id });
|
||
|
||
// Clear current transcript and static items
|
||
buffersRef.current.byId.clear();
|
||
buffersRef.current.order = [];
|
||
buffersRef.current.tokenCount = 0;
|
||
emittedIdsRef.current.clear();
|
||
setStaticItems([]);
|
||
setStaticRenderEpoch((e) => e + 1);
|
||
|
||
// Reset turn counter for memory reminders
|
||
turnCountRef.current = 0;
|
||
|
||
// Update agent state
|
||
agentIdRef.current = agent.id;
|
||
setAgentId(agent.id);
|
||
setAgentState(agent);
|
||
setAgentName(agent.name);
|
||
setLlmConfig(agent.llm_config);
|
||
|
||
// Build success message with hints
|
||
const agentUrl = `https://app.letta.com/projects/default-project/agents/${agent.id}`;
|
||
const successOutput = [
|
||
`Created **${agent.name || agent.id}** (use /pin to save)`,
|
||
`⎿ ${agentUrl}`,
|
||
`⎿ Tip: use /init to initialize your agent's memory system!`,
|
||
].join("\n");
|
||
|
||
const separator = {
|
||
kind: "separator" as const,
|
||
id: uid("sep"),
|
||
};
|
||
const successItem: StaticItem = {
|
||
kind: "command",
|
||
id: uid("cmd"),
|
||
input: inputCmd,
|
||
output: successOutput,
|
||
phase: "finished",
|
||
success: true,
|
||
};
|
||
|
||
setStaticItems([separator, successItem]);
|
||
// Sync lines display after clearing buffers
|
||
setLines(toLines(buffersRef.current));
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: inputCmd,
|
||
output: `Failed to create agent: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
},
|
||
[refreshDerived, agentId, setCommandRunning],
|
||
);
|
||
|
||
// Handle bash mode command submission
|
||
// Uses the same shell runner as the Bash tool for consistency
|
||
const handleBashSubmit = useCallback(
|
||
async (command: string) => {
|
||
const cmdId = uid("bash");
|
||
|
||
// Add running bash_command line
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "bash_command",
|
||
id: cmdId,
|
||
input: command,
|
||
output: "",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
try {
|
||
// Use the same spawnCommand as the Bash tool for consistent behavior
|
||
const { spawnCommand } = await import("../tools/impl/Bash.js");
|
||
const { getShellEnv } = await import("../tools/impl/shellEnv.js");
|
||
|
||
const result = await spawnCommand(command, {
|
||
cwd: process.cwd(),
|
||
env: getShellEnv(),
|
||
timeout: 30000, // 30 second timeout
|
||
});
|
||
|
||
// Combine stdout and stderr for output
|
||
const output = (result.stdout + result.stderr).trim();
|
||
const success = result.exitCode === 0;
|
||
|
||
// Update line with output
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "bash_command",
|
||
id: cmdId,
|
||
input: command,
|
||
output: output || (success ? "" : `Exit code: ${result.exitCode}`),
|
||
phase: "finished",
|
||
success,
|
||
});
|
||
|
||
// Cache for next user message
|
||
bashCommandCacheRef.current.push({
|
||
input: command,
|
||
output: output || (success ? "" : `Exit code: ${result.exitCode}`),
|
||
});
|
||
} catch (error: unknown) {
|
||
// Handle command errors (timeout, abort, etc.)
|
||
const errOutput =
|
||
error instanceof Error
|
||
? (error as { stderr?: string; stdout?: string }).stderr ||
|
||
(error as { stdout?: string }).stdout ||
|
||
error.message
|
||
: String(error);
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "bash_command",
|
||
id: cmdId,
|
||
input: command,
|
||
output: errOutput,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
|
||
// Still cache for next user message (even failures are visible to agent)
|
||
bashCommandCacheRef.current.push({ input: command, output: errOutput });
|
||
}
|
||
|
||
refreshDerived();
|
||
},
|
||
[refreshDerived],
|
||
);
|
||
|
||
// biome-ignore lint/correctness/useExhaustiveDependencies: refs read .current dynamically, complex callback with intentional deps
|
||
const onSubmit = useCallback(
|
||
async (message?: string): Promise<{ submitted: boolean }> => {
|
||
const msg = message?.trim() ?? "";
|
||
|
||
// Handle profile load confirmation (Enter to continue)
|
||
if (profileConfirmPending && !msg) {
|
||
// User pressed Enter with empty input - proceed with loading
|
||
const { name, agentId: targetAgentId, cmdId } = profileConfirmPending;
|
||
buffersRef.current.byId.delete(cmdId);
|
||
const orderIdx = buffersRef.current.order.indexOf(cmdId);
|
||
if (orderIdx !== -1) buffersRef.current.order.splice(orderIdx, 1);
|
||
refreshDerived();
|
||
setProfileConfirmPending(null);
|
||
await handleAgentSelect(targetAgentId, { profileName: name });
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Cancel profile confirmation if user types something else
|
||
if (profileConfirmPending && msg) {
|
||
const { cmdId } = profileConfirmPending;
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/profile load ${profileConfirmPending.name}`,
|
||
output: "Cancelled",
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
setProfileConfirmPending(null);
|
||
// Continue processing the new message
|
||
}
|
||
|
||
if (!msg) return { submitted: false };
|
||
|
||
// Track user input (agent_id automatically added from telemetry.currentAgentId)
|
||
telemetry.trackUserInput(msg, "user", currentModelId || "unknown");
|
||
|
||
// Block submission if waiting for explicit user action (approvals)
|
||
// In this case, input is hidden anyway, so this shouldn't happen
|
||
if (pendingApprovals.length > 0) {
|
||
return { submitted: false };
|
||
}
|
||
|
||
// Queue message if agent is busy (streaming, executing tool, or running command)
|
||
// This allows messages to queue up while agent is working
|
||
|
||
// Reset cancellation flag before queue check - this ensures queued messages
|
||
// can be dequeued even if the user just cancelled. The dequeue effect checks
|
||
// userCancelledRef.current, so we must clear it here to prevent blocking.
|
||
userCancelledRef.current = false;
|
||
|
||
if (isAgentBusy()) {
|
||
setMessageQueue((prev) => {
|
||
const newQueue = [...prev, msg];
|
||
|
||
// For slash commands, just queue and wait - don't interrupt the agent.
|
||
// For regular messages, cancel the stream so the new message can be sent.
|
||
const isSlashCommand = msg.startsWith("/");
|
||
|
||
if (
|
||
!isSlashCommand &&
|
||
streamingRef.current &&
|
||
!waitingForQueueCancelRef.current
|
||
) {
|
||
waitingForQueueCancelRef.current = true;
|
||
queueSnapshotRef.current = [...newQueue];
|
||
|
||
// Send cancel request to backend (fire-and-forget)
|
||
getClient()
|
||
.then((client) => client.agents.messages.cancel(agentId))
|
||
.then(() => {})
|
||
.catch(() => {
|
||
// Reset flag if cancel fails
|
||
waitingForQueueCancelRef.current = false;
|
||
});
|
||
}
|
||
|
||
return newQueue;
|
||
});
|
||
return { submitted: true }; // Clears input
|
||
}
|
||
|
||
// Note: userCancelledRef.current was already reset above before the queue check
|
||
// to ensure the dequeue effect isn't blocked by a stale cancellation flag.
|
||
|
||
let aliasedMsg = msg;
|
||
if (msg === "exit" || msg === "quit") {
|
||
aliasedMsg = "/exit";
|
||
}
|
||
|
||
// Handle commands (messages starting with "/")
|
||
if (aliasedMsg.startsWith("/")) {
|
||
const trimmed = aliasedMsg.trim();
|
||
|
||
// Special handling for /model command - opens selector
|
||
if (trimmed === "/model") {
|
||
setActiveOverlay("model");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /toolset command - opens selector
|
||
if (trimmed === "/toolset") {
|
||
setActiveOverlay("toolset");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /ade command - open agent in browser
|
||
if (trimmed === "/ade") {
|
||
const adeUrl = `https://app.letta.com/agents/${agentId}`;
|
||
const cmdId = uid("cmd");
|
||
|
||
// Fire-and-forget browser open
|
||
import("open")
|
||
.then(({ default: open }) => open(adeUrl, { wait: false }))
|
||
.catch(() => {
|
||
// Silently ignore - user can use the URL from the output
|
||
});
|
||
|
||
// Always show the URL in case browser doesn't open
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/ade",
|
||
output: `Opening ADE...\n→ ${adeUrl}`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /system command - opens system prompt selector
|
||
if (trimmed === "/system") {
|
||
setActiveOverlay("system");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /subagents command - opens subagent manager
|
||
if (trimmed === "/subagents") {
|
||
setActiveOverlay("subagent");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /memory command - opens memory viewer
|
||
if (trimmed === "/memory") {
|
||
setActiveOverlay("memory");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /mcp command - manage MCP servers
|
||
if (msg.trim().startsWith("/mcp")) {
|
||
const mcpCtx: McpCommandContext = {
|
||
buffersRef,
|
||
refreshDerived,
|
||
setCommandRunning,
|
||
};
|
||
|
||
// Check for subcommand by looking at the first word after /mcp
|
||
const afterMcp = msg.trim().slice(4).trim(); // Remove "/mcp" prefix
|
||
const firstWord = afterMcp.split(/\s+/)[0]?.toLowerCase();
|
||
|
||
// /mcp - open MCP server selector
|
||
if (!firstWord) {
|
||
setActiveOverlay("mcp");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// /mcp add --transport <type> <name> <url/command> [options]
|
||
if (firstWord === "add") {
|
||
// Pass the full command string after "add" to preserve quotes
|
||
const afterAdd = afterMcp.slice(firstWord.length).trim();
|
||
await handleMcpAdd(mcpCtx, msg, afterAdd);
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Unknown subcommand
|
||
handleMcpUsage(mcpCtx, msg);
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /connect command - OAuth connection
|
||
if (msg.trim().startsWith("/connect")) {
|
||
const parts = msg.trim().split(/\s+/);
|
||
const provider = parts[1]?.toLowerCase();
|
||
const hasCode = parts.length > 2;
|
||
|
||
// If no code provided and provider is claude, show the OAuth dialog
|
||
if (provider === "claude" && !hasCode) {
|
||
setActiveOverlay("oauth");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Otherwise (with code or invalid provider), use existing handler
|
||
const { handleConnect } = await import("./commands/connect");
|
||
await handleConnect(
|
||
{
|
||
buffersRef,
|
||
refreshDerived,
|
||
setCommandRunning,
|
||
},
|
||
msg,
|
||
);
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /disconnect command - remove OAuth connection
|
||
if (msg.trim().startsWith("/disconnect")) {
|
||
const { handleDisconnect } = await import("./commands/connect");
|
||
await handleDisconnect(
|
||
{
|
||
buffersRef,
|
||
refreshDerived,
|
||
setCommandRunning,
|
||
},
|
||
msg,
|
||
);
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /help command - opens help dialog
|
||
if (trimmed === "/help") {
|
||
setActiveOverlay("help");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /usage command - show session stats
|
||
if (trimmed === "/usage") {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: trimmed,
|
||
output: "Fetching usage statistics...",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
// Fetch balance and display stats asynchronously
|
||
(async () => {
|
||
try {
|
||
const stats = sessionStatsRef.current.getSnapshot();
|
||
|
||
// Try to fetch balance info (only works for Letta Cloud)
|
||
// Silently skip if endpoint not available (not deployed yet or self-hosted)
|
||
let balance:
|
||
| {
|
||
total_balance: number;
|
||
monthly_credit_balance: number;
|
||
purchased_credit_balance: number;
|
||
billing_tier: string;
|
||
}
|
||
| undefined;
|
||
|
||
try {
|
||
const settings = settingsManager.getSettings();
|
||
const baseURL =
|
||
process.env.LETTA_BASE_URL ||
|
||
settings.env?.LETTA_BASE_URL ||
|
||
"https://api.letta.com";
|
||
const apiKey =
|
||
process.env.LETTA_API_KEY || settings.env?.LETTA_API_KEY;
|
||
|
||
const balanceResponse = await fetch(
|
||
`${baseURL}/v1/metadata/balance`,
|
||
{
|
||
headers: {
|
||
"Content-Type": "application/json",
|
||
Authorization: `Bearer ${apiKey}`,
|
||
"X-Letta-Source": "letta-code",
|
||
},
|
||
},
|
||
);
|
||
|
||
if (balanceResponse.ok) {
|
||
balance = (await balanceResponse.json()) as {
|
||
total_balance: number;
|
||
monthly_credit_balance: number;
|
||
purchased_credit_balance: number;
|
||
billing_tier: string;
|
||
};
|
||
}
|
||
} catch {
|
||
// Silently skip balance info if endpoint not available
|
||
}
|
||
|
||
const output = formatUsageStats({
|
||
stats,
|
||
balance,
|
||
});
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: trimmed,
|
||
output,
|
||
phase: "finished",
|
||
success: true,
|
||
dimOutput: true,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: trimmed,
|
||
output: `Error fetching usage: ${error instanceof Error ? error.message : String(error)}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
}
|
||
})();
|
||
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /exit command - exit without stats
|
||
if (trimmed === "/exit") {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: trimmed,
|
||
output: "See ya!",
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
handleExit();
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /logout command - clear credentials and exit
|
||
if (trimmed === "/logout") {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "Logging out...",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
const { settingsManager } = await import("../settings-manager");
|
||
const currentSettings =
|
||
await settingsManager.getSettingsWithSecureTokens();
|
||
|
||
// Revoke refresh token on server if we have one
|
||
if (currentSettings.refreshToken) {
|
||
const { revokeToken } = await import("../auth/oauth");
|
||
await revokeToken(currentSettings.refreshToken);
|
||
}
|
||
|
||
// Clear all credentials including secrets
|
||
await settingsManager.logout();
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output:
|
||
"✓ Logged out successfully. Run 'letta' to re-authenticate.",
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
|
||
saveLastAgentBeforeExit();
|
||
|
||
// Track session end explicitly (before exit) with stats
|
||
const stats = sessionStatsRef.current.getSnapshot();
|
||
telemetry.trackSessionEnd(stats, "logout");
|
||
|
||
// Flush telemetry before exit
|
||
await telemetry.flush();
|
||
|
||
// Exit after a brief delay to show the message
|
||
setTimeout(() => process.exit(0), 500);
|
||
} catch (error) {
|
||
let errorOutput = formatErrorDetails(error, agentId);
|
||
|
||
// Add helpful tip for summarization failures
|
||
if (errorOutput.includes("Summarization failed")) {
|
||
errorOutput +=
|
||
"\n\nTip: Use /clear instead to clear the current message buffer.";
|
||
}
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorOutput}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /stream command - toggle and save
|
||
if (msg.trim() === "/stream") {
|
||
const newValue = !tokenStreamingEnabled;
|
||
|
||
// Immediately add command to transcript with "running" phase and loading message
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `${newValue ? "Enabling" : "Disabling"} token streaming...`,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
// Lock input during async operation
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
setTokenStreamingEnabled(newValue);
|
||
|
||
// Save to settings
|
||
const { settingsManager } = await import("../settings-manager");
|
||
settingsManager.updateSettings({ tokenStreaming: newValue });
|
||
|
||
// Update the same command with final result
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Token streaming ${newValue ? "enabled" : "disabled"}`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
// Mark command as failed
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
// Unlock input
|
||
setCommandRunning(false);
|
||
}
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /clear command - reset conversation
|
||
if (msg.trim() === "/clear") {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "Clearing conversation...",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
const client = await getClient();
|
||
await client.agents.messages.reset(agentId, {
|
||
add_default_initial_messages: false,
|
||
});
|
||
|
||
// Clear local buffers and static items
|
||
// buffersRef.current.byId.clear();
|
||
// buffersRef.current.order = [];
|
||
// buffersRef.current.tokenCount = 0;
|
||
// emittedIdsRef.current.clear();
|
||
// setStaticItems([]);
|
||
|
||
// Reset turn counter for memory reminders
|
||
turnCountRef.current = 0;
|
||
|
||
// Update command with success
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "Conversation cleared",
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /compact command - summarize conversation history
|
||
if (msg.trim() === "/compact") {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "Compacting conversation history...",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
const client = await getClient();
|
||
// SDK types are out of date - compact returns CompactionResponse, not void
|
||
const result = (await client.agents.messages.compact(
|
||
agentId,
|
||
)) as unknown as {
|
||
num_messages_before: number;
|
||
num_messages_after: number;
|
||
summary: string;
|
||
};
|
||
|
||
// Format success message with before/after counts and summary
|
||
const outputLines = [
|
||
`Compaction completed. Message buffer length reduced from ${result.num_messages_before} to ${result.num_messages_after}.`,
|
||
"",
|
||
`Summary: ${result.summary}`,
|
||
];
|
||
|
||
// Update command with success
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: outputLines.join("\n"),
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
let errorOutput: string;
|
||
|
||
// Check for summarization failure - format it cleanly
|
||
const apiError = error as {
|
||
status?: number;
|
||
error?: { detail?: string };
|
||
};
|
||
const detail = apiError?.error?.detail;
|
||
if (
|
||
apiError?.status === 400 &&
|
||
detail?.includes("Summarization failed")
|
||
) {
|
||
// Clean format for this specific error, but preserve raw JSON
|
||
const cleanDetail = detail.replace(/^\d{3}:\s*/, "");
|
||
const rawJson = JSON.stringify(apiError.error);
|
||
errorOutput = [
|
||
`Request failed (code=400)`,
|
||
`Raw: ${rawJson}`,
|
||
`Detail: ${cleanDetail}`,
|
||
"",
|
||
"Tip: Use /clear instead to clear the current message buffer.",
|
||
].join("\n");
|
||
} else {
|
||
errorOutput = formatErrorDetails(error, agentId);
|
||
}
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorOutput}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /rename command - rename the agent
|
||
if (msg.trim().startsWith("/rename")) {
|
||
const parts = msg.trim().split(/\s+/);
|
||
const newName = parts.slice(1).join(" ");
|
||
|
||
if (!newName) {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "Please provide a new name: /rename <name>",
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Validate the name before sending to API
|
||
const validationError = validateAgentName(newName);
|
||
if (validationError) {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: validationError,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
return { submitted: true };
|
||
}
|
||
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Renaming agent to "${newName}"...`,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
const client = await getClient();
|
||
await client.agents.update(agentId, { name: newName });
|
||
setAgentName(newName);
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Agent renamed to "${newName}"`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /description command - update agent description
|
||
if (msg.trim().startsWith("/description")) {
|
||
const parts = msg.trim().split(/\s+/);
|
||
const newDescription = parts.slice(1).join(" ");
|
||
|
||
if (!newDescription) {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "Please provide a description: /description <text>",
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
return { submitted: true };
|
||
}
|
||
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "Updating description...",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
const client = await getClient();
|
||
await client.agents.update(agentId, {
|
||
description: newDescription,
|
||
});
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Description updated to "${newDescription}"`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /agents command - show agent browser
|
||
// /resume, /pinned, /profiles are hidden aliases
|
||
if (
|
||
msg.trim() === "/agents" ||
|
||
msg.trim() === "/resume" ||
|
||
msg.trim() === "/pinned" ||
|
||
msg.trim() === "/profiles"
|
||
) {
|
||
setActiveOverlay("resume");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /search command - show message search
|
||
if (msg.trim() === "/search") {
|
||
setActiveOverlay("search");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /profile command - manage local profiles
|
||
if (msg.trim().startsWith("/profile")) {
|
||
const parts = msg.trim().split(/\s+/);
|
||
const subcommand = parts[1]?.toLowerCase();
|
||
const profileName = parts.slice(2).join(" ");
|
||
|
||
const profileCtx: ProfileCommandContext = {
|
||
buffersRef,
|
||
refreshDerived,
|
||
agentId,
|
||
agentName: agentName || "",
|
||
setCommandRunning,
|
||
setAgentName,
|
||
};
|
||
|
||
// /profile - open agent browser (now points to /agents)
|
||
if (!subcommand) {
|
||
setActiveOverlay("resume");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// /profile save <name>
|
||
if (subcommand === "save") {
|
||
await handleProfileSave(profileCtx, msg, profileName);
|
||
return { submitted: true };
|
||
}
|
||
|
||
// /profile load <name>
|
||
if (subcommand === "load") {
|
||
const validation = validateProfileLoad(
|
||
profileCtx,
|
||
msg,
|
||
profileName,
|
||
);
|
||
if (validation.errorMessage) {
|
||
return { submitted: true };
|
||
}
|
||
|
||
if (validation.needsConfirmation && validation.targetAgentId) {
|
||
// Show warning and wait for confirmation
|
||
const cmdId = addCommandResult(
|
||
buffersRef,
|
||
refreshDerived,
|
||
msg,
|
||
"Warning: Current agent is not saved to any profile.\nPress Enter to continue, or type anything to cancel.",
|
||
false,
|
||
"running",
|
||
);
|
||
setProfileConfirmPending({
|
||
name: profileName,
|
||
agentId: validation.targetAgentId,
|
||
cmdId,
|
||
});
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Current agent is saved, proceed with loading
|
||
if (validation.targetAgentId) {
|
||
await handleAgentSelect(validation.targetAgentId, {
|
||
profileName,
|
||
});
|
||
}
|
||
return { submitted: true };
|
||
}
|
||
|
||
// /profile delete <name>
|
||
if (subcommand === "delete") {
|
||
handleProfileDelete(profileCtx, msg, profileName);
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Unknown subcommand
|
||
handleProfileUsage(profileCtx, msg);
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /new command - create new agent dialog
|
||
if (msg.trim() === "/new") {
|
||
setActiveOverlay("new");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /pin command - pin current agent to project (or globally with -g)
|
||
if (msg.trim() === "/pin" || msg.trim().startsWith("/pin ")) {
|
||
const argsStr = msg.trim().slice(4).trim();
|
||
|
||
// Parse args to check if name was provided
|
||
const parts = argsStr.split(/\s+/).filter(Boolean);
|
||
let hasNameArg = false;
|
||
let isLocal = false;
|
||
|
||
for (const part of parts) {
|
||
if (part === "-l" || part === "--local") {
|
||
isLocal = true;
|
||
} else {
|
||
hasNameArg = true;
|
||
}
|
||
}
|
||
|
||
// If no name provided, show the pin dialog
|
||
if (!hasNameArg) {
|
||
setPinDialogLocal(isLocal);
|
||
setActiveOverlay("pin");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Name was provided, use existing behavior
|
||
const profileCtx: ProfileCommandContext = {
|
||
buffersRef,
|
||
refreshDerived,
|
||
agentId,
|
||
agentName: agentName || "",
|
||
setCommandRunning,
|
||
setAgentName,
|
||
};
|
||
await handlePin(profileCtx, msg, argsStr);
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /unpin command - unpin current agent from project (or globally with -g)
|
||
if (msg.trim() === "/unpin" || msg.trim().startsWith("/unpin ")) {
|
||
const profileCtx: ProfileCommandContext = {
|
||
buffersRef,
|
||
refreshDerived,
|
||
agentId,
|
||
agentName: agentName || "",
|
||
setCommandRunning,
|
||
setAgentName,
|
||
};
|
||
const argsStr = msg.trim().slice(6).trim();
|
||
handleUnpin(profileCtx, msg, argsStr);
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /bg command - show background shell processes
|
||
if (msg.trim() === "/bg") {
|
||
const { backgroundProcesses } = await import(
|
||
"../tools/impl/process_manager"
|
||
);
|
||
const cmdId = uid("cmd");
|
||
|
||
let output: string;
|
||
if (backgroundProcesses.size === 0) {
|
||
output = "No background processes running";
|
||
} else {
|
||
const lines = ["Background processes:"];
|
||
for (const [id, proc] of backgroundProcesses) {
|
||
const status =
|
||
proc.status === "running"
|
||
? "running"
|
||
: proc.status === "completed"
|
||
? `completed (exit ${proc.exitCode})`
|
||
: `failed (exit ${proc.exitCode})`;
|
||
lines.push(` ${id}: ${proc.command} [${status}]`);
|
||
}
|
||
output = lines.join("\n");
|
||
}
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /download command - download agent file
|
||
if (msg.trim() === "/download") {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "Downloading agent file...",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
const client = await getClient();
|
||
const fileContent = await client.agents.exportFile(agentId);
|
||
const fileName = `${agentId}.af`;
|
||
writeFileSync(fileName, JSON.stringify(fileContent, null, 2));
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `AgentFile downloaded to ${fileName}`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /skill command - enter skill creation mode
|
||
if (trimmed.startsWith("/skill")) {
|
||
const cmdId = uid("cmd");
|
||
|
||
// Extract optional description after `/skill`
|
||
const [, ...rest] = trimmed.split(/\s+/);
|
||
const description = rest.join(" ").trim();
|
||
|
||
const initialOutput = description
|
||
? `Starting skill creation for: ${description}`
|
||
: "Starting skill creation. I’ll load the creating-skills skill and ask a few questions about the skill you want to build...";
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: initialOutput,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
// Import the skill-creation prompt
|
||
const { SKILL_CREATOR_PROMPT } = await import(
|
||
"../agent/promptAssets.js"
|
||
);
|
||
|
||
// Build system-reminder content for skill creation
|
||
const userDescriptionLine = description
|
||
? `\n\nUser-provided skill description:\n${description}`
|
||
: "\n\nThe user did not provide a description with /skill. Ask what kind of skill they want to create before proceeding.";
|
||
|
||
const skillMessage = `<system-reminder>\n${SKILL_CREATOR_PROMPT}${userDescriptionLine}\n</system-reminder>`;
|
||
|
||
// Mark command as finished before sending message
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output:
|
||
"Entered skill creation mode. Answer the assistant’s questions to design your new skill.",
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
|
||
// Process conversation with the skill-creation prompt
|
||
await processConversation([
|
||
{
|
||
type: "message",
|
||
role: "user",
|
||
content: skillMessage,
|
||
},
|
||
]);
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /remember command - remember something from conversation
|
||
if (trimmed.startsWith("/remember")) {
|
||
const cmdId = uid("cmd");
|
||
|
||
// Extract optional description after `/remember`
|
||
const [, ...rest] = trimmed.split(/\s+/);
|
||
const userText = rest.join(" ").trim();
|
||
|
||
const initialOutput = userText
|
||
? "Storing to memory..."
|
||
: "Processing memory request...";
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: initialOutput,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
// Import the remember prompt
|
||
const { REMEMBER_PROMPT } = await import(
|
||
"../agent/promptAssets.js"
|
||
);
|
||
|
||
// Build system-reminder content for memory request
|
||
const rememberMessage = userText
|
||
? `<system-reminder>\n${REMEMBER_PROMPT}\n</system-reminder>${userText}`
|
||
: `<system-reminder>\n${REMEMBER_PROMPT}\n\nThe user did not specify what to remember. Look at the recent conversation context to identify what they likely want you to remember, or ask them to clarify.\n</system-reminder>`;
|
||
|
||
// Mark command as finished before sending message
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: userText
|
||
? "Storing to memory..."
|
||
: "Processing memory request from conversation context...",
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
|
||
// Process conversation with the remember prompt
|
||
await processConversation([
|
||
{
|
||
type: "message",
|
||
role: "user",
|
||
content: rememberMessage,
|
||
},
|
||
]);
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
|
||
return { submitted: true };
|
||
}
|
||
|
||
// Special handling for /init command - initialize agent memory
|
||
if (trimmed === "/init") {
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "Gathering project context...",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
// Gather git context if available
|
||
let gitContext = "";
|
||
try {
|
||
const { execSync } = await import("node:child_process");
|
||
const cwd = process.cwd();
|
||
|
||
// Check if we're in a git repo
|
||
try {
|
||
execSync("git rev-parse --git-dir", {
|
||
cwd,
|
||
stdio: "pipe",
|
||
});
|
||
|
||
// Gather git info
|
||
const branch = execSync("git branch --show-current", {
|
||
cwd,
|
||
encoding: "utf-8",
|
||
}).trim();
|
||
const mainBranch = execSync(
|
||
"git symbolic-ref refs/remotes/origin/HEAD 2>/dev/null | sed 's@^refs/remotes/origin/@@' || echo 'main'",
|
||
{ cwd, encoding: "utf-8", shell: "/bin/bash" },
|
||
).trim();
|
||
const status = execSync("git status --short", {
|
||
cwd,
|
||
encoding: "utf-8",
|
||
}).trim();
|
||
const recentCommits = execSync(
|
||
"git log --oneline -10 2>/dev/null || echo 'No commits yet'",
|
||
{ cwd, encoding: "utf-8" },
|
||
).trim();
|
||
|
||
gitContext = `
|
||
## Current Project Context
|
||
|
||
**Working directory**: ${cwd}
|
||
|
||
### Git Status
|
||
- **Current branch**: ${branch}
|
||
- **Main branch**: ${mainBranch}
|
||
- **Status**:
|
||
${status || "(clean working tree)"}
|
||
|
||
### Recent Commits
|
||
${recentCommits}
|
||
`;
|
||
} catch {
|
||
// Not a git repo, just include working directory
|
||
gitContext = `
|
||
## Current Project Context
|
||
|
||
**Working directory**: ${cwd}
|
||
**Git**: Not a git repository
|
||
`;
|
||
}
|
||
} catch {
|
||
// execSync import failed, skip git context
|
||
}
|
||
|
||
// Mark command as finished before sending message
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output:
|
||
"Assimilating project context and defragmenting memories...",
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
|
||
// Send trigger message instructing agent to load the initializing-memory skill
|
||
const initMessage = `<system-reminder>
|
||
The user has requested memory initialization via /init.
|
||
|
||
## 1. Load the initializing-memory skill
|
||
|
||
First, check your \`loaded_skills\` memory block. If the \`initializing-memory\` skill is not already loaded:
|
||
1. Use the \`Skill\` tool with \`command: "load", skills: ["initializing-memory"]\`
|
||
2. The skill contains comprehensive instructions for memory initialization
|
||
|
||
If the skill fails to load, proceed with your best judgment based on these guidelines:
|
||
- Ask upfront questions (research depth, identity, related repos, workflow style)
|
||
- Research the project based on chosen depth
|
||
- Create/update memory blocks incrementally
|
||
- Reflect and verify completeness
|
||
|
||
## 2. Follow the loaded skill instructions
|
||
|
||
Once loaded, follow the instructions in the \`initializing-memory\` skill to complete the initialization.
|
||
${gitContext}
|
||
</system-reminder>`;
|
||
|
||
// Process conversation with the init prompt
|
||
await processConversation([
|
||
{
|
||
type: "message",
|
||
role: "user",
|
||
content: initMessage,
|
||
},
|
||
]);
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
return { submitted: true };
|
||
}
|
||
|
||
if (trimmed.startsWith("/feedback")) {
|
||
const maybeMsg = msg.slice("/feedback".length).trim();
|
||
setFeedbackPrefill(maybeMsg);
|
||
setActiveOverlay("feedback");
|
||
return { submitted: true };
|
||
}
|
||
|
||
// === Custom command handling ===
|
||
// Check BEFORE falling through to executeCommand()
|
||
const { findCustomCommand, substituteArguments, expandBashCommands } =
|
||
await import("./commands/custom.js");
|
||
const commandName = trimmed.split(/\s+/)[0]?.slice(1) || ""; // e.g., "review" from "/review arg"
|
||
const matchedCustom = await findCustomCommand(commandName);
|
||
|
||
if (matchedCustom) {
|
||
const cmdId = uid("cmd");
|
||
|
||
// Extract arguments (everything after command name)
|
||
const args = trimmed.slice(`/${matchedCustom.id}`.length).trim();
|
||
|
||
// Build prompt: 1) substitute args, 2) expand bash commands
|
||
let prompt = substituteArguments(matchedCustom.content, args);
|
||
prompt = await expandBashCommands(prompt);
|
||
|
||
// Show command in transcript (running phase for visual feedback)
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: trimmed,
|
||
output: `Running /${matchedCustom.id}...`,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
// Mark command as finished BEFORE sending to agent
|
||
// (matches /remember pattern - command succeeded in triggering agent)
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: trimmed,
|
||
output: `Running custom command...`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
|
||
// Send prompt to agent
|
||
// NOTE: Unlike /remember, we DON'T append args separately because
|
||
// they're already substituted into the prompt via $ARGUMENTS
|
||
await processConversation([
|
||
{
|
||
type: "message",
|
||
role: "user",
|
||
content: `<system-reminder>\n${prompt}\n</system-reminder>`,
|
||
},
|
||
]);
|
||
} catch (error) {
|
||
// Only catch errors from processConversation setup, not agent execution
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: trimmed,
|
||
output: `Failed to run command: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
setCommandRunning(false);
|
||
}
|
||
|
||
return { submitted: true };
|
||
}
|
||
// === END custom command handling ===
|
||
|
||
// Immediately add command to transcript with "running" phase
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: "",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
// Lock input during async operation
|
||
setCommandRunning(true);
|
||
|
||
try {
|
||
const { executeCommand } = await import("./commands/registry");
|
||
const result = await executeCommand(msg);
|
||
|
||
// Update the same command with result
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: result.output,
|
||
phase: "finished",
|
||
success: result.success,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
// Mark command as failed if executeCommand throws
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: msg,
|
||
output: `Failed: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
} finally {
|
||
// Unlock input
|
||
setCommandRunning(false);
|
||
}
|
||
return { submitted: true }; // Don't send commands to Letta agent
|
||
}
|
||
|
||
// Build message content from display value (handles placeholders for text/images)
|
||
const contentParts = buildMessageContentFromDisplay(msg);
|
||
|
||
// Prepend plan mode reminder if in plan mode
|
||
const planModeReminder = getPlanModeReminder();
|
||
|
||
// Prepend skill unload reminder if skills are loaded (using cached flag)
|
||
const skillUnloadReminder = getSkillUnloadReminder();
|
||
|
||
// Prepend session context on first message of CLI session (if enabled)
|
||
let sessionContextReminder = "";
|
||
const sessionContextEnabled = settingsManager.getSetting(
|
||
"sessionContextEnabled",
|
||
);
|
||
if (!hasSentSessionContextRef.current && sessionContextEnabled) {
|
||
const { buildSessionContext } = await import(
|
||
"./helpers/sessionContext"
|
||
);
|
||
sessionContextReminder = buildSessionContext({
|
||
agentInfo: {
|
||
id: agentId,
|
||
name: agentName,
|
||
description: agentDescription,
|
||
lastRunAt: agentLastRunAt,
|
||
},
|
||
});
|
||
hasSentSessionContextRef.current = true;
|
||
}
|
||
|
||
// Build bash command prefix if there are cached commands
|
||
let bashCommandPrefix = "";
|
||
if (bashCommandCacheRef.current.length > 0) {
|
||
bashCommandPrefix = `<system-reminder>
|
||
The messages below were generated by the user while running local commands using "bash mode" in the Letta Code CLI tool.
|
||
DO NOT respond to these messages or otherwise consider them in your response unless the user explicitly asks you to.
|
||
</system-reminder>
|
||
`;
|
||
for (const cmd of bashCommandCacheRef.current) {
|
||
bashCommandPrefix += `<bash-input>${cmd.input}</bash-input>\n<bash-output>${cmd.output}</bash-output>\n`;
|
||
}
|
||
// Clear the cache after building the prefix
|
||
bashCommandCacheRef.current = [];
|
||
}
|
||
|
||
// Build memory reminder if interval is set and we've reached the Nth turn
|
||
const memoryReminderContent = await buildMemoryReminder(
|
||
turnCountRef.current,
|
||
);
|
||
|
||
// Increment turn count for next iteration
|
||
turnCountRef.current += 1;
|
||
|
||
// Combine reminders with content (session context first, then plan mode, then skill unload, then bash commands, then memory reminder)
|
||
const allReminders =
|
||
sessionContextReminder +
|
||
planModeReminder +
|
||
skillUnloadReminder +
|
||
bashCommandPrefix +
|
||
memoryReminderContent;
|
||
const messageContent =
|
||
allReminders && typeof contentParts === "string"
|
||
? allReminders + contentParts
|
||
: Array.isArray(contentParts) && allReminders
|
||
? [{ type: "text" as const, text: allReminders }, ...contentParts]
|
||
: contentParts;
|
||
|
||
// Append the user message to transcript IMMEDIATELY (optimistic update)
|
||
const userId = uid("user");
|
||
buffersRef.current.byId.set(userId, {
|
||
kind: "user",
|
||
id: userId,
|
||
text: msg,
|
||
});
|
||
buffersRef.current.order.push(userId);
|
||
|
||
// Reset token counter for this turn (only count the agent's response)
|
||
buffersRef.current.tokenCount = 0;
|
||
// Clear interrupted flag from previous turn
|
||
buffersRef.current.interrupted = false;
|
||
// Rotate to a new thinking message for this turn
|
||
setThinkingMessage(getRandomThinkingVerb());
|
||
// Show streaming state immediately for responsiveness (pending approval check takes ~100ms)
|
||
setStreaming(true);
|
||
refreshDerived();
|
||
|
||
// Check for pending approvals before sending message (skip if we already have
|
||
// a queued approval response to send first).
|
||
if (CHECK_PENDING_APPROVALS_BEFORE_SEND && !queuedApprovalResults) {
|
||
try {
|
||
const client = await getClient();
|
||
// Fetch fresh agent state to check for pending approvals with accurate in-context messages
|
||
const agent = await client.agents.retrieve(agentId);
|
||
const { pendingApprovals: existingApprovals } = await getResumeData(
|
||
client,
|
||
agent,
|
||
);
|
||
|
||
// Check if user cancelled while we were fetching approval state
|
||
if (
|
||
userCancelledRef.current ||
|
||
abortControllerRef.current?.signal.aborted
|
||
) {
|
||
// User hit ESC during the check - abort and clean up
|
||
buffersRef.current.byId.delete(userId);
|
||
const orderIndex = buffersRef.current.order.indexOf(userId);
|
||
if (orderIndex !== -1) {
|
||
buffersRef.current.order.splice(orderIndex, 1);
|
||
}
|
||
setStreaming(false);
|
||
refreshDerived();
|
||
return { submitted: false };
|
||
}
|
||
|
||
if (existingApprovals && existingApprovals.length > 0) {
|
||
// There are pending approvals - check permissions first (respects yolo mode)
|
||
const approvalResults = await Promise.all(
|
||
existingApprovals.map(async (approvalItem) => {
|
||
if (!approvalItem.toolName) {
|
||
return {
|
||
approval: approvalItem,
|
||
permission: {
|
||
decision: "deny" as const,
|
||
reason: "Tool call incomplete - missing name",
|
||
},
|
||
context: null,
|
||
};
|
||
}
|
||
const parsedArgs = safeJsonParseOr<Record<string, unknown>>(
|
||
approvalItem.toolArgs,
|
||
{},
|
||
);
|
||
const permission = await checkToolPermission(
|
||
approvalItem.toolName,
|
||
parsedArgs,
|
||
);
|
||
const context = await analyzeToolApproval(
|
||
approvalItem.toolName,
|
||
parsedArgs,
|
||
);
|
||
return { approval: approvalItem, permission, context };
|
||
}),
|
||
);
|
||
|
||
// Check if user cancelled during permission check
|
||
if (
|
||
userCancelledRef.current ||
|
||
abortControllerRef.current?.signal.aborted
|
||
) {
|
||
buffersRef.current.byId.delete(userId);
|
||
const orderIndex = buffersRef.current.order.indexOf(userId);
|
||
if (orderIndex !== -1) {
|
||
buffersRef.current.order.splice(orderIndex, 1);
|
||
}
|
||
setStreaming(false);
|
||
refreshDerived();
|
||
return { submitted: false };
|
||
}
|
||
|
||
// Categorize by permission decision
|
||
const needsUserInput: typeof approvalResults = [];
|
||
const autoAllowed: typeof approvalResults = [];
|
||
const autoDenied: typeof approvalResults = [];
|
||
|
||
for (const ac of approvalResults) {
|
||
const { approval, permission } = ac;
|
||
let decision = permission.decision;
|
||
|
||
// Some tools always need user input regardless of yolo mode
|
||
if (
|
||
alwaysRequiresUserInput(approval.toolName) &&
|
||
decision === "allow"
|
||
) {
|
||
decision = "ask";
|
||
}
|
||
|
||
if (decision === "ask") {
|
||
needsUserInput.push(ac);
|
||
} else if (decision === "deny") {
|
||
autoDenied.push(ac);
|
||
} else {
|
||
autoAllowed.push(ac);
|
||
}
|
||
}
|
||
|
||
// If all approvals can be auto-handled (yolo mode), process them immediately
|
||
if (needsUserInput.length === 0) {
|
||
// Precompute diffs for file edit tools before execution (both auto-allowed and needs-user-input)
|
||
for (const ac of [...autoAllowed, ...needsUserInput]) {
|
||
const toolName = ac.approval.toolName;
|
||
const toolCallId = ac.approval.toolCallId;
|
||
try {
|
||
const args = JSON.parse(ac.approval.toolArgs || "{}");
|
||
|
||
if (isFileWriteTool(toolName)) {
|
||
const filePath = args.file_path as string | undefined;
|
||
if (filePath) {
|
||
const result = computeAdvancedDiff({
|
||
kind: "write",
|
||
filePath,
|
||
content: (args.content as string) || "",
|
||
});
|
||
if (result.mode === "advanced") {
|
||
precomputedDiffsRef.current.set(toolCallId, result);
|
||
}
|
||
}
|
||
} else if (isFileEditTool(toolName)) {
|
||
const filePath = args.file_path as string | undefined;
|
||
if (filePath) {
|
||
// Check if it's a multi-edit (has edits array) or single edit
|
||
if (args.edits && Array.isArray(args.edits)) {
|
||
const result = computeAdvancedDiff({
|
||
kind: "multi_edit",
|
||
filePath,
|
||
edits: args.edits as Array<{
|
||
old_string: string;
|
||
new_string: string;
|
||
replace_all?: boolean;
|
||
}>,
|
||
});
|
||
if (result.mode === "advanced") {
|
||
precomputedDiffsRef.current.set(toolCallId, result);
|
||
}
|
||
} else {
|
||
const result = computeAdvancedDiff({
|
||
kind: "edit",
|
||
filePath,
|
||
oldString: (args.old_string as string) || "",
|
||
newString: (args.new_string as string) || "",
|
||
replaceAll: args.replace_all as boolean | undefined,
|
||
});
|
||
if (result.mode === "advanced") {
|
||
precomputedDiffsRef.current.set(toolCallId, result);
|
||
}
|
||
}
|
||
}
|
||
} else if (isPatchTool(toolName) && args.input) {
|
||
// Patch tools - parse hunks directly (patches ARE diffs)
|
||
const operations = parsePatchOperations(
|
||
args.input as string,
|
||
);
|
||
for (const op of operations) {
|
||
const key = `${toolCallId}:${op.path}`;
|
||
if (op.kind === "add" || op.kind === "update") {
|
||
const result = parsePatchToAdvancedDiff(
|
||
op.patchLines,
|
||
op.path,
|
||
);
|
||
if (result) {
|
||
precomputedDiffsRef.current.set(key, result);
|
||
}
|
||
}
|
||
// Delete operations don't need diffs
|
||
}
|
||
}
|
||
} catch {
|
||
// Ignore errors in diff computation for auto-allowed tools
|
||
}
|
||
}
|
||
|
||
// Execute auto-allowed tools (sequential for writes, parallel for reads)
|
||
const autoAllowedResults = await executeAutoAllowedTools(
|
||
autoAllowed,
|
||
(chunk) => onChunk(buffersRef.current, chunk),
|
||
);
|
||
|
||
// Create denial results for auto-denied and update UI
|
||
const autoDeniedResults = autoDenied.map((ac) => {
|
||
// Prefer the detailed reason over the short matchedRule name
|
||
const reason = ac.permission.reason
|
||
? `Permission denied: ${ac.permission.reason}`
|
||
: "matchedRule" in ac.permission && ac.permission.matchedRule
|
||
? `Permission denied by rule: ${ac.permission.matchedRule}`
|
||
: "Permission denied: Unknown";
|
||
|
||
// Update buffers with denial for UI
|
||
onChunk(buffersRef.current, {
|
||
message_type: "tool_return_message",
|
||
id: "dummy",
|
||
date: new Date().toISOString(),
|
||
tool_call_id: ac.approval.toolCallId,
|
||
tool_return: `Error: request to call tool denied. User reason: ${reason}`,
|
||
status: "error",
|
||
stdout: null,
|
||
stderr: null,
|
||
});
|
||
|
||
return {
|
||
type: "approval" as const,
|
||
tool_call_id: ac.approval.toolCallId,
|
||
approve: false,
|
||
reason,
|
||
};
|
||
});
|
||
|
||
refreshDerived();
|
||
|
||
// Combine results and send directly with the user's message
|
||
// (can't use state here as it won't be available until next render)
|
||
const recoveryApprovalResults = [
|
||
...autoAllowedResults.map((ar) => ({
|
||
type: "approval" as const,
|
||
tool_call_id: ar.toolCallId,
|
||
approve: true,
|
||
tool_return: ar.result.toolReturn,
|
||
})),
|
||
...autoDeniedResults,
|
||
];
|
||
|
||
// Build and send initialInput directly
|
||
const initialInput: Array<MessageCreate | ApprovalCreate> = [
|
||
{
|
||
type: "approval",
|
||
approvals: recoveryApprovalResults,
|
||
},
|
||
{
|
||
type: "message",
|
||
role: "user",
|
||
content:
|
||
messageContent as unknown as MessageCreate["content"],
|
||
},
|
||
];
|
||
|
||
await processConversation(initialInput);
|
||
clearPlaceholdersInText(msg);
|
||
return { submitted: true };
|
||
} else {
|
||
// Some approvals need user input - show dialog
|
||
// Remove the optimistic user message from transcript
|
||
buffersRef.current.byId.delete(userId);
|
||
const orderIndex = buffersRef.current.order.indexOf(userId);
|
||
if (orderIndex !== -1) {
|
||
buffersRef.current.order.splice(orderIndex, 1);
|
||
}
|
||
|
||
setStreaming(false);
|
||
setPendingApprovals(needsUserInput.map((ac) => ac.approval));
|
||
setApprovalContexts(
|
||
needsUserInput
|
||
.map((ac) => ac.context)
|
||
.filter(Boolean) as ApprovalContext[],
|
||
);
|
||
|
||
// Precompute diffs for file edit tools before execution (both auto-allowed and needs-user-input)
|
||
for (const ac of [...autoAllowed, ...needsUserInput]) {
|
||
const toolName = ac.approval.toolName;
|
||
const toolCallId = ac.approval.toolCallId;
|
||
try {
|
||
const args = JSON.parse(ac.approval.toolArgs || "{}");
|
||
|
||
if (isFileWriteTool(toolName)) {
|
||
const filePath = args.file_path as string | undefined;
|
||
if (filePath) {
|
||
const result = computeAdvancedDiff({
|
||
kind: "write",
|
||
filePath,
|
||
content: (args.content as string) || "",
|
||
});
|
||
if (result.mode === "advanced") {
|
||
precomputedDiffsRef.current.set(toolCallId, result);
|
||
}
|
||
}
|
||
} else if (isFileEditTool(toolName)) {
|
||
const filePath = args.file_path as string | undefined;
|
||
if (filePath) {
|
||
// Check if it's a multi-edit (has edits array) or single edit
|
||
if (args.edits && Array.isArray(args.edits)) {
|
||
const result = computeAdvancedDiff({
|
||
kind: "multi_edit",
|
||
filePath,
|
||
edits: args.edits as Array<{
|
||
old_string: string;
|
||
new_string: string;
|
||
replace_all?: boolean;
|
||
}>,
|
||
});
|
||
if (result.mode === "advanced") {
|
||
precomputedDiffsRef.current.set(toolCallId, result);
|
||
}
|
||
} else {
|
||
const result = computeAdvancedDiff({
|
||
kind: "edit",
|
||
filePath,
|
||
oldString: (args.old_string as string) || "",
|
||
newString: (args.new_string as string) || "",
|
||
replaceAll: args.replace_all as boolean | undefined,
|
||
});
|
||
if (result.mode === "advanced") {
|
||
precomputedDiffsRef.current.set(toolCallId, result);
|
||
}
|
||
}
|
||
}
|
||
} else if (isPatchTool(toolName) && args.input) {
|
||
// Patch tools - parse hunks directly (patches ARE diffs)
|
||
const operations = parsePatchOperations(
|
||
args.input as string,
|
||
);
|
||
for (const op of operations) {
|
||
const key = `${toolCallId}:${op.path}`;
|
||
if (op.kind === "add" || op.kind === "update") {
|
||
const result = parsePatchToAdvancedDiff(
|
||
op.patchLines,
|
||
op.path,
|
||
);
|
||
if (result) {
|
||
precomputedDiffsRef.current.set(key, result);
|
||
}
|
||
}
|
||
// Delete operations don't need diffs
|
||
}
|
||
}
|
||
} catch {
|
||
// Ignore errors in diff computation for auto-allowed tools
|
||
}
|
||
}
|
||
|
||
// Execute auto-allowed tools (sequential for writes, parallel for reads)
|
||
const autoAllowedWithResults = await executeAutoAllowedTools(
|
||
autoAllowed,
|
||
(chunk) => onChunk(buffersRef.current, chunk),
|
||
);
|
||
|
||
// Create denial reasons for auto-denied and update UI
|
||
const autoDeniedWithReasons = autoDenied.map((ac) => {
|
||
// Prefer the detailed reason over the short matchedRule name
|
||
const reason = ac.permission.reason
|
||
? `Permission denied: ${ac.permission.reason}`
|
||
: "matchedRule" in ac.permission && ac.permission.matchedRule
|
||
? `Permission denied by rule: ${ac.permission.matchedRule}`
|
||
: "Permission denied: Unknown";
|
||
|
||
// Update buffers with denial for UI
|
||
onChunk(buffersRef.current, {
|
||
message_type: "tool_return_message",
|
||
id: "dummy",
|
||
date: new Date().toISOString(),
|
||
tool_call_id: ac.approval.toolCallId,
|
||
tool_return: `Error: request to call tool denied. User reason: ${reason}`,
|
||
status: "error",
|
||
stdout: null,
|
||
stderr: null,
|
||
});
|
||
|
||
return {
|
||
approval: ac.approval,
|
||
reason,
|
||
};
|
||
});
|
||
|
||
// Store auto-handled results to send along with user decisions
|
||
setAutoHandledResults(autoAllowedWithResults);
|
||
setAutoDeniedApprovals(autoDeniedWithReasons);
|
||
|
||
refreshDerived();
|
||
return { submitted: false };
|
||
}
|
||
}
|
||
} catch (_error) {
|
||
// If check fails, proceed anyway (don't block user)
|
||
}
|
||
}
|
||
|
||
// Start the conversation loop. If we have queued approval results from an interrupted
|
||
// client-side execution, send them first before the new user message.
|
||
const initialInput: Array<MessageCreate | ApprovalCreate> = [];
|
||
|
||
if (queuedApprovalResults) {
|
||
initialInput.push({
|
||
type: "approval",
|
||
approvals: queuedApprovalResults,
|
||
});
|
||
setQueuedApprovalResults(null);
|
||
}
|
||
|
||
initialInput.push({
|
||
type: "message",
|
||
role: "user",
|
||
content: messageContent as unknown as MessageCreate["content"],
|
||
});
|
||
|
||
await processConversation(initialInput);
|
||
|
||
// Clean up placeholders after submission
|
||
clearPlaceholdersInText(msg);
|
||
|
||
return { submitted: true };
|
||
},
|
||
[
|
||
streaming,
|
||
commandRunning,
|
||
processConversation,
|
||
refreshDerived,
|
||
agentId,
|
||
agentName,
|
||
agentDescription,
|
||
agentLastRunAt,
|
||
handleExit,
|
||
isExecutingTool,
|
||
queuedApprovalResults,
|
||
pendingApprovals,
|
||
profileConfirmPending,
|
||
handleAgentSelect,
|
||
tokenStreamingEnabled,
|
||
isAgentBusy,
|
||
setStreaming,
|
||
setCommandRunning,
|
||
],
|
||
);
|
||
|
||
const onSubmitRef = useRef(onSubmit);
|
||
useEffect(() => {
|
||
onSubmitRef.current = onSubmit;
|
||
}, [onSubmit]);
|
||
|
||
// Process queued messages when streaming ends
|
||
useEffect(() => {
|
||
if (
|
||
!streaming &&
|
||
messageQueue.length > 0 &&
|
||
pendingApprovals.length === 0 &&
|
||
!commandRunning &&
|
||
!isExecutingTool &&
|
||
!anySelectorOpen && // Don't dequeue while a selector/overlay is open
|
||
!waitingForQueueCancelRef.current && // Don't dequeue while waiting for cancel
|
||
!userCancelledRef.current // Don't dequeue if user just cancelled
|
||
) {
|
||
const [firstMessage, ...rest] = messageQueue;
|
||
setMessageQueue(rest);
|
||
|
||
// Submit the first message using the normal submit flow
|
||
// This ensures all setup (reminders, UI updates, etc.) happens correctly
|
||
onSubmitRef.current(firstMessage);
|
||
}
|
||
}, [
|
||
streaming,
|
||
messageQueue,
|
||
pendingApprovals,
|
||
commandRunning,
|
||
isExecutingTool,
|
||
anySelectorOpen,
|
||
]);
|
||
|
||
// Helper to send all approval results when done
|
||
const sendAllResults = useCallback(
|
||
async (
|
||
additionalDecision?:
|
||
| { type: "approve"; approval: ApprovalRequest }
|
||
| { type: "deny"; approval: ApprovalRequest; reason: string },
|
||
) => {
|
||
try {
|
||
// Don't send results if user has already cancelled
|
||
if (
|
||
userCancelledRef.current ||
|
||
abortControllerRef.current?.signal.aborted
|
||
) {
|
||
setStreaming(false);
|
||
setIsExecutingTool(false);
|
||
setPendingApprovals([]);
|
||
setApprovalContexts([]);
|
||
setApprovalResults([]);
|
||
setAutoHandledResults([]);
|
||
setAutoDeniedApprovals([]);
|
||
return;
|
||
}
|
||
|
||
// Snapshot current state before clearing dialog
|
||
const approvalResultsSnapshot = [...approvalResults];
|
||
const autoHandledSnapshot = [...autoHandledResults];
|
||
const autoDeniedSnapshot = [...autoDeniedApprovals];
|
||
const pendingSnapshot = [...pendingApprovals];
|
||
|
||
// Clear dialog state immediately so UI updates right away
|
||
setPendingApprovals([]);
|
||
setApprovalContexts([]);
|
||
setApprovalResults([]);
|
||
setAutoHandledResults([]);
|
||
setAutoDeniedApprovals([]);
|
||
|
||
// Show "thinking" state and lock input while executing approved tools client-side
|
||
setStreaming(true);
|
||
// Ensure interrupted flag is cleared for this execution
|
||
buffersRef.current.interrupted = false;
|
||
|
||
const approvalAbortController = new AbortController();
|
||
toolAbortControllerRef.current = approvalAbortController;
|
||
|
||
// Combine all decisions using snapshots
|
||
const allDecisions = [
|
||
...approvalResultsSnapshot,
|
||
...(additionalDecision ? [additionalDecision] : []),
|
||
];
|
||
|
||
// Execute approved tools and format results using shared function
|
||
const { executeApprovalBatch } = await import(
|
||
"../agent/approval-execution"
|
||
);
|
||
const executedResults = await executeApprovalBatch(
|
||
allDecisions,
|
||
(chunk) => {
|
||
onChunk(buffersRef.current, chunk);
|
||
// Also log errors to the UI error display
|
||
if (
|
||
chunk.status === "error" &&
|
||
chunk.message_type === "tool_return_message"
|
||
) {
|
||
const isToolError = chunk.tool_return?.startsWith(
|
||
"Error executing tool:",
|
||
);
|
||
if (isToolError) {
|
||
appendError(chunk.tool_return);
|
||
}
|
||
}
|
||
// Flush UI so completed tools show up while the batch continues
|
||
refreshDerived();
|
||
},
|
||
{ abortSignal: approvalAbortController.signal },
|
||
);
|
||
|
||
// Combine with auto-handled and auto-denied results using snapshots
|
||
const allResults = [
|
||
...autoHandledSnapshot.map((ar) => ({
|
||
type: "tool" as const,
|
||
tool_call_id: ar.toolCallId,
|
||
tool_return: ar.result.toolReturn,
|
||
status: ar.result.status,
|
||
stdout: ar.result.stdout,
|
||
stderr: ar.result.stderr,
|
||
})),
|
||
...autoDeniedSnapshot.map((ad) => ({
|
||
type: "approval" as const,
|
||
tool_call_id: ad.approval.toolCallId,
|
||
approve: false,
|
||
reason: ad.reason,
|
||
})),
|
||
...executedResults,
|
||
];
|
||
|
||
// Dev-only validation: ensure outgoing IDs match expected IDs (using snapshots)
|
||
if (process.env.NODE_ENV !== "production") {
|
||
// Include ALL tool call IDs: auto-handled, auto-denied, and pending approvals
|
||
const expectedIds = new Set([
|
||
...autoHandledSnapshot.map((ar) => ar.toolCallId),
|
||
...autoDeniedSnapshot.map((ad) => ad.approval.toolCallId),
|
||
...pendingSnapshot.map((a) => a.toolCallId),
|
||
]);
|
||
const sendingIds = new Set(
|
||
allResults.map((r) => r.tool_call_id).filter(Boolean),
|
||
);
|
||
|
||
const setsEqual = (a: Set<string>, b: Set<string>) =>
|
||
a.size === b.size && [...a].every((id) => b.has(id));
|
||
|
||
if (!setsEqual(expectedIds, sendingIds)) {
|
||
console.error("[BUG] Approval ID mismatch detected");
|
||
console.error("Expected IDs:", Array.from(expectedIds));
|
||
console.error("Sending IDs:", Array.from(sendingIds));
|
||
throw new Error(
|
||
"Approval ID mismatch - refusing to send mismatched IDs",
|
||
);
|
||
}
|
||
}
|
||
|
||
// Rotate to a new thinking message
|
||
setThinkingMessage(getRandomThinkingVerb());
|
||
refreshDerived();
|
||
|
||
const wasAborted = approvalAbortController.signal.aborted;
|
||
const userCancelled =
|
||
userCancelledRef.current ||
|
||
abortControllerRef.current?.signal.aborted;
|
||
|
||
if (wasAborted || userCancelled) {
|
||
// Queue results to send alongside the next user message (if not cancelled entirely)
|
||
if (!userCancelled) {
|
||
setQueuedApprovalResults(allResults as ApprovalResult[]);
|
||
}
|
||
setStreaming(false);
|
||
} else {
|
||
// Continue conversation with all results
|
||
await processConversation([
|
||
{
|
||
type: "approval",
|
||
approvals: allResults as ApprovalResult[],
|
||
},
|
||
]);
|
||
}
|
||
} finally {
|
||
// Always release the execution guard, even if an error occurred
|
||
setIsExecutingTool(false);
|
||
toolAbortControllerRef.current = null;
|
||
}
|
||
},
|
||
[
|
||
approvalResults,
|
||
autoHandledResults,
|
||
autoDeniedApprovals,
|
||
pendingApprovals,
|
||
processConversation,
|
||
refreshDerived,
|
||
appendError,
|
||
setStreaming,
|
||
],
|
||
);
|
||
|
||
// Handle approval callbacks - sequential review
|
||
const handleApproveCurrent = useCallback(
|
||
async (diffs?: Map<string, AdvancedDiffSuccess>) => {
|
||
if (isExecutingTool) return;
|
||
|
||
const currentIndex = approvalResults.length;
|
||
const currentApproval = pendingApprovals[currentIndex];
|
||
|
||
if (!currentApproval) return;
|
||
|
||
// Store precomputed diffs before execution
|
||
if (diffs) {
|
||
for (const [key, diff] of diffs) {
|
||
precomputedDiffsRef.current.set(key, diff);
|
||
}
|
||
}
|
||
|
||
setIsExecutingTool(true);
|
||
|
||
try {
|
||
// Store approval decision (don't execute yet - batch execute after all approvals)
|
||
const decision = {
|
||
type: "approve" as const,
|
||
approval: currentApproval,
|
||
};
|
||
|
||
// Check if we're done with all approvals
|
||
if (currentIndex + 1 >= pendingApprovals.length) {
|
||
// All approvals collected, execute and send to backend
|
||
// sendAllResults owns the lock release via its finally block
|
||
await sendAllResults(decision);
|
||
} else {
|
||
// Not done yet, store decision and show next approval
|
||
setApprovalResults((prev) => [...prev, decision]);
|
||
setIsExecutingTool(false);
|
||
}
|
||
} catch (e) {
|
||
const errorDetails = formatErrorDetails(e, agentId);
|
||
appendError(errorDetails);
|
||
setStreaming(false);
|
||
setIsExecutingTool(false);
|
||
}
|
||
},
|
||
[
|
||
agentId,
|
||
pendingApprovals,
|
||
approvalResults,
|
||
sendAllResults,
|
||
appendError,
|
||
isExecutingTool,
|
||
setStreaming,
|
||
],
|
||
);
|
||
|
||
const handleApproveAlways = useCallback(
|
||
async (
|
||
scope?: "project" | "session",
|
||
diffs?: Map<string, AdvancedDiffSuccess>,
|
||
) => {
|
||
if (isExecutingTool) return;
|
||
|
||
if (pendingApprovals.length === 0 || approvalContexts.length === 0)
|
||
return;
|
||
|
||
const currentIndex = approvalResults.length;
|
||
const approvalContext = approvalContexts[currentIndex];
|
||
if (!approvalContext) return;
|
||
|
||
const rule = approvalContext.recommendedRule;
|
||
const actualScope = scope || approvalContext.defaultScope;
|
||
|
||
// Save the permission rule
|
||
await savePermissionRule(rule, "allow", actualScope);
|
||
|
||
// Show confirmation in transcript
|
||
const scopeText =
|
||
actualScope === "session" ? " (session only)" : " (project)";
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/approve-always",
|
||
output: `Added permission: ${rule}${scopeText}`,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
// Re-check remaining approvals against the newly saved permission
|
||
// This allows subsequent approvals that match the new rule to be auto-allowed
|
||
const remainingApprovals = pendingApprovals.slice(currentIndex + 1);
|
||
if (remainingApprovals.length > 0) {
|
||
const recheckResults = await Promise.all(
|
||
remainingApprovals.map(async (approval) => {
|
||
const parsedArgs = safeJsonParseOr<Record<string, unknown>>(
|
||
approval.toolArgs,
|
||
{},
|
||
);
|
||
const permission = await checkToolPermission(
|
||
approval.toolName,
|
||
parsedArgs,
|
||
);
|
||
return { approval, permission };
|
||
}),
|
||
);
|
||
|
||
const nowAutoAllowed = recheckResults.filter(
|
||
(r) => r.permission.decision === "allow",
|
||
);
|
||
const stillNeedAsking = recheckResults.filter(
|
||
(r) => r.permission.decision === "ask",
|
||
);
|
||
|
||
// Only auto-handle if ALL remaining are now allowed
|
||
// (avoids complex state synchronization issues with partial batches)
|
||
if (stillNeedAsking.length === 0 && nowAutoAllowed.length > 0) {
|
||
const currentApproval = pendingApprovals[currentIndex];
|
||
if (!currentApproval) return;
|
||
|
||
// Store diffs before execution
|
||
if (diffs) {
|
||
for (const [key, diff] of diffs) {
|
||
precomputedDiffsRef.current.set(key, diff);
|
||
}
|
||
}
|
||
|
||
setIsExecutingTool(true);
|
||
|
||
// Build ALL decisions: current + auto-allowed remaining
|
||
const allDecisions: Array<{
|
||
type: "approve";
|
||
approval: ApprovalRequest;
|
||
}> = [
|
||
{ type: "approve", approval: currentApproval },
|
||
...nowAutoAllowed.map((r) => ({
|
||
type: "approve" as const,
|
||
approval: r.approval,
|
||
})),
|
||
];
|
||
|
||
// Clear dialog state immediately
|
||
setPendingApprovals([]);
|
||
setApprovalContexts([]);
|
||
setApprovalResults([]);
|
||
setAutoHandledResults([]);
|
||
setAutoDeniedApprovals([]);
|
||
|
||
setStreaming(true);
|
||
buffersRef.current.interrupted = false;
|
||
|
||
try {
|
||
// Execute ALL decisions together
|
||
const { executeApprovalBatch } = await import(
|
||
"../agent/approval-execution"
|
||
);
|
||
const executedResults = await executeApprovalBatch(
|
||
allDecisions,
|
||
(chunk) => {
|
||
onChunk(buffersRef.current, chunk);
|
||
refreshDerived();
|
||
},
|
||
);
|
||
|
||
setThinkingMessage(getRandomThinkingVerb());
|
||
refreshDerived();
|
||
|
||
// Continue conversation with all results
|
||
await processConversation([
|
||
{
|
||
type: "approval",
|
||
approvals: executedResults as ApprovalResult[],
|
||
},
|
||
]);
|
||
} finally {
|
||
setIsExecutingTool(false);
|
||
}
|
||
return; // Don't call handleApproveCurrent - we handled everything
|
||
}
|
||
}
|
||
|
||
// Fallback: proceed with normal flow (will prompt for remaining approvals)
|
||
await handleApproveCurrent(diffs);
|
||
},
|
||
[
|
||
approvalResults,
|
||
approvalContexts,
|
||
pendingApprovals,
|
||
handleApproveCurrent,
|
||
processConversation,
|
||
refreshDerived,
|
||
isExecutingTool,
|
||
setStreaming,
|
||
],
|
||
);
|
||
|
||
const handleDenyCurrent = useCallback(
|
||
async (reason: string) => {
|
||
if (isExecutingTool) return;
|
||
|
||
const currentIndex = approvalResults.length;
|
||
const currentApproval = pendingApprovals[currentIndex];
|
||
|
||
if (!currentApproval) return;
|
||
|
||
setIsExecutingTool(true);
|
||
|
||
try {
|
||
// Store denial decision
|
||
const decision = {
|
||
type: "deny" as const,
|
||
approval: currentApproval,
|
||
reason: reason || "User denied the tool execution",
|
||
};
|
||
|
||
// Check if we're done with all approvals
|
||
if (currentIndex + 1 >= pendingApprovals.length) {
|
||
// All approvals collected, execute and send to backend
|
||
// sendAllResults owns the lock release via its finally block
|
||
setThinkingMessage(getRandomThinkingVerb());
|
||
await sendAllResults(decision);
|
||
} else {
|
||
// Not done yet, store decision and show next approval
|
||
setApprovalResults((prev) => [...prev, decision]);
|
||
setIsExecutingTool(false);
|
||
}
|
||
} catch (e) {
|
||
const errorDetails = formatErrorDetails(e, agentId);
|
||
appendError(errorDetails);
|
||
setStreaming(false);
|
||
setIsExecutingTool(false);
|
||
}
|
||
},
|
||
[
|
||
agentId,
|
||
pendingApprovals,
|
||
approvalResults,
|
||
sendAllResults,
|
||
appendError,
|
||
isExecutingTool,
|
||
setStreaming,
|
||
],
|
||
);
|
||
|
||
// Cancel all pending approvals - queue denials to send with next message
|
||
// Similar to interrupt flow during tool execution
|
||
const handleCancelApprovals = useCallback(() => {
|
||
if (pendingApprovals.length === 0) return;
|
||
|
||
// Create denial results for all pending approvals and queue for next message
|
||
const denialResults = pendingApprovals.map((approval) => ({
|
||
type: "approval" as const,
|
||
tool_call_id: approval.toolCallId,
|
||
approve: false,
|
||
reason: "User cancelled the approval",
|
||
}));
|
||
setQueuedApprovalResults(denialResults);
|
||
|
||
// Mark the pending approval tool calls as cancelled in the buffers
|
||
markIncompleteToolsAsCancelled(buffersRef.current);
|
||
refreshDerived();
|
||
|
||
// Clear all approval state
|
||
setPendingApprovals([]);
|
||
setApprovalContexts([]);
|
||
setApprovalResults([]);
|
||
setAutoHandledResults([]);
|
||
setAutoDeniedApprovals([]);
|
||
}, [pendingApprovals, refreshDerived]);
|
||
|
||
const handleModelSelect = useCallback(
|
||
async (modelId: string) => {
|
||
await withCommandLock(async () => {
|
||
// Declare cmdId outside try block so it's accessible in catch
|
||
let cmdId: string | null = null;
|
||
|
||
try {
|
||
// Find the selected model from models.json first (for loading message)
|
||
const { models } = await import("../agent/model");
|
||
let selectedModel = models.find((m) => m.id === modelId);
|
||
|
||
// If not found in static list, it might be a BYOK model where id === handle
|
||
if (!selectedModel && modelId.includes("/")) {
|
||
// Treat it as a BYOK model - the modelId is actually the handle
|
||
// Look up the context window from the API-cached model info
|
||
const { getModelContextWindow } = await import(
|
||
"../agent/available-models"
|
||
);
|
||
const apiContextWindow = getModelContextWindow(modelId);
|
||
|
||
selectedModel = {
|
||
id: modelId,
|
||
handle: modelId,
|
||
label: modelId.split("/").pop() ?? modelId,
|
||
description: "Custom model",
|
||
updateArgs: apiContextWindow
|
||
? { context_window: apiContextWindow }
|
||
: undefined,
|
||
} as unknown as (typeof models)[number];
|
||
}
|
||
|
||
if (!selectedModel) {
|
||
// Create a failed command in the transcript
|
||
cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/model ${modelId}`,
|
||
output: `Model not found: ${modelId}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
|
||
// Immediately add command to transcript with "running" phase and loading message
|
||
cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/model ${modelId}`,
|
||
output: `Switching model to ${selectedModel.label}...`,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
// Update the agent with new model and config args
|
||
const { updateAgentLLMConfig } = await import("../agent/modify");
|
||
|
||
const updatedConfig = await updateAgentLLMConfig(
|
||
agentId,
|
||
selectedModel.handle,
|
||
selectedModel.updateArgs,
|
||
);
|
||
setLlmConfig(updatedConfig);
|
||
setCurrentModelId(modelId);
|
||
|
||
// After switching models, only switch toolset if it actually changes
|
||
const { isOpenAIModel, isGeminiModel } = await import(
|
||
"../tools/manager"
|
||
);
|
||
const targetToolset:
|
||
| "codex"
|
||
| "codex_snake"
|
||
| "default"
|
||
| "gemini"
|
||
| "gemini_snake"
|
||
| "none" = isOpenAIModel(selectedModel.handle ?? "")
|
||
? "codex"
|
||
: isGeminiModel(selectedModel.handle ?? "")
|
||
? "gemini"
|
||
: "default";
|
||
|
||
let toolsetName:
|
||
| "codex"
|
||
| "codex_snake"
|
||
| "default"
|
||
| "gemini"
|
||
| "gemini_snake"
|
||
| "none"
|
||
| null = null;
|
||
if (currentToolset !== targetToolset) {
|
||
const { switchToolsetForModel } = await import("../tools/toolset");
|
||
toolsetName = await switchToolsetForModel(
|
||
selectedModel.handle ?? "",
|
||
agentId,
|
||
);
|
||
setCurrentToolset(toolsetName);
|
||
}
|
||
|
||
// Update the same command with final result (include toolset info only if changed)
|
||
const autoToolsetLine = toolsetName
|
||
? `Automatically switched toolset to ${toolsetName}. Use /toolset to change back if desired.\nConsider switching to a different system prompt using /system to match.`
|
||
: null;
|
||
const outputLines = [
|
||
`Switched to ${selectedModel.label}`,
|
||
...(autoToolsetLine ? [autoToolsetLine] : []),
|
||
].join("\n");
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/model ${modelId}`,
|
||
output: outputLines,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
// Mark command as failed (only if cmdId was created)
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
if (cmdId) {
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/model ${modelId}`,
|
||
output: `Failed to switch model: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
}
|
||
}
|
||
});
|
||
},
|
||
[agentId, refreshDerived, currentToolset, withCommandLock],
|
||
);
|
||
|
||
const handleSystemPromptSelect = useCallback(
|
||
async (promptId: string) => {
|
||
await withCommandLock(async () => {
|
||
const cmdId = uid("cmd");
|
||
|
||
try {
|
||
// Find the selected prompt
|
||
const { SYSTEM_PROMPTS } = await import("../agent/promptAssets");
|
||
const selectedPrompt = SYSTEM_PROMPTS.find((p) => p.id === promptId);
|
||
|
||
if (!selectedPrompt) {
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/system ${promptId}`,
|
||
output: `System prompt not found: ${promptId}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
return;
|
||
}
|
||
|
||
// Immediately add command to transcript with "running" phase
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/system ${promptId}`,
|
||
output: `Switching system prompt to ${selectedPrompt.label}...`,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
// Update the agent's system prompt
|
||
const { updateAgentSystemPromptRaw } = await import(
|
||
"../agent/modify"
|
||
);
|
||
const result = await updateAgentSystemPromptRaw(
|
||
agentId,
|
||
selectedPrompt.content,
|
||
);
|
||
|
||
if (result.success) {
|
||
setCurrentSystemPromptId(promptId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/system ${promptId}`,
|
||
output: `Switched system prompt to ${selectedPrompt.label}`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
} else {
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/system ${promptId}`,
|
||
output: result.message,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
}
|
||
refreshDerived();
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/system ${promptId}`,
|
||
output: `Failed to switch system prompt: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
}
|
||
});
|
||
},
|
||
[agentId, refreshDerived, withCommandLock],
|
||
);
|
||
|
||
const handleToolsetSelect = useCallback(
|
||
async (
|
||
toolsetId:
|
||
| "codex"
|
||
| "codex_snake"
|
||
| "default"
|
||
| "gemini"
|
||
| "gemini_snake"
|
||
| "none",
|
||
) => {
|
||
await withCommandLock(async () => {
|
||
const cmdId = uid("cmd");
|
||
|
||
try {
|
||
// Immediately add command to transcript with "running" phase
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/toolset ${toolsetId}`,
|
||
output: `Switching toolset to ${toolsetId}...`,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
// Force switch to the selected toolset
|
||
const { forceToolsetSwitch } = await import("../tools/toolset");
|
||
await forceToolsetSwitch(toolsetId, agentId);
|
||
setCurrentToolset(toolsetId);
|
||
|
||
// Update the command with final result
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/toolset ${toolsetId}`,
|
||
output: `Switched toolset to ${toolsetId}`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/toolset ${toolsetId}`,
|
||
output: `Failed to switch toolset: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
}
|
||
});
|
||
},
|
||
[agentId, refreshDerived, withCommandLock],
|
||
);
|
||
|
||
// Handle escape when profile confirmation is pending
|
||
const handleFeedbackSubmit = useCallback(
|
||
async (message: string) => {
|
||
closeOverlay();
|
||
|
||
await withCommandLock(async () => {
|
||
const cmdId = uid("cmd");
|
||
|
||
try {
|
||
const resolvedMessage = resolvePlaceholders(message);
|
||
|
||
// Immediately add command to transcript with "running" phase
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/feedback",
|
||
output: "Sending feedback...",
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
const settings = settingsManager.getSettings();
|
||
const apiKey =
|
||
process.env.LETTA_API_KEY || settings.env?.LETTA_API_KEY;
|
||
|
||
// Only send anonymized, safe settings for debugging
|
||
const {
|
||
env: _env,
|
||
refreshToken: _refreshToken,
|
||
anthropicOAuth: _anthropicOAuth,
|
||
...safeSettings
|
||
} = settings;
|
||
|
||
const response = await fetch(
|
||
"https://api.letta.com/v1/metadata/feedback",
|
||
{
|
||
method: "POST",
|
||
headers: {
|
||
"Content-Type": "application/json",
|
||
...(apiKey ? { Authorization: `Bearer ${apiKey}` } : {}),
|
||
"X-Letta-Source": "letta-code",
|
||
"X-Letta-Code-Device-ID": settingsManager.getOrCreateDeviceId(),
|
||
},
|
||
body: JSON.stringify({
|
||
message: resolvedMessage,
|
||
feature: "letta-code",
|
||
agent_id: agentId,
|
||
session_id: telemetry.getSessionId(),
|
||
version: process.env.npm_package_version || "unknown",
|
||
platform: process.platform,
|
||
settings: JSON.stringify(safeSettings),
|
||
}),
|
||
},
|
||
);
|
||
|
||
if (!response.ok) {
|
||
const errorText = await response.text();
|
||
throw new Error(
|
||
`Failed to send feedback (${response.status}): ${errorText}`,
|
||
);
|
||
}
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/feedback",
|
||
output:
|
||
"Feedback submitted! To chat with the Letta dev team live, join our Discord (https://discord.gg/letta).",
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
refreshDerived();
|
||
} catch (error) {
|
||
const errorDetails = formatErrorDetails(error, agentId);
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/feedback",
|
||
output: `Failed to send feedback: ${errorDetails}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
}
|
||
});
|
||
},
|
||
[agentId, refreshDerived, withCommandLock, closeOverlay],
|
||
);
|
||
|
||
const handleProfileEscapeCancel = useCallback(() => {
|
||
if (profileConfirmPending) {
|
||
const { cmdId, name } = profileConfirmPending;
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: `/profile load ${name}`,
|
||
output: "Cancelled",
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
refreshDerived();
|
||
setProfileConfirmPending(null);
|
||
}
|
||
}, [profileConfirmPending, refreshDerived]);
|
||
|
||
// Track permission mode changes for UI updates
|
||
const [uiPermissionMode, setUiPermissionMode] = useState(
|
||
permissionMode.getMode(),
|
||
);
|
||
|
||
// Handle permission mode changes from the Input component (e.g., shift+tab cycling)
|
||
const handlePermissionModeChange = useCallback((mode: PermissionMode) => {
|
||
// When entering plan mode via tab cycling, generate and set the plan file path
|
||
if (mode === "plan") {
|
||
const planPath = generatePlanFilePath();
|
||
permissionMode.setPlanFilePath(planPath);
|
||
}
|
||
// permissionMode.setMode() is called in InputRich.tsx before this callback
|
||
setUiPermissionMode(mode);
|
||
}, []);
|
||
|
||
const handlePlanApprove = useCallback(
|
||
async (acceptEdits: boolean = false) => {
|
||
const currentIndex = approvalResults.length;
|
||
const approval = pendingApprovals[currentIndex];
|
||
if (!approval) return;
|
||
|
||
const isLast = currentIndex + 1 >= pendingApprovals.length;
|
||
|
||
// Capture plan file path BEFORE exiting plan mode (for post-approval rendering)
|
||
const planFilePath = permissionMode.getPlanFilePath();
|
||
lastPlanFilePathRef.current = planFilePath;
|
||
|
||
// Exit plan mode
|
||
const newMode = acceptEdits ? "acceptEdits" : "default";
|
||
permissionMode.setMode(newMode);
|
||
setUiPermissionMode(newMode);
|
||
|
||
try {
|
||
// Execute ExitPlanMode tool to get the result
|
||
const parsedArgs = safeJsonParseOr<Record<string, unknown>>(
|
||
approval.toolArgs,
|
||
{},
|
||
);
|
||
const toolResult = await executeTool("ExitPlanMode", parsedArgs);
|
||
|
||
// Update buffers with tool return
|
||
onChunk(buffersRef.current, {
|
||
message_type: "tool_return_message",
|
||
id: "dummy",
|
||
date: new Date().toISOString(),
|
||
tool_call_id: approval.toolCallId,
|
||
tool_return: toolResult.toolReturn,
|
||
status: toolResult.status,
|
||
stdout: toolResult.stdout,
|
||
stderr: toolResult.stderr,
|
||
});
|
||
|
||
setThinkingMessage(getRandomThinkingVerb());
|
||
refreshDerived();
|
||
|
||
const decision = {
|
||
type: "approve" as const,
|
||
approval,
|
||
precomputedResult: toolResult,
|
||
};
|
||
|
||
if (isLast) {
|
||
setIsExecutingTool(true);
|
||
await sendAllResults(decision);
|
||
} else {
|
||
setApprovalResults((prev) => [...prev, decision]);
|
||
}
|
||
} catch (e) {
|
||
const errorDetails = formatErrorDetails(e, agentId);
|
||
appendError(errorDetails);
|
||
setStreaming(false);
|
||
}
|
||
},
|
||
[
|
||
agentId,
|
||
pendingApprovals,
|
||
approvalResults,
|
||
sendAllResults,
|
||
appendError,
|
||
refreshDerived,
|
||
setStreaming,
|
||
],
|
||
);
|
||
|
||
const handlePlanKeepPlanning = useCallback(
|
||
async (reason: string) => {
|
||
const currentIndex = approvalResults.length;
|
||
const approval = pendingApprovals[currentIndex];
|
||
if (!approval) return;
|
||
|
||
const isLast = currentIndex + 1 >= pendingApprovals.length;
|
||
|
||
// Stay in plan mode
|
||
const denialReason =
|
||
reason ||
|
||
"The user doesn't want to proceed with this tool use. The tool use was rejected (eg. if it was a file edit, the new_string was NOT written to the file). STOP what you are doing and wait for the user to tell you how to proceed.";
|
||
|
||
const decision = {
|
||
type: "deny" as const,
|
||
approval,
|
||
reason: denialReason,
|
||
};
|
||
|
||
if (isLast) {
|
||
setIsExecutingTool(true);
|
||
await sendAllResults(decision);
|
||
} else {
|
||
setApprovalResults((prev) => [...prev, decision]);
|
||
}
|
||
},
|
||
[pendingApprovals, approvalResults, sendAllResults],
|
||
);
|
||
|
||
// Auto-reject ExitPlanMode if plan file doesn't exist
|
||
useEffect(() => {
|
||
const currentIndex = approvalResults.length;
|
||
const approval = pendingApprovals[currentIndex];
|
||
if (approval?.toolName === "ExitPlanMode" && !planFileExists()) {
|
||
const planFilePath = permissionMode.getPlanFilePath();
|
||
handlePlanKeepPlanning(
|
||
`You must write your plan to the plan file before exiting plan mode.\n` +
|
||
`Plan file path: ${planFilePath || "not set"}\n` +
|
||
`Use a write tool (e.g. Write, ApplyPatch, etc.) to create your plan, then call ExitPlanMode again.`,
|
||
);
|
||
}
|
||
}, [pendingApprovals, approvalResults.length, handlePlanKeepPlanning]);
|
||
|
||
const handleQuestionSubmit = useCallback(
|
||
async (answers: Record<string, string>) => {
|
||
const currentIndex = approvalResults.length;
|
||
const approval = pendingApprovals[currentIndex];
|
||
if (!approval) return;
|
||
|
||
const isLast = currentIndex + 1 >= pendingApprovals.length;
|
||
|
||
// Get questions from approval args
|
||
const questions = getQuestionsFromApproval(approval);
|
||
|
||
// Check for memory preference question and update setting
|
||
parseMemoryPreference(questions, answers);
|
||
|
||
// Format the answer string like Claude Code does
|
||
const answerParts = questions.map((q) => {
|
||
const answer = answers[q.question] || "";
|
||
return `"${q.question}"="${answer}"`;
|
||
});
|
||
const toolReturn = `User has answered your questions: ${answerParts.join(", ")}. You can now continue with the user's answers in mind.`;
|
||
|
||
const precomputedResult: ToolExecutionResult = {
|
||
toolReturn,
|
||
status: "success",
|
||
};
|
||
|
||
// Update buffers with tool return
|
||
onChunk(buffersRef.current, {
|
||
message_type: "tool_return_message",
|
||
id: "dummy",
|
||
date: new Date().toISOString(),
|
||
tool_call_id: approval.toolCallId,
|
||
tool_return: toolReturn,
|
||
status: "success",
|
||
stdout: null,
|
||
stderr: null,
|
||
});
|
||
|
||
setThinkingMessage(getRandomThinkingVerb());
|
||
refreshDerived();
|
||
|
||
const decision = {
|
||
type: "approve" as const,
|
||
approval,
|
||
precomputedResult,
|
||
};
|
||
|
||
if (isLast) {
|
||
setIsExecutingTool(true);
|
||
await sendAllResults(decision);
|
||
} else {
|
||
setApprovalResults((prev) => [...prev, decision]);
|
||
}
|
||
},
|
||
[pendingApprovals, approvalResults, sendAllResults, refreshDerived],
|
||
);
|
||
|
||
const handleEnterPlanModeApprove = useCallback(async () => {
|
||
const currentIndex = approvalResults.length;
|
||
const approval = pendingApprovals[currentIndex];
|
||
if (!approval) return;
|
||
|
||
const isLast = currentIndex + 1 >= pendingApprovals.length;
|
||
|
||
// Generate plan file path
|
||
const planFilePath = generatePlanFilePath();
|
||
|
||
// Toggle plan mode on and store plan file path
|
||
permissionMode.setMode("plan");
|
||
permissionMode.setPlanFilePath(planFilePath);
|
||
setUiPermissionMode("plan");
|
||
|
||
// Get the tool return message from the implementation
|
||
const toolReturn = `Entered plan mode. You should now focus on exploring the codebase and designing an implementation approach.
|
||
|
||
In plan mode, you should:
|
||
1. Thoroughly explore the codebase to understand existing patterns
|
||
2. Identify similar features and architectural approaches
|
||
3. Consider multiple approaches and their trade-offs
|
||
4. Use AskUserQuestion if you need to clarify the approach
|
||
5. Design a concrete implementation strategy
|
||
6. When ready, use ExitPlanMode to present your plan for approval
|
||
|
||
Remember: DO NOT write or edit any files yet. This is a read-only exploration and planning phase.
|
||
|
||
Plan file path: ${planFilePath}`;
|
||
|
||
const precomputedResult: ToolExecutionResult = {
|
||
toolReturn,
|
||
status: "success",
|
||
};
|
||
|
||
// Update buffers with tool return
|
||
onChunk(buffersRef.current, {
|
||
message_type: "tool_return_message",
|
||
id: "dummy",
|
||
date: new Date().toISOString(),
|
||
tool_call_id: approval.toolCallId,
|
||
tool_return: toolReturn,
|
||
status: "success",
|
||
stdout: null,
|
||
stderr: null,
|
||
});
|
||
|
||
setThinkingMessage(getRandomThinkingVerb());
|
||
refreshDerived();
|
||
|
||
const decision = {
|
||
type: "approve" as const,
|
||
approval,
|
||
precomputedResult,
|
||
};
|
||
|
||
if (isLast) {
|
||
setIsExecutingTool(true);
|
||
await sendAllResults(decision);
|
||
} else {
|
||
setApprovalResults((prev) => [...prev, decision]);
|
||
}
|
||
}, [pendingApprovals, approvalResults, sendAllResults, refreshDerived]);
|
||
|
||
const handleEnterPlanModeReject = useCallback(async () => {
|
||
const currentIndex = approvalResults.length;
|
||
const approval = pendingApprovals[currentIndex];
|
||
if (!approval) return;
|
||
|
||
const isLast = currentIndex + 1 >= pendingApprovals.length;
|
||
|
||
const rejectionReason =
|
||
"User chose to skip plan mode and start implementing directly.";
|
||
|
||
const decision = {
|
||
type: "deny" as const,
|
||
approval,
|
||
reason: rejectionReason,
|
||
};
|
||
|
||
if (isLast) {
|
||
setIsExecutingTool(true);
|
||
await sendAllResults(decision);
|
||
} else {
|
||
setApprovalResults((prev) => [...prev, decision]);
|
||
}
|
||
}, [pendingApprovals, approvalResults, sendAllResults]);
|
||
|
||
// Live area shows only in-progress items
|
||
const liveItems = useMemo(() => {
|
||
return lines.filter((ln) => {
|
||
if (!("phase" in ln)) return false;
|
||
if (ln.kind === "command" || ln.kind === "bash_command") {
|
||
return ln.phase === "running";
|
||
}
|
||
if (ln.kind === "tool_call") {
|
||
// Task tool_calls need special handling:
|
||
// - Only include if pending approval (phase: "ready" or "streaming")
|
||
// - Running/finished Task tools are handled by SubagentGroupDisplay
|
||
if (ln.name && isTaskTool(ln.name)) {
|
||
// Only show Task tools that are awaiting approval (not running/finished)
|
||
return ln.phase === "ready" || ln.phase === "streaming";
|
||
}
|
||
// Always show other tool calls in progress
|
||
return ln.phase !== "finished";
|
||
}
|
||
if (!tokenStreamingEnabled && ln.phase === "streaming") return false;
|
||
return ln.phase === "streaming";
|
||
});
|
||
}, [lines, tokenStreamingEnabled]);
|
||
|
||
// Commit welcome snapshot once when ready for fresh sessions (no history)
|
||
// Wait for agentProvenance to be available for new agents (continueSession=false)
|
||
useEffect(() => {
|
||
if (
|
||
loadingState === "ready" &&
|
||
!welcomeCommittedRef.current &&
|
||
messageHistory.length === 0
|
||
) {
|
||
// For new agents, wait until provenance is available
|
||
// For resumed agents, provenance stays null (that's expected)
|
||
if (!continueSession && !agentProvenance) {
|
||
return; // Wait for provenance to be set
|
||
}
|
||
welcomeCommittedRef.current = true;
|
||
setStaticItems((prev) => [
|
||
...prev,
|
||
{
|
||
kind: "welcome",
|
||
id: `welcome-${Date.now().toString(36)}`,
|
||
snapshot: {
|
||
continueSession,
|
||
agentState,
|
||
agentProvenance,
|
||
terminalWidth: columns,
|
||
},
|
||
},
|
||
]);
|
||
|
||
// Add status line showing agent info
|
||
const statusId = `status-agent-${Date.now().toString(36)}`;
|
||
|
||
// Get short path for display
|
||
const cwd = process.cwd();
|
||
const shortCwd = cwd.startsWith(process.env.HOME || "")
|
||
? `~${cwd.slice((process.env.HOME || "").length)}`
|
||
: cwd;
|
||
|
||
// Check if agent is pinned (locally or globally)
|
||
const isPinned = agentState?.id
|
||
? settingsManager.getLocalPinnedAgents().includes(agentState.id) ||
|
||
settingsManager.getGlobalPinnedAgents().includes(agentState.id)
|
||
: false;
|
||
|
||
// Build status message based on session type
|
||
const agentName = agentState?.name || "Unnamed Agent";
|
||
const headerMessage = continueSession
|
||
? `Connecting to **${agentName}** (last used in ${shortCwd})`
|
||
: "Creating a new agent";
|
||
|
||
// Command hints - for pinned agents show /memory, for unpinned show /pin
|
||
const commandHints = isPinned
|
||
? [
|
||
"→ **/memory** view your agent's memory blocks",
|
||
"→ **/init** initialize your agent's memory",
|
||
"→ **/remember** teach your agent",
|
||
"→ **/agents** list agents",
|
||
"→ **/ade** open in the browser (web UI)",
|
||
]
|
||
: [
|
||
"→ **/pin** save + name your agent",
|
||
"→ **/init** initialize your agent's memory",
|
||
"→ **/remember** teach your agent",
|
||
"→ **/agents** list agents",
|
||
"→ **/ade** open in the browser (web UI)",
|
||
];
|
||
|
||
const statusLines = [headerMessage, ...commandHints];
|
||
|
||
buffersRef.current.byId.set(statusId, {
|
||
kind: "status",
|
||
id: statusId,
|
||
lines: statusLines,
|
||
});
|
||
buffersRef.current.order.push(statusId);
|
||
refreshDerived();
|
||
}
|
||
}, [
|
||
loadingState,
|
||
continueSession,
|
||
messageHistory.length,
|
||
columns,
|
||
agentProvenance,
|
||
agentState,
|
||
refreshDerived,
|
||
]);
|
||
|
||
return (
|
||
<Box key={resumeKey} flexDirection="column" gap={1}>
|
||
<Static
|
||
key={staticRenderEpoch}
|
||
items={staticItems}
|
||
style={{ flexDirection: "column" }}
|
||
>
|
||
{(item: StaticItem, index: number) => (
|
||
<Box key={item.id} marginTop={index > 0 ? 1 : 0}>
|
||
{item.kind === "welcome" ? (
|
||
<WelcomeScreen loadingState="ready" {...item.snapshot} />
|
||
) : item.kind === "user" ? (
|
||
<UserMessage line={item} />
|
||
) : item.kind === "reasoning" ? (
|
||
<ReasoningMessage line={item} />
|
||
) : item.kind === "assistant" ? (
|
||
<AssistantMessage line={item} />
|
||
) : item.kind === "tool_call" ? (
|
||
<ToolCallMessage
|
||
line={item}
|
||
precomputedDiffs={precomputedDiffsRef.current}
|
||
lastPlanFilePath={lastPlanFilePathRef.current}
|
||
/>
|
||
) : item.kind === "subagent_group" ? (
|
||
<SubagentGroupStatic agents={item.agents} />
|
||
) : item.kind === "error" ? (
|
||
<ErrorMessage line={item} />
|
||
) : item.kind === "status" ? (
|
||
<StatusMessage line={item} />
|
||
) : item.kind === "separator" ? (
|
||
<Text dimColor>{"─".repeat(columns)}</Text>
|
||
) : item.kind === "command" ? (
|
||
<CommandMessage line={item} />
|
||
) : item.kind === "bash_command" ? (
|
||
<BashCommandMessage line={item} />
|
||
) : item.kind === "approval_preview" ? (
|
||
<ApprovalPreview
|
||
toolName={item.toolName}
|
||
toolArgs={item.toolArgs}
|
||
precomputedDiff={item.precomputedDiff}
|
||
allDiffs={precomputedDiffsRef.current}
|
||
planContent={item.planContent}
|
||
planFilePath={item.planFilePath}
|
||
toolCallId={item.toolCallId}
|
||
/>
|
||
) : null}
|
||
</Box>
|
||
)}
|
||
</Static>
|
||
|
||
<Box flexDirection="column" gap={1}>
|
||
{/* Loading screen / intro text */}
|
||
{loadingState !== "ready" && (
|
||
<WelcomeScreen
|
||
loadingState={loadingState}
|
||
continueSession={continueSession}
|
||
agentState={agentState}
|
||
/>
|
||
)}
|
||
|
||
{loadingState === "ready" && (
|
||
<>
|
||
{/* Transcript */}
|
||
{/* Show liveItems always - all approvals now render inline */}
|
||
{liveItems.length > 0 && (
|
||
<Box flexDirection="column">
|
||
{liveItems.map((ln) => {
|
||
// Check if this tool call matches the current ExitPlanMode approval
|
||
const isExitPlanModeApproval =
|
||
ln.kind === "tool_call" &&
|
||
currentApproval?.toolName === "ExitPlanMode" &&
|
||
ln.toolCallId === currentApproval?.toolCallId;
|
||
|
||
// Check if this tool call matches a file edit/write/patch approval
|
||
const isFileEditApproval =
|
||
ln.kind === "tool_call" &&
|
||
currentApproval &&
|
||
(isFileEditTool(currentApproval.toolName) ||
|
||
isFileWriteTool(currentApproval.toolName) ||
|
||
isPatchTool(currentApproval.toolName)) &&
|
||
ln.toolCallId === currentApproval.toolCallId;
|
||
|
||
// Check if this tool call matches a bash/shell approval
|
||
const isBashApproval =
|
||
ln.kind === "tool_call" &&
|
||
currentApproval &&
|
||
isShellTool(currentApproval.toolName) &&
|
||
ln.toolCallId === currentApproval.toolCallId;
|
||
|
||
// Check if this tool call matches an EnterPlanMode approval
|
||
const isEnterPlanModeApproval =
|
||
ln.kind === "tool_call" &&
|
||
currentApproval?.toolName === "EnterPlanMode" &&
|
||
ln.toolCallId === currentApproval?.toolCallId;
|
||
|
||
// Check if this tool call matches an AskUserQuestion approval
|
||
const isAskUserQuestionApproval =
|
||
ln.kind === "tool_call" &&
|
||
currentApproval?.toolName === "AskUserQuestion" &&
|
||
ln.toolCallId === currentApproval?.toolCallId;
|
||
|
||
// Check if this tool call matches a Task tool approval
|
||
const isTaskToolApproval =
|
||
ln.kind === "tool_call" &&
|
||
currentApproval &&
|
||
isTaskTool(currentApproval.toolName) &&
|
||
ln.toolCallId === currentApproval.toolCallId;
|
||
|
||
// Parse file edit info from approval args
|
||
const getFileEditInfo = () => {
|
||
if (!isFileEditApproval || !currentApproval) return null;
|
||
try {
|
||
const args = JSON.parse(currentApproval.toolArgs || "{}");
|
||
|
||
// For patch tools, use the input field
|
||
if (isPatchTool(currentApproval.toolName)) {
|
||
return {
|
||
toolName: currentApproval.toolName,
|
||
filePath: "", // Patch can have multiple files
|
||
patchInput: args.input as string | undefined,
|
||
toolCallId: ln.toolCallId,
|
||
};
|
||
}
|
||
|
||
// For regular file edit/write tools
|
||
return {
|
||
toolName: currentApproval.toolName,
|
||
filePath: String(args.file_path || ""),
|
||
content: args.content as string | undefined,
|
||
oldString: args.old_string as string | undefined,
|
||
newString: args.new_string as string | undefined,
|
||
replaceAll: args.replace_all as boolean | undefined,
|
||
edits: args.edits as
|
||
| Array<{
|
||
old_string: string;
|
||
new_string: string;
|
||
replace_all?: boolean;
|
||
}>
|
||
| undefined,
|
||
toolCallId: ln.toolCallId,
|
||
};
|
||
} catch {
|
||
return null;
|
||
}
|
||
};
|
||
|
||
const fileEditInfo = getFileEditInfo();
|
||
|
||
// Parse bash info from approval args
|
||
const getBashInfo = () => {
|
||
if (!isBashApproval || !currentApproval) return null;
|
||
try {
|
||
const args = JSON.parse(currentApproval.toolArgs || "{}");
|
||
const t = currentApproval.toolName.toLowerCase();
|
||
|
||
// Handle different bash tool arg formats
|
||
let command = "";
|
||
let description = "";
|
||
|
||
if (t === "shell") {
|
||
// Shell tool uses command array and justification
|
||
const cmdVal = args.command;
|
||
command = Array.isArray(cmdVal)
|
||
? cmdVal.join(" ")
|
||
: typeof cmdVal === "string"
|
||
? cmdVal
|
||
: "(no command)";
|
||
description =
|
||
typeof args.justification === "string"
|
||
? args.justification
|
||
: "";
|
||
} else {
|
||
// Bash/shell_command uses command string and description
|
||
command =
|
||
typeof args.command === "string"
|
||
? args.command
|
||
: "(no command)";
|
||
description =
|
||
typeof args.description === "string"
|
||
? args.description
|
||
: "";
|
||
}
|
||
|
||
return {
|
||
toolName: currentApproval.toolName,
|
||
command,
|
||
description,
|
||
};
|
||
} catch {
|
||
return null;
|
||
}
|
||
};
|
||
|
||
const bashInfo = getBashInfo();
|
||
|
||
// Parse Task tool info from approval args
|
||
const getTaskInfo = () => {
|
||
if (!isTaskToolApproval || !currentApproval) return null;
|
||
try {
|
||
const args = JSON.parse(currentApproval.toolArgs || "{}");
|
||
return {
|
||
subagentType:
|
||
typeof args.subagent_type === "string"
|
||
? args.subagent_type
|
||
: "unknown",
|
||
description:
|
||
typeof args.description === "string"
|
||
? args.description
|
||
: "(no description)",
|
||
prompt:
|
||
typeof args.prompt === "string"
|
||
? args.prompt
|
||
: "(no prompt)",
|
||
model:
|
||
typeof args.model === "string"
|
||
? args.model
|
||
: undefined,
|
||
};
|
||
} catch {
|
||
return null;
|
||
}
|
||
};
|
||
|
||
const taskInfo = getTaskInfo();
|
||
|
||
return (
|
||
<Box key={ln.id} flexDirection="column" marginTop={1}>
|
||
{/* For ExitPlanMode awaiting approval: render StaticPlanApproval */}
|
||
{/* Plan preview is eagerly committed to staticItems, so this only shows options */}
|
||
{isExitPlanModeApproval ? (
|
||
<StaticPlanApproval
|
||
onApprove={() => handlePlanApprove(false)}
|
||
onApproveAndAcceptEdits={() =>
|
||
handlePlanApprove(true)
|
||
}
|
||
onKeepPlanning={handlePlanKeepPlanning}
|
||
isFocused={true}
|
||
/>
|
||
) : isFileEditApproval && fileEditInfo ? (
|
||
<InlineFileEditApproval
|
||
fileEdit={fileEditInfo}
|
||
precomputedDiff={
|
||
ln.toolCallId
|
||
? precomputedDiffsRef.current.get(ln.toolCallId)
|
||
: undefined
|
||
}
|
||
allDiffs={precomputedDiffsRef.current}
|
||
onApprove={(diffs) => handleApproveCurrent(diffs)}
|
||
onApproveAlways={(scope, diffs) =>
|
||
handleApproveAlways(scope, diffs)
|
||
}
|
||
onDeny={(reason) => handleDenyCurrent(reason)}
|
||
onCancel={handleCancelApprovals}
|
||
isFocused={true}
|
||
approveAlwaysText={
|
||
currentApprovalContext?.approveAlwaysText
|
||
}
|
||
allowPersistence={
|
||
currentApprovalContext?.allowPersistence ?? true
|
||
}
|
||
/>
|
||
) : isBashApproval && bashInfo ? (
|
||
<InlineBashApproval
|
||
bashInfo={bashInfo}
|
||
onApprove={() => handleApproveCurrent()}
|
||
onApproveAlways={(scope) =>
|
||
handleApproveAlways(scope)
|
||
}
|
||
onDeny={(reason) => handleDenyCurrent(reason)}
|
||
onCancel={handleCancelApprovals}
|
||
isFocused={true}
|
||
approveAlwaysText={
|
||
currentApprovalContext?.approveAlwaysText
|
||
}
|
||
allowPersistence={
|
||
currentApprovalContext?.allowPersistence ?? true
|
||
}
|
||
/>
|
||
) : isEnterPlanModeApproval ? (
|
||
<InlineEnterPlanModeApproval
|
||
onApprove={handleEnterPlanModeApprove}
|
||
onReject={handleEnterPlanModeReject}
|
||
isFocused={true}
|
||
/>
|
||
) : isAskUserQuestionApproval ? (
|
||
<InlineQuestionApproval
|
||
questions={getQuestionsFromApproval(currentApproval)}
|
||
onSubmit={handleQuestionSubmit}
|
||
onCancel={handleCancelApprovals}
|
||
isFocused={true}
|
||
/>
|
||
) : isTaskToolApproval && taskInfo ? (
|
||
<InlineTaskApproval
|
||
taskInfo={taskInfo}
|
||
onApprove={() => handleApproveCurrent()}
|
||
onApproveAlways={(scope) =>
|
||
handleApproveAlways(scope)
|
||
}
|
||
onDeny={(reason) => handleDenyCurrent(reason)}
|
||
onCancel={handleCancelApprovals}
|
||
isFocused={true}
|
||
approveAlwaysText={
|
||
currentApprovalContext?.approveAlwaysText
|
||
}
|
||
allowPersistence={
|
||
currentApprovalContext?.allowPersistence ?? true
|
||
}
|
||
/>
|
||
) : ln.kind === "tool_call" &&
|
||
currentApproval &&
|
||
ln.toolCallId === currentApproval.toolCallId ? (
|
||
// Generic fallback for any other tool needing approval
|
||
<InlineGenericApproval
|
||
toolName={currentApproval.toolName}
|
||
toolArgs={currentApproval.toolArgs}
|
||
onApprove={() => handleApproveCurrent()}
|
||
onApproveAlways={(scope) =>
|
||
handleApproveAlways(scope)
|
||
}
|
||
onDeny={(reason) => handleDenyCurrent(reason)}
|
||
onCancel={handleCancelApprovals}
|
||
isFocused={true}
|
||
approveAlwaysText={
|
||
currentApprovalContext?.approveAlwaysText
|
||
}
|
||
allowPersistence={
|
||
currentApprovalContext?.allowPersistence ?? true
|
||
}
|
||
/>
|
||
) : ln.kind === "user" ? (
|
||
<UserMessage line={ln} />
|
||
) : ln.kind === "reasoning" ? (
|
||
<ReasoningMessage line={ln} />
|
||
) : ln.kind === "assistant" ? (
|
||
<AssistantMessage line={ln} />
|
||
) : ln.kind === "tool_call" &&
|
||
ln.toolCallId &&
|
||
queuedIds.has(ln.toolCallId) ? (
|
||
// Render stub for queued (decided but not executed) approval
|
||
<PendingApprovalStub
|
||
toolName={
|
||
approvalMap.get(ln.toolCallId)?.toolName ||
|
||
ln.name ||
|
||
"Unknown"
|
||
}
|
||
description={stubDescriptions.get(ln.toolCallId)}
|
||
decision={queuedDecisions.get(ln.toolCallId)}
|
||
/>
|
||
) : ln.kind === "tool_call" &&
|
||
ln.toolCallId &&
|
||
pendingIds.has(ln.toolCallId) ? (
|
||
// Render stub for pending (undecided) approval
|
||
<PendingApprovalStub
|
||
toolName={
|
||
approvalMap.get(ln.toolCallId)?.toolName ||
|
||
ln.name ||
|
||
"Unknown"
|
||
}
|
||
description={stubDescriptions.get(ln.toolCallId)}
|
||
/>
|
||
) : ln.kind === "tool_call" ? (
|
||
<ToolCallMessage
|
||
line={ln}
|
||
precomputedDiffs={precomputedDiffsRef.current}
|
||
lastPlanFilePath={lastPlanFilePathRef.current}
|
||
/>
|
||
) : ln.kind === "error" ? (
|
||
<ErrorMessage line={ln} />
|
||
) : ln.kind === "status" ? (
|
||
<StatusMessage line={ln} />
|
||
) : ln.kind === "command" ? (
|
||
<CommandMessage line={ln} />
|
||
) : ln.kind === "bash_command" ? (
|
||
<BashCommandMessage line={ln} />
|
||
) : null}
|
||
</Box>
|
||
);
|
||
})}
|
||
</Box>
|
||
)}
|
||
|
||
{/* Subagent group display - shows running/completed subagents */}
|
||
<SubagentGroupDisplay />
|
||
|
||
{/* Exit stats - shown when exiting via double Ctrl+C */}
|
||
{showExitStats && (
|
||
<Box flexDirection="column" marginTop={1}>
|
||
<Text dimColor>
|
||
{formatUsageStats({
|
||
stats: sessionStatsRef.current.getSnapshot(),
|
||
})}
|
||
</Text>
|
||
<Text dimColor>Resume this agent with:</Text>
|
||
<Text color={colors.link.url}>
|
||
{/* Show -n "name" if agent has name and is pinned, otherwise --agent */}
|
||
{agentName &&
|
||
(settingsManager.getLocalPinnedAgents().includes(agentId) ||
|
||
settingsManager.getGlobalPinnedAgents().includes(agentId))
|
||
? `letta -n "${agentName}"`
|
||
: `letta --agent ${agentId}`}
|
||
</Text>
|
||
</Box>
|
||
)}
|
||
|
||
{/* Input row - always mounted to preserve state */}
|
||
<Box marginTop={liveItems.length > 0 ? 0 : 1}>
|
||
<Input
|
||
visible={
|
||
!showExitStats &&
|
||
pendingApprovals.length === 0 &&
|
||
!anySelectorOpen
|
||
}
|
||
streaming={
|
||
streaming && !abortControllerRef.current?.signal.aborted
|
||
}
|
||
tokenCount={tokenCount}
|
||
thinkingMessage={thinkingMessage}
|
||
onSubmit={onSubmit}
|
||
onBashSubmit={handleBashSubmit}
|
||
permissionMode={uiPermissionMode}
|
||
onPermissionModeChange={handlePermissionModeChange}
|
||
onExit={handleExit}
|
||
onInterrupt={handleInterrupt}
|
||
interruptRequested={interruptRequested}
|
||
agentId={agentId}
|
||
agentName={agentName}
|
||
currentModel={currentModelDisplay}
|
||
currentModelProvider={currentModelProvider}
|
||
messageQueue={messageQueue}
|
||
onEnterQueueEditMode={handleEnterQueueEditMode}
|
||
onEscapeCancel={
|
||
profileConfirmPending ? handleProfileEscapeCancel : undefined
|
||
}
|
||
/>
|
||
</Box>
|
||
|
||
{/* Model Selector - conditionally mounted as overlay */}
|
||
{activeOverlay === "model" && (
|
||
<ModelSelector
|
||
currentModelId={currentModelId ?? undefined}
|
||
onSelect={handleModelSelect}
|
||
onCancel={closeOverlay}
|
||
/>
|
||
)}
|
||
|
||
{/* Toolset Selector - conditionally mounted as overlay */}
|
||
{activeOverlay === "toolset" && (
|
||
<ToolsetSelector
|
||
currentToolset={currentToolset ?? undefined}
|
||
onSelect={handleToolsetSelect}
|
||
onCancel={closeOverlay}
|
||
/>
|
||
)}
|
||
|
||
{/* System Prompt Selector - conditionally mounted as overlay */}
|
||
{activeOverlay === "system" && (
|
||
<SystemPromptSelector
|
||
currentPromptId={currentSystemPromptId ?? undefined}
|
||
onSelect={handleSystemPromptSelect}
|
||
onCancel={closeOverlay}
|
||
/>
|
||
)}
|
||
|
||
{/* Agent Selector - conditionally mounted as overlay */}
|
||
{activeOverlay === "agent" && (
|
||
<AgentSelector
|
||
currentAgentId={agentId}
|
||
onSelect={handleAgentSelect}
|
||
onCancel={closeOverlay}
|
||
/>
|
||
)}
|
||
|
||
{/* Subagent Manager - for managing custom subagents */}
|
||
{activeOverlay === "subagent" && (
|
||
<SubagentManager onClose={closeOverlay} />
|
||
)}
|
||
|
||
{/* Resume Selector - conditionally mounted as overlay */}
|
||
{activeOverlay === "resume" && (
|
||
<ResumeSelector
|
||
currentAgentId={agentId}
|
||
onSelect={async (id) => {
|
||
closeOverlay();
|
||
await handleAgentSelect(id);
|
||
}}
|
||
onCancel={closeOverlay}
|
||
/>
|
||
)}
|
||
|
||
{/* Message Search - conditionally mounted as overlay */}
|
||
{activeOverlay === "search" && (
|
||
<MessageSearch onClose={closeOverlay} />
|
||
)}
|
||
|
||
{/* Feedback Dialog - conditionally mounted as overlay */}
|
||
{activeOverlay === "feedback" && (
|
||
<FeedbackDialog
|
||
onSubmit={handleFeedbackSubmit}
|
||
onCancel={closeOverlay}
|
||
initialValue={feedbackPrefill}
|
||
/>
|
||
)}
|
||
|
||
{/* Memory Viewer - conditionally mounted as overlay */}
|
||
{activeOverlay === "memory" && (
|
||
<MemoryViewer
|
||
blocks={agentState?.memory?.blocks || []}
|
||
agentId={agentId}
|
||
agentName={agentName}
|
||
onClose={closeOverlay}
|
||
/>
|
||
)}
|
||
|
||
{/* MCP Server Selector - conditionally mounted as overlay */}
|
||
{activeOverlay === "mcp" && (
|
||
<McpSelector
|
||
agentId={agentId}
|
||
onAdd={() => {
|
||
// Close overlay and prompt user to use /mcp add command
|
||
closeOverlay();
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/mcp",
|
||
output:
|
||
"Use /mcp add --transport <http|sse|stdio> <name> <url|command> [...] to add a new server",
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
}}
|
||
onCancel={closeOverlay}
|
||
/>
|
||
)}
|
||
|
||
{/* Help Dialog - conditionally mounted as overlay */}
|
||
{activeOverlay === "help" && <HelpDialog onClose={closeOverlay} />}
|
||
|
||
{/* OAuth Code Dialog - for Claude OAuth connection */}
|
||
{activeOverlay === "oauth" && (
|
||
<OAuthCodeDialog
|
||
onComplete={(success, message) => {
|
||
closeOverlay();
|
||
const cmdId = uid("cmd");
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/connect claude",
|
||
output: message,
|
||
phase: "finished",
|
||
success,
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
}}
|
||
onCancel={closeOverlay}
|
||
onModelSwitch={async (modelHandle: string) => {
|
||
const { updateAgentLLMConfig } = await import(
|
||
"../agent/modify"
|
||
);
|
||
const { getModelUpdateArgs, getModelInfo } = await import(
|
||
"../agent/model"
|
||
);
|
||
const updateArgs = getModelUpdateArgs(modelHandle);
|
||
await updateAgentLLMConfig(agentId, modelHandle, updateArgs);
|
||
// Update current model display - use model id for correct "(current)" indicator
|
||
const modelInfo = getModelInfo(modelHandle);
|
||
setCurrentModelId(modelInfo?.id || modelHandle);
|
||
}}
|
||
/>
|
||
)}
|
||
|
||
{/* New Agent Dialog - for naming new agent before creation */}
|
||
{activeOverlay === "new" && (
|
||
<NewAgentDialog
|
||
onSubmit={handleCreateNewAgent}
|
||
onCancel={closeOverlay}
|
||
/>
|
||
)}
|
||
|
||
{/* Pin Dialog - for naming agent before pinning */}
|
||
{activeOverlay === "pin" && (
|
||
<PinDialog
|
||
currentName={agentName || ""}
|
||
local={pinDialogLocal}
|
||
onSubmit={async (newName) => {
|
||
closeOverlay();
|
||
setCommandRunning(true);
|
||
|
||
const cmdId = uid("cmd");
|
||
const scopeText = pinDialogLocal
|
||
? "to this project"
|
||
: "globally";
|
||
const displayName =
|
||
newName || agentName || agentId.slice(0, 12);
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/pin",
|
||
output: `Pinning "${displayName}" ${scopeText}...`,
|
||
phase: "running",
|
||
});
|
||
buffersRef.current.order.push(cmdId);
|
||
refreshDerived();
|
||
|
||
try {
|
||
const client = await getClient();
|
||
|
||
// Rename if new name provided
|
||
if (newName && newName !== agentName) {
|
||
await client.agents.update(agentId, { name: newName });
|
||
setAgentName(newName);
|
||
}
|
||
|
||
// Pin the agent
|
||
if (pinDialogLocal) {
|
||
settingsManager.pinLocal(agentId);
|
||
} else {
|
||
settingsManager.pinGlobal(agentId);
|
||
}
|
||
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/pin",
|
||
output: `Pinned "${newName || agentName || agentId.slice(0, 12)}" ${scopeText}.`,
|
||
phase: "finished",
|
||
success: true,
|
||
});
|
||
} catch (error) {
|
||
buffersRef.current.byId.set(cmdId, {
|
||
kind: "command",
|
||
id: cmdId,
|
||
input: "/pin",
|
||
output: `Failed to pin: ${error}`,
|
||
phase: "finished",
|
||
success: false,
|
||
});
|
||
} finally {
|
||
setCommandRunning(false);
|
||
refreshDerived();
|
||
}
|
||
}}
|
||
onCancel={closeOverlay}
|
||
/>
|
||
)}
|
||
|
||
{/* Plan Mode Dialog - NOW RENDERED INLINE with tool call (see liveItems above) */}
|
||
{/* ExitPlanMode approval is handled by InlinePlanApproval component */}
|
||
|
||
{/* AskUserQuestion now rendered inline via InlineQuestionApproval */}
|
||
{/* EnterPlanMode now rendered inline in liveItems above */}
|
||
{/* ApprovalDialog removed - all approvals now render inline via InlineGenericApproval fallback */}
|
||
</>
|
||
)}
|
||
</Box>
|
||
</Box>
|
||
);
|
||
}
|