From abd0d39aeb95d41a97b37f0876d35f573d9bc5ae Mon Sep 17 00:00:00 2001
From: Aarnav Tale <aarnavtale@icloud.com>
Date: Thu, 24 Apr 2025 12:37:19 -0400
Subject: [PATCH] fix: only return queried machines from an agent

---
 app/server/web/agent.ts | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/app/server/web/agent.ts b/app/server/web/agent.ts
index 311442f..0fc901c 100644
--- a/app/server/web/agent.ts
+++ b/app/server/web/agent.ts
@@ -252,7 +252,16 @@ class AgentManager {
 			await this.requestData(missing);
 		}
 
-		return entries;
+		return Object.entries(entries).reduce<Record<string, HostInfo>>(
+			(acc, [key, value]) => {
+				if (nodeIds.includes(key)) {
+					acc[key] = value;
+				}
+
+				return acc;
+			},
+			{},
+		);
 	}
 
 	// Request data from the internal agent by sending a message to the process