codeprobe 0.3.8__tar.gz → 0.3.9__tar.gz

{codeprobe-0.3.8 → codeprobe-0.3.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codeprobe
-Version: 0.3.8
+Version: 0.3.9
 Summary: Benchmark AI coding agents against your own codebase. Mine real tasks from repo history, run agents, interpret results.
 Author: codeprobe contributors
 License-Expression: Apache-2.0

{codeprobe-0.3.8 → codeprobe-0.3.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codeprobe"
-version = "0.3.8"
+version = "0.3.9"
 description = "Benchmark AI coding agents against your own codebase. Mine real tasks from repo history, run agents, interpret results."
 readme = "README.md"
 license = "Apache-2.0"

{codeprobe-0.3.8 → codeprobe-0.3.9}/src/codeprobe/__init__.py

@@ -1,3 +1,3 @@
 """codeprobe — Benchmark AI coding agents against your own codebase."""
 
-__version__ = "0.3.8"
+__version__ = "0.3.9"

{codeprobe-0.3.8 → codeprobe-0.3.9}/src/codeprobe/cli/init_cmd.py

@@ -220,51 +220,64 @@ def _extract_sourcegraph_mcp(
     return None
 
 
+def _load_discovered_config(path: Path) -> dict | None:
+    """Load a discovered MCP config file and return its mcpServers dict."""
+    import json
+
+    try:
+        data = json.loads(path.read_text(encoding="utf-8"))
+    except (json.JSONDecodeError, OSError):
+        return None
+    servers = data.get("mcpServers", {})
+    if servers:
+        return {"mcpServers": servers}
+    return None
+
+
 def _goal_mcp(agents: list[str], name: str) -> _Result:
     """Goal 1: MCP comparison prompts."""
     agent = _prompt_agent(agents)
     model = _prompt_model()
 
-    # Check if Sourcegraph is available in discovered MCP configs
     discovered = discover_mcp_configs()
 
-    if _detect_sourcegraph_in_mcp(discovered):
+    # Auto-detect: if any MCP configs are discovered, let the user pick one
+    if discovered:
+        click.echo()
+        click.echo("Discovered MCP configurations:")
+        for i, (p, servers) in enumerate(discovered, 1):
+            server_list = ", ".join(servers)
+            click.echo(f"  {i}. {p}  ({server_list})")
+        manual_idx = len(discovered) + 1
+        click.echo(f"  {manual_idx}. Enter a Sourcegraph token manually")
         click.echo()
-        click.echo("Detected Sourcegraph MCP server in your configuration.")
-        if click.confirm("Use this Sourcegraph config?", default=True):
-            sg_config = _extract_sourcegraph_mcp(discovered)
-            if sg_config:
+
+        choice = click.prompt(
+            "Select MCP config",
+            type=click.IntRange(1, manual_idx),
+            default=1,
+        )
+        if choice <= len(discovered):
+            path = discovered[choice - 1][0]
+            mcp_config = _load_discovered_config(path)
+            if mcp_config:
+                click.echo(f"  Using {path}")
                 return ask_mcp_comparison(
                     experiment_name=name,
                     agent=agent,
                     model=model,
-                    mcp_config=sg_config,
+                    mcp_config=mcp_config,
                 )
 
-    click.echo()
-    click.echo("MCP server options:")
-    click.echo("  1. Use a Sourcegraph access token (PAT)")
-    click.echo("  2. Use an existing MCP config file")
-    choice = click.prompt("Choose", type=click.IntRange(1, 2), default=1)
-
-    if choice == 1:
-        token = _prompt_sourcegraph_token()
-        sg_url = _prompt_sourcegraph_url()
-        return ask_mcp_comparison(
-            experiment_name=name,
-            agent=agent,
-            model=model,
-            sourcegraph_token=token,
-            sourcegraph_url=sg_url,
-        )
-
-    # Fall back to generic MCP config path
-    mcp_path = _prompt_mcp_config()
+    # No discovered configs or user chose manual entry
+    token = _prompt_sourcegraph_token()
+    sg_url = _prompt_sourcegraph_url()
     return ask_mcp_comparison(
         experiment_name=name,
         agent=agent,
         model=model,
-        mcp_config_path=mcp_path,
+        sourcegraph_token=token,
+        sourcegraph_url=sg_url,
     )
 
 

{codeprobe-0.3.8 → codeprobe-0.3.9}/src/codeprobe.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codeprobe
-Version: 0.3.8
+Version: 0.3.9
 Summary: Benchmark AI coding agents against your own codebase. Mine real tasks from repo history, run agents, interpret results.
 Author: codeprobe contributors
 License-Expression: Apache-2.0

{codeprobe-0.3.8 → codeprobe-0.3.9}/tests/test_init_wizard.py

@@ -465,16 +465,22 @@ class TestInitCliIntegration:
     def test_goal1_mcp_flow(
         self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
     ) -> None:
+        """When a discovered config exists, selecting it skips token prompt."""
         mcp_file = tmp_path / "mcp.json"
-        mcp_file.write_text(json.dumps({"mcpServers": {}}))
+        mcp_file.write_text(
+            json.dumps({"mcpServers": {"my-server": {"type": "stdio", "cmd": "echo"}}})
+        )
 
-        # Patch discovery so the prompt falls through to manual path entry
-        monkeypatch.setattr("codeprobe.cli.init_cmd.discover_mcp_configs", lambda: [])
+        # Patch discovery to return the test config file
+        monkeypatch.setattr(
+            "codeprobe.cli.init_cmd.discover_mcp_configs",
+            lambda: [(mcp_file, ["my-server"])],
+        )
 
         runner = CliRunner()
-        # Inputs: goal=1, experiment name (enter=default), agent (enter=default),
-        # model (enter=skip), choose=2 (MCP config file), mcp config path
-        input_text = f"1\n\nclaude\n\n2\n{mcp_file}\n"
+        # Inputs: goal=1, name=default, agent=claude, model=skip,
+        # select=1 (the discovered config)
+        input_text = "1\n\nclaude\n\n1\n"
         result = runner.invoke(main, ["init", str(tmp_path)], input=input_text)
         assert result.exit_code == 0, result.output
         assert not (tmp_path / ".evalrc.yaml").exists()