graphsnip 0.2.0__tar.gz → 0.2.1__tar.gz

{graphsnip-0.2.0/src/graphsnip.egg-info → graphsnip-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphsnip
-Version: 0.2.0
+Version: 0.2.1
 Summary: Token-saving layer on top of Graphify: symbol-level code slicing, context packs, a savings ledger with an honest baseline, hooks and MCP for Claude Code, Copilot and Codex.
 License-Expression: Apache-2.0
 Project-URL: Repository, https://github.com/naubadnithin7/codegraf

{graphsnip-0.2.0 → graphsnip-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "graphsnip"
-version = "0.2.0"
+version = "0.2.1"
 description = "Token-saving layer on top of Graphify: symbol-level code slicing, context packs, a savings ledger with an honest baseline, hooks and MCP for Claude Code, Copilot and Codex."
 readme = "README.md"
 license = "Apache-2.0"

{graphsnip-0.2.0 → graphsnip-0.2.1}/src/graphsnip/__init__.py

@@ -1,3 +1,3 @@
 """graphsnip: a token-saving layer on top of Graphify."""
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"

{graphsnip-0.2.0 → graphsnip-0.2.1}/src/graphsnip/cli.py

@@ -444,6 +444,25 @@ def cmd_doctor(a) -> int:
                       f"Re-run `graphsnip install` (without --portable) to write the full path.")
             elif not os.path.exists(c):
                 print(f"! {label}: server command {c} does not exist; re-run `graphsnip install`.")
+    if not getattr(a, "no_probe", False):
+        from graphsnip.probe import probe_mcp
+
+        for label, f, key in (("claude", mcp_claude, "mcpServers"), ("copilot", p.root / ".vscode" / "mcp.json", "servers")):
+            try:
+                ent = json.loads(f.read_text(encoding="utf-8")).get(key, {}).get("graphsnip")
+            except Exception:
+                ent = None
+            if not ent:
+                continue
+            r = probe_mcp(str(ent.get("command", "")), [str(x) for x in ent.get("args", [])], str(p.root))
+            if r["ok"]:
+                line(True, f"{label}: the MCP server starts and lists {len(r['tools'])} tools ({r['seconds']}s)")
+            else:
+                line(False, f"{label}: the MCP server does NOT work: {r['error']}", "fix the command in the config, then reload the app")
+                if r["stderr"]:
+                    print("    server said: " + r["stderr"].replace("\n", "\n    "))
+        print("  If every server above starts but the assistant still has no graphsnip tools, the app has not started or trusted it:")
+        print("  Claude Code: run /mcp and approve it. VS Code: Command Palette > 'MCP: List Servers' > start/trust graphsnip, then 'Configure Tools'.")
     hook_on = st.exists() and HOOK_MARK in st.read_text(encoding="utf-8", errors="ignore")
     print(f"{'✓' if hook_on else '·'} claude hook: {'installed' if hook_on else 'not installed'}  (mode: {p.config()['mode']})")
     print("\nAll good." if ok else "\nSome checks failed; see arrows above.")
@@ -550,7 +569,8 @@ def main(argv=None) -> int:
     sp = add("uninstall", cmd_uninstall, "remove everything install added")
     sp.add_argument("--clients", default="all")
     sp.add_argument("--codex-scope", choices=["project", "user"], default="project")
-    add("doctor", cmd_doctor, "check the setup")
+    sp = add("doctor", cmd_doctor, "check the setup (also starts the configured MCP servers to prove they work)")
+    sp.add_argument("--no-probe", action="store_true", help="skip starting the MCP servers")
 
     args = ap.parse_args(argv)
     if not getattr(args, "fn", None):

graphsnip-0.2.1/src/graphsnip/probe.py

@@ -0,0 +1,83 @@
+"""Start an MCP server exactly as an assistant would and check that it answers, so `doctor` can say why tools are missing."""
+from __future__ import annotations
+
+import json
+import queue
+import subprocess
+import threading
+import time
+
+
+def probe_mcp(command: str, args: list[str], cwd: str, timeout: float = 25.0) -> dict:
+    t0 = time.time()
+    try:
+        proc = subprocess.Popen([command, *args], cwd=cwd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
+                                text=True, bufsize=1)
+    except FileNotFoundError:
+        return {"ok": False, "error": f"command not found: {command}", "stderr": "", "tools": [], "seconds": 0.0}
+    except OSError as e:
+        return {"ok": False, "error": f"could not start {command}: {e}", "stderr": "", "tools": [], "seconds": 0.0}
+
+    out: queue.Queue = queue.Queue()
+    err_lines: list[str] = []
+
+    def pump(stream, sink):
+        for ln in stream:
+            sink(ln)
+
+    threading.Thread(target=pump, args=(proc.stdout, out.put), daemon=True).start()
+    threading.Thread(target=pump, args=(proc.stderr, err_lines.append), daemon=True).start()
+
+    def send(obj):
+        try:
+            proc.stdin.write(json.dumps(obj) + "\n")
+            proc.stdin.flush()
+        except OSError:
+            pass
+
+    def wait_for(mid):
+        end = t0 + timeout
+        while time.time() < end:
+            try:
+                ln = out.get(timeout=0.2)
+            except queue.Empty:
+                if proc.poll() is not None and out.empty():
+                    return None
+                continue
+            try:
+                msg = json.loads(ln)
+            except Exception:
+                continue
+            if isinstance(msg, dict) and msg.get("id") == mid:
+                return msg
+        return None
+
+    res = {"ok": False, "error": "", "stderr": "", "tools": [], "seconds": 0.0}
+    try:
+        send({"jsonrpc": "2.0", "id": 1, "method": "initialize",
+              "params": {"protocolVersion": "2025-03-26", "capabilities": {}, "clientInfo": {"name": "graphsnip-doctor", "version": "0"}}})
+        init = wait_for(1)
+        if init is None:
+            res["error"] = ("the server exited before answering" if proc.poll() is not None else f"no answer within {timeout:.0f}s")
+        else:
+            send({"jsonrpc": "2.0", "method": "notifications/initialized"})
+            send({"jsonrpc": "2.0", "id": 2, "method": "tools/list"})
+            tl = wait_for(2)
+            names = [t.get("name") for t in ((tl or {}).get("result") or {}).get("tools", [])]
+            if names:
+                res.update(ok=True, tools=names)
+            else:
+                res["error"] = "the server started but listed no tools"
+    finally:
+        try:
+            proc.stdin.close()
+        except OSError:
+            pass
+        try:
+            proc.terminate()
+            proc.wait(timeout=3)
+        except Exception:
+            proc.kill()
+    res["seconds"] = round(time.time() - t0, 1)
+    res["stderr"] = "".join(err_lines)[-800:].strip()
+    return res

{graphsnip-0.2.0 → graphsnip-0.2.1/src/graphsnip.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphsnip
-Version: 0.2.0
+Version: 0.2.1
 Summary: Token-saving layer on top of Graphify: symbol-level code slicing, context packs, a savings ledger with an honest baseline, hooks and MCP for Claude Code, Copilot and Codex.
 License-Expression: Apache-2.0
 Project-URL: Repository, https://github.com/naubadnithin7/codegraf

{graphsnip-0.2.0 → graphsnip-0.2.1}/src/graphsnip.egg-info/SOURCES.txt

@@ -14,6 +14,7 @@ src/graphsnip/hooks.py
 src/graphsnip/installer.py
 src/graphsnip/ledger.py
 src/graphsnip/mcp_server.py
+src/graphsnip/probe.py
 src/graphsnip/project.py
 src/graphsnip/py.typed
 src/graphsnip/retrieval.py

{graphsnip-0.2.0 → graphsnip-0.2.1}/tests/test_integrations.py

@@ -331,3 +331,33 @@ def test_copilot_config_uses_absolute_path_even_when_portable(tmp_path, monkeypa
     cl = json.loads((tmp_path / ".mcp.json").read_text())["mcpServers"]["graphsnip"]
     assert vs["command"] == "/opt/bin/graphsnip"
     assert cl["command"] == "graphsnip"
+
+
+def test_probe_reports_a_working_and_a_broken_server(tmp_path):
+    import sys
+
+    from graphsnip.probe import probe_mcp
+
+    (tmp_path / "a.py").write_text("def f():\n    return 1\n")
+    ok = probe_mcp(sys.executable, ["-m", "graphsnip", "serve", "--root", "."], str(tmp_path))
+    assert ok["ok"] and "gs_pack" in ok["tools"] and len(ok["tools"]) == 7
+    bad = probe_mcp("/definitely/not/here/graphsnip", ["serve"], str(tmp_path))
+    assert not bad["ok"] and "not found" in bad["error"]
+    crash = probe_mcp(sys.executable, ["-c", "import sys; sys.stderr.write('boom: ImportError x'); sys.exit(1)"], str(tmp_path), timeout=5)
+    assert not crash["ok"] and "boom" in crash["stderr"]
+
+
+def test_doctor_probes_configured_servers(tmp_path, capsys):
+    import json
+    import sys
+
+    from graphsnip import cli
+
+    (tmp_path / "a.py").write_text("def f():\n    return 1\n")
+    (tmp_path / ".mcp.json").write_text(json.dumps({"mcpServers": {"graphsnip": {"command": sys.executable, "args": ["-m", "graphsnip", "serve", "--root", "."]}}}))
+    (tmp_path / ".vscode").mkdir()
+    (tmp_path / ".vscode" / "mcp.json").write_text(json.dumps({"servers": {"graphsnip": {"type": "stdio", "command": "/nope/graphsnip", "args": ["serve"]}}}))
+    cli.main(["doctor", "--root", str(tmp_path)])
+    out = capsys.readouterr().out
+    assert "claude: the MCP server starts and lists 7 tools" in out
+    assert "copilot: the MCP server does NOT work" in out and "command not found" in out