feat: add TUI wizard streaming steps and docs for memory record streaming

Author: Hweinstock

docs/memory.md

@@ -229,6 +229,13 @@ pushed to a delivery target in your account, enabling event-driven architectures
 
 ### Enabling Streaming
 
+Via the interactive wizard:
+
+```bash
+agentcore add memory
+# Select "Yes" when prompted for streaming, then provide the data stream ARN and content level
+```
+
 Via CLI flags:
 
 ```bash

src/cli/tui/hooks/useCreateMemory.ts

@@ -8,6 +8,7 @@ interface CreateMemoryConfig {
   name: string;
   eventExpiryDuration: number;
   strategies: { type: string }[];
+  streaming?: { dataStreamArn: string; contentLevel: string };
 }
 
 interface CreateStatus<T> {
@@ -27,6 +28,8 @@ export function useCreateMemory() {
         name: config.name,
         expiry: config.eventExpiryDuration,
         strategies: strategiesStr || undefined,
+        dataStreamArn: config.streaming?.dataStreamArn,
+        contentLevel: config.streaming?.contentLevel,
       });
       if (!addResult.success) {
         throw new Error(addResult.error ?? 'Failed to create memory');

src/cli/tui/screens/memory/AddMemoryScreen.tsx

@@ -1,4 +1,4 @@
-import type { MemoryStrategyType } from '../../../../schema';
+import type { MemoryStrategyType, StreamContentLevel } from '../../../../schema';
 import { AgentNameSchema } from '../../../../schema';
 import {
   ConfirmReview,
@@ -14,7 +14,7 @@ import { HELP_TEXT } from '../../constants';
 import { useListNavigation, useMultiSelectNavigation } from '../../hooks';
 import { generateUniqueName } from '../../utils';
 import type { AddMemoryConfig } from './types';
-import { EVENT_EXPIRY_OPTIONS, MEMORY_STEP_LABELS, MEMORY_STRATEGY_OPTIONS } from './types';
+import { CONTENT_LEVEL_OPTIONS, EVENT_EXPIRY_OPTIONS, MEMORY_STEP_LABELS, MEMORY_STRATEGY_OPTIONS } from './types';
 import { useAddMemoryWizard } from './useAddMemoryWizard';
 import React, { useMemo } from 'react';
 
@@ -24,6 +24,11 @@ interface AddMemoryScreenProps {
   existingMemoryNames: string[];
 }
 
+const STREAMING_OPTIONS: SelectableItem[] = [
+  { id: 'yes', title: 'Yes', description: 'Stream memory record events to a delivery target (e.g. Kinesis)' },
+  { id: 'no', title: 'No', description: 'No streaming' },
+];
+
 export function AddMemoryScreen({ onComplete, onExit, existingMemoryNames }: AddMemoryScreenProps) {
   const wizard = useAddMemoryWizard();
 
@@ -37,9 +42,17 @@ export function AddMemoryScreen({ onComplete, onExit, existingMemoryNames }: Add
     []
   );
 
+  const contentLevelItems: SelectableItem[] = useMemo(
+    () => CONTENT_LEVEL_OPTIONS.map(opt => ({ id: opt.id, title: opt.title, description: opt.description })),
+    []
+  );
+
   const isNameStep = wizard.step === 'name';
   const isExpiryStep = wizard.step === 'expiry';
   const isStrategiesStep = wizard.step === 'strategies';
+  const isStreamingStep = wizard.step === 'streaming';
+  const isStreamArnStep = wizard.step === 'streamArn';
+  const isContentLevelStep = wizard.step === 'contentLevel';
   const isConfirmStep = wizard.step === 'confirm';
 
   const expiryNav = useListNavigation({
@@ -58,6 +71,20 @@ export function AddMemoryScreen({ onComplete, onExit, existingMemoryNames }: Add
     requireSelection: false,
   });
 
+  const streamingNav = useListNavigation({
+    items: STREAMING_OPTIONS,
+    onSelect: item => wizard.setStreamingEnabled(item.id === 'yes'),
+    onExit: () => wizard.goBack(),
+    isActive: isStreamingStep,
+  });
+
+  const contentLevelNav = useListNavigation({
+    items: contentLevelItems,
+    onSelect: item => wizard.setContentLevel(item.id as StreamContentLevel),
+    onExit: () => wizard.goBack(),
+    isActive: isContentLevelStep,
+  });
+
   useListNavigation({
     items: [{ id: 'confirm', title: 'Confirm' }],
     onSelect: () => onComplete(wizard.config),
@@ -67,14 +94,29 @@ export function AddMemoryScreen({ onComplete, onExit, existingMemoryNames }: Add
 
   const helpText = isStrategiesStep
     ? 'Space toggle · Enter confirm · Esc back'
-    : isExpiryStep
+    : isExpiryStep || isStreamingStep || isContentLevelStep
       ? HELP_TEXT.NAVIGATE_SELECT
       : isConfirmStep
         ? HELP_TEXT.CONFIRM_CANCEL
         : HELP_TEXT.TEXT_INPUT;
 
   const headerContent = <StepIndicator steps={wizard.steps} currentStep={wizard.step} labels={MEMORY_STEP_LABELS} />;
 
+  const confirmFields = useMemo(
+    () => [
+      { label: 'Name', value: wizard.config.name },
+      { label: 'Event Expiry', value: `${wizard.config.eventExpiryDuration} days` },
+      { label: 'Strategies', value: wizard.config.strategies.map(s => s.type).join(', ') || 'None' },
+      ...(wizard.config.streaming
+        ? [
+            { label: 'Stream ARN', value: wizard.config.streaming.dataStreamArn },
+            { label: 'Content Level', value: wizard.config.streaming.contentLevel },
+          ]
+        : [{ label: 'Streaming', value: 'Disabled' }]),
+    ],
+    [wizard.config]
+  );
+
   return (
     <Screen title="Add Memory" onExit={onExit} helpText={helpText} headerContent={headerContent}>
       <Panel>
@@ -109,15 +151,36 @@ export function AddMemoryScreen({ onComplete, onExit, existingMemoryNames }: Add
           />
         )}
 
-        {isConfirmStep && (
-          <ConfirmReview
-            fields={[
-              { label: 'Name', value: wizard.config.name },
-              { label: 'Event Expiry', value: `${wizard.config.eventExpiryDuration} days` },
-              { label: 'Strategies', value: wizard.config.strategies.map(s => s.type).join(', ') || 'None' },
-            ]}
+        {isStreamingStep && (
+          <WizardSelect
+            title="Enable memory record streaming?"
+            description="Stream memory record lifecycle events to a delivery target"
+            items={STREAMING_OPTIONS}
+            selectedIndex={streamingNav.selectedIndex}
+          />
+        )}
+
+        {isStreamArnStep && (
+          <TextInput
+            key="streamArn"
+            prompt="Delivery target ARN (e.g. Kinesis stream)"
+            initialValue=""
+            onSubmit={wizard.setStreamArn}
+            onCancel={() => wizard.goBack()}
+            customValidation={value => value.startsWith('arn:') || 'Must be a valid ARN (starts with arn:)'}
           />
         )}
+
+        {isContentLevelStep && (
+          <WizardSelect
+            title="Stream content level"
+            description="What data to include in stream events"
+            items={contentLevelItems}
+            selectedIndex={contentLevelNav.selectedIndex}
+          />
+        )}
+
+        {isConfirmStep && <ConfirmReview fields={confirmFields} />}
       </Panel>
     </Screen>
   );

src/cli/tui/screens/memory/types.ts

@@ -1,26 +1,35 @@
-import type { MemoryStrategyType } from '../../../../schema';
+import type { MemoryStrategyType, StreamContentLevel } from '../../../../schema';
 import { MemoryStrategyTypeSchema } from '../../../../schema';
 
 // ─────────────────────────────────────────────────────────────────────────────
 // Memory Flow Types
 // ─────────────────────────────────────────────────────────────────────────────
 
-export type AddMemoryStep = 'name' | 'expiry' | 'strategies' | 'confirm';
+export type AddMemoryStep = 'name' | 'expiry' | 'strategies' | 'streaming' | 'streamArn' | 'contentLevel' | 'confirm';
 
 export interface AddMemoryStrategyConfig {
   type: MemoryStrategyType;
 }
 
+export interface AddMemoryStreamingConfig {
+  dataStreamArn: string;
+  contentLevel: StreamContentLevel;
+}
+
 export interface AddMemoryConfig {
   name: string;
   eventExpiryDuration: number;
   strategies: AddMemoryStrategyConfig[];
+  streaming?: AddMemoryStreamingConfig;
 }
 
 export const MEMORY_STEP_LABELS: Record<AddMemoryStep, string> = {
   name: 'Name',
   expiry: 'Expiry',
   strategies: 'Strategies',
+  streaming: 'Streaming',
+  streamArn: 'Stream ARN',
+  contentLevel: 'Content Level',
   confirm: 'Confirm',
 };
 
@@ -49,6 +58,11 @@ export const EVENT_EXPIRY_OPTIONS = [
   { id: 365, title: '365 days', description: 'Maximum retention' },
 ] as const;
 
+export const CONTENT_LEVEL_OPTIONS = [
+  { id: 'FULL_CONTENT', title: 'Full content', description: 'Include memory record text in stream events' },
+  { id: 'METADATA_ONLY', title: 'Metadata only', description: 'Only include metadata (IDs, timestamps, namespaces)' },
+] as const;
+
 // ─────────────────────────────────────────────────────────────────────────────
 // Defaults
 // ─────────────────────────────────────────────────────────────────────────────

src/cli/tui/screens/memory/useAddMemoryWizard.ts

@@ -1,9 +1,12 @@
-import type { MemoryStrategyType } from '../../../../schema';
+import type { MemoryStrategyType, StreamContentLevel } from '../../../../schema';
 import type { AddMemoryConfig, AddMemoryStep, AddMemoryStrategyConfig } from './types';
 import { DEFAULT_EVENT_EXPIRY } from './types';
-import { useCallback, useState } from 'react';
+import { useCallback, useMemo, useState } from 'react';
 
-const ALL_STEPS: AddMemoryStep[] = ['name', 'expiry', 'strategies', 'confirm'];
+const BASE_STEPS = ['name', 'expiry', 'strategies', 'streaming'] as const;
+const STREAMING_STEPS = ['streamArn', 'contentLevel'] as const;
+const FIRST_STREAMING_STEP = STREAMING_STEPS[0];
+const CONFIRM_STEP = 'confirm' as const;
 
 function getDefaultConfig(): AddMemoryConfig {
   return {
@@ -16,18 +19,27 @@ function getDefaultConfig(): AddMemoryConfig {
 export function useAddMemoryWizard() {
   const [config, setConfig] = useState<AddMemoryConfig>(getDefaultConfig);
   const [step, setStep] = useState<AddMemoryStep>('name');
+  const [enableStreaming, setEnableStreaming] = useState(false);
 
-  const currentIndex = ALL_STEPS.indexOf(step);
+  const allSteps = useMemo(
+    () => (enableStreaming ? [...BASE_STEPS, ...STREAMING_STEPS, CONFIRM_STEP] : [...BASE_STEPS, CONFIRM_STEP]),
+    [enableStreaming]
+  );
+  const currentIndex = allSteps.indexOf(step);
 
   const goBack = useCallback(() => {
-    const prevStep = ALL_STEPS[currentIndex - 1];
+    const idx = allSteps.indexOf(step);
+    const prevStep = allSteps[idx - 1];
     if (prevStep) setStep(prevStep);
-  }, [currentIndex]);
+  }, [allSteps, step]);
 
-  const nextStep = useCallback((currentStep: AddMemoryStep): AddMemoryStep | undefined => {
-    const idx = ALL_STEPS.indexOf(currentStep);
-    return ALL_STEPS[idx + 1];
-  }, []);
+  const nextStep = useCallback(
+    (currentStep: AddMemoryStep): AddMemoryStep | undefined => {
+      const idx = allSteps.indexOf(currentStep);
+      return allSteps[idx + 1];
+    },
+    [allSteps]
+  );
 
   const setName = useCallback(
     (name: string) => {
@@ -57,20 +69,65 @@ export function useAddMemoryWizard() {
     [nextStep]
   );
 
+  const setStreamingEnabled = useCallback((enabled: boolean) => {
+    setEnableStreaming(enabled);
+    if (enabled) {
+      // Can't use nextStep() here — allSteps hasn't updated yet since
+      // setEnableStreaming is queued. Hardcode the known next step.
+      setStep(FIRST_STREAMING_STEP);
+    } else {
+      setConfig(c => ({ ...c, streaming: undefined }));
+      setStep(CONFIRM_STEP);
+    }
+  }, []);
+
+  const setStreamArn = useCallback(
+    (dataStreamArn: string) => {
+      setConfig(c => ({
+        ...c,
+        streaming: { dataStreamArn, contentLevel: c.streaming?.contentLevel ?? 'FULL_CONTENT' },
+      }));
+      const next = nextStep(FIRST_STREAMING_STEP);
+      if (next) setStep(next);
+    },
+    [nextStep]
+  );
+
+  const setContentLevel = useCallback(
+    (contentLevel: StreamContentLevel) => {
+      setConfig(c => {
+        if (!c.streaming?.dataStreamArn) {
+          throw new Error('Cannot set content level without a data stream ARN');
+        }
+        return {
+          ...c,
+          streaming: { dataStreamArn: c.streaming.dataStreamArn, contentLevel },
+        };
+      });
+      const next = nextStep('contentLevel');
+      if (next) setStep(next);
+    },
+    [nextStep]
+  );
+
   const reset = useCallback(() => {
     setConfig(getDefaultConfig());
     setStep('name');
+    setEnableStreaming(false);
   }, []);
 
   return {
     config,
     step,
-    steps: ALL_STEPS,
+    steps: allSteps,
     currentIndex,
     goBack,
     setName,
     setExpiry,
     setStrategyTypes,
+    setStreamingEnabled,
+    setStreamArn,
+    setContentLevel,
     reset,
   };
 }