1

src/services/aiService.ts

@@ -629,6 +629,104 @@ export async function sendCustomWorldAgentMessage(
   );
 }
 
+export async function streamCustomWorldAgentMessage(
+  sessionId: string,
+  payload: SendCustomWorldAgentMessageRequest,
+  options: TextStreamOptions = {},
+) {
+  const response = await fetchWithApiAuth(
+    `${RUNTIME_API_BASE}/custom-world/agent/sessions/${encodeURIComponent(sessionId)}/messages/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, '发送共创消息失败'));
+  }
+
+  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 finalSession: CustomWorldAgentSessionSnapshot | 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 CustomWorldAgentSessionSnapshot;
+        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;
+}
+
 export async function executeCustomWorldAgentAction(
   sessionId: string,
   payload: CustomWorldAgentActionRequest,