Merge pull request #478 from PR0M3TH3AN/scheduler-agent-selection-13421489652897686129

Add scheduler agent selection script

scripts/agent/select-agent.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+import json
+import os
+import sys
+import glob
+import re
+
+def get_latest_log_file(log_dir):
+    files = glob.glob(os.path.join(log_dir, "*.md"))
+    if not files:
+        return None
+    return max(files, key=os.path.getmtime)
+
+def get_previous_agent(latest_file):
+    if not latest_file:
+        return None
+
+    # Try parsing frontmatter
+    try:
+        with open(latest_file, 'r') as f:
+            content = f.read()
+            # Simple frontmatter parser
+            match = re.search(r'^agent:\s*(.+)$', content, re.MULTILINE)
+            if match:
+                return match.group(1).strip()
+    except Exception:
+        pass
+
+    # Fallback to filename
+    filename = os.path.basename(latest_file)
+    parts = filename.split('__')
+    if len(parts) >= 2:
+        return parts[1]
+
+    return None
+
+def main():
+    roster_path = 'src/prompts/roster.json'
+    log_dir = 'task-logs/daily/'
+    config_path = 'torch-config.json'
+
+    # Get excluded agents from stdin (JSON)
+    excluded = set()
+    try:
+        input_data = sys.stdin.read()
+
+        # Skip potential npm noise/headers before the JSON object
+        json_start = input_data.find('{')
+        if json_start != -1:
+            input_data = input_data[json_start:]
+            lock_data = json.loads(input_data)
+            excluded = set(lock_data.get('excluded', []))
+            # Also add currently locked/completed to exclusion if not already
+            excluded.update(lock_data.get('locked', []))
+            excluded.update(lock_data.get('completed', []))
+
+            # Also consider explicit "paused" list if present
+            excluded.update(lock_data.get('paused', []))
+    except Exception as e:
+        sys.stderr.write(f"Warning: Failed to parse input JSON: {e}\n")
+
+    with open(roster_path, 'r') as f:
+        roster_data = json.load(f)
+        roster = roster_data.get('daily', [])
+
+    if not roster:
+        print("Error: Empty roster")
+        sys.exit(1)
+
+    latest_file = get_latest_log_file(log_dir)
+    previous_agent = get_previous_agent(latest_file)
+
+    start_index = 0
+    if previous_agent and previous_agent in roster:
+         start_index = (roster.index(previous_agent) + 1) % len(roster)
+    else:
+        # First run fallback
+        try:
+            with open(config_path, 'r') as f:
+                config = json.load(f)
+                first_agent = config.get('scheduler', {}).get('firstPromptByCadence', {}).get('daily')
+                if first_agent and first_agent in roster:
+                    start_index = roster.index(first_agent)
+        except Exception:
+            pass
+
+    # Round-robin selection
+    for i in range(len(roster)):
+        idx = (start_index + i) % len(roster)
+        agent = roster[idx]
+        if agent not in excluded:
+            print(agent)
+            return
+
+    print("All roster tasks currently claimed by other agents")
+    sys.exit(0)
+
+if __name__ == "__main__":
+    main()