fix: get refreshed code block context (#2612)

Author: Kitenite

apps/web/client/public/onlook-preload-script.js

@@ -52,8 +52,8 @@ export class ChatContext {
                     return c;
                 }
                 return { ...c, content: fileContent.content } satisfies FileMessageContext;
-            } else if (c.type === MessageContextType.HIGHLIGHT) {
-                const codeBlock = await this.editorEngine.sandbox.getCodeBlock(c.path);
+            } else if (c.type === MessageContextType.HIGHLIGHT && c.oid) {
+                const codeBlock = await this.editorEngine.sandbox.getCodeBlock(c.oid);
                 if (codeBlock === null) {
                     console.error('No code block found for node', c.path);
                     return c;
@@ -119,6 +119,7 @@ export class ChatContext {
                 content: codeBlock,
                 start: templateNode.startTag.start.line,
                 end: templateNode.endTag?.end.line || templateNode.startTag.start.line,
+                oid,
             });
             fileNames.add(templateNode.path);
         }

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

@@ -22,6 +22,7 @@ export type HighlightMessageContext = BaseMessageContext & {
     path: string;
     start: number;
     end: number;
+    oid?: string;
 };
 
 export type ImageMessageContext = BaseMessageContext & {