From bf1c6314cb45ece1efc4ee9389a2a70e88db80d3 Mon Sep 17 00:00:00 2001
From: Graham McIntire <graham@mcintire.me>
Date: Fri, 30 Jan 2026 17:25:08 -0600
Subject: [PATCH] agent version tracking

---
 src/poller/scheduler.rs | 2 +-
 src/version.rs          | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/poller/scheduler.rs b/src/poller/scheduler.rs
index cd69a5b..dc708c5 100644
--- a/src/poller/scheduler.rs
+++ b/src/poller/scheduler.rs
@@ -161,7 +161,7 @@ impl Scheduler {
             .unwrap_or_else(|| "unknown".to_string());
 
         let metadata = HeartbeatMetadata {
-            version: env!("CARGO_PKG_VERSION").to_string(),
+            version: crate::version::current_version().to_string(),
             hostname,
             uptime_seconds: uptime,
         };
diff --git a/src/version.rs b/src/version.rs
index f1f8121..ba367b7 100644
--- a/src/version.rs
+++ b/src/version.rs
@@ -1,5 +1,5 @@
-// Get version at runtime - prefers BUILD_VERSION from build.rs, falls back to Cargo.toml
-fn current_version() -> &'static str {
+/// Get version at runtime - prefers BUILD_VERSION from build.rs, falls back to Cargo.toml
+pub fn current_version() -> &'static str {
     option_env!("BUILD_VERSION").unwrap_or(env!("CARGO_PKG_VERSION"))
 }