clean up

Author: Kitenite

apps/web/client/src/components/store/editor/chat/context.ts

@@ -140,16 +140,13 @@ export class ChatContext {
         highlightedContext: HighlightMessageContext[],
         frames: FrameData[],
     ): BranchMessageContext[] {
-        // Get unique branch IDs from highlighted context
-        const uniqueBranchIds = new Set<string>();
+        // Get unique branch IDs from selected elements and frames context
+        const uniqueBranchIds = new Set<string>(frames.map(frame => frame.frame.branchId));
+
         highlightedContext.forEach(highlight => {
             uniqueBranchIds.add(highlight.branchId);
         });
 
-        frames.forEach((frame) => {
-            uniqueBranchIds.add(frame.frame.branchId);
-        });
-
         // Get branch objects for each unique branch ID
         const branchContext: BranchMessageContext[] = [];
         uniqueBranchIds.forEach(branchId => {