@lvce-editor/chat-view 3.2.0 → 3.4.0

package/dist/chatViewWorkerMain.js

@@ -1118,6 +1118,9 @@ const sendMessagePortToExtensionHostWorker$1 = async (port, rpcId = 0) => {
   const command = 'HandleMessagePort.handleMessagePort2';
   await invokeAndTransfer('SendMessagePortToExtensionHostWorker.sendMessagePortToExtensionHostWorker', port, command, rpcId);
 };
+const sendMessagePortToChatNetworkWorker = async port => {
+  await invokeAndTransfer('SendMessagePortToExtensionHostWorker.sendMessagePortToChatNetworkWorker', port, 'HandleMessagePort.handleMessagePort');
+};
 const activateByEvent$1 = (event, assetDir, platform) => {
   return invoke('ExtensionHostManagement.activateByEvent', event, assetDir, platform);
 };
@@ -1275,6 +1278,9 @@ const getComposerHeight = async (state, value, width = state.width) => {
     composerLineHeight,
     maxComposerRows
   } = state;
+  if (value === '') {
+    return composerLineHeight;
+  }
   const minimumHeight = getMinComposerHeight(composerLineHeight);
   const maximumHeight = getMaxComposerHeight(composerLineHeight, maxComposerRows);
   const content = value || ' ';
@@ -1463,6 +1469,7 @@ const createDefaultState = () => {
   const composerFontSize = 13;
   const composerLineHeight = 20;
   return {
+    aiSessionTitleGenerationEnabled: false,
     assetDir: '',
     chatListScrollTop: 0,
     chatMessageFontFamily: 'system-ui',
@@ -5015,6 +5022,10 @@ const handleToolCallsChunkFunction = async (uid, assistantMessageId, toolCalls,
   };
 };
 
+const slashCommandRegex = /^\/(clear|export|help|new)(?:\s+.*)?$/;
+const mentionRegex = /(^|\s)@([^\s]+)/g;
+const maxMentionCount = 5;
+const maxMentionTextLength = 8000;
 const appendMessageToSelectedSession = (sessions, selectedSessionId, message) => {
   return sessions.map(session => {
     if (session.id !== selectedSessionId) {
@@ -5068,8 +5079,208 @@ const isStreamingFunctionCallEvent = parsed => {
 const getSseEventType = value => {
   return value && typeof value === 'object' && Reflect.get(value, 'type') === 'response.completed' ? 'sse-response-completed' : 'sse-response-part';
 };
+const getCommandHelpText = () => {
+  return ['Available commands:', '/new - Create and switch to a new chat session.', '/clear - Clear messages in the selected chat session.', '/export - Export current chat session as Markdown.', '/help - Show this help.'].join('\n');
+};
+const withClearedComposer = state => {
+  return focusInput({
+    ...state,
+    composerHeight: getMinComposerHeightForState(state),
+    composerValue: '',
+    inputSource: 'script'
+  });
+};
+const toMarkdownTranscript = session => {
+  const lines = [`# ${session.title}`, ''];
+  for (const message of session.messages) {
+    const role = message.role === 'assistant' ? 'Assistant' : 'User';
+    lines.push(`## ${role}`);
+    lines.push(message.text || '(empty)');
+    lines.push('');
+  }
+  return lines.join('\n').trim();
+};
+const executeSlashCommand = async (state, command) => {
+  if (command === 'new') {
+    const nextState = await createSession(state);
+    return withClearedComposer({
+      ...nextState,
+      viewMode: 'detail'
+    });
+  }
+  const selectedSession = state.sessions.find(session => session.id === state.selectedSessionId);
+  if (!selectedSession) {
+    return withClearedComposer(state);
+  }
+  if (command === 'clear') {
+    const updatedSessions = state.sessions.map(session => {
+      if (session.id !== state.selectedSessionId) {
+        return session;
+      }
+      return {
+        ...session,
+        messages: []
+      };
+    });
+    const updatedSelectedSession = updatedSessions.find(session => session.id === state.selectedSessionId);
+    if (updatedSelectedSession) {
+      await saveChatSession(updatedSelectedSession);
+    }
+    return withClearedComposer({
+      ...state,
+      sessions: updatedSessions
+    });
+  }
+  const assistantText = command === 'help' ? getCommandHelpText() : ['```md', toMarkdownTranscript(selectedSession), '```'].join('\n');
+  const assistantMessage = {
+    id: crypto.randomUUID(),
+    role: 'assistant',
+    text: assistantText,
+    time: new Date().toLocaleTimeString([], {
+      hour: '2-digit',
+      minute: '2-digit'
+    })
+  };
+  const updatedSessions = appendMessageToSelectedSession(state.sessions, state.selectedSessionId, assistantMessage);
+  const updatedSelectedSession = updatedSessions.find(session => session.id === state.selectedSessionId);
+  if (updatedSelectedSession) {
+    await saveChatSession(updatedSelectedSession);
+  }
+  return withClearedComposer({
+    ...state,
+    sessions: updatedSessions
+  });
+};
+const getSlashCommand = value => {
+  const trimmed = value.trim();
+  const match = trimmed.match(slashCommandRegex);
+  if (!match) {
+    return undefined;
+  }
+  return match[1];
+};
+const parseMentionedPaths = value => {
+  const matches = value.matchAll(mentionRegex);
+  const paths = [];
+  for (const match of matches) {
+    const rawPath = match[2] || '';
+    const cleanedPath = rawPath.replaceAll(/[),.;:!?]+$/g, '');
+    if (!cleanedPath || isPathTraversalAttempt(cleanedPath)) {
+      continue;
+    }
+    const normalizedPath = normalizeRelativePath(cleanedPath);
+    if (paths.includes(normalizedPath)) {
+      continue;
+    }
+    paths.push(normalizedPath);
+    if (paths.length >= maxMentionCount) {
+      break;
+    }
+  }
+  return paths;
+};
+const getMentionContextMessage = async value => {
+  const paths = parseMentionedPaths(value);
+  if (paths.length === 0) {
+    return undefined;
+  }
+  const sections = [];
+  for (const path of paths) {
+    try {
+      const fileContent = await readFile(path);
+      const truncatedContent = fileContent.length > maxMentionTextLength ? `${fileContent.slice(0, maxMentionTextLength)}\n... [truncated]` : fileContent;
+      sections.push([`File: ${path}`, '```text', truncatedContent, '```'].join('\n'));
+    } catch (error) {
+      sections.push([`File: ${path}`, `Error: ${String(error)}`].join('\n'));
+    }
+  }
+  return {
+    id: crypto.randomUUID(),
+    role: 'user',
+    text: ['Referenced file context:', ...sections].join('\n\n'),
+    time: new Date().toLocaleTimeString([], {
+      hour: '2-digit',
+      minute: '2-digit'
+    })
+  };
+};
+const isDefaultSessionTitle = title => {
+  return /^Chat \d+$/.test(title);
+};
+const sanitizeGeneratedTitle = value => {
+  return value.replace(/^title:\s*/i, '').replaceAll(/^['"`\s]+|['"`\s]+$/g, '').replaceAll(/\s+/g, ' ').trim().slice(0, 80);
+};
+const updateSessionTitle = (sessions, selectedSessionId, title) => {
+  return sessions.map(session => {
+    if (session.id !== selectedSessionId) {
+      return session;
+    }
+    return {
+      ...session,
+      title
+    };
+  });
+};
+const getAiSessionTitle = async (state, userText, assistantText) => {
+  const {
+    models,
+    openApiApiBaseUrl,
+    openApiApiKey,
+    openRouterApiBaseUrl,
+    openRouterApiKey,
+    selectedModelId,
+    useMockApi
+  } = state;
+  if (useMockApi) {
+    return '';
+  }
+  const usesOpenApiModel = isOpenApiModel(selectedModelId, models);
+  const usesOpenRouterModel = isOpenRouterModel(selectedModelId, models);
+  if (usesOpenApiModel && !openApiApiKey) {
+    return '';
+  }
+  if (usesOpenRouterModel && !openRouterApiKey) {
+    return '';
+  }
+  if (!usesOpenApiModel && !usesOpenRouterModel) {
+    return '';
+  }
+  const titlePrompt = `Create a concise title (max 6 words) for this conversation. Respond only with the title, no punctuation at the end.
+User: ${userText}
+Assistant: ${assistantText}`;
+  const promptMessage = {
+    id: crypto.randomUUID(),
+    role: 'user',
+    text: titlePrompt,
+    time: new Date().toLocaleTimeString([], {
+      hour: '2-digit',
+      minute: '2-digit'
+    })
+  };
+  const titleResponse = await getAiResponse({
+    assetDir: state.assetDir,
+    messages: [promptMessage],
+    mockAiResponseDelay: state.mockAiResponseDelay,
+    mockApiCommandId: state.mockApiCommandId,
+    models,
+    openApiApiBaseUrl,
+    openApiApiKey,
+    openRouterApiBaseUrl,
+    openRouterApiKey,
+    passIncludeObfuscation: state.passIncludeObfuscation,
+    platform: state.platform,
+    selectedModelId,
+    streamingEnabled: false,
+    useMockApi,
+    userText: titlePrompt,
+    webSearchEnabled: false
+  });
+  const title = sanitizeGeneratedTitle(titleResponse.text);
+  return title && !isDefaultSessionTitle(title) ? title : '';
+};
 const handleSubmit = async state => {
   const {
+    aiSessionTitleGenerationEnabled,
     assetDir,
     composerValue,
     emitStreamingFunctionCallEvents,
@@ -5095,6 +5306,10 @@ const handleSubmit = async state => {
   if (!userText) {
     return state;
   }
+  const slashCommand = getSlashCommand(userText);
+  if (slashCommand) {
+    return executeSlashCommand(state, slashCommand);
+  }
   const userTime = new Date().toLocaleTimeString([], {
     hour: '2-digit',
     minute: '2-digit'
@@ -5131,6 +5346,7 @@ const handleSubmit = async state => {
     }
   }
   let optimisticState;
+  const createsNewSession = viewMode === 'list';
   if (viewMode === 'list') {
     const newSessionId = generateSessionId();
     await appendChatViewEvent({
@@ -5188,13 +5404,15 @@ const handleSubmit = async state => {
   };
   const selectedOptimisticSession = optimisticState.sessions.find(session => session.id === optimisticState.selectedSessionId);
   const messages = (selectedOptimisticSession?.messages ?? []).filter(message => !message.inProgress);
+  const mentionContextMessage = await getMentionContextMessage(userText);
+  const messagesWithMentionContext = mentionContextMessage ? [...messages, mentionContextMessage] : messages;
   const handleTextChunkFunctionRef = streamingEnabled ? async chunk => {
     handleTextChunkState = await handleTextChunkFunction(state.uid, assistantMessageId, chunk, handleTextChunkState);
   } : undefined;
   const assistantMessage = await getAiResponse({
     assetDir,
     messageId: assistantMessageId,
-    messages,
+    messages: messagesWithMentionContext,
     mockAiResponseDelay,
     mockApiCommandId,
     models,
@@ -5239,7 +5457,16 @@ const handleSubmit = async state => {
   const {
     latestState
   } = handleTextChunkState;
-  const updatedSessions = streamingEnabled ? updateMessageTextInSelectedSession(latestState.sessions, latestState.selectedSessionId, assistantMessageId, assistantMessage.text, false) : appendMessageToSelectedSession(latestState.sessions, latestState.selectedSessionId, assistantMessage);
+  let updatedSessions = streamingEnabled ? updateMessageTextInSelectedSession(latestState.sessions, latestState.selectedSessionId, assistantMessageId, assistantMessage.text, false) : appendMessageToSelectedSession(latestState.sessions, latestState.selectedSessionId, assistantMessage);
+  if (aiSessionTitleGenerationEnabled && createsNewSession) {
+    const selectedSession = updatedSessions.find(session => session.id === latestState.selectedSessionId);
+    if (selectedSession && isDefaultSessionTitle(selectedSession.title)) {
+      const generatedTitle = await getAiSessionTitle(latestState, userText, assistantMessage.text);
+      if (generatedTitle) {
+        updatedSessions = updateSessionTitle(updatedSessions, latestState.selectedSessionId, generatedTitle);
+      }
+    }
+  }
   const selectedSession = updatedSessions.find(session => session.id === latestState.selectedSessionId);
   if (selectedSession) {
     await saveChatSession(selectedSession);
@@ -5441,6 +5668,9 @@ const handleClickSettings = async () => {
 };
 
 const handleInput = async (state, name, value, inputSource = 'user') => {
+  const {
+    selectedSessionId
+  } = state;
   if (name === OpenApiApiKeyInput) {
     return {
       ...state,
@@ -5456,9 +5686,9 @@ const handleInput = async (state, name, value, inputSource = 'user') => {
   if (name !== Composer) {
     return state;
   }
-  if (state.selectedSessionId) {
+  if (selectedSessionId) {
     await appendChatViewEvent({
-      sessionId: state.selectedSessionId,
+      sessionId: selectedSessionId,
       timestamp: new Date().toISOString(),
       type: 'handle-input',
       value
@@ -5705,6 +5935,15 @@ const getSavedViewMode = savedState => {
   return viewMode;
 };
 
+const loadAiSessionTitleGenerationEnabled = async () => {
+  try {
+    const savedAiSessionTitleGenerationEnabled = await get('chatView.aiSessionTitleGenerationEnabled');
+    return typeof savedAiSessionTitleGenerationEnabled === 'boolean' ? savedAiSessionTitleGenerationEnabled : false;
+  } catch {
+    return false;
+  }
+};
+
 const loadEmitStreamingFunctionCallEvents = async () => {
   try {
     const savedEmitStreamingFunctionCallEvents = await get('chatView.emitStreamingFunctionCallEvents');
@@ -5759,12 +5998,9 @@ const loadStreamingEnabled = async () => {
 };
 
 const loadPreferences = async () => {
-  const openApiApiKey = await loadOpenApiApiKey();
-  const openRouterApiKey = await loadOpenRouterApiKey();
-  const emitStreamingFunctionCallEvents = await loadEmitStreamingFunctionCallEvents();
-  const streamingEnabled = await loadStreamingEnabled();
-  const passIncludeObfuscation = await loadPassIncludeObfuscation();
+  const [aiSessionTitleGenerationEnabled, openApiApiKey, openRouterApiKey, emitStreamingFunctionCallEvents, streamingEnabled, passIncludeObfuscation] = await Promise.all([loadAiSessionTitleGenerationEnabled(), loadOpenApiApiKey(), loadOpenRouterApiKey(), loadEmitStreamingFunctionCallEvents(), loadStreamingEnabled(), loadPassIncludeObfuscation()]);
   return {
+    aiSessionTitleGenerationEnabled,
     emitStreamingFunctionCallEvents,
     openApiApiKey,
     openRouterApiKey,
@@ -5800,6 +6036,7 @@ const loadContent = async (state, savedState) => {
   const savedSelectedModelId = getSavedSelectedModelId(savedState);
   const savedViewMode = getSavedViewMode(savedState);
   const {
+    aiSessionTitleGenerationEnabled,
     emitStreamingFunctionCallEvents,
     openApiApiKey,
     openRouterApiKey,
@@ -5832,6 +6069,7 @@ const loadContent = async (state, savedState) => {
   const viewMode = sessions.length === 0 || !selectedSessionId ? 'list' : preferredViewMode === 'detail' ? 'detail' : 'list';
   return {
     ...state,
+    aiSessionTitleGenerationEnabled,
     chatListScrollTop,
     emitStreamingFunctionCallEvents,
     initial: false,
@@ -5903,6 +6141,13 @@ const openMockSession = async (state, mockSessionId, mockChatMessages) => {
   };
 };
 
+const registerMockResponse = (state, mockResponse) => {
+  reset$1();
+  pushChunk(mockResponse.text);
+  finish();
+  return state;
+};
+
 const getCss = (composerHeight, listItemHeight, chatMessageFontSize, chatMessageLineHeight, chatMessageFontFamily, renderHtmlCss) => {
   const baseCss = `:root {
   --ChatInputBoxHeight: ${composerHeight}px;
@@ -5968,6 +6213,17 @@ const getCss = (composerHeight, listItemHeight, chatMessageFontSize, chatMessage
   cursor: pointer;
 }
 
+.ChatOrderedList,
+.ChatUnorderedList {
+  margin: 6px 0;
+  padding-inline-start: 20px;
+}
+
+.ChatOrderedListItem,
+.ChatUnorderedListItem {
+  margin: 2px 0;
+}
+
 .MarkdownTable {
   width: 100%;
   margin: 6px 0;
@@ -6072,6 +6328,8 @@ const ChatToolCallRenderHtmlBody = 'ChatToolCallRenderHtmlBody';
 const ChatMessageLink = 'ChatMessageLink';
 const ChatOrderedList = 'ChatOrderedList';
 const ChatOrderedListItem = 'ChatOrderedListItem';
+const ChatUnorderedList = 'ChatUnorderedList';
+const ChatUnorderedListItem = 'ChatUnorderedListItem';
 const MessageUser = 'MessageUser';
 const MessageAssistant = 'MessageAssistant';
 const MultilineInputBox = 'MultilineInputBox';
@@ -6331,6 +6589,13 @@ const getOrderedListItemDom = item => {
     type: Li
   }, ...item.children.flatMap(getInlineNodeDom)];
 };
+const getUnorderedListItemDom = item => {
+  return [{
+    childCount: item.children.length,
+    className: ChatUnorderedListItem,
+    type: Li
+  }, ...item.children.flatMap(getInlineNodeDom)];
+};
 const getTableHeadCellDom = cell => {
   return [{
     childCount: cell.children.length,
@@ -6365,6 +6630,28 @@ const getTableDom = node => {
     type: TBody
   }, ...node.rows.flatMap(getTableRowDom)];
 };
+const getHeadingElementType = level => {
+  switch (level) {
+    case 1:
+      return H1;
+    case 2:
+      return H2;
+    case 3:
+      return H3;
+    case 4:
+      return H4;
+    case 5:
+      return H5;
+    case 6:
+      return H6;
+  }
+};
+const getHeadingDom = node => {
+  return [{
+    childCount: node.children.length,
+    type: getHeadingElementType(node.level)
+  }, ...node.children.flatMap(getInlineNodeDom)];
+};
 const getMessageNodeDom = node => {
   if (node.type === 'text') {
     return [{
@@ -6379,11 +6666,21 @@ const getMessageNodeDom = node => {
   if (node.type === 'code-block') {
     return getCodeBlockDom(node);
   }
+  if (node.type === 'heading') {
+    return getHeadingDom(node);
+  }
+  if (node.type === 'ordered-list') {
+    return [{
+      childCount: node.items.length,
+      className: ChatOrderedList,
+      type: Ol
+    }, ...node.items.flatMap(getOrderedListItemDom)];
+  }
   return [{
     childCount: node.items.length,
-    className: ChatOrderedList,
-    type: Ol
-  }, ...node.items.flatMap(getOrderedListItemDom)];
+    className: ChatUnorderedList,
+    type: Ul
+  }, ...node.items.flatMap(getUnorderedListItemDom)];
 };
 
 const getMessageContentDom = nodes => {
@@ -6929,9 +7226,11 @@ const getToolCallsDom = message => {
 };
 
 const orderedListItemRegex = /^\s*\d+\.\s+(.*)$/;
+const unorderedListItemRegex = /^\s*[-*]\s+(.*)$/;
 const markdownInlineRegex = /\[([^\]]+)\]\(([^)]+)\)|\*\*([^*]+)\*\*/g;
 const markdownTableSeparatorCellRegex = /^:?-{3,}:?$/;
 const fencedCodeBlockRegex = /^```/;
+const markdownHeadingRegex = /^\s*(#{1,6})\s+(.*)$/;
 const normalizeInlineTables = value => {
   return value.split(/\r?\n/).map(line => {
     if (!line.includes('|')) {
@@ -7034,6 +7333,7 @@ const parseMessageContent = rawMessage => {
   const nodes = [];
   let paragraphLines = [];
   let listItems = [];
+  let listType = '';
   const flushParagraph = () => {
     if (paragraphLines.length === 0) {
       return;
@@ -7050,9 +7350,10 @@ const parseMessageContent = rawMessage => {
     }
     nodes.push({
       items: listItems,
-      type: 'list'
+      type: listType || 'ordered-list'
     });
     listItems = [];
+    listType = '';
   };
   for (let i = 0; i < lines.length; i++) {
     const line = lines[i];
@@ -7099,15 +7400,43 @@ const parseMessageContent = rawMessage => {
         continue;
       }
     }
-    const match = line.match(orderedListItemRegex);
-    if (match) {
+    const orderedMatch = line.match(orderedListItemRegex);
+    if (orderedMatch) {
+      if (listType && listType !== 'ordered-list') {
+        flushList();
+      }
       flushParagraph();
+      listType = 'ordered-list';
       listItems.push({
-        children: parseInlineNodes(match[1]),
+        children: parseInlineNodes(orderedMatch[1]),
         type: 'list-item'
       });
       continue;
     }
+    const unorderedMatch = line.match(unorderedListItemRegex);
+    if (unorderedMatch) {
+      if (listType && listType !== 'unordered-list') {
+        flushList();
+      }
+      flushParagraph();
+      listType = 'unordered-list';
+      listItems.push({
+        children: parseInlineNodes(unorderedMatch[1]),
+        type: 'list-item'
+      });
+      continue;
+    }
+    const headingMatch = line.match(markdownHeadingRegex);
+    if (headingMatch) {
+      flushList();
+      flushParagraph();
+      nodes.push({
+        children: parseInlineNodes(headingMatch[2]),
+        level: headingMatch[1].length,
+        type: 'heading'
+      });
+      continue;
+    }
     flushList();
     paragraphLines.push(line);
   }
@@ -7561,6 +7890,7 @@ const commandMap = {
   'Chat.mockOpenApiStreamPushChunk': wrapCommand(mockOpenApiStreamPushChunk),
   'Chat.mockOpenApiStreamReset': wrapCommand(mockOpenApiStreamReset),
   'Chat.openMockSession': wrapCommand(openMockSession),
+  'Chat.registerMockResponse': wrapCommand(registerMockResponse),
   'Chat.render2': render2,
   'Chat.renderEventListeners': renderEventListeners,
   'Chat.rerender': wrapCommand(rerender),
@@ -7576,7 +7906,7 @@ const commandMap = {
 };
 
 const send = port => {
-  return invokeAndTransfer('SendMessagePortToExtensionHostWorker.sendMessagePortToChatNetworkWorker', port, 'HandleMessagePort.handleMessagePort');
+  return sendMessagePortToChatNetworkWorker(port);
 };
 const initializeChatNetworkWorker = async () => {
   const rpc = await create$4({

package/package.json

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