openrunner-sdk 2.20.0__tar.gz → 2.20.2__tar.gz

{openrunner_sdk-2.20.0 → openrunner_sdk-2.20.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.20.0
+Version: 2.20.2
 Summary: OpenRunner SDK - W&B-compatible ML experiment tracking client
 Project-URL: Homepage, https://github.com/jqueguiner/openrunner
 Project-URL: Repository, https://github.com/jqueguiner/openrunner

{openrunner_sdk-2.20.0 → openrunner_sdk-2.20.2}/openrunner/__init__.py

@@ -120,7 +120,7 @@ launch.from_run = _launch_from_run  # type: ignore[attr-defined]
 # openrunner.trace.patch_openai() syntax
 trace.patch_openai = _patch_openai  # type: ignore[attr-defined]
 
-__version__ = "2.20.0"
+__version__ = "2.20.2"
 
 logger = logging.getLogger("openrunner")
 

{openrunner_sdk-2.20.0 → openrunner_sdk-2.20.2}/openrunner/install_commands.py

@@ -791,20 +791,19 @@ allowed-tools:
   - AskUserQuestion
 ---
 <objective>
-Configure which training metrics appear in the Claude Code statusline footer.
-Fetches available metrics from the most recent run and lets the user pick.
+Configure which project, run, and metrics appear in the Claude Code statusline footer.
+Interactive setup: project → run → metrics.
 </objective>
 
 <process>
-1. Fetch available metrics from the latest run in the configured project.
-2. Present them to the user as a numbered list.
-3. Ask which metric to use for the sparkline (primary).
-4. Ask which additional metrics to show (comma-separated numbers).
-5. Save to ~/.openrunner/statusline.json.
-6. Confirm and show preview.
+1. List projects. Ask user which one (propose current if configured).
+2. List recent runs in that project. Ask user which run (propose latest).
+3. List available metrics for that run. Ask user which for sparkline + which extra.
+4. Save to ~/.openrunner/statusline.json.
+5. Confirm and show preview.
 </process>
 
-### Step 1: Fetch available metrics
+### Step 1: List projects and runs
 
 ```bash
 python3 -c "
@@ -820,70 +819,130 @@ if not config.get('api_key'):
     print('NOT_CONFIGURED'); sys.exit(1)
 
 client = APIClient(base_url=config['base_url'], api_key=config['api_key'])
-pname = config.get('project', '')
+
+# List projects
+projects = client.list_projects()
+print('PROJECTS:')
+current = config.get('project', '')
+for i, p in enumerate(projects, 1):
+    full = '%s/%s' % (p.get('org_name',''), p.get('name',''))
+    marker = ' (current)' if full.lower() == current.lower() else ''
+    print('  [%d] %s%s' % (i, full, marker))
+
+# Show current statusline config
+cfg_path = Path.home() / '.openrunner' / 'statusline.json'
+if cfg_path.exists():
+    cfg = json.loads(cfg_path.read_text())
+    print('\\nCURRENT_CONFIG: project=%s run=%s sparkline=%s metrics=%s' % (
+        cfg.get('project','auto'), cfg.get('run','latest'),
+        cfg.get('sparkline','auto'), cfg.get('metrics',[])))
+else:
+    print('\\nCURRENT_CONFIG: not configured')
+
+client.close()
+"
+```
+
+### Step 2: Ask project
+
+Ask the user: "Which project? (number, or Enter for current)"
+Default = current project.
+
+### Step 3: List runs for chosen project
+
+```bash
+python3 -c "
+import sys, os, json
+from pathlib import Path
+for p in ([os.path.join(d, 'site-packages') for d in __import__('glob').glob(os.path.expanduser('~/.local/lib/python3.*/')) + __import__('glob').glob(os.path.expanduser('~/Library/Python/3.*/lib/python/'))]):
+    if os.path.isdir(p): sys.path.insert(0, p)
+from openrunner.session import get_session_config
+from openrunner.api_client import APIClient
+
+config = get_session_config()
+client = APIClient(base_url=config['base_url'], api_key=config['api_key'])
+
+# PROJECT_NAME will be replaced by the chosen project
+pname = 'PROJECT_NAME'
 pid = None
 for p in client.list_projects():
-    full = f'{{p.get(\"org_name\",\"\")}}/{{p.get(\"name\",\"\")}}'
+    full = '%s/%s' % (p.get('org_name',''), p.get('name',''))
     if full.lower() == pname.lower() or (p.get('name') or '').lower() == pname.lower():
         pid = p.get('id'); break
 
 if not pid:
     print('PROJECT_NOT_FOUND'); client.close(); sys.exit(1)
 
-# Get latest run
-resp = client._request('get', f'/projects/{{pid}}/runs', params={{'limit': 1, 'order': 'desc'}})
+resp = client._request('get', '/projects/%s/runs' % pid, params={{'limit': 10, 'order': 'desc'}})
 runs_data = resp.json()
 runs = runs_data.get('runs', []) if isinstance(runs_data, dict) else runs_data
+
+print('RUNS in %s:' % pname)
+for i, r in enumerate(runs, 1):
+    name = r.get('display_name') or r.get('id','?')
+    state = r.get('state', '?')
+    icon = ' [running]' if state == 'running' else ''
+    print('  [%d] %s (%s)%s' % (i, name, r.get('id','?')[:8], icon))
+
 if not runs:
-    print('NO_RUNS'); client.close(); sys.exit(1)
+    print('  (no runs)')
 
-rid = runs[0].get('id', '')
-run_name = runs[0].get('display_name') or rid
+print('\\n  [0] latest (auto-follow newest run)')
+client.close()
+"
+```
+
+### Step 4: Ask run
+
+Ask: "Which run to track? (number, 0=latest auto-follow)" — propose 0 (latest) as default.
+
+### Step 5: List metrics for chosen run
+
+```bash
+python3 -c "
+import sys, os, json
+from pathlib import Path
+for p in ([os.path.join(d, 'site-packages') for d in __import__('glob').glob(os.path.expanduser('~/.local/lib/python3.*/')) + __import__('glob').glob(os.path.expanduser('~/Library/Python/3.*/lib/python/'))]):
+    if os.path.isdir(p): sys.path.insert(0, p)
+from openrunner.session import get_session_config
+from openrunner.api_client import APIClient
+
+config = get_session_config()
+client = APIClient(base_url=config['base_url'], api_key=config['api_key'])
 
-# Get all metrics for this run
-resp = client._request('get', f'/runs/{{rid}}/metrics', params={{'max_points': 10000}})
+# RUN_ID will be replaced
+rid = 'RUN_ID'
+resp = client._request('get', '/runs/%s/metrics' % rid, params={{'max_points': 10000}})
 data = resp.json()
 metrics = data.get('metrics', data) if isinstance(data, dict) else {{}}
-if 'detail' in metrics:
-    metrics = {{}}
+if 'detail' in metrics: metrics = {{}}
 
 keys = sorted(metrics.keys())
-if not keys:
-    print('NO_METRICS'); client.close(); sys.exit(1)
-
-print(f'METRICS_FROM_RUN={{run_name}}')
+print('METRICS (%d available):' % len(keys))
 for i, k in enumerate(keys, 1):
-    pts = metrics[k]
+    pts = metrics.get(k, [])
     n = len(pts) if isinstance(pts, list) else 0
-    last = ''
+    last_str = ''
     if n > 0:
         lp = pts[-1]
         v = lp.get('value', lp) if isinstance(lp, dict) else lp
         if isinstance(v, float):
-            last = f' (latest: {{v:.4f}}, {{n}} pts)'
+            last_str = ' (%.4f, %d pts)' % (v, n)
         else:
-            last = f' (latest: {{v}}, {{n}} pts)'
-    print(f'  [{{i}}] {{k}}{{last}}')
-
-# Also show current config if exists
-cfg_path = Path.home() / '.openrunner' / 'statusline.json'
-if cfg_path.exists():
-    cfg = json.loads(cfg_path.read_text())
-    print('\\nCURRENT_CONFIG: sparkline=%s metrics=%s' % (cfg.get('sparkline','auto'), cfg.get('metrics',[])))
-else:
-    print('\\nCURRENT_CONFIG: not configured (auto-detect)')
+            last_str = ' (%s, %d pts)' % (v, n)
+    print('  [%d] %s%s' % (i, k, last_str))
 
 client.close()
 "
 ```
 
-### Step 2: Ask the user
+### Step 6: Ask metrics
 
-After showing the list, ask:
-- "Which metric for the sparkline? (number or name)" — this is the primary one shown as a mini chart
-- "Which additional metrics to display? (comma-separated numbers, or 'auto')"
+Ask:
+- "Which metric for the sparkline? (number)" — primary, shown as mini chart
+- "Which additional metrics? (comma-separated numbers, or 'auto')"
 
-### Step 3: Save config
+### Step 7: Save config
 
 ```bash
 python3 -c "
@@ -891,17 +950,19 @@ import json
 from pathlib import Path
 cfg_path = Path.home() / '.openrunner' / 'statusline.json'
 cfg_path.parent.mkdir(parents=True, exist_ok=True)
-cfg = {{'sparkline': 'SPARKLINE_KEY', 'metrics': METRICS_LIST}}
+cfg = {{'project': 'PROJECT_NAME', 'run': 'RUN_VALUE', 'sparkline': 'SPARKLINE_KEY', 'metrics': METRICS_LIST}}
 cfg_path.write_text(json.dumps(cfg, indent=2))
-print(f'Saved: {{cfg_path}}')
-print(f'  Sparkline: {{cfg[\"sparkline\"]}}')
-print(f'  Metrics: {{cfg[\"metrics\"]}}')
+print('Saved: %s' % cfg_path)
+print('  Project: %s' % cfg['project'])
+print('  Run: %s' % cfg['run'])
+print('  Sparkline: %s' % cfg['sparkline'])
+print('  Metrics: %s' % cfg['metrics'])
 print()
-print('Footer will show on next statusline refresh (30s).')
+print('Footer will update on next statusline refresh (30s).')
 "
 ```
 
-Replace SPARKLINE_KEY with the user's chosen metric key and METRICS_LIST with the Python list of chosen keys.
+Replace PROJECT_NAME, RUN_VALUE (run_id or 'latest'), SPARKLINE_KEY, and METRICS_LIST with user's choices.
 """
 
 MCP_CMD = """---
@@ -977,15 +1038,27 @@ except ImportError:
 cfg = get_session_config()
 if not cfg.get('api_key'): print("not configured"); sys.exit(0)
 client = APIClient(base_url=cfg['base_url'], api_key=cfg['api_key'])
-pname = cfg.get('project', '')
+# Read statusline config for project/run override
+import json as _json
+sl_cfg = {}
+sl_path = os.path.expanduser('~/.openrunner/statusline.json')
+if os.path.isfile(sl_path):
+    try: sl_cfg = _json.loads(open(sl_path).read())
+    except: pass
+pname = sl_cfg.get('project') or cfg.get('project', '')
 pid = None
 for p in client.list_projects():
     full = f"{p.get('org_name','')}/{p.get('name','')}"
     if full.lower() == pname.lower() or (p.get('name') or '').lower() == pname.lower():
         pid = p.get('id'); break
 if not pid: print("?"); client.close(); sys.exit(0)
-runs = client._request('get', f'/projects/{pid}/runs', params={'limit': 3, 'order': 'desc'}).json()
-if isinstance(runs, dict): runs = runs.get('runs', [])
+# If specific run configured, only show that one
+configured_run = sl_cfg.get('run', 'latest')
+if configured_run and configured_run != 'latest':
+    runs = [{'id': configured_run, 'state': 'unknown'}]
+else:
+    runs = client._request('get', f'/projects/{pid}/runs', params={'limit': 3, 'order': 'desc'}).json()
+    if isinstance(runs, dict): runs = runs.get('runs', [])
 parts = []
 for run in runs[:3]:
     rid = run.get('id', '?'); state = run.get('state', '?')
@@ -1133,9 +1206,49 @@ When the user asks about papers, research, or sessions:
     statusline_script.chmod(0o755)
     files.append(str(statusline_script))
 
-    # Convention: OpenRunner writes to /tmp/claude-statusline-openrunner.txt
-    # Any statusline runner that reads /tmp/claude-statusline-*.txt will pick it up.
-    # No patching of other plugins — purely convention-based.
+    # Auto-patch any JS statusline to read /tmp/claude-statusline-*.txt plugin files.
+    # Generic: finds the statusline command from settings.json, patches if it's a node script.
+    import json as _json
+    settings_file = Path.home() / ".claude" / "settings.json"
+    if settings_file.exists():
+        try:
+            settings = _json.loads(settings_file.read_text())
+            sl_cmd = settings.get("statusLine", {}).get("command", "")
+            # Extract JS file path from command like 'node "/path/to/script.js"'
+            import shlex
+            parts = shlex.split(sl_cmd) if sl_cmd else []
+            js_file = None
+            for p in parts:
+                if p.endswith(".js") and Path(p).exists():
+                    js_file = Path(p)
+                    break
+            if js_file:
+                content = js_file.read_text()
+                if "claude-statusline-" not in content and "// Output" in content:
+                    patch = """    // Plugin statusline segments: read /tmp/claude-statusline-*.txt
+    let plugins = '';
+    try {
+      const tmpDir = os.tmpdir();
+      const pfiles = fs.readdirSync(tmpDir)
+        .filter(f => f.startsWith('claude-statusline-') && f.endsWith('.txt'));
+      for (const f of pfiles) {
+        try {
+          const fp = path.join(tmpDir, f);
+          const st = fs.statSync(fp);
+          if ((Date.now() - st.mtimeMs) / 1000 > 120) continue;
+          const c = fs.readFileSync(fp, 'utf8').trim();
+          if (c && c.length < 200) plugins += ' \\u2502 ' + c;
+        } catch (e) {}
+      }
+    } catch (e) {}
+
+"""
+                    content = content.replace("// Output", patch + "    // Output")
+                    content = content.replace("${ctx}`)", "${ctx}${plugins}`)")
+                    js_file.write_text(content)
+                    files.append(str(js_file))
+        except Exception:
+            pass
 
     return files
 

{openrunner_sdk-2.20.0 → openrunner_sdk-2.20.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "openrunner-sdk"
-version = "2.20.0"
+version = "2.20.2"
 description = "OpenRunner SDK - W&B-compatible ML experiment tracking client"
 readme = "README.md"
 license = {text = "MIT"}