feat: unify creation agent chat fill

2026-04-25 10:50:19 +08:00
parent c06bf84d0a
commit 31f350d499
26 changed files with 540 additions and 153 deletions
--- a/src/components/big-fish-creation/BigFishAgentWorkspace.interaction.test.tsx
+++ b/src/components/big-fish-creation/BigFishAgentWorkspace.interaction.test.tsx
@@ -1,4 +1,4 @@
-/* @vitest-environment jsdom */
+/* @vitest-environment jsdom */

 import { render, screen } from '@testing-library/react';
 import userEvent from '@testing-library/user-event';
@@ -81,11 +81,12 @@ test('big fish workspace submits quick keyword fill request after two turns', as
    />,
  );

-  await user.click(screen.getByRole('button', { name: '补充剩余关键字' }));
+  await user.click(screen.getByRole('button', { name: '补充剩余设定' }));

  expect(onSubmitMessage).toHaveBeenCalledWith(
    expect.objectContaining({
-      text: '请补充剩余关键字。',
+      text: '请补充剩余设定。',
+      quickFillRequested: true,
    }),
  );
 });
@@ -100,5 +101,5 @@ test('big fish workspace hides keyword fill before two turns', () => {
    />,
  );

-  expect(screen.queryByRole('button', { name: '补充剩余关键字' })).toBeNull();
+  expect(screen.queryByRole('button', { name: '补充剩余设定' })).toBeNull();
 });
--- a/src/components/big-fish-creation/BigFishAgentWorkspace.tsx
+++ b/src/components/big-fish-creation/BigFishAgentWorkspace.tsx
@@ -4,7 +4,12 @@ import type {
  ExecuteBigFishActionRequest,
  SendBigFishMessageRequest,
 } from '../../../packages/shared/src/contracts/bigFish';
-import { createCreationAgentClientMessageId } from '../../services/creation-agent';
+import {
+  buildCreationAgentChatMessage,
+  createCreationAgentChatQuickActions,
+  createCreationAgentClientMessageId,
+  resolveCreationAgentQuickActionMessage,
+} from '../../services/creation-agent';
 import {
  type CreationAgentAnchorView,
  type CreationAgentSessionView,
@@ -84,35 +89,30 @@ export function BigFishAgentWorkspace({
      isStreamingReply={Boolean(streamingReplyText)}
      isBusy={isBusy}
      error={error}
-      quickActions={[
-        {
-          key: 'summarize',
-          label: '总结当前设定',
-        },
-        {
-          key: 'quickFill',
-          label: '补充剩余关键字',
-          minTurn: 2,
-        },
-      ]}
+      quickActions={createCreationAgentChatQuickActions()}
      onBack={onBack}
      onSubmitText={(text) => {
-        onSubmitMessage({
-          clientMessageId: createCreationAgentClientMessageId('big-fish'),
-          text,
-        });
+        onSubmitMessage(
+          buildCreationAgentChatMessage({
+            clientMessageId: createCreationAgentClientMessageId('big-fish'),
+            text,
+          }),
+        );
      }}
      onPrimaryAction={() => {
        onExecuteAction({ action: 'big_fish_compile_draft' });
      }}
      onQuickAction={(action) => {
-        onSubmitMessage({
-          clientMessageId: createCreationAgentClientMessageId('big-fish'),
-          text:
-            action.key === 'quickFill'
-              ? '请补充剩余关键字。'
-              : '请总结一下当前已经成形的大鱼吃小鱼设定。',
-        });
+        const quickActionMessage = resolveCreationAgentQuickActionMessage(
+          action.key,
+          '请总结一下当前已经成形的大鱼吃小鱼设定。',
+        );
+        onSubmitMessage(
+          buildCreationAgentChatMessage({
+            clientMessageId: createCreationAgentClientMessageId('big-fish'),
+            ...quickActionMessage,
+          }),
+        );
      }}
    />
  );
--- a/src/components/creation-agent/CreationAgentWorkspace.test.tsx
+++ b/src/components/creation-agent/CreationAgentWorkspace.test.tsx
@@ -1,8 +1,10 @@
-/* @vitest-environment jsdom */
+/* @vitest-environment jsdom */

 import { fireEvent, render, screen } from '@testing-library/react';
 import { afterEach, expect, test, vi } from 'vitest';

+import { createCreationAgentChatQuickActions } from '../../services/creation-agent';
+
 import {
  type CreationAgentTheme,
  CreationAgentWorkspace,
@@ -243,17 +245,7 @@ test('creation agent workspace shows primary and progress actions at completed p
      loadingText="正在准备"
      composerPlaceholder="输入消息"
      primaryActionLabel="生成结果页"
-      quickActions={[
-        {
-          key: 'summarize',
-          label: '总结当前设定',
-        },
-        {
-          key: 'quickFill',
-          label: '补全剩余设定',
-          minTurn: 2,
-        },
-      ]}
+      quickActions={createCreationAgentChatQuickActions()}
      onBack={() => {}}
      onSubmitText={() => {}}
      onPrimaryAction={() => {}}
@@ -262,7 +254,7 @@ test('creation agent workspace shows primary and progress actions at completed p

  expect(screen.getByRole('button', { name: '生成结果页' })).toBeTruthy();
  expect(screen.getByRole('button', { name: '总结当前设定' })).toBeTruthy();
-  expect(screen.getByRole('button', { name: '补全剩余设定' })).toBeTruthy();
+  expect(screen.getByRole('button', { name: '补充剩余设定' })).toBeTruthy();
 });

 test('creation agent workspace hides hero copy area when title and summary are absent', () => {
--- a/src/components/custom-world-agent/CustomWorldAgentWorkspace.interaction.test.tsx
+++ b/src/components/custom-world-agent/CustomWorldAgentWorkspace.interaction.test.tsx
@@ -1,4 +1,4 @@
-/* @vitest-environment jsdom */
+/* @vitest-environment jsdom */

 import { render, screen } from '@testing-library/react';
 import userEvent from '@testing-library/user-event';
@@ -108,11 +108,11 @@ test('workspace enables quick fill after at least two turns and submits quick fi
    />,
  );

-  await user.click(screen.getByRole('button', { name: '补全剩余设定' }));
+  await user.click(screen.getByRole('button', { name: '补充剩余设定' }));

  expect(onSubmitMessage).toHaveBeenCalledWith(
    expect.objectContaining({
-      text: '请补全剩余设定。',
+      text: '请补充剩余设定。',
      quickFillRequested: true,
    }),
  );
--- a/src/components/custom-world-agent/CustomWorldAgentWorkspace.tsx
+++ b/src/components/custom-world-agent/CustomWorldAgentWorkspace.tsx
@@ -5,8 +5,11 @@ import type {
  SendCustomWorldAgentMessageRequest,
 } from '../../../packages/shared/src/contracts/customWorldAgent';
 import {
+  buildCreationAgentChatMessage,
+  createCreationAgentChatQuickActions,
  createCreationAgentClientMessageId,
  isCreationAgentOperationBusy,
+  resolveCreationAgentQuickActionMessage,
 } from '../../services/creation-agent';
 import {
  type CreationAgentAnchorView,
@@ -166,13 +169,17 @@ export function CustomWorldAgentWorkspace({
    isCreationAgentOperationBusy(activeOperation) || isStreamingReply;

  const submitMessage = (text: string, quickFillRequested = false) => {
-    onSubmitMessage({
-      clientMessageId: createCreationAgentClientMessageId('custom-world'),
-      text,
-      quickFillRequested,
-      focusCardId: null,
-      selectedCardIds: [],
-    });
+    onSubmitMessage(
+      buildCreationAgentChatMessage({
+        clientMessageId: createCreationAgentClientMessageId('custom-world'),
+        text,
+        quickFillRequested,
+        extraPayload: {
+          focusCardId: null,
+          selectedCardIds: [],
+        },
+      }),
+    );
  };

  return (
@@ -186,17 +193,7 @@ export function CustomWorldAgentWorkspace({
      streamingReplyText={streamingReplyText}
      isStreamingReply={isStreamingReply}
      isBusy={isBusy}
-      quickActions={[
-        {
-          key: 'summarize',
-          label: '总结当前设定',
-        },
-        {
-          key: 'quickFill',
-          label: '补全剩余设定',
-          minTurn: 2,
-        },
-      ]}
+      quickActions={createCreationAgentChatQuickActions()}
      onBack={onBack}
      onSubmitText={(text) => {
        submitMessage(text);
@@ -207,12 +204,14 @@ export function CustomWorldAgentWorkspace({
        });
      }}
      onQuickAction={(action) => {
-        if (action.key === 'quickFill') {
-          submitMessage('请补全剩余设定。', true);
-          return;
-        }
-
-        submitMessage('请总结一下当前已经成形的世界设定。');
+        const quickActionMessage = resolveCreationAgentQuickActionMessage(
+          action.key,
+          '请总结一下当前已经成形的世界设定。',
+        );
+        submitMessage(
+          quickActionMessage.text,
+          quickActionMessage.quickFillRequested,
+        );
      }}
    />
  );
--- a/src/components/puzzle-agent/PuzzleAgentWorkspace.interaction.test.tsx
+++ b/src/components/puzzle-agent/PuzzleAgentWorkspace.interaction.test.tsx
@@ -1,4 +1,4 @@
-/* @vitest-environment jsdom */
+/* @vitest-environment jsdom */

 import { render, screen } from '@testing-library/react';
 import userEvent from '@testing-library/user-event';
@@ -80,11 +80,12 @@ test('puzzle workspace submits quick keyword fill request after two turns', asyn
    />,
  );

-  await user.click(screen.getByRole('button', { name: '补充剩余关键字' }));
+  await user.click(screen.getByRole('button', { name: '补充剩余设定' }));

  expect(onSubmitMessage).toHaveBeenCalledWith(
    expect.objectContaining({
-      text: '请补充剩余关键字。',
+      text: '请补充剩余设定。',
+      quickFillRequested: true,
    }),
  );
 });
@@ -99,5 +100,5 @@ test('puzzle workspace hides keyword fill before two turns', () => {
    />,
  );

-  expect(screen.queryByRole('button', { name: '补充剩余关键字' })).toBeNull();
+  expect(screen.queryByRole('button', { name: '补充剩余设定' })).toBeNull();
 });
--- a/src/components/puzzle-agent/PuzzleAgentWorkspace.tsx
+++ b/src/components/puzzle-agent/PuzzleAgentWorkspace.tsx
@@ -6,7 +6,12 @@ import type {
  PuzzleAgentSessionSnapshot,
  SendPuzzleAgentMessageRequest,
 } from '../../../packages/shared/src/contracts/puzzleAgentSession';
-import { createCreationAgentClientMessageId } from '../../services/creation-agent';
+import {
+  buildCreationAgentChatMessage,
+  createCreationAgentChatQuickActions,
+  createCreationAgentClientMessageId,
+  resolveCreationAgentQuickActionMessage,
+} from '../../services/creation-agent';
 import {
  type CreationAgentOperationView,
  type CreationAgentSessionView,
@@ -100,35 +105,30 @@ export function PuzzleAgentWorkspace({
      isStreamingReply={Boolean(streamingReplyText)}
      isBusy={isBusy}
      error={error}
-      quickActions={[
-        {
-          key: 'summarize',
-          label: '总结当前设定',
-        },
-        {
-          key: 'quickFill',
-          label: '补充剩余关键字',
-          minTurn: 2,
-        },
-      ]}
+      quickActions={createCreationAgentChatQuickActions()}
      onBack={onBack}
      onSubmitText={(text) => {
-        onSubmitMessage({
-          clientMessageId: createCreationAgentClientMessageId('puzzle'),
-          text,
-        });
+        onSubmitMessage(
+          buildCreationAgentChatMessage({
+            clientMessageId: createCreationAgentClientMessageId('puzzle'),
+            text,
+          }),
+        );
      }}
      onPrimaryAction={() => {
        onExecuteAction({ action: 'compile_puzzle_draft' });
      }}
      onQuickAction={(action) => {
-        onSubmitMessage({
-          clientMessageId: createCreationAgentClientMessageId('puzzle'),
-          text:
-            action.key === 'quickFill'
-              ? '请补充剩余关键字。'
-              : '请总结一下当前已经成形的拼图设定。',
-        });
+        const quickActionMessage = resolveCreationAgentQuickActionMessage(
+          action.key,
+          '请总结一下当前已经成形的拼图设定。',
+        );
+        onSubmitMessage(
+          buildCreationAgentChatMessage({
+            clientMessageId: createCreationAgentClientMessageId('puzzle'),
+            ...quickActionMessage,
+          }),
+        );
      }}
    />
  );