claude-code/src/components/agents/new-agent-creation/wizard-steps/MemoryStep.tsx

import { type ReactNode } from 'react';
import { Box, Byline, KeyboardShortcutHint } from '@anthropic/ink';
import { useKeybinding } from '../../../../keybindings/useKeybinding.js';
import { isAutoMemoryEnabled } from '../../../../memdir/paths.js';
import {
  type AgentMemoryScope,
  loadAgentMemoryPrompt,
} from '@claude-code-best/builtin-tools/tools/AgentTool/agentMemory.js';
import { ConfigurableShortcutHint } from '../../../ConfigurableShortcutHint.js';
import { Select } from '../../../CustomSelect/select.js';
import { useWizard } from '../../../wizard/index.js';
import { WizardDialogLayout } from '../../../wizard/WizardDialogLayout.js';
import type { AgentWizardData } from '../types.js';

type MemoryOption = {
  label: string;
  value: AgentMemoryScope | 'none';
};

export function MemoryStep(): ReactNode {
  const { goNext, goBack, updateWizardData, wizardData } = useWizard<AgentWizardData>();

  useKeybinding('confirm:no', goBack, { context: 'Confirmation' });

  const isUserScope = wizardData.location === 'userSettings';

  // Build options with the recommended default first, then alternatives
  // The recommended scope matches the agent's location (project agent → project memory, user agent → user memory)
  const memoryOptions: MemoryOption[] = isUserScope
    ? [
        {
          label: 'User scope (~/.claude/agent-memory/) (Recommended)',
          value: 'user',
        },
        { label: 'None (no persistent memory)', value: 'none' },
        { label: 'Project scope (.claude/agent-memory/)', value: 'project' },
        { label: 'Local scope (.claude/agent-memory-local/)', value: 'local' },
      ]
    : [
        {
          label: 'Project scope (.claude/agent-memory/) (Recommended)',
          value: 'project',
        },
        { label: 'None (no persistent memory)', value: 'none' },
        { label: 'User scope (~/.claude/agent-memory/)', value: 'user' },
        { label: 'Local scope (.claude/agent-memory-local/)', value: 'local' },
      ];

  const handleSelect = (value: string): void => {
    const memory = value === 'none' ? undefined : (value as AgentMemoryScope);
    const agentType = wizardData.finalAgent?.agentType;
    updateWizardData({
      selectedMemory: memory,
      // Update finalAgent with memory and rewire getSystemPrompt to include memory loading.
      // Explicitly set memory (not conditional spread) so selecting 'none' after going back clears it.
      finalAgent: wizardData.finalAgent
        ? {
            ...wizardData.finalAgent,
            memory,
            getSystemPrompt:
              isAutoMemoryEnabled() && memory && agentType
                ? () => wizardData.systemPrompt! + '\n\n' + loadAgentMemoryPrompt(agentType, memory)
                : () => wizardData.systemPrompt!,
          }
        : undefined,
    });
    goNext();
  };

  return (
    <WizardDialogLayout
      subtitle="Configure agent memory"
      footerText={
        <Byline>
          <KeyboardShortcutHint shortcut="↑↓" action="navigate" />
          <KeyboardShortcutHint shortcut="Enter" action="select" />
          <ConfigurableShortcutHint action="confirm:no" context="Confirmation" fallback="Esc" description="go back" />
        </Byline>
      }
    >
      <Box>
        <Select key="memory-select" options={memoryOptions} onChange={handleSelect} onCancel={goBack} />
      </Box>
    </WizardDialogLayout>
  );
}