1

src/services/rpg-creation/rpgCreationAgentClient.ts

@@ -9,6 +9,7 @@ import type {
 } from '../../../packages/shared/src';
 import type { RpgAgentActionRequest } from '../../../packages/shared/src';
 import type { TextStreamOptions } from '../aiTypes';
+import { readCreationAgentSessionFromSse } from '../creation-agent';
 import {
   openRpgCreationSsePost,
   requestRpgCreationPostJson,
@@ -63,84 +64,11 @@ export async function streamRpgCreationMessage(
     '发送共创消息失败',
   );
 
-  const streamBody = response.body;
-  if (!streamBody) {
-    throw new Error('streaming response body is unavailable');
-  }
-
-  const reader = streamBody.getReader();
-  const decoder = new TextDecoder('utf-8');
-  let buffer = '';
-  let finalSession: RpgAgentSessionSnapshot | 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);
-
-      let eventName = 'message';
-      const dataLines: string[] = [];
-
-      for (const rawLine of eventBlock.split(/\r?\n/u)) {
-        const line = rawLine.trim();
-        if (line.startsWith('event:')) {
-          eventName = line.slice(6).trim() || 'message';
-          continue;
-        }
-        if (line.startsWith('data:')) {
-          dataLines.push(line.slice(5).trim());
-        }
-      }
-
-      if (dataLines.length === 0) {
-        continue;
-      }
-
-      const data = dataLines.join('\n');
-      let parsed: Record<string, unknown> | null = null;
-
-      try {
-        parsed = JSON.parse(data) as Record<string, unknown>;
-      } catch {
-        parsed = null;
-      }
-
-      if (eventName === 'reply_delta' && parsed) {
-        const text = parsed.text;
-        if (typeof text === 'string') {
-          options.onUpdate?.(text);
-        }
-        continue;
-      }
-
-      if (eventName === 'session' && parsed?.session) {
-        finalSession = parsed.session as RpgAgentSessionSnapshot;
-        continue;
-      }
-
-      if (eventName === 'error' && parsed) {
-        const message =
-          typeof parsed.message === 'string' && parsed.message.trim()
-            ? parsed.message.trim()
-            : '发送共创消息失败';
-        throw new Error(message);
-      }
-    }
-  }
-
-  if (!finalSession) {
-    throw new Error('共创消息流式结果不完整');
-  }
-
-  return finalSession;
+  return readCreationAgentSessionFromSse<RpgAgentSessionSnapshot>(response, {
+    ...options,
+    fallbackMessage: '发送共创消息失败',
+    incompleteMessage: '共创消息流式结果不完整',
+  });
 }
 
 export async function executeRpgCreationAction(