Merge pull request #78 from ai-action/feat/history

Author: remarkablemark

src/components/Chat/Chat.test.tsx

@@ -8,10 +8,12 @@ import { prewarmCodeBlocks } from '../CodeBlock';
 
 const mockState = vi.hoisted(() => ({
   handler: undefined as ((value: string) => void) | undefined,
+  history: [] as string[],
   testInput: '',
   shouldReset: false,
   clear() {
     this.handler = undefined;
+    this.history = [];
     this.testInput = '';
     this.shouldReset = true;
   },
@@ -92,8 +94,9 @@ vi.mock('../../utils', async () => ({
   },
 }));
 
-vi.mock('./Input', () => ({
-  Input: (props: {
+vi.mock('./ChatInput', () => ({
+  ChatInput: (props: {
+    history?: string[];
     onSubmit?: (value: string) => void;
     onInterrupt?: () => void;
     isDisabled?: boolean;
@@ -102,6 +105,8 @@ vi.mock('./Input', () => ({
       mockState.handler = props.onSubmit;
     }
 
+    mockState.history = props.history ?? [];
+
     if (props.onInterrupt) {
       interruptState.handler = props.onInterrupt;
     }
@@ -216,6 +221,28 @@ describe('Chat', () => {
     expect(frame).toContain('hello');
   }, 10_000);
 
+  it('derives prompt history from user messages and excludes slash commands', async () => {
+    render(
+      <Chat
+        initialMessages={[
+          { role: 'user', content: 'first prompt' },
+          { role: 'assistant', content: 'response' },
+          { role: 'user', content: '/model' },
+          { role: 'user', content: 'second prompt' },
+        ]}
+        model="gemma4"
+        onCommand={vi.fn()}
+        mode={MODE.SAFE}
+        onModeChange={onModeChange}
+        sessionId="0"
+      />,
+    );
+
+    await time.tick();
+
+    expect(mockState.history).toEqual(['first prompt', 'second prompt']);
+  });
+
   it('does not add blank messages', async () => {
     const chat = (
       <Chat

src/components/Chat/Chat.tsx

@@ -1,5 +1,5 @@
 import { Box, Text } from 'ink';
-import { useCallback, useEffect, useRef, useState } from 'react';
+import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
 
 import { DECISION, MODE, PROMPT, ROLE } from '../../constants';
 import type { Decision, Mode, ToolName, ToolResult } from '../../types';
@@ -9,13 +9,13 @@ import { Messages } from '../Messages';
 import { TURN_ABORTED_MESSAGE } from '../Messages/constants';
 import { PlanApproval } from '../PlanApproval';
 import { ToolApproval } from '../ToolApproval';
+import { ChatInput } from './ChatInput';
 import {
   ACTION_NOT_PERFORMED,
   INTERRUPT_REASON,
   PLAN_CHECKLIST_REMINDER,
   PLAN_EXECUTION_REMINDER,
 } from './constants';
-import { Input } from './Input';
 import { hasExecutablePlan } from './plan';
 
 interface Props {
@@ -38,6 +38,13 @@ export function Chat({
   sessionId,
 }: Props) {
   const sessionMessages = initialMessages ?? [];
+  const history = useMemo(
+    () =>
+      sessionMessages.flatMap(({ role, content }) =>
+        role === ROLE.USER && !content.startsWith('/') ? [content] : [],
+      ),
+    [sessionMessages],
+  );
   const [messages, setMessages] = useState<ollama.Message[]>(sessionMessages);
   const [streamingMessage, setStreamingMessage] =
     useState<ollama.Message | null>(null);
@@ -585,7 +592,8 @@ export function Chat({
 
       {!pendingPlan && !pendingToolCall && (
         <Box marginTop={1}>
-          <Input
+          <ChatInput
+            history={history}
             isDisabled={isLoading}
             onInterrupt={handleInterrupt}
             // eslint-disable-next-line @typescript-eslint/no-misused-promises