machineconfig 4.1__tar.gz → 4.4__tar.gz

{machineconfig-4.1/src/machineconfig.egg-info → machineconfig-4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: machineconfig
-Version: 4.1
+Version: 4.4
 Summary: Dotfiles management package
 Author-email: Alex Al-Saffar <programmer@usa.com>
 License: Apache 2.0
@@ -26,6 +26,7 @@ Requires-Dist: rclone-python>=0.1.23
 Requires-Dist: pyjson5>=1.6.9
 Requires-Dist: typer-slim>=0.19.2
 Requires-Dist: questionary>=2.1.1
+Requires-Dist: typer>=0.19.2
 Provides-Extra: windows
 Requires-Dist: pywin32; extra == "windows"
 Provides-Extra: docs

{machineconfig-4.1 → machineconfig-4.4}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "machineconfig"
-version = "4.01"
+version = "4.04"
 description = "Dotfiles management package"
 readme = "README.md"
 license = {text = "Apache 2.0"}
@@ -35,6 +35,7 @@ dependencies = [
     "pyjson5>=1.6.9",
     "typer-slim>=0.19.2",
     "questionary>=2.1.1",
+    "typer>=0.19.2",
 ]
 
 [project.optional-dependencies]
@@ -53,7 +54,7 @@ docs = [
 [project.scripts]
 devops = "machineconfig.scripts.python.devops:app"
 fire = "machineconfig.scripts.python.fire_jobs:main_from_parser"
-fire_agents = "machineconfig.scripts.python.fire_agents:main"
+fire_agents = "machineconfig.scripts.python.fire_agents:app"
 initai = "machineconfig.scripts.python.ai.initai:main"
 
 choose_wezterm_theme = "machineconfig.scripts.python.choose_wezterm_theme:main"

machineconfig-4.4/src/machineconfig/cluster/sessions_managers/utils/load_balancer.py

@@ -0,0 +1,53 @@
+
+from machineconfig.utils.schemas.layouts.layout_types import TabConfig, LayoutConfig
+# from machineconfig.utils.accessories import split_list
+from typing import Literal, Protocol
+from machineconfig.cluster.sessions_managers.utils.load_balancer_helper import restrict_num_tabs_helper1, restrict_num_tabs_helper2, restrict_num_tabs_helper3, restrict_num_tabs_helper4
+
+class COMMAND_SPLITTER(Protocol):
+    def __call__(self, command: str, to: int) -> list[str]: ...
+
+
+def limit_tab_num(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["number", "weight"], breaking_method: Literal["moreLayouts", "combineTabs"]) -> list[LayoutConfig]:
+    match threshold_type, breaking_method:
+        case "number", "moreLayouts":
+            return restrict_num_tabs_helper1(layout_configs=layout_configs, max_thresh=max_thresh, threshold_type=threshold_type, breaking_method=breaking_method)
+        case "number", "combineTabs":
+            return restrict_num_tabs_helper2(layout_configs=layout_configs, max_thresh=max_thresh, threshold_type=threshold_type, breaking_method=breaking_method)
+        case "weight", "moreLayouts":
+            return restrict_num_tabs_helper3(layout_configs=layout_configs, max_thresh=max_thresh, threshold_type=threshold_type, breaking_method=breaking_method)
+        case "weight", "combineTabs":
+            return restrict_num_tabs_helper4(layout_configs=layout_configs, max_thresh=max_thresh, threshold_type=threshold_type, breaking_method=breaking_method)
+        case _:
+            raise NotImplementedError(f"The combination {threshold_type}, {breaking_method} is not implemented")
+def limit_tab_weight(layout_configs: list[LayoutConfig], max_weight: int, command_splitter: COMMAND_SPLITTER) -> list[LayoutConfig]:
+    new_layout_configs: list[LayoutConfig] = []
+    for a_layout_config in layout_configs:
+        new_tabs: list[TabConfig] = []
+        for tab in a_layout_config["layoutTabs"]:
+            tab_weight = tab.get("tabWeight", 1)
+            if tab_weight > max_weight:
+                print(f"Tab '{tab['tabName']}' in layout '{a_layout_config['layoutName']}' has too much weight ({tab_weight} > {max_weight}). Splitting command.")
+                split_commands = command_splitter(tab["command"], to=max_weight)
+                for idx, cmd in enumerate(split_commands):
+                    new_tabs.append({
+                        "tabName": f"{tab['tabName']}_part{idx+1}",
+                        "startDir": tab["startDir"],
+                        "command": cmd,
+                        "tabWeight": max_weight
+                    })
+            else:
+                new_tabs.append(tab)
+        new_layout_configs.append({
+            "layoutName": a_layout_config["layoutName"],
+            "layoutTabs": new_tabs
+        })
+    return new_layout_configs
+
+
+def run(layouts: list[LayoutConfig]):
+    from machineconfig.cluster.sessions_managers.zellij_local_manager import ZellijLocalManager
+    manager = ZellijLocalManager(session_layouts=layouts)
+    manager.start_all_sessions(poll_interval=2, poll_seconds=2)
+    manager.run_monitoring_routine(wait_ms=2000)
+    manager.kill_all_sessions()

machineconfig-4.1/src/machineconfig/cluster/sessions_managers/utils/load_balancer.py → machineconfig-4.4/src/machineconfig/cluster/sessions_managers/utils/load_balancer_helper.py

@@ -1,8 +1,7 @@
-
-from machineconfig.utils.schemas.layouts.layout_types import TabConfig, LayoutConfig
 from machineconfig.utils.accessories import split_list
-from typing import Literal
+from machineconfig.utils.schemas.layouts.layout_types import TabConfig, LayoutConfig
 
+from typing import Literal
 
 
 def split_tabs_by_weight(tabs: list[TabConfig], max_weight: int) -> list[list[TabConfig]]:
@@ -73,21 +72,7 @@ def combine_tabs_by_weight_into_super_tabs(tabs: list[TabConfig], max_weight: in
     return super_tabs
 
 
-def restrict_num_tabs(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["number", "weight"], breaking_method: Literal["moreLayouts", "combineTabs"]) -> list[LayoutConfig]:
-    match threshold_type, breaking_method:
-        case "number", "moreLayouts":
-            return _restrict_num_tabs_helper1(layout_configs=layout_configs, max_thresh=max_thresh, threshold_type=threshold_type, breaking_method=breaking_method)
-        case "number", "combineTabs":
-            return _restrict_num_tabs_helper2(layout_configs=layout_configs, max_thresh=max_thresh, threshold_type=threshold_type, breaking_method=breaking_method)
-        case "weight", "moreLayouts":
-            return _restrict_num_tabs_helper3(layout_configs=layout_configs, max_thresh=max_thresh, threshold_type=threshold_type, breaking_method=breaking_method)
-        case "weight", "combineTabs":
-            return _restrict_num_tabs_helper4(layout_configs=layout_configs, max_thresh=max_thresh, threshold_type=threshold_type, breaking_method=breaking_method)
-        case _:
-            raise NotImplementedError(f"The combination {threshold_type}, {breaking_method} is not implemented")
-
-
-def _restrict_num_tabs_helper1(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["number"], breaking_method: Literal["moreLayouts"]) -> list[LayoutConfig]:
+def restrict_num_tabs_helper1(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["number"], breaking_method: Literal["moreLayouts"]) -> list[LayoutConfig]:
     """When threshold is exceeded, create more layouts with max_thresh tabs each."""
     new_layout_configs: list[LayoutConfig] = []
     for a_layout_config in layout_configs:
@@ -105,7 +90,7 @@ def _restrict_num_tabs_helper1(layout_configs: list[LayoutConfig], max_thresh: i
     return new_layout_configs
 
 
-def _restrict_num_tabs_helper2(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["number"], breaking_method: Literal["combineTabs"]) -> list[LayoutConfig]:
+def restrict_num_tabs_helper2(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["number"], breaking_method: Literal["combineTabs"]) -> list[LayoutConfig]:
     """When threshold is exceeded, combine tabs into super tabs to reduce count to max_thresh."""
     new_layout_configs: list[LayoutConfig] = []
     for a_layout_config in layout_configs:
@@ -123,7 +108,7 @@ def _restrict_num_tabs_helper2(layout_configs: list[LayoutConfig], max_thresh: i
     return new_layout_configs
 
 
-def _restrict_num_tabs_helper3(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["weight"], breaking_method: Literal["moreLayouts"]) -> list[LayoutConfig]:
+def restrict_num_tabs_helper3(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["weight"], breaking_method: Literal["moreLayouts"]) -> list[LayoutConfig]:
     """When threshold is exceeded, create more layouts with max_thresh total weight each."""
     new_layout_configs: list[LayoutConfig] = []
     for a_layout_config in layout_configs:
@@ -142,7 +127,7 @@ def _restrict_num_tabs_helper3(layout_configs: list[LayoutConfig], max_thresh: i
     return new_layout_configs
 
 
-def _restrict_num_tabs_helper4(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["weight"], breaking_method: Literal["combineTabs"]) -> list[LayoutConfig]:
+def restrict_num_tabs_helper4(layout_configs: list[LayoutConfig], max_thresh: int, threshold_type: Literal["weight"], breaking_method: Literal["combineTabs"]) -> list[LayoutConfig]:
     """When threshold is exceeded, combine tabs into super tabs with weight <= max_thresh."""
     new_layout_configs: list[LayoutConfig] = []
     for a_layout_config in layout_configs:
@@ -158,11 +143,3 @@ def _restrict_num_tabs_helper4(layout_configs: list[LayoutConfig], max_thresh: i
             print(f"Layout '{a_layout_config['layoutName']}' has acceptable total weight ({layout_weight} <= {max_thresh}). Keeping as is.")
             new_layout_configs.append(a_layout_config)
     return new_layout_configs
-
-
-def run(layouts: list[LayoutConfig]):
-    from machineconfig.cluster.sessions_managers.zellij_local_manager import ZellijLocalManager
-    manager = ZellijLocalManager(session_layouts=layouts)
-    manager.start_all_sessions(poll_interval=2, poll_seconds=2)
-    manager.run_monitoring_routine(wait_ms=2000)
-    manager.kill_all_sessions()

machineconfig-4.4/src/machineconfig/scripts/python/fire_agents.py

@@ -0,0 +1,160 @@
+"""Utilitfrom pathlib import Path
+from typing import cast, get_args, Iterable, Optional, TypeAlias, Literal
+import json
+import typer
+
+from machineconfig.scripts.python.fire_agents_help_launch import prep_agent_launch, get_agents_launch_layout, AGENTS
+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.accessories import get_repo_rootch multiple AI agent prompts in a Zellij session.
+
+Improved design notes:
+  * Clear separation of: input collection, prompt preparation, agent launch.
+  * Configurable max agent cap (default 15) with interactive confirmation if exceeded.
+  * Added type aliases + docstrings for maintainability.
+  * Preserves original core behavior & command generation for each agent type.
+"""
+
+from pathlib import Path
+from typing import cast, Iterable, Optional, TypeAlias, Literal, get_args
+import json
+import time
+import typer
+
+from machineconfig.scripts.python.fire_agents_help_launch import prep_agent_launch, get_agents_launch_layout, AGENTS
+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
+
+SEARCH_STRATEGIES: TypeAlias = Literal["file_path", "keyword_search", "filename_pattern"]
+
+app = typer.Typer()
+
+
+def _write_list_file(target: Path, files: Iterable[Path]) -> None:
+    target.parent.mkdir(parents=True, exist_ok=True)
+    target.write_text("\n".join(str(f) for f in files), encoding="utf-8")
+
+
+@app.command()
+def create(
+    context_path: Optional[Path] = typer.Option(None, help="Path to the context file"),
+    keyword_search: Optional[str] = typer.Option(None, help="Keyword to search in Python files"),
+    filename_pattern: Optional[str] = typer.Option(None, help="Filename pattern to match"),
+    separator: str = typer.Option("\n", help="Separator for context"),
+    tasks_per_prompt: int = typer.Option(13, help="Number of tasks per prompt"),
+    agent: AGENTS = typer.Option(..., help=f"Agent type. One of {', '.join(get_args(AGENTS))}"),
+    prompt: Optional[str] = typer.Option(None, help="Prompt prefix as string"),
+    prompt_path: Optional[Path] = typer.Option(None, help="Path to prompt file"),
+    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"),
+):
+    # validate mutual exclusive
+    context_options = [context_path, keyword_search, filename_pattern]
+    provided_context = [opt for opt in context_options if opt is not None]
+    if len(provided_context) != 1:
+        raise typer.BadParameter("Exactly one of --context-path, --keyword-search, --filename-pattern must be provided")
+    
+    prompt_options = [prompt, prompt_path]
+    provided_prompt = [opt for opt in prompt_options if opt is not None]
+    if len(provided_prompt) != 1:
+        raise typer.BadParameter("Exactly one of --prompt or --prompt-path must be provided")
+    
+    repo_root = get_repo_root(Path.cwd())
+    if repo_root is None:
+        typer.echo("💥 Could not determine the repository root. Please run this script from within a git repository.")
+        raise typer.Exit(1)
+    typer.echo(f"Operating @ {repo_root}")
+    
+    search_strategy = ""
+    prompt_material_path = Path("")
+    
+    if context_path is not None:
+        search_strategy = "file_path"
+        target_file_path = context_path.expanduser().resolve()
+        if not target_file_path.exists() or not target_file_path.is_file():
+            raise typer.BadParameter(f"Invalid file path: {target_file_path}")
+        prompt_material_path = target_file_path
+    elif keyword_search is not None:
+        search_strategy = "keyword_search"
+        matching_files = search_python_files(repo_root, keyword_search)
+        if not matching_files:
+            typer.echo(f"💥 No .py files found containing keyword: {keyword_search}")
+            raise typer.Exit(1)
+        target_file_path = repo_root / ".ai" / "target_file.txt"
+        _write_list_file(target_file_path, matching_files)
+        prompt_material_path = target_file_path
+    elif filename_pattern is not None:
+        search_strategy = "filename_pattern"
+        matching_files = search_files_by_pattern(repo_root, filename_pattern)
+        if not matching_files:
+            typer.echo(f"💥 No files found matching pattern: {filename_pattern}")
+            raise typer.Exit(1)
+        target_file_path = repo_root / ".ai" / "target_file.txt"
+        _write_list_file(target_file_path, matching_files)
+        prompt_material_path = target_file_path
+    
+    if prompt_path is not None:
+        prompt_prefix = prompt_path.read_text(encoding="utf-8")
+    else:
+        prompt_prefix = cast(str, prompt)
+    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)
+    layoutfile = get_agents_launch_layout(session_root=agents_dir)    
+    regenerate_py_code = f"""
+#!/usr/bin/env uv run --python 3.13 --with machineconfig
+#!/usr/bin/env uv run --no-dev --project $HOME/code/machineconfig
+fire_agents create --context-path "{prompt_material_path}" \\
+    --{search_strategy} "{context_path or keyword_search or filename_pattern}" \\
+    --prompt-path "{prompt_path or ''}" \\
+    --agent "{agent_selected}" \\
+    --job-name "{job_name}" \\
+    --tasks-per-prompt {tasks_per_prompt} \\
+    --separator "{separator}" \\
+    {"--keep-separate" if keep_material_in_separate_file_input else ""}
+"""
+    (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.write_text(data=json.dumps(layoutfile, indent=4), encoding="utf-8")
+    typer.echo(f"Created agents in {agents_dir}")
+
+
+@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")):
+    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)
+    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 i < len(layoutfile["layouts"]) - 1:  # Don't sleep after the last layout
+            time.sleep(sleep_between_layouts)
+
+
+@app.command(help="Adjust layout file to limit max tabs per layout, etc.")
+def load_balance(layout_path: Path = typer.Argument(..., help="Path to the layout.json file"),
+           max_thresh: int = typer.Option(..., help="Maximum tabs per layout"),
+           thresh_type: Literal['number', 'weight'] = typer.Option(..., help="Threshold type"),
+           breaking_method: Literal['moreLayouts', 'combineTabs'] = typer.Option(..., help="Breaking method"),
+           output_path: Optional[Path] = typer.Option(None, help="Path to write the adjusted layout.json file")):
+    layoutfile: LayoutsFile = json.loads(layout_path.read_text())
+    layout_configs = layoutfile["layouts"]
+    from machineconfig.cluster.sessions_managers.utils.load_balancer import limit_tab_num
+    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.write_text(data=json.dumps(layoutfile, indent=4), encoding="utf-8")
+    typer.echo(f"Adjusted layout saved to {target_file}")
+
+
+if __name__ == "__main__":  # pragma: no cover
+    app()

{machineconfig-4.1 → machineconfig-4.4}/src/machineconfig/scripts/python/fire_agents_help_launch.py

@@ -7,7 +7,7 @@ from typing import Literal, TypeAlias
 
 
 AGENTS: TypeAlias = Literal[
-    "cursor-agent", "gemini", "crush", "q", "onlyPrepPromptFiles"
+    "cursor-agent", "gemini", "crush", "q"
     # warp terminal
 ]
 AGENT_NAME_FORMATTER = "agent_{idx}_cmd.sh"  # e.g., agent_0_cmd.sh
@@ -34,7 +34,6 @@ def prep_agent_launch(repo_root: Path, prompts_material: list[str], prompt_prefi
     prompt_folder = session_root / "prompts"
     prompt_folder.mkdir(parents=True, exist_ok=True)
 
-    all_materials_scripts: list[Path] = []
     for idx, a_prompt_material in enumerate(prompts_material):
         prompt_root = prompt_folder / f"agent_{idx}"
         prompt_root.mkdir(parents=True, exist_ok=True)
@@ -43,7 +42,6 @@ def prep_agent_launch(repo_root: Path, prompts_material: list[str], prompt_prefi
             prompt_material_path = prompt_root / f"agent_{idx}_material.txt"
             prompt_material_path.write_text(a_prompt_material, encoding="utf-8")
             prompt_path.write_text(prompt_prefix + f"""\nPlease only look @ {prompt_material_path}. You don't need to do any other work beside the content of this material file.""", encoding="utf-8")
-            all_materials_scripts.append(prompt_material_path)
         else:
             prompt_material_path = prompt_path
             prompt_path.write_text(prompt_prefix + """\nPlease only look @ the following:\n""" + a_prompt_material, encoding="utf-8")
@@ -87,7 +85,7 @@ sleep 0.1
                 cmd = f"""
 export GEMINI_API_KEY={shlex.quote(api_key)}
 echo "Using Gemini API key $GEMINI_API_KEY"
-bash -lc 'cat {safe_path} | gemini {model_arg} --yolo --prompt'
+gemini {model_arg} --yolo --prompt {safe_path}
 """
             case "cursor-agent":
                 # As originally implemented
@@ -103,10 +101,6 @@ crush run {prompt_path}
             case "q":
                 cmd = f"""
 q chat --no-interactive --trust-all-tools {prompt_path}
-"""
-            case "onlyPrepPromptFiles":
-                cmd = f"""
-echo "Prepared prompt file at {prompt_path}"
 """
             case _:
                 raise ValueError(f"Unsupported agent type: {agent}")
@@ -116,11 +110,6 @@ echo "---------END OF AGENT OUTPUT---------"
 """
         agent_cmd_launch_path.write_text(cmd_prefix + cmd + cmd_postfix, encoding="utf-8")
 
-    # print(f"Launching a template with #{len(tab_config)} agents")
-    if len(all_materials_scripts) > 0:
-        all_materials_list_path = session_root / "all_materials_redistributed.txt"
-        all_materials_list_path.write_text("\n".join(str(p) for p in all_materials_scripts), encoding="utf-8")
-        print(f"All prompt materials listed @ {all_materials_list_path}")
     return session_root
 
 
@@ -130,14 +119,11 @@ def get_agents_launch_layout(session_root: Path):
     tab_config: list[TabConfig] = []
     prompt_root = session_root / "prompts"
     all_dirs_under_prompts = [d for d in prompt_root.iterdir() if d.is_dir()]
-    launch_agents_squentially = ""
     for a_prompt_dir in all_dirs_under_prompts:
         idx = a_prompt_dir.name.split("_")[-1]  # e.g., agent_0 -> 0
         agent_cmd_path = a_prompt_dir / AGENT_NAME_FORMATTER.format(idx=idx)
         fire_cmd = f"bash {shlex.quote(str(agent_cmd_path))}"
         tab_config.append(TabConfig(tabName=f"Agent{idx}", startDir=str(session_root.parent.parent.parent), command=fire_cmd))
-        launch_agents_squentially += f". {shlex.quote(str(agent_cmd_path))}\n"
     layout = LayoutConfig(layoutName="Agents", layoutTabs=tab_config)
-    (session_root / "launch_all_agents_sequentially.sh").write_text(launch_agents_squentially, encoding="utf-8")
     layouts_file: LayoutsFile = LayoutsFile(version="1.0", layouts=[layout])
     return layouts_file

machineconfig-4.4/src/machineconfig/scripts/python/fire_agents_load_balancer.py

@@ -0,0 +1,22 @@
+# from math import ceil
+from pathlib import Path
+
+
+def chunk_prompts(prompt_material_path: Path, joiner: str, *, tasks_per_prompt: int) -> list[str]:
+    """Chunk prompts based on splitting strategy.
+
+    Args:
+        prompts: List of prompts to chunk
+        strategy: Either 'agent_cap' or 'task_rows'
+        agent_cap: Maximum number of agents (used with 'agent_cap' strategy)
+        task_rows: Number of rows/tasks per agent (used with 'task_rows' strategy)
+    """
+    prompts = [p for p in prompt_material_path.read_text(encoding="utf-8", errors="ignore").split(joiner) if p.strip() != ""]  # drop blank entries
+    if tasks_per_prompt >= len(prompts):
+        print("No need to chunk prompts, as tasks_per_prompt >= total prompts.", f"({tasks_per_prompt} >= {len(prompts)})")
+        return prompts
+    print(f"Chunking {len(prompts)} prompts into groups of {tasks_per_prompt} rows/tasks each.")
+    grouped: list[str] = []
+    for i in range(0, len(prompts), tasks_per_prompt):
+        grouped.append(joiner.join(prompts[i : i + tasks_per_prompt]))
+    return grouped

{machineconfig-4.1 → machineconfig-4.4}/src/machineconfig/utils/schemas/fire_agents/fire_agents_input.py

@@ -6,7 +6,6 @@ capturing all user inputs collected during interactive execution.
 
 from pathlib import Path
 from typing import TypedDict, Literal, NotRequired
-from machineconfig.scripts.python.fire_agents_load_balancer import SPLITTING_STRATEGY
 from machineconfig.scripts.python.fire_agents_help_launch import AGENTS
 
 SEARCH_STRATEGIES = Literal["file_path", "keyword_search", "filename_pattern"]
@@ -21,26 +20,17 @@ class FilePathSearchInput(TypedDict):
 
 class KeywordSearchInput(TypedDict):
     """Input for keyword_search strategy."""
-
     keyword: str
 
 
 class FilenamePatternSearchInput(TypedDict):
     """Input for filename_pattern search strategy."""
-
     pattern: str  # e.g., '*.py', '*test*', 'config.*'
 
 
-class AgentCapSplittingInput(TypedDict):
-    """Input for agent_cap splitting strategy."""
-
-    agent_cap: int  # Default: 6
-
-
-class TaskRowsSplittingInput(TypedDict):
-    """Input for task_rows splitting strategy."""
-
-    task_rows: int  # Default: 13
+class TasksPerPromptSplittingInput(TypedDict):
+    """Input for tasks_per_prompt splitting strategy."""
+    tasks_per_prompt: int  # Default: 13
 
 
 class FireAgentsMainInput(TypedDict):
@@ -49,7 +39,6 @@ class FireAgentsMainInput(TypedDict):
     # Core configuration
     repo_root: Path
     search_strategy: SEARCH_STRATEGIES
-    splitting_strategy: SPLITTING_STRATEGY
     agent_selected: AGENTS
     prompt_prefix: str
     job_name: str  # Default: "AI_Agents"
@@ -62,8 +51,7 @@ class FireAgentsMainInput(TypedDict):
     filename_pattern_input: NotRequired[FilenamePatternSearchInput]
 
     # Splitting strategy specific inputs (only one will be present based on splitting_strategy)
-    agent_cap_input: NotRequired[AgentCapSplittingInput]
-    task_rows_input: NotRequired[TaskRowsSplittingInput]
+    tasks_per_prompt: TasksPerPromptSplittingInput
 
 
 class FireAgentsRuntimeData(TypedDict):

{machineconfig-4.1 → machineconfig-4.4/src/machineconfig.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: machineconfig
-Version: 4.1
+Version: 4.4
 Summary: Dotfiles management package
 Author-email: Alex Al-Saffar <programmer@usa.com>
 License: Apache 2.0
@@ -26,6 +26,7 @@ Requires-Dist: rclone-python>=0.1.23
 Requires-Dist: pyjson5>=1.6.9
 Requires-Dist: typer-slim>=0.19.2
 Requires-Dist: questionary>=2.1.1
+Requires-Dist: typer>=0.19.2
 Provides-Extra: windows
 Requires-Dist: pywin32; extra == "windows"
 Provides-Extra: docs

{machineconfig-4.1 → machineconfig-4.4}/src/machineconfig.egg-info/SOURCES.txt

@@ -30,6 +30,7 @@ src/machineconfig/cluster/sessions_managers/zellij_remote.py
 src/machineconfig/cluster/sessions_managers/zellij_remote_manager.py
 src/machineconfig/cluster/sessions_managers/utils/enhanced_command_runner.py
 src/machineconfig/cluster/sessions_managers/utils/load_balancer.py
+src/machineconfig/cluster/sessions_managers/utils/load_balancer_helper.py
 src/machineconfig/cluster/sessions_managers/wt_utils/layout_generator.py
 src/machineconfig/cluster/sessions_managers/wt_utils/process_monitor.py
 src/machineconfig/cluster/sessions_managers/wt_utils/remote_executor.py

{machineconfig-4.1 → machineconfig-4.4}/src/machineconfig.egg-info/entry_points.txt

@@ -7,7 +7,7 @@ cloud_sync = machineconfig.scripts.python.cloud_sync:arg_parser
 croshell = machineconfig.scripts.python.croshell:arg_parser
 devops = machineconfig.scripts.python.devops:app
 fire = machineconfig.scripts.python.fire_jobs:main_from_parser
-fire_agents = machineconfig.scripts.python.fire_agents:main
+fire_agents = machineconfig.scripts.python.fire_agents:app
 ftpx = machineconfig.scripts.python.ftpx:main_from_parser
 initai = machineconfig.scripts.python.ai.initai:main
 kill_process = machineconfig.utils.procs:main

{machineconfig-4.1 → machineconfig-4.4}/src/machineconfig.egg-info/requires.txt

@@ -13,6 +13,7 @@ rclone-python>=0.1.23
 pyjson5>=1.6.9
 typer-slim>=0.19.2
 questionary>=2.1.1
+typer>=0.19.2
 
 [docs]
 pdoc>=15.0.2