@lvce-editor/chat-view 2.4.0 → 2.6.0

package/dist/chatViewWorkerMain.js

@@ -1078,6 +1078,8 @@ const {
   set: set$3
 } = create$2(6002);
 
+const Div$1 = 4;
+
 const ClientX = 'event.clientX';
 const ClientY = 'event.clientY';
 const Key = 'event.key';
@@ -2739,22 +2741,166 @@ const readNextChunk = async () => {
   return promise;
 };
 
-const getMockOpenApiAssistantText = async (stream, onTextChunk) => {
+const parseSseDataLines = eventChunk => {
+  const lines = eventChunk.split('\n');
+  const dataLines = [];
+  for (const line of lines) {
+    if (!line.startsWith('data:')) {
+      continue;
+    }
+    dataLines.push(line.slice(5).trimStart());
+  }
+  return dataLines;
+};
+const emitToolCalls = async (toolCallAccumulator, onToolCallsChunk) => {
+  if (!onToolCallsChunk) {
+    return;
+  }
+  const toolCalls = Object.entries(toolCallAccumulator).toSorted((a, b) => Number(a[0]) - Number(b[0])).map(entry => entry[1]).filter(toolCall => !!toolCall.name);
+  if (toolCalls.length === 0) {
+    return;
+  }
+  await onToolCallsChunk(toolCalls);
+};
+const getMockOpenApiAssistantText = async (stream, onTextChunk, onToolCallsChunk, onDataEvent, onEventStreamFinished) => {
   const error = takeErrorResponse();
   if (error) {
     return error;
   }
   let text = '';
+  let remainder = '';
+  let toolCallAccumulator = {};
+  let finishedNotified = false;
+  const notifyFinished = async () => {
+    if (finishedNotified) {
+      return;
+    }
+    finishedNotified = true;
+    if (onEventStreamFinished) {
+      await onEventStreamFinished();
+    }
+  };
+  const handleParsedSseEvent = async parsed => {
+    if (onDataEvent) {
+      await onDataEvent(parsed);
+    }
+    if (!parsed || typeof parsed !== 'object') {
+      return;
+    }
+    const eventType = Reflect.get(parsed, 'type');
+    if (eventType === 'response.completed') {
+      await notifyFinished();
+      return;
+    }
+    if (eventType === 'response.output_text.delta') {
+      const delta = Reflect.get(parsed, 'delta');
+      if (typeof delta !== 'string' || !delta) {
+        return;
+      }
+      text += delta;
+      if (stream && onTextChunk) {
+        await onTextChunk(delta);
+      }
+      return;
+    }
+    if (eventType === 'response.output_item.added') {
+      const outputIndex = Reflect.get(parsed, 'output_index');
+      const item = Reflect.get(parsed, 'item');
+      if (typeof outputIndex !== 'number' || !item || typeof item !== 'object') {
+        return;
+      }
+      if (Reflect.get(item, 'type') !== 'function_call') {
+        return;
+      }
+      const name = Reflect.get(item, 'name');
+      const argumentsValue = Reflect.get(item, 'arguments');
+      const callId = Reflect.get(item, 'call_id');
+      toolCallAccumulator = {
+        ...toolCallAccumulator,
+        [outputIndex]: {
+          arguments: typeof argumentsValue === 'string' ? argumentsValue : '',
+          ...(typeof callId === 'string' ? {
+            id: callId
+          } : {}),
+          name: typeof name === 'string' ? name : ''
+        }
+      };
+      await emitToolCalls(toolCallAccumulator, onToolCallsChunk);
+      return;
+    }
+    if (eventType === 'response.function_call_arguments.delta' || eventType === 'response.function_call_arguments.done') {
+      const outputIndex = Reflect.get(parsed, 'output_index');
+      if (typeof outputIndex !== 'number') {
+        return;
+      }
+      const current = toolCallAccumulator[outputIndex] || {
+        arguments: '',
+        name: ''
+      };
+      const delta = Reflect.get(parsed, 'delta');
+      const argumentsValue = Reflect.get(parsed, 'arguments');
+      const name = Reflect.get(parsed, 'name');
+      const callId = Reflect.get(parsed, 'call_id');
+      const next = {
+        arguments: typeof argumentsValue === 'string' ? argumentsValue : typeof delta === 'string' ? `${current.arguments}${delta}` : current.arguments,
+        ...(typeof callId === 'string' ? {
+          id: callId
+        } : current.id ? {
+          id: current.id
+        } : {}),
+        name: typeof name === 'string' && name ? name : current.name
+      };
+      toolCallAccumulator = {
+        ...toolCallAccumulator,
+        [outputIndex]: next
+      };
+      await emitToolCalls(toolCallAccumulator, onToolCallsChunk);
+    }
+  };
+  const consumeSseDataLines = async dataLines => {
+    for (const line of dataLines) {
+      if (line === '[DONE]') {
+        await notifyFinished();
+        continue;
+      }
+      let parsed;
+      try {
+        parsed = JSON.parse(line);
+      } catch {
+        continue;
+      }
+      await handleParsedSseEvent(parsed);
+    }
+  };
   while (true) {
     const chunk = await readNextChunk();
     if (typeof chunk !== 'string') {
       break;
     }
+    if (chunk.startsWith('data:')) {
+      remainder += chunk;
+      while (true) {
+        const separatorIndex = remainder.indexOf('\n\n');
+        if (separatorIndex === -1) {
+          break;
+        }
+        const rawEvent = remainder.slice(0, separatorIndex);
+        remainder = remainder.slice(separatorIndex + 2);
+        const dataLines = parseSseDataLines(rawEvent);
+        await consumeSseDataLines(dataLines);
+      }
+      continue;
+    }
     text += chunk;
     if (stream && onTextChunk) {
       await onTextChunk(chunk);
     }
   }
+  if (remainder) {
+    const dataLines = parseSseDataLines(remainder);
+    await consumeSseDataLines(dataLines);
+  }
+  await notifyFinished();
   return {
     text,
     type: 'success'
@@ -4079,7 +4225,7 @@ const getAiResponse = async ({
   const usesOpenRouterModel = isOpenRouterModel(selectedModelId, models);
   if (usesOpenApiModel) {
     if (useMockApi) {
-      const result = await getMockOpenApiAssistantText(streamingEnabled, onTextChunk);
+      const result = await getMockOpenApiAssistantText(streamingEnabled, onTextChunk, onToolCallsChunk, onDataEvent, onEventStreamFinished);
       if (result.type === 'success') {
         const {
           text: assistantText
@@ -5641,6 +5787,27 @@ const getToolCallArgumentPreview = rawArguments => {
   return rawArguments;
 };
 
+const getToolCallDom = toolCall => {
+  const argumentPreview = getToolCallArgumentPreview(toolCall.arguments);
+  const label = `${toolCall.name} ${argumentPreview}`;
+  return [{
+    childCount: 1,
+    className: Markdown,
+    type: P
+  }, text(label)];
+};
+
+const getToolCallsDom = message => {
+  if (message.role !== 'assistant' || !message.toolCalls || message.toolCalls.length === 0) {
+    return [];
+  }
+  return [{
+    childCount: message.toolCalls.length,
+    className: 'ToolCallsContainer',
+    type: Div$1
+  }, ...message.toolCalls.flatMap(getToolCallDom)];
+};
+
 const orderedListItemRegex = /^\s*\d+\.\s+(.*)$/;
 const parseMessageContent = rawMessage => {
   if (rawMessage === '') {
@@ -5721,20 +5888,6 @@ const getMessageContentDom = nodes => {
   });
 };
 
-const getToolCallsDom = message => {
-  if (message.role !== 'assistant' || !message.toolCalls || message.toolCalls.length === 0) {
-    return [];
-  }
-  return message.toolCalls.flatMap(toolCall => {
-    const argumentPreview = getToolCallArgumentPreview(toolCall.arguments);
-    const label = `${toolCall.name} ${argumentPreview}`;
-    return [{
-      childCount: 1,
-      className: Markdown,
-      type: P
-    }, text(label)];
-  });
-};
 const getChatMessageDom = (message, openRouterApiKeyInput, openApiApiKeyInput = '', openRouterApiKeyState = 'idle') => {
   const roleClassName = message.role === 'user' ? MessageUser : MessageAssistant;
   const isOpenApiApiKeyMissingMessage = message.role === 'assistant' && message.text === openApiApiKeyRequiredMessage;
@@ -5744,7 +5897,8 @@ const getChatMessageDom = (message, openRouterApiKeyInput, openApiApiKeyInput =
   const messageIntermediate = parseMessageContent(message.text);
   const messageDom = getMessageContentDom(messageIntermediate);
   const toolCallsDom = getToolCallsDom(message);
-  const extraChildCount = isOpenApiApiKeyMissingMessage || isOpenRouterApiKeyMissingMessage || isOpenRouterRequestFailedMessage || isOpenRouterTooManyRequestsMessage ? messageIntermediate.length + 1 + toolCallsDom.length : messageIntermediate.length + toolCallsDom.length;
+  const toolCallsChildCount = toolCallsDom.length > 0 ? 1 : 0;
+  const extraChildCount = isOpenApiApiKeyMissingMessage || isOpenRouterApiKeyMissingMessage || isOpenRouterRequestFailedMessage || isOpenRouterTooManyRequestsMessage ? messageIntermediate.length + 1 + toolCallsChildCount : messageIntermediate.length + toolCallsChildCount;
   return [{
     childCount: 1,
     className: mergeClassNames(Message, roleClassName),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lvce-editor/chat-view",
-  "version": "2.4.0",
+  "version": "2.6.0",
   "description": "Chat View Worker",
   "repository": {
     "type": "git",