From 7ee514518cab17ce2a111aca79ce282e5af42545 Mon Sep 17 00:00:00 2001
From: Jonathan Tran <jonnytran@gmail.com>
Date: Wed, 29 Jan 2025 20:11:00 -0500
Subject: [PATCH] Add updating the artifact graph after mock execution

---
 src/lang/KclSingleton.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/lang/KclSingleton.ts b/src/lang/KclSingleton.ts
index fe95edd949..40be1b0268 100644
--- a/src/lang/KclSingleton.ts
+++ b/src/lang/KclSingleton.ts
@@ -442,6 +442,7 @@ export class KclManager {
       this.lastSuccessfulProgramMemory = execState.memory
       this.lastSuccessfulOperations = execState.operations
     }
+    this.engineCommandManager.updateArtifactGraph(execState.artifactGraph)
   }
   cancelAllExecutions() {
     this._cancelTokens.forEach((_, key) => {