1

src/services/aiService.ts

@@ -23,6 +23,8 @@ import type {
   CharacterChatSuggestionsRequest,
   CharacterChatSummaryRequest,
   NpcChatDialogueRequest,
+  NpcChatTurnRequest,
+  NpcChatTurnResult,
   NpcRecruitDialogueRequest,
   PlainTextResponse,
 } from '../../packages/shared/src/contracts/story';
@@ -156,6 +158,37 @@ async function requestPlainTextStream(
   return accumulatedText.trim();
 }
 
+type ParsedSseEvent = {
+  event: string | null;
+  data: string;
+};
+
+function parseSseEventBlock(eventBlock: string): ParsedSseEvent | null {
+  let eventName: string | null = null;
+  const dataLines: string[] = [];
+
+  for (const rawLine of eventBlock.split(/\r?\n/u)) {
+    const line = rawLine.trim();
+    if (!line) continue;
+    if (line.startsWith('event:')) {
+      eventName = line.slice(6).trim() || null;
+      continue;
+    }
+    if (line.startsWith('data:')) {
+      dataLines.push(line.slice(5).trim());
+    }
+  }
+
+  if (dataLines.length === 0) {
+    return null;
+  }
+
+  return {
+    event: eventName,
+    data: dataLines.join('\n'),
+  };
+}
+
 export async function generateInitialStory(
   world: WorldType,
   character: Character,
@@ -893,6 +926,109 @@ export async function streamNpcChatDialogue(
   return dialogue.trim();
 }
 
+export async function streamNpcChatTurn(
+  world: WorldType,
+  character: Character,
+  encounter: Encounter,
+  monsters: SceneHostileNpc[],
+  history: StoryMoment[],
+  context: StoryGenerationContext,
+  conversationHistory: StoryMoment['dialogue'],
+  playerMessage: string,
+  npcState: Record<string, unknown>,
+  options: {
+    onReplyUpdate?: (text: string) => void;
+  } = {},
+) {
+  const payload = {
+    worldType: world,
+    character,
+    encounter,
+    monsters,
+    history,
+    context,
+    conversationHistory: conversationHistory ?? [],
+    playerMessage,
+    npcState,
+  } satisfies NpcChatTurnRequest;
+
+  const response = await fetchWithApiAuth(`${RUNTIME_API_BASE}/chat/npc/turn/stream`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify(payload),
+  });
+
+  if (!response.ok) {
+    const responseText = await response.text();
+    throw new Error(parseApiErrorMessage(responseText, 'NPC 聊天续写失败'));
+  }
+
+  if (!response.body) {
+    throw new Error('streaming response body is unavailable');
+  }
+
+  const reader = response.body.getReader();
+  const decoder = new TextDecoder('utf-8');
+  let buffer = '';
+  let accumulatedReply = '';
+  let completedResult: NpcChatTurnResult | null = null;
+
+  for (;;) {
+    const { done, value } = await reader.read();
+    if (done) {
+      break;
+    }
+
+    buffer += decoder.decode(value, { stream: true });
+
+    while (buffer.includes('\n\n')) {
+      const boundary = buffer.indexOf('\n\n');
+      const eventBlock = buffer.slice(0, boundary);
+      buffer = buffer.slice(boundary + 2);
+
+      const parsedEvent = parseSseEventBlock(eventBlock);
+      if (!parsedEvent) {
+        continue;
+      }
+
+      if (parsedEvent.data === '[DONE]') {
+        continue;
+      }
+
+      if (parsedEvent.event === 'reply_delta') {
+        const payloadRecord = JSON.parse(parsedEvent.data) as Record<string, unknown>;
+        const nextText =
+          typeof payloadRecord.text === 'string' ? payloadRecord.text : '';
+        accumulatedReply = nextText;
+        options.onReplyUpdate?.(accumulatedReply);
+        continue;
+      }
+
+      if (parsedEvent.event === 'complete') {
+        completedResult = JSON.parse(parsedEvent.data) as NpcChatTurnResult;
+        accumulatedReply = completedResult.npcReply;
+        options.onReplyUpdate?.(accumulatedReply);
+        continue;
+      }
+
+      if (parsedEvent.event === 'error') {
+        const payloadRecord = JSON.parse(parsedEvent.data) as Record<string, unknown>;
+        throw new Error(
+          typeof payloadRecord.message === 'string'
+            ? payloadRecord.message
+            : 'NPC 聊天续写失败',
+        );
+      }
+    }
+  }
+
+  if (!completedResult) {
+    throw new Error('NPC 聊天续写结果为空');
+  }
+
+  return completedResult;
+}
+
 export async function streamNpcRecruitDialogue(
   world: WorldType,
   character: Character,