Bugfix, plus some QoL upgrades from the Async branch

services/agenthandler.py

@@ -86,12 +86,8 @@ def findAllAgents(api):
     policies = [Policy(**row["data"]) for _, row in api.policy_find_all().iterrows()]
     agents = [Agent(**row["data"]) for _, row in api.agent_find_all().iterrows()]
 
-    # Step 2: Create groupid → groupname map
-    groupid_to_name = {policy.groupid: policy.name for policy in policies}
-
-    # Step 3: Enrich agents
     for agent in agents:
-        agent.enrich(groupid_to_name)
+        agent.enrich_with_policies(policies)
 
     return agents
 
@@ -120,7 +116,7 @@ def findAgents(api, return_dataframe):
     if user_input == 'y':
         timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
         filename = f"agentsearch_{timestamp}.csv"
-        file_path = os.path.join(working_dir, filename)
+        file_path = os.path.join(str(working_dir), filename)
 
         agent_df.to_csv(file_path, index=False)
         logging.info(f"Exported DataFrame to {file_path}")
@@ -138,7 +134,7 @@ def findAgents(api, return_dataframe):
 def selectAgents(api: AirlockAPIWrapper) -> List[Agent]:
     print(colorText("🔍 Device Search", "cyan"))
     print(colorText("Enter the device hostnames you'd like to search for, one per line.", "cyan"))
-    print(colorText("When you're done, press Enter twice.\n", "cyan"))
+    print(colorText("When you're done, press Enter twice (Three times if you have a single device).\n", "cyan"))
     print(colorText("Example:", "cyan"))
     print(colorText("H00000", "cyan"))
     print(colorText("UTN00000", "cyan"))