From 15a56b545d8d32a203c7b2f6d4f2baa666a4bb08 Mon Sep 17 00:00:00 2001
From: JR-Morgan <45512892+JR-Morgan@users.noreply.github.com>
Date: Tue, 11 Nov 2025 16:07:10 +0300
Subject: [PATCH] mark version received

---
 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 @@ public async Task<Base> ReceiveVersion(CancellationToken cancellationToken = def
       );
     }
 
-    Base? rootObject = await operations
+    Base rootObject = await operations
       .Receive2(
         SpeckleClient.ServerUrl,
         AutomationRunData.ProjectId,
@@ -74,6 +74,10 @@ public async Task<Base> ReceiveVersion(CancellationToken cancellationToken = def
       )
       .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;
   }