openrunner-sdk 2.16.0__tar.gz → 2.16.2__tar.gz

{openrunner_sdk-2.16.0 → openrunner_sdk-2.16.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.16.0
+Version: 2.16.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.16.0 → openrunner_sdk-2.16.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.16.0"
+__version__ = "2.16.2"
 
 logger = logging.getLogger("openrunner")
 

{openrunner_sdk-2.16.0 → openrunner_sdk-2.16.2}/openrunner/install_commands.py

@@ -467,6 +467,56 @@ Report version and installed commands to user. Remind them to enable the MCP plu
 """
 
 
+MCP_CMD = """---
+name: {prefix}mcp
+description: Install and activate OpenRunner MCP plugin for native tool access
+allowed-tools:
+  - Bash
+  - AskUserQuestion
+---
+<objective>
+Help user install, activate, and understand the OpenRunner MCP plugin that gives Claude Code native access to search sessions, runs, artifacts, papers, research, metrics, and code.
+</objective>
+
+<process>
+1. Install the MCP plugin config file
+2. Explain what tools become available
+3. Tell user to enable plugin and restart session
+</process>
+
+Install MCP plugin:
+```bash
+python3 -c "
+import json; from pathlib import Path
+mcp_dir = Path.home() / '.claude' / 'plugins' / 'marketplaces' / 'claude-plugins-official' / 'external_plugins' / 'openrunner'
+mcp_dir.mkdir(parents=True, exist_ok=True)
+mcp_file = mcp_dir / '.mcp.json'
+mcp_file.write_text(json.dumps({{'openrunner': {{'command': 'openrunner', 'args': ['mcp', 'serve']}}}}, indent=2))
+print('MCP plugin installed.')
+print()
+print('10 tools available after activation:')
+print('  openrunner_search_sessions  — search AI sessions by keyword')
+print('  openrunner_search_papers    — search papers (title, abstract, sections)')
+print('  openrunner_search_research  — search research plans (SOTA, gaps, experiments)')
+print('  openrunner_search_runs      — search runs by name/tags')
+print('  openrunner_search_code      — search code artifacts')
+print('  openrunner_get_session      — get full session detail')
+print('  openrunner_list_runs        — list recent runs')
+print('  openrunner_get_run_metrics  — get metrics for a run')
+print('  openrunner_list_artifacts   — list project artifacts')
+print('  openrunner_log_note         — log a research note')
+print()
+print('TO ACTIVATE:')
+print('  1. Run /plugins in Claude Code')
+print('  2. Enable openrunner')
+print('  3. Restart this Claude Code session')
+"
+```
+
+Tell the user the activation steps clearly.
+"""
+
+
 def install_claude_code() -> list[str]:
     """Install commands in ~/.claude/commands/openrunner/."""
     cmd_dir = Path.home() / ".claude" / "commands" / "openrunner"
@@ -481,6 +531,7 @@ def install_claude_code() -> list[str]:
         "login.md": LOGIN_CMD,
         "code-upload.md": CODE_UPLOAD_CMD,
         "update.md": UPDATE_CMD,
+        "mcp.md": MCP_CMD,
     }
 
     files = []
@@ -505,6 +556,7 @@ def _install_all_commands_to_dir(cmd_dir: Path, prefix: str = "openrunner:") ->
         "login.md": LOGIN_CMD,
         "code-upload.md": CODE_UPLOAD_CMD,
         "update.md": UPDATE_CMD,
+        "mcp.md": MCP_CMD,
     }
 
     files = []

{openrunner_sdk-2.16.0 → openrunner_sdk-2.16.2}/openrunner/mcp_server.py

@@ -259,6 +259,42 @@ def create_server():
                     "required": [],
                 },
             ),
+            Tool(
+                name="openrunner_search_papers",
+                description="Search papers by title, abstract, or section content.",
+                inputSchema={
+                    "type": "object",
+                    "properties": {
+                        "query": {
+                            "type": "string",
+                            "description": "Search term (matches title, abstract, section content)",
+                        },
+                        "project": {
+                            "type": "string",
+                            "description": "Optional: project name",
+                        },
+                    },
+                    "required": ["query"],
+                },
+            ),
+            Tool(
+                name="openrunner_search_research",
+                description="Search research plans (literature, SOTA analysis, gaps, experiments, protocols).",
+                inputSchema={
+                    "type": "object",
+                    "properties": {
+                        "query": {
+                            "type": "string",
+                            "description": "Search term (matches domain, SOTA, gaps, experiments)",
+                        },
+                        "project": {
+                            "type": "string",
+                            "description": "Optional: project name",
+                        },
+                    },
+                    "required": ["query"],
+                },
+            ),
         ]
 
     @server.call_tool()
@@ -281,6 +317,10 @@ def create_server():
                 result = await _tool_list_artifacts(arguments)
             elif name == "openrunner_search_runs":
                 result = await _tool_search_runs(arguments)
+            elif name == "openrunner_search_papers":
+                result = await _tool_search_papers(arguments)
+            elif name == "openrunner_search_research":
+                result = await _tool_search_research(arguments)
             else:
                 result = f"Unknown tool: {name}"
         except RuntimeError as e:
@@ -606,6 +646,123 @@ async def _tool_search_runs(args: dict) -> str:
         return f"Error: {e}"
 
 
+async def _tool_search_papers(args: dict) -> str:
+    """Search papers by title, abstract, section content."""
+    query = args["query"].lower()
+    project = args.get("project")
+    settings = _load_settings()
+    if not project:
+        project = settings.get("project")
+    if not project:
+        return "Error: No project specified."
+
+    try:
+        data = await _api_get("/projects", params={})
+        projects = data if isinstance(data, list) else []
+        project_id = None
+        for p in projects:
+            org = p.get("org_name", "")
+            name = p.get("name", "")
+            slug = p.get("slug", "")
+            if f"{org}/{name}" == project or f"{org}/{slug}" == project or name == project:
+                project_id = p.get("id")
+                break
+        if not project_id:
+            return f"Project '{project}' not found"
+
+        papers_data = await _api_get(f"/projects/{project_id}/papers")
+        papers = papers_data.get("papers", []) if isinstance(papers_data, dict) else []
+
+        results = []
+        for paper in papers:
+            matches = []
+            if query in (paper.get("title") or "").lower():
+                matches.append("title")
+            if query in (paper.get("abstract") or "").lower():
+                matches.append("abstract")
+            for section in (paper.get("sections") or []):
+                if query in (section.get("content") or "").lower():
+                    matches.append(f"section:{section.get('title', '?')}")
+            if matches:
+                results.append((paper, matches))
+
+        if not results:
+            return f"No papers matching '{args['query']}'"
+
+        lines = [f"Found {len(results)} paper(s):\n"]
+        for paper, matches in results:
+            lines.append(f"- **{paper.get('title', 'Untitled')}** (status: {paper.get('status', '?')})")
+            lines.append(f"  Matched in: {', '.join(matches)}")
+            if paper.get("abstract"):
+                lines.append(f"  Abstract: {paper['abstract'][:150]}...")
+            lines.append(f"  ID: {paper.get('id')}")
+            lines.append("")
+        return "\n".join(lines)
+    except Exception as e:
+        return f"Error: {e}"
+
+
+async def _tool_search_research(args: dict) -> str:
+    """Search research plans (literature, SOTA, gaps, experiments, protocols)."""
+    query = args["query"].lower()
+    project = args.get("project")
+    settings = _load_settings()
+    if not project:
+        project = settings.get("project")
+    if not project:
+        return "Error: No project specified."
+
+    try:
+        data = await _api_get("/projects", params={})
+        projects = data if isinstance(data, list) else []
+        project_id = None
+        for p in projects:
+            org = p.get("org_name", "")
+            name = p.get("name", "")
+            slug = p.get("slug", "")
+            if f"{org}/{name}" == project or f"{org}/{slug}" == project or name == project:
+                project_id = p.get("id")
+                break
+        if not project_id:
+            return f"Project '{project}' not found"
+
+        research_data = await _api_get(f"/projects/{project_id}/research")
+        plans = research_data.get("plans", []) if isinstance(research_data, dict) else []
+
+        results = []
+        for plan in plans:
+            matches = []
+            if query in (plan.get("domain") or "").lower():
+                matches.append("domain")
+            if query in (plan.get("sota_analysis") or "").lower():
+                matches.append("sota")
+            if query in str(plan.get("gaps") or []).lower():
+                matches.append("gaps")
+            if query in str(plan.get("proposed_experiments") or []).lower():
+                matches.append("experiments")
+            if query in (plan.get("protocol") or "").lower():
+                matches.append("protocol")
+            if query in (plan.get("title") or "").lower():
+                matches.append("title")
+            if matches:
+                results.append((plan, matches))
+
+        if not results:
+            return f"No research plans matching '{args['query']}'"
+
+        lines = [f"Found {len(results)} research plan(s):\n"]
+        for plan, matches in results:
+            lines.append(f"- **{plan.get('title', 'Untitled')}** (domain: {plan.get('domain', '?')}, status: {plan.get('status', '?')})")
+            lines.append(f"  Matched in: {', '.join(matches)}")
+            if plan.get("sota_analysis"):
+                lines.append(f"  SOTA: {plan['sota_analysis'][:150]}...")
+            lines.append(f"  ID: {plan.get('id')}")
+            lines.append("")
+        return "\n".join(lines)
+    except Exception as e:
+        return f"Error: {e}"
+
+
 # ---------------------------------------------------------------------------
 # Server runner
 # ---------------------------------------------------------------------------

{openrunner_sdk-2.16.0 → openrunner_sdk-2.16.2}/pyproject.toml

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