guard-cli 0.1.3__tar.gz → 0.1.5__tar.gz

{guard_cli-0.1.3 → guard_cli-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: guard-cli
-Version: 0.1.3
+Version: 0.1.5
 Summary: Graph-powered safety layer for AI coding agents built on FalkorDB
 License-Expression: MIT
 Project-URL: Homepage, https://github.com/thisisvaishnav/guard

{guard_cli-0.1.3 → guard_cli-0.1.5}/guard_cli/api.py

@@ -396,12 +396,13 @@ def run(
 def covering_tests(
     function_id: str,
     min_confidence: float = 0.0,
-    graph_name: str = "guard",
+    graph_name: str | None = None,
+    repo_path: str | Path | None = None,
     strict: bool = False,
 ) -> list[dict[str, Any]]:
     """Find tests covering a given function, ordered by confidence."""
     try:
-        g = db.get_graph(graph_name)
+        g = db.get_graph(graph_name, repo_path=repo_path)
         return db.get_covering_tests(g, function_id, min_confidence=min_confidence)
     except Exception as exc:
         if strict:
@@ -414,12 +415,13 @@ def callers(
     function_id: str,
     hops: int = 4,
     min_confidence: float = 0.0,
-    graph_name: str = "guard",
+    graph_name: str | None = None,
+    repo_path: str | Path | None = None,
     strict: bool = False,
 ) -> list[dict[str, Any]]:
     """Traverse inbound CALLS edges up to N hops filtered by min_confidence."""
     try:
-        g = db.get_graph(graph_name)
+        g = db.get_graph(graph_name, repo_path=repo_path)
         return db.get_callers(g, function_id, hops=hops, min_confidence=min_confidence)
     except Exception as exc:
         if strict:
@@ -432,12 +434,13 @@ def report_miss(
     function_id: str,
     test_id: str,
     new_confidence: float = 1.0,
-    graph_name: str = "guard",
+    graph_name: str | None = None,
+    repo_path: str | Path | None = None,
     strict: bool = False,
 ) -> dict[str, Any]:
     """Records a missed failure by repairing the graph with a miss_repair edge."""
     try:
-        g = db.get_graph(graph_name)
+        g = db.get_graph(graph_name, repo_path=repo_path)
         return feedback.record_miss(
             g,
             function_id=function_id,

{guard_cli-0.1.3 → guard_cli-0.1.5}/guard_cli/cli.py

@@ -239,7 +239,7 @@ def main(argv=None) -> int:
 
     p = argparse.ArgumentParser(prog="guard")
     p.add_argument("-v", "--version", action="version", version=f"guard {guard_cli.__version__}")
-    p.add_argument("--graph", default="guard")
+    p.add_argument("--graph", default=None, help="FalkorDB graph name (defaults to repository graph from .guard/config.toml)")
     sub = p.add_subparsers(dest="cmd", required=True)
 
     ini = sub.add_parser("init", help="Initialize Guard for repository")

{guard_cli-0.1.3 → guard_cli-0.1.5}/guard_cli/db.py

@@ -47,7 +47,9 @@ def get_graph(
     if backend is None:
         backend = get_backend(repo_path=repo_path)
     if name is None:
-        name = os.getenv("GUARD_GRAPH", GRAPH_NAME)
+        from guard_cli.backend import load_config
+        cfg = load_config(repo_path)
+        name = cfg.get("backend", {}).get("graph_name") or os.getenv("GUARD_GRAPH", GRAPH_NAME)
     return backend.get_graph(name)
 
 

{guard_cli-0.1.3 → guard_cli-0.1.5}/guard_cli/hook.py

@@ -530,16 +530,11 @@ def run_stop_hook(
         if is_antigravity:
             resp: dict[str, Any] = {"decision": "stop", "reason": reason or ""}
             payload = json.dumps(resp) + "\n"
-            if saved_stdout is not None:
+            target_out = real_out if real_out is not None else saved_stdout
+            if target_out is not None:
                 try:
-                    saved_stdout.write(payload)
-                    saved_stdout.flush()
-                except Exception:
-                    pass
-            if real_out is not None and real_out is not saved_stdout:
-                try:
-                    real_out.write(payload)
-                    real_out.flush()
+                    target_out.write(payload)
+                    target_out.flush()
                 except Exception:
                     pass
         return 0
@@ -548,16 +543,11 @@ def run_stop_hook(
         if is_antigravity:
             resp: dict[str, Any] = {"decision": "continue", "reason": reason_text}
             payload = json.dumps(resp) + "\n"
-            if saved_stdout is not None:
-                try:
-                    saved_stdout.write(payload)
-                    saved_stdout.flush()
-                except Exception:
-                    pass
-            if real_out is not None and real_out is not saved_stdout:
+            target_out = real_out if real_out is not None else saved_stdout
+            if target_out is not None:
                 try:
-                    real_out.write(payload)
-                    real_out.flush()
+                    target_out.write(payload)
+                    target_out.flush()
                 except Exception:
                     pass
             sys.stderr.write(reason_text)

{guard_cli-0.1.3 → guard_cli-0.1.5}/guard_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: guard-cli
-Version: 0.1.3
+Version: 0.1.5
 Summary: Graph-powered safety layer for AI coding agents built on FalkorDB
 License-Expression: MIT
 Project-URL: Homepage, https://github.com/thisisvaishnav/guard

{guard_cli-0.1.3 → guard_cli-0.1.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "guard-cli"
-version = "0.1.3"
+version = "0.1.5"
 description = "Graph-powered safety layer for AI coding agents built on FalkorDB"
 readme = "README.md"
 license = "MIT"

{guard_cli-0.1.3 → guard_cli-0.1.5}/tests/test_api_and_hook.py

@@ -58,10 +58,16 @@ def seeded_guard_graph():
     """Seeds a dedicated graph so check() exercises the 'select' path without
     depending on an externally pre-populated 'guard' graph (which may lack
     COVERED_BY evidence and silently fall back to run-all)."""
-    g = db.get_graph("guard_api_check_test")
-    db.reset_graph(g)
-    g = db.get_graph("guard_api_check_test")
-    db.ensure_schema(g)
+    backend = db.get_backend()
+    if not backend.is_healthy():
+        pytest.skip("FalkorDB server is not reachable")
+    try:
+        g = db.get_graph("guard_api_check_test", backend=backend)
+        db.reset_graph(g)
+        g = db.get_graph("guard_api_check_test", backend=backend)
+        db.ensure_schema(g)
+    except Exception as exc:
+        pytest.skip(f"FalkorDB server error: {exc}")
     db.write_static(
         g,
         StaticResult(
@@ -74,7 +80,10 @@ def seeded_guard_graph():
     )
     db.write_coverage(g, {"tests/test_mod.py::test_calc": ["pkg/mod.py::calc"]}, confidence=0.9)
     yield g
-    db.reset_graph(g)
+    try:
+        db.reset_graph(g)
+    except Exception:
+        pass
 
 
 def test_api_check_schema_version_and_keys(temp_repo, seeded_guard_graph):
@@ -452,3 +461,22 @@ def test_cli_hook_subcommands(temp_repo, capsys):
     ret = main(["hook", "uninstall", "--repo", str(temp_repo), "--antigravity"])
     assert ret == 0
 
+
+def test_cli_default_graph_resolution(temp_repo):
+    """Verify that when graph name is omitted, db and api resolve graph_name from .guard/config.toml."""
+    from guard_cli.backend import save_config
+    save_config(temp_repo, {"backend": {"type": "server", "graph_name": "custom_repo_graph"}})
+
+    with patch("guard_cli.db.get_backend") as mock_get_backend:
+        mock_backend = MagicMock()
+        mock_get_backend.return_value = mock_backend
+        db.get_graph(name=None, repo_path=temp_repo)
+        mock_backend.get_graph.assert_called_with("custom_repo_graph")
+
+    with patch("guard_cli.db.get_backend") as mock_get_backend, patch("guard_cli.db.get_covering_tests") as mock_get_cov:
+        mock_backend = MagicMock()
+        mock_get_backend.return_value = mock_backend
+        mock_get_cov.return_value = []
+        api.covering_tests("pkg/mod.py::calc", repo_path=temp_repo)
+        mock_backend.get_graph.assert_called_with("custom_repo_graph")
+

{guard_cli-0.1.3 → guard_cli-0.1.5}/tests/test_coverage.py

@@ -51,12 +51,21 @@ def test_compute_flakiness():
 
 @pytest.fixture
 def clean_test_graph():
-    g = db.get_graph("guard_cov_test")
-    db.reset_graph(g)
-    g = db.get_graph("guard_cov_test")
-    db.ensure_schema(g)
+    backend = db.get_backend()
+    if not backend.is_healthy():
+        pytest.skip("FalkorDB server is not reachable")
+    try:
+        g = db.get_graph("guard_cov_test", backend=backend)
+        db.reset_graph(g)
+        g = db.get_graph("guard_cov_test", backend=backend)
+        db.ensure_schema(g)
+    except Exception as exc:
+        pytest.skip(f"FalkorDB server error: {exc}")
     yield g
-    db.reset_graph(g)
+    try:
+        db.reset_graph(g)
+    except Exception:
+        pass
 
 
 def test_coverage_write_to_graph(clean_test_graph):

{guard_cli-0.1.3 → guard_cli-0.1.5}/tests/test_db.py

@@ -6,12 +6,21 @@ from guard.ingest.static import StaticResult, FuncDef
 @pytest.fixture
 def test_graph():
     """Provides a fresh test graph in FalkorDB."""
-    g = db.get_graph("guard_test")
-    db.reset_graph(g)
-    g = db.get_graph("guard_test")
-    db.ensure_schema(g)
+    backend = db.get_backend()
+    if not backend.is_healthy():
+        pytest.skip("FalkorDB server is not reachable")
+    try:
+        g = db.get_graph("guard_test", backend=backend)
+        db.reset_graph(g)
+        g = db.get_graph("guard_test", backend=backend)
+        db.ensure_schema(g)
+    except Exception as exc:
+        pytest.skip(f"FalkorDB server error: {exc}")
     yield g
-    db.reset_graph(g)
+    try:
+        db.reset_graph(g)
+    except Exception:
+        pass
 
 
 def test_ensure_schema_creates_indexes(test_graph):

{guard_cli-0.1.3 → guard_cli-0.1.5}/tests/test_feedback.py

@@ -7,12 +7,21 @@ from guard.ingest.static import StaticResult, FuncDef
 
 @pytest.fixture
 def feedback_test_graph():
-    g = db.get_graph("guard_feedback_test")
-    db.reset_graph(g)
-    g = db.get_graph("guard_feedback_test")
-    db.ensure_schema(g)
+    backend = db.get_backend()
+    if not backend.is_healthy():
+        pytest.skip("FalkorDB server is not reachable")
+    try:
+        g = db.get_graph("guard_feedback_test", backend=backend)
+        db.reset_graph(g)
+        g = db.get_graph("guard_feedback_test", backend=backend)
+        db.ensure_schema(g)
+    except Exception as exc:
+        pytest.skip(f"FalkorDB server error: {exc}")
     yield g
-    db.reset_graph(g)
+    try:
+        db.reset_graph(g)
+    except Exception:
+        pass
 
 
 def test_record_miss_creates_repair_edge(feedback_test_graph):

{guard_cli-0.1.3 → guard_cli-0.1.5}/tests/test_thin_slice.py

@@ -52,12 +52,21 @@ def test_thin_slice_decision_fallback_on_test_files_touched():
 
 @pytest.fixture
 def slice_test_graph():
-    g = db.get_graph("guard_slice_test")
-    db.reset_graph(g)
-    g = db.get_graph("guard_slice_test")
-    db.ensure_schema(g)
+    backend = db.get_backend()
+    if not backend.is_healthy():
+        pytest.skip("FalkorDB server is not reachable")
+    try:
+        g = db.get_graph("guard_slice_test", backend=backend)
+        db.reset_graph(g)
+        g = db.get_graph("guard_slice_test", backend=backend)
+        db.ensure_schema(g)
+    except Exception as exc:
+        pytest.skip(f"FalkorDB server error: {exc}")
     yield g
-    db.reset_graph(g)
+    try:
+        db.reset_graph(g)
+    except Exception:
+        pass
 
 
 def test_thin_slice_selects_covering_tests(slice_test_graph):