From e01360ad0319e600acad8061619bd6b0fbe770ec Mon Sep 17 00:00:00 2001
From: Jedd Morgan <45512892+JR-Morgan@users.noreply.github.com>
Date: Mon, 24 Nov 2025 22:53:09 +0300
Subject: [PATCH] mark version received (#419)

---
 src/Speckle.Automate.Sdk/AutomationContext.cs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/Speckle.Automate.Sdk/AutomationContext.cs b/src/Speckle.Automate.Sdk/AutomationContext.cs
index b3b8a638..7bd0cc1e 100644
--- a/src/Speckle.Automate.Sdk/AutomationContext.cs
+++ b/src/Speckle.Automate.Sdk/AutomationContext.cs
@@ -63,7 +63,7 @@ internal sealed class AutomationContext(IOperations operations) : IAutomationCon
       );
     }
 
-    Base? rootObject = await operations
+    Base rootObject = await operations
       .Receive2(
         SpeckleClient.ServerUrl,
         AutomationRunData.ProjectId,
@@ -74,6 +74,10 @@ internal sealed class AutomationContext(IOperations operations) : IAutomationCon
       )
       .ConfigureAwait(false);
 
+    await SpeckleClient
+      .Version.Received(new(version.id, AutomationRunData.ProjectId, "automate_function"), cancellationToken)
+      .ConfigureAwait(false);
+
     Console.WriteLine($"It took {Elapsed.TotalSeconds} seconds to receive the speckle version {versionId}");
     return rootObject;
   }