machineconfig 4.4__py3-none-any.whl → 4.5__py3-none-any.whl

machineconfig/cluster/sessions_managers/zellij_local_manager.py

@@ -120,12 +120,9 @@ class ZellijLocalManager:
                 session_name = manager.session_name
                 if session_name is None:
                     continue  # Skip managers without a session name
-
                 cmd = f"zellij delete-session --force {session_name}"
-
                 logger.info(f"Killing session '{session_name}'")
                 result = subprocess.run(cmd, shell=True, capture_output=True, text=True, timeout=10)
-
                 results[session_name] = {"success": result.returncode == 0, "message": "Session killed" if result.returncode == 0 else result.stderr}
 
             except Exception as e:

machineconfig/scripts/python/fire_agents.py

@@ -25,7 +25,7 @@ from machineconfig.scripts.python.fire_agents_help_launch import prep_agent_laun
 from machineconfig.scripts.python.fire_agents_help_search import search_files_by_pattern, search_python_files
 from machineconfig.scripts.python.fire_agents_load_balancer import chunk_prompts
 from machineconfig.utils.schemas.layouts.layout_types import LayoutsFile
-from machineconfig.utils.accessories import get_repo_root
+from machineconfig.utils.accessories import get_repo_root, randstr
 
 SEARCH_STRATEGIES: TypeAlias = Literal["file_path", "keyword_search", "filename_pattern"]
 
@@ -50,6 +50,7 @@ def create(
     job_name: str = typer.Option("AI_Agents", help="Job name"),
     keep_separate: bool = typer.Option(True, help="Keep prompt material in separate file to the context."),
     output_path: Optional[Path] = typer.Option(None, help="Path to write the layout.json file"),
+    agents_dir: Optional[Path] = typer.Option(None, help="Directory to store agent files. If not provided, will be constructed automatically."),
 ):
     # validate mutual exclusive
     context_options = [context_path, keyword_search, filename_pattern]
@@ -103,7 +104,12 @@ def create(
     agent_selected = agent
     keep_material_in_separate_file_input = keep_separate
     prompt_material_re_splitted = chunk_prompts(prompt_material_path, tasks_per_prompt=tasks_per_prompt, joiner=separator)
-    agents_dir = prep_agent_launch(repo_root=repo_root, prompts_material=prompt_material_re_splitted, keep_material_in_separate_file=keep_material_in_separate_file_input, prompt_prefix=prompt_prefix, agent=agent_selected, job_name=job_name)
+    if agents_dir is None: agents_dir = repo_root / ".ai" / f"tmp_prompts/{job_name}_{randstr()}"
+    else:
+        import shutil
+        if agents_dir.exists():
+            shutil.rmtree(agents_dir)
+    prep_agent_launch(agents_dir=agents_dir, prompts_material=prompt_material_re_splitted, keep_material_in_separate_file=keep_material_in_separate_file_input, prompt_prefix=prompt_prefix, agent=agent_selected, job_name=job_name)
     layoutfile = get_agents_launch_layout(session_root=agents_dir)    
     regenerate_py_code = f"""
 #!/usr/bin/env uv run --python 3.13 --with machineconfig
@@ -119,25 +125,34 @@ fire_agents create --context-path "{prompt_material_path}" \\
 """
     (agents_dir / "aa_agents_relaunch.py").write_text(data=regenerate_py_code, encoding="utf-8")
     layout_output_path = output_path if output_path is not None else agents_dir / "layout.json"
+    layout_output_path.parent.mkdir(parents=True, exist_ok=True)
     layout_output_path.write_text(data=json.dumps(layoutfile, indent=4), encoding="utf-8")
     typer.echo(f"Created agents in {agents_dir}")
+    typer.echo(f"Ceated layout in {layout_output_path}")
 
 
 @app.command()
 def run(layout_path: Path = typer.Argument(..., help="Path to the layout.json file"),
         max_tabs: int = typer.Option(6, help="Maximum number of tabs to launch"),
-        sleep_between_layouts: float = typer.Option(1.0, help="Sleep time in seconds between launching layouts")):
+        sleep_inbetween: float = typer.Option(1.0, help="Sleep time in seconds between launching layouts"),
+        kill_upon_completion: bool = typer.Option(False, help="Kill the layout sessions upon completion")):
     layoutfile: LayoutsFile = json.loads(layout_path.read_text())
-    if len(layoutfile["layouts"][0]["layoutTabs"]) > max_tabs:
-        typer.echo(f"Too many tabs (>{max_tabs}) to launch. Skipping launch.")
-        raise typer.Exit(0)
+    for a_layout in layoutfile["layouts"]:
+        if len(a_layout["layoutTabs"]) > max_tabs:
+            typer.echo(f"Layout '{a_layout.get('layoutName', 'Unnamed')}' has {len(a_layout['layoutTabs'])} tabs which exceeds the max of {max_tabs}.")
+            confirm = typer.confirm("Do you want to proceed with launching this layout?", default=False)
+            if not confirm:
+                typer.echo("Aborting launch.")
+                raise typer.Exit(0)
     from machineconfig.cluster.sessions_managers.zellij_local_manager import ZellijLocalManager
     for i, a_layouts in enumerate(layoutfile["layouts"]):
         manager = ZellijLocalManager(session_layouts=[a_layouts])
         manager.start_all_sessions(poll_interval=2, poll_seconds=2)
         manager.run_monitoring_routine(wait_ms=2000)
+        if kill_upon_completion:
+            manager.kill_all_sessions()
         if i < len(layoutfile["layouts"]) - 1:  # Don't sleep after the last layout
-            time.sleep(sleep_between_layouts)
+            time.sleep(sleep_inbetween)
 
 
 @app.command(help="Adjust layout file to limit max tabs per layout, etc.")
@@ -152,9 +167,52 @@ def load_balance(layout_path: Path = typer.Argument(..., help="Path to the layou
     new_layouts = limit_tab_num(layout_configs=layout_configs, max_thresh=max_thresh, threshold_type=thresh_type, breaking_method=breaking_method)
     layoutfile["layouts"] = new_layouts
     target_file = output_path if output_path is not None else layout_path.parent / f'{layout_path.stem}_adjusted_{max_thresh}_{thresh_type}_{breaking_method}.json'
+    target_file.parent.mkdir(parents=True, exist_ok=True)
     target_file.write_text(data=json.dumps(layoutfile, indent=4), encoding="utf-8")
     typer.echo(f"Adjusted layout saved to {target_file}")
 
 
+@app.command()
+def collect(
+    agent_dir: Path = typer.Argument(..., help="Path to the agent directory containing the prompts folder"),
+    output_path: Path = typer.Argument(..., help="Path to write the concatenated material files"),
+    separator: str = typer.Option("\n", help="Separator to use when concatenating material files"),
+) -> None:
+    """Collect all material files from an agent directory and concatenate them."""
+    if not agent_dir.exists() or not agent_dir.is_dir():
+        raise typer.BadParameter(f"Agent directory does not exist or is not a directory: {agent_dir}")
+    
+    prompts_dir = agent_dir / "prompts"
+    if not prompts_dir.exists():
+        raise typer.BadParameter(f"Prompts directory not found: {prompts_dir}")
+    
+    material_files = []
+    for agent_subdir in prompts_dir.iterdir():
+        if agent_subdir.is_dir() and agent_subdir.name.startswith("agent_"):
+            material_file = agent_subdir / f"{agent_subdir.name}_material.txt"
+            if material_file.exists():
+                material_files.append(material_file)
+    
+    if not material_files:
+        typer.echo("No material files found in the agent directory.")
+        return
+    
+    # Sort by agent index for consistent ordering
+    material_files.sort(key=lambda x: int(x.parent.name.split("_")[-1]))
+    
+    # Read and concatenate all material files
+    concatenated_content = []
+    for material_file in material_files:
+        content = material_file.read_text(encoding="utf-8")
+        concatenated_content.append(content)
+    
+    result = separator.join(concatenated_content)
+    
+    # Write to output file
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+    output_path.write_text(result, encoding="utf-8")
+    typer.echo(f"Concatenated material written to {output_path}")
+
+
 if __name__ == "__main__":  # pragma: no cover
     app()

machineconfig/scripts/python/fire_agents_help_launch.py

@@ -1,4 +1,3 @@
-from machineconfig.utils.accessories import randstr
 
 import random
 import shlex
@@ -28,10 +27,9 @@ def get_gemini_api_keys() -> list[str]:
     return res
 
 
-def prep_agent_launch(repo_root: Path, prompts_material: list[str], prompt_prefix: str, keep_material_in_separate_file: bool, agent: AGENTS, *, job_name: str) -> Path:
-    session_root = repo_root / ".ai" / f"tmp_prompts/{job_name}_{randstr()}"
-    session_root.mkdir(parents=True, exist_ok=True)
-    prompt_folder = session_root / "prompts"
+def prep_agent_launch(agents_dir: Path, prompts_material: list[str], prompt_prefix: str, keep_material_in_separate_file: bool, agent: AGENTS, *, job_name: str) -> None:
+    agents_dir.mkdir(parents=True, exist_ok=True)
+    prompt_folder = agents_dir / "prompts"
     prompt_folder.mkdir(parents=True, exist_ok=True)
 
     for idx, a_prompt_material in enumerate(prompts_material):
@@ -110,7 +108,7 @@ echo "---------END OF AGENT OUTPUT---------"
 """
         agent_cmd_launch_path.write_text(cmd_prefix + cmd + cmd_postfix, encoding="utf-8")
 
-    return session_root
+    return None
 
 
 def get_agents_launch_layout(session_root: Path):

{machineconfig-4.4.dist-info → machineconfig-4.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: machineconfig
-Version: 4.4
+Version: 4.5
 Summary: Dotfiles management package
 Author-email: Alex Al-Saffar <programmer@usa.com>
 License: Apache 2.0

{machineconfig-4.4.dist-info → machineconfig-4.5.dist-info}/RECORD

@@ -16,7 +16,7 @@ machineconfig/cluster/sessions_managers/wt_local_manager.py,sha256=LqOuA9fqVDu5m
 machineconfig/cluster/sessions_managers/wt_remote.py,sha256=XmZV9rLubwxND5UYAS15mAcpzDdXvm4KyubVGYkVBmo,8743
 machineconfig/cluster/sessions_managers/wt_remote_manager.py,sha256=CnnOtPiwLx0pIfs_KlcKnclLSqiDTNzfm8t8sW6nXf0,19764
 machineconfig/cluster/sessions_managers/zellij_local.py,sha256=T2wXkW9ug1nzdnhlD-1G2HPlBpBaNw6rdM9rk4KKeHQ,26137
-machineconfig/cluster/sessions_managers/zellij_local_manager.py,sha256=4ap0Gtd8wOybFc461OItA2kU-1jpgPQUX03s8Zjyk_4,24028
+machineconfig/cluster/sessions_managers/zellij_local_manager.py,sha256=HaofZHTvjp1xFmcwJ5XaMfDVUCRdYk27eBlTKrqVr0E,24025
 machineconfig/cluster/sessions_managers/zellij_remote.py,sha256=3gz-wgqVW7B-4CgvJq0fiDh8691h4yeK3Xh3CfSPc2s,8749
 machineconfig/cluster/sessions_managers/zellij_remote_manager.py,sha256=T-j1KMV7mDTeGSHC5To0_JmqNtjSR_LVZT9VanP4lyI,8313
 machineconfig/cluster/sessions_managers/utils/enhanced_command_runner.py,sha256=3vcQVg-HHa_WTxBGPtKMAdoSqJVa2EO5KAtrY8a6I3c,5264
@@ -155,8 +155,8 @@ machineconfig/scripts/python/devops_backup_retrieve.py,sha256=jZe5Vki7E2GCMG8hvq
 machineconfig/scripts/python/devops_devapps_install.py,sha256=QbRQhNdDRLLtgJwaRl2pbLmWvajb1b_Xte2ql8N3JRs,9096
 machineconfig/scripts/python/devops_update_repos.py,sha256=c5qBc9cuTGDEqDHufkjDT4d_vvJsswv3tlqk9MAulYk,8063
 machineconfig/scripts/python/dotfile.py,sha256=SRcX-9Ak1jRvF-killBTTm2IWcsNxfiLucH6ZsytAFA,2202
-machineconfig/scripts/python/fire_agents.py,sha256=An6OE2dVyni_XGt3V-5kMhPxDSfXOnJuO2N6AokQHyo,8860
-machineconfig/scripts/python/fire_agents_help_launch.py,sha256=TpuARLC_9b3SNMYeYIi_h1qKCSifT074QEvZ-ZK967M,5264
+machineconfig/scripts/python/fire_agents.py,sha256=lBnA4uYZ2tFmkhEmixAlIagAvl-nJ-AE-p-w0M2KfOY,11653
+machineconfig/scripts/python/fire_agents_help_launch.py,sha256=UO-rV0z0G6WoL3RtnbqjIo97fq-FQO8ynmQ2lLrIdSg,5124
 machineconfig/scripts/python/fire_agents_help_search.py,sha256=qIfSS_su2YJ1Gb0_lu4cbjlJlYMBw0v52NTGiSrGjk8,2991
 machineconfig/scripts/python/fire_agents_load_balancer.py,sha256=mpqx3uaQdBXYieuvhdK-qsvLepf9oIMo3pwPj9mSEDI,1079
 machineconfig/scripts/python/fire_jobs.py,sha256=qD_9JtoNTjOjAyX0_QPysfHNQ-qq55t-Tv9ctEK17MI,20569
@@ -406,8 +406,8 @@ machineconfig/utils/schemas/fire_agents/fire_agents_input.py,sha256=pTxvLzIpD5RF
 machineconfig/utils/schemas/installer/installer_types.py,sha256=DLagmIe0G5-xg7HZ9VrlFCDk1gIbwvX7O4gZjwq0wh0,1326
 machineconfig/utils/schemas/layouts/layout_types.py,sha256=TcqlZdGVoH8htG5fHn1KWXhRdPueAcoyApppZsPAPto,2020
 machineconfig/utils/schemas/repos/repos_types.py,sha256=ECVr-3IVIo8yjmYmVXX2mnDDN1SLSwvQIhx4KDDQHBQ,405
-machineconfig-4.4.dist-info/METADATA,sha256=lyni_bLaEQayMHJkaaRhSCbb0g7svm5l25Ek_EaQ7lw,7060
-machineconfig-4.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-machineconfig-4.4.dist-info/entry_points.txt,sha256=fhSTJGrT7JZ7PcN-F3nrb5xXIohicZo57rbM23Mdblk,1065
-machineconfig-4.4.dist-info/top_level.txt,sha256=porRtB8qms8fOIUJgK-tO83_FeH6Bpe12oUVC670teA,14
-machineconfig-4.4.dist-info/RECORD,,
+machineconfig-4.5.dist-info/METADATA,sha256=hwREinHY-RApFYtMH7odfRdK32LRkZ9mKm05Cry8uBk,7060
+machineconfig-4.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+machineconfig-4.5.dist-info/entry_points.txt,sha256=fhSTJGrT7JZ7PcN-F3nrb5xXIohicZo57rbM23Mdblk,1065
+machineconfig-4.5.dist-info/top_level.txt,sha256=porRtB8qms8fOIUJgK-tO83_FeH6Bpe12oUVC670teA,14
+machineconfig-4.5.dist-info/RECORD,,