praetorian-cli 2.2.10__py3-none-any.whl → 2.2.11__py3-none-any.whl

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/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/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.10.dist-info → praetorian_cli-2.2.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praetorian-cli
-Version: 2.2.10
+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.10.dist-info → praetorian_cli-2.2.11.dist-info}/RECORD

@@ -85,20 +85,20 @@ praetorian_cli/sdk/test/utils.py,sha256=svxMpzlaW4FRCij05cPgJFrTUEELVdt8G7SPKEds
 praetorian_cli/ui/__init__.py,sha256=wEgkrgIaoOguH1VVp2FndaGIxWmZ5CfAynXtNtZ6iTo,81
 praetorian_cli/ui/aegis/__init__.py,sha256=1HE5aQ6C6Qruf1X-KmKvDw1M3lzaUbPZoz9QOp4Lofs,123
 praetorian_cli/ui/aegis/constants.py,sha256=trr62uBwdBwm4LrFVms7_moThdtOPJi2uzCeIv4m3VU,578
-praetorian_cli/ui/aegis/menu.py,sha256=8byIJAafsL5Atu8UrEQkLKF4VOpBC7fKkFz_qzqF7sk,14470
-praetorian_cli/ui/aegis/utils.py,sha256=l5NiWUU0jIWNJG-LSg_Fa6MdWNtIXHWAd1YcbtqIYkg,5052
+praetorian_cli/ui/aegis/menu.py,sha256=vLdIBJxDOSFClW319YMtPMwqHD2_qPo79xkVk2BsyEU,14633
+praetorian_cli/ui/aegis/utils.py,sha256=2sXVwSzXcwPNhB-Rg80kc-4hECRko1wQBQWzu8cBcgI,5233
 praetorian_cli/ui/aegis/commands/__init__.py,sha256=Af3AE-KqPevE_N-NWFTpTPaUkRhkaMAug6hCr_soHEU,39
 praetorian_cli/ui/aegis/commands/help.py,sha256=ucalA21yUy5EFtN1RbDBwTjfFQnrwWwd36HFzzqaSRg,3366
 praetorian_cli/ui/aegis/commands/info.py,sha256=AcvaIeZqnugk1BFUXB25xOVkYYe0UzTiFiT_2m8tgIE,5516
 praetorian_cli/ui/aegis/commands/job.py,sha256=BUcp47K-4PBIYq9nyUQTKhOakZwXM4fOf2TSWIDKoLE,14591
 praetorian_cli/ui/aegis/commands/list.py,sha256=puIiy0skYE59Q2hVSMsla1tKiYYAFTOaz8oPM17tCyI,352
-praetorian_cli/ui/aegis/commands/set.py,sha256=ODa9u_6yW2dbKVQBuV9YeiS_ty_R_Xk_vPz7YrW3OWs,1101
+praetorian_cli/ui/aegis/commands/set.py,sha256=WJ0Qt30fBa_hyWEaawyA3h3rSeWHFPbJSKY8A0k6SC0,1176
 praetorian_cli/ui/aegis/commands/ssh.py,sha256=KGsNlN0i-Cwp6gWyr-cjML9_L13oE7xFenysF2pC8Rc,3045
 praetorian_cli/ui/conversation/__init__.py,sha256=sNhNN_ZG1Va_7OLTaoXlIFL6ageKHWdufFVYw6F_aV8,90
 praetorian_cli/ui/conversation/textual_chat.py,sha256=gGgEs7HhNAto9rTVrGbz62mG10OqmtArI-aKxFLSfVs,24405
-praetorian_cli-2.2.10.dist-info/licenses/LICENSE,sha256=Zv97QripiVALv-WokW_Elsiz9vtOfbtNt1aLZhhk67I,1067
-praetorian_cli-2.2.10.dist-info/METADATA,sha256=Ph8KnJnjGdNkTwbcS0HubrSrf2Zbr9-3NVJw_kVRHDI,7752
-praetorian_cli-2.2.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-praetorian_cli-2.2.10.dist-info/entry_points.txt,sha256=uJbDvZdkYaLiCh2DMvXPUGKFm2p5ZfzJCizUK3-PUEE,56
-praetorian_cli-2.2.10.dist-info/top_level.txt,sha256=QbUdRPGEj_TyHO-E7AD5BxFfR8ore37i273jP4Gn43c,15
-praetorian_cli-2.2.10.dist-info/RECORD,,
+praetorian_cli-2.2.11.dist-info/licenses/LICENSE,sha256=Zv97QripiVALv-WokW_Elsiz9vtOfbtNt1aLZhhk67I,1067
+praetorian_cli-2.2.11.dist-info/METADATA,sha256=S38eZTVQ_ckoHoD6gFKzDanvXEcyOFjyJEWcIuFFZnU,7752
+praetorian_cli-2.2.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+praetorian_cli-2.2.11.dist-info/entry_points.txt,sha256=uJbDvZdkYaLiCh2DMvXPUGKFm2p5ZfzJCizUK3-PUEE,56
+praetorian_cli-2.2.11.dist-info/top_level.txt,sha256=QbUdRPGEj_TyHO-E7AD5BxFfR8ore37i273jP4Gn43c,15
+praetorian_cli-2.2.11.dist-info/RECORD,,