@lobehub/lobehub 2.0.0-next.361 → 2.0.0-next.362

package/src/locales/default/chat.ts

@@ -234,8 +234,9 @@ export default {
   'operation.sendMessage': 'Sending message',
   'owner': 'Group owner',
   'pageCopilot.title': 'Page Agent',
-  'pageCopilot.welcome':
-    '**Clearer, sharper writing**\n\nDraft, rewrite, or polish—tell me your intent and I’ll refine the rest.',
+  'pageCopilot.welcome': `**Clearer, sharper writing**\n\nDraft, rewrite, or polish—tell me your intent and I'll refine the rest.`,
+  'pageSelection.lines': 'Lines {{start}}-{{end}}',
+  'pageSelection.reference': 'Selected Text',
   'pin': 'Pin',
   'pinOff': 'Unpin',
   'prompts.summaryExpert':

package/src/server/routers/lambda/aiChat.ts

@@ -123,11 +123,18 @@ export const aiChatRouter = router({
 
       // create user message
       log('creating user message with content length: %d', input.newUserMessage.content.length);
+
+      // Build user message metadata with pageSelections if present
+      const userMessageMetadata = input.newUserMessage.pageSelections?.length
+        ? { pageSelections: input.newUserMessage.pageSelections }
+        : undefined;
+
       const userMessageItem = await ctx.messageModel.create({
         agentId: input.agentId,
         content: input.newUserMessage.content,
         files: input.newUserMessage.files,
         groupId: input.groupId,
+        metadata: userMessageMetadata,
         parentId: input.newUserMessage.parentId,
         role: 'user',
         sessionId,

package/src/store/chat/slices/aiChat/actions/conversationLifecycle.ts

@@ -77,11 +77,11 @@ export const conversationLifecycle: StateCreator<
   sendMessage: async ({
     message,
     files,
-    contexts,
     onlyAddUserMessage,
     context,
     messages: inputMessages,
     parentId: inputParentId,
+    pageSelections,
   }) => {
     const { internal_execAgentRuntime, mainInputEditor } = get();
 
@@ -186,6 +186,8 @@ export const conversationLifecycle: StateCreator<
         threadId: operationContext.threadId ?? undefined,
         imageList: tempImages.length > 0 ? tempImages : undefined,
         videoList: tempVideos.length > 0 ? tempVideos : undefined,
+        // Pass pageSelections metadata for immediate display
+        metadata: pageSelections?.length ? { pageSelections } : undefined,
       },
       { operationId, tempMessageId: tempId },
     );
@@ -222,7 +224,7 @@ export const conversationLifecycle: StateCreator<
       const topicId = operationContext.topicId;
       data = await aiChatService.sendMessageInServer(
         {
-          newUserMessage: { content: message, files: fileIdList, parentId },
+          newUserMessage: { content: message, files: fileIdList, pageSelections, parentId },
           // if there is topicId，then add topicId to message
           topicId: topicId ?? undefined,
           threadId: operationContext.threadId ?? undefined,
@@ -372,26 +374,10 @@ export const conversationLifecycle: StateCreator<
       messageMapKey(execContext),
     )(get());
 
-    const contextMessages =
-      contexts?.map((item, index) => {
-        const now = Date.now();
-        const title = item.title ? `${item.title}\n` : '';
-        return {
-          content: `Context ${index + 1}:\n${title}${item.content}`,
-          createdAt: now,
-          id: `ctx_${tempId}_${index}`,
-          role: 'system' as const,
-          updatedAt: now,
-        };
-      }) ?? [];
-
-    const runtimeMessages =
-      contextMessages.length > 0 ? [...displayMessages, ...contextMessages] : displayMessages;
-
     try {
       await internal_execAgentRuntime({
         context: execContext,
-        messages: runtimeMessages,
+        messages: displayMessages,
         parentMessageId: data.assistantMessageId,
         parentMessageType: 'assistant',
         parentOperationId: operationId, // Pass as parent operation