praetorian-cli 2.2.9__tar.gz → 2.2.11__tar.gz

{praetorian_cli-2.2.9/praetorian_cli.egg-info → praetorian_cli-2.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praetorian-cli
-Version: 2.2.9
+Version: 2.2.11
 Summary: For interacting with the Chariot API
 Home-page: https://github.com/praetorian-inc/praetorian-cli
 Author: Praetorian

{praetorian_cli-2.2.9 → praetorian_cli-2.2.11}/praetorian_cli/sdk/entities/aegis.py

@@ -59,7 +59,7 @@ class Aegis:
             - is_online: Boolean indicating if agent is currently online
         """
         try:
-            agents_data = self.api.get('/aegis/agent')
+            agents_data = self.api.get('/agent/enhanced')
             
             # Return Agent objects
             agents = []

{praetorian_cli-2.2.9 → praetorian_cli-2.2.11}/praetorian_cli/ui/aegis/commands/set.py

@@ -9,15 +9,16 @@ def handle_set(menu, args):
         return
 
     selection = args[0]
-    selected_agent = parse_agent_identifier(selection, menu.agents)
+    selected_agent = parse_agent_identifier(selection, menu.displayed_agents, menu.agents)
 
     if selected_agent:
         menu.selected_agent = selected_agent
         hostname = selected_agent.hostname
         menu.console.print(f"\n  Selected: {hostname}\n")
     else:
+        displayed_count = len(menu.displayed_agents)
         menu.console.print(f"\n[red]  Agent not found:[/red] {selection}")
-        menu.console.print(f"[dim]  Use agent number (1-{len(menu.agents)}), client ID, or hostname[/dim]\n")
+        menu.console.print(f"[dim]  Use agent number (1-{displayed_count}), client ID, or hostname[/dim]\n")
         menu.pause()
 
 

{praetorian_cli-2.2.9 → praetorian_cli-2.2.11}/praetorian_cli/ui/aegis/menu.py

@@ -54,7 +54,8 @@ class AegisMenu:
         self.selected_agent: Optional[Agent] = None  
         self._first_render = True
         self.agent_computed_data = {}  
-        self.current_prompt = "> " 
+        self.current_prompt = "> "
+        self.displayed_agents: List[Agent] = []  # Track currently displayed agents 
         
         self.user_email, self.username = self.sdk.get_current_user()
         
@@ -173,6 +174,8 @@ class AegisMenu:
         if show_offline:
             display_agents = display_agents + offline_agents
         
+        self.displayed_agents = [agent for _, agent in display_agents]
+        
         self.console.print()
 
         if not display_agents:
@@ -392,4 +395,4 @@ class AegisMenu:
 def run_aegis_menu(sdk: Chariot) -> None:
     """Run the Aegis menu interface"""
     menu = AegisMenu(sdk)
-    menu.run()
+    menu.run()

{praetorian_cli-2.2.9 → praetorian_cli-2.2.11}/praetorian_cli/ui/aegis/utils.py

@@ -132,31 +132,32 @@ def get_agent_display_style(group: str, colors: dict) -> dict:
         }
 
 
-def parse_agent_identifier(identifier: str, agents: List[Agent]) -> Optional[Agent]:
-    """Parse agent identifier and return matching agent"""
-    if not agents:
+def parse_agent_identifier(identifier: str, displayed_agents: List[Agent], all_agents: Optional[List[Agent]] = None) -> Optional[Agent]:
+    """Parse agent identifier and return matching agent
+    """
+    if not displayed_agents:
         return None
     
-    # Try numeric index first
+    # Use all_agents for fallback searches, or displayed_agents if not provided
+    search_agents = all_agents if all_agents is not None else displayed_agents
+    
     if identifier.isdigit():
         agent_num = int(identifier)
-        if 1 <= agent_num <= len(agents):
-            return agents[agent_num - 1]
+        if 1 <= agent_num <= len(displayed_agents):
+            return displayed_agents[agent_num - 1]
     
-    # Try client ID match
-    for agent in agents:
+    for agent in search_agents:
         try:
             if agent.client_id and agent.client_id.lower() == identifier.lower():
                 return agent
         except AttributeError:
             continue
     
-    # Try hostname match  
-    for agent in agents:
+    for agent in search_agents:
         try:
             if agent.hostname and agent.hostname.lower() == identifier.lower():
                 return agent
         except AttributeError:
             continue
     
-    return None
+    return None

{praetorian_cli-2.2.9 → praetorian_cli-2.2.11/praetorian_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praetorian-cli
-Version: 2.2.9
+Version: 2.2.11
 Summary: For interacting with the Chariot API
 Home-page: https://github.com/praetorian-inc/praetorian-cli
 Author: Praetorian

{praetorian_cli-2.2.9 → praetorian_cli-2.2.11}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = praetorian-cli
-version = 2.2.9
+version = 2.2.11
 author = Praetorian
 author_email = support@praetorian.com
 description = For interacting with the Chariot API