openrunner-sdk 2.17.0__tar.gz → 2.17.2__tar.gz

{openrunner_sdk-2.17.0 → openrunner_sdk-2.17.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.17.0
+Version: 2.17.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.17.2/openrunner/CLAUDE.md

@@ -0,0 +1,15 @@
+# OpenRunner MCP Integration
+
+When the user asks about training metrics, loss charts, run data, or experiment results:
+- Use `mcp__openrunner__openrunner_watch_run` to show ASCII sparkline charts
+- Use `mcp__openrunner__openrunner_get_run_metrics` for raw metrics
+- Use `mcp__openrunner__openrunner_list_runs` to find runs
+- Use `mcp__openrunner__openrunner_search_sessions` to search AI session history
+
+Do NOT parse log files or grep for metrics when OpenRunner MCP tools are available.
+
+When the user asks about papers, research, or sessions:
+- Use `mcp__openrunner__openrunner_search_papers` for paper content
+- Use `mcp__openrunner__openrunner_search_research` for research plans
+- Use `mcp__openrunner__openrunner_search_sessions` for AI session history
+- Use `mcp__openrunner__openrunner_update_paper` to modify paper sections

{openrunner_sdk-2.17.0 → openrunner_sdk-2.17.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.17.0"
+__version__ = "2.17.2"
 
 logger = logging.getLogger("openrunner")
 

{openrunner_sdk-2.17.0 → openrunner_sdk-2.17.2}/openrunner/install_commands.py

@@ -543,6 +543,33 @@ def install_claude_code() -> list[str]:
         path.write_text(template.format(prefix="openrunner:"))
         files.append(str(path))
 
+    # Install CLAUDE.md with MCP usage instructions
+    claude_md = Path.home() / ".claude" / "CLAUDE.md"
+    mcp_instructions = """# OpenRunner MCP Integration
+
+When the user asks about training metrics, loss charts, run data, or experiment results:
+- Use `mcp__openrunner__openrunner_watch_run` to show ASCII sparkline charts
+- Use `mcp__openrunner__openrunner_get_run_metrics` for raw metrics
+- Use `mcp__openrunner__openrunner_list_runs` to find runs
+- Use `mcp__openrunner__openrunner_search_sessions` to search AI session history
+
+Do NOT parse log files or grep for metrics when OpenRunner MCP tools are available.
+
+When the user asks about papers, research, or sessions:
+- Use `mcp__openrunner__openrunner_search_papers` for paper content
+- Use `mcp__openrunner__openrunner_search_research` for research plans
+- Use `mcp__openrunner__openrunner_search_sessions` for AI session history
+- Use `mcp__openrunner__openrunner_update_paper` to modify paper sections
+"""
+    if claude_md.exists():
+        content = claude_md.read_text()
+        if "openrunner" not in content.lower():
+            claude_md.write_text(content + "\n" + mcp_instructions)
+            files.append(str(claude_md))
+    else:
+        claude_md.write_text(mcp_instructions)
+        files.append(str(claude_md))
+
     return files
 
 

{openrunner_sdk-2.17.0 → openrunner_sdk-2.17.2}/openrunner/mcp_server.py

@@ -292,6 +292,19 @@ def create_server():
                     "required": ["query"],
                 },
             ),
+            Tool(
+                name="openrunner_watch_run",
+                description="Get latest metrics for a run and render as ASCII chart. Use to monitor training progress.",
+                inputSchema={
+                    "type": "object",
+                    "properties": {
+                        "run_id": {"type": "string", "description": "Run ID to watch"},
+                        "metric": {"type": "string", "description": "Metric key (default: train/loss)", "default": "train/loss"},
+                        "width": {"type": "integer", "description": "Chart width in chars (default: 60)", "default": 60},
+                    },
+                    "required": ["run_id"],
+                },
+            ),
             Tool(
                 name="openrunner_update_paper",
                 description="Update a paper's title, abstract, or section content (LaTeX).",
@@ -352,6 +365,8 @@ def create_server():
                 result = await _tool_search_papers(arguments)
             elif name == "openrunner_update_paper":
                 result = await _tool_update_paper(arguments)
+            elif name == "openrunner_watch_run":
+                result = await _tool_watch_run(arguments)
             elif name == "openrunner_search_research":
                 result = await _tool_search_research(arguments)
             else:
@@ -735,6 +750,64 @@ async def _tool_search_papers(args: dict) -> str:
         return f"Error: {e}"
 
 
+async def _tool_watch_run(args: dict) -> str:
+    """Get metrics for a run and render ASCII sparkline chart."""
+    run_id = args["run_id"]
+    metric_key = args.get("metric", "train/loss")
+    width = args.get("width", 60)
+
+    try:
+        data = await _api_get(f"/runs/{run_id}/metrics", params={"keys": metric_key})
+        points = data.get("metrics", {}).get(metric_key, [])
+        if not points:
+            # Try without namespace
+            all_metrics = data.get("metrics", {})
+            if all_metrics:
+                # Show available keys
+                keys = list(all_metrics.keys())
+                return f"No '{metric_key}' data. Available: {', '.join(keys[:20])}"
+            return f"No metrics yet for run {run_id}. Training may still be starting."
+
+        # Extract values
+        values = [p.get("value", p) if isinstance(p, dict) else p for p in points]
+        steps = [p.get("step", i) if isinstance(p, dict) else i for i, p in enumerate(points)]
+
+        if not values:
+            return "No numeric values found."
+
+        # ASCII chart
+        min_v = min(values)
+        max_v = max(values)
+        range_v = max_v - min_v or 1
+        height = 10
+        chart_width = min(width, len(values))
+
+        # Sample if too many points
+        if len(values) > chart_width:
+            step_size = len(values) / chart_width
+            sampled = [values[int(i * step_size)] for i in range(chart_width)]
+        else:
+            sampled = values
+
+        # Build chart
+        bars = "▁▂▃▄▅▆▇█"
+        sparkline = ""
+        for v in sampled:
+            idx = int((v - min_v) / range_v * (len(bars) - 1))
+            sparkline += bars[idx]
+
+        lines = []
+        lines.append(f"📊 {metric_key} — run {run_id}")
+        lines.append(f"   Steps: {steps[0]}→{steps[-1]} ({len(values)} points)")
+        lines.append(f"   Range: {min_v:.4f} → {max_v:.4f} (latest: {values[-1]:.4f})")
+        lines.append(f"   {sparkline}")
+        lines.append(f"   {'↑' if values[-1] > values[0] else '↓'} {'rising' if values[-1] > values[0] else 'falling'} ({((values[-1] - values[0]) / (values[0] or 1)) * 100:.1f}%)")
+
+        return "\n".join(lines)
+    except Exception as e:
+        return f"Error fetching metrics: {e}"
+
+
 async def _tool_update_paper(args: dict) -> str:
     """Update a paper's title, abstract, or section content."""
     paper_id = args["paper_id"]

{openrunner_sdk-2.17.0 → openrunner_sdk-2.17.2}/pyproject.toml

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