memgraph-context-graph 0.7.1__tar.gz → 0.7.2__tar.gz

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: memgraph-context-graph
-Version: 0.7.1
+Version: 0.7.2
 Summary: Connect agent SDKs to context-graph components (actions-graph, skills-graph, etc.)
 License: MIT
 License-File: LICENSE

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/docs/command-hooks.md

@@ -205,7 +205,7 @@ session end, so adapters emit two different events:
 | Runtime | Turn end | Session end |
 |---|---|---|
 | Claude Code | `Stop` (records `last_assistant_message`) | `SessionEnd` |
-| Codex | `Stop` (records `last_assistant_message`) | none |
+| Codex | `Stop` (records `last_assistant_message`) | `SessionEnd`, only when Codex ends the session; with its shared background server `/exit` just detaches |
 | Copilot CLI | `agentStop` | `sessionEnd` |
 | Cursor | `stop` | `sessionEnd` |
 | OpenCode | `session.execution.succeeded` / `.failed` | `session.deleted` |

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "memgraph-context-graph"
-version = "0.7.1"
+version = "0.7.2"
 description = "Connect agent SDKs to context-graph components (actions-graph, skills-graph, etc.)"
 readme = "README.md"
 license = { text = "MIT" }

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/src/memgraph_context_graph/adapters/codex.py

@@ -11,6 +11,7 @@ from memgraph_context_graph.adapters._spec import (
     SpecAdapter,
     SpecPlugin,
     pre_tool_use_context,
+    session_end,
     session_start,
     tool_end,
     tool_start,
@@ -40,8 +41,7 @@ def _permission(context: EventContext) -> MessageEvent:
 
 
 def _turn_end(context: EventContext) -> list[Event]:
-    # Codex has no session-end hook, and Stop fires after every turn, so it is
-    # only ever a turn end.
+    # Stop fires after every turn; SessionEnd, registered separately, ends the session.
     return turn_end(context, reply=context.value("reply"))
 
 
@@ -50,7 +50,7 @@ SPEC = RuntimeSpec(
     name="codex",
     source_sdk="codex",
     event_key="hook_event_name",
-    hooks=("SessionStart", "UserPromptSubmit", "PreToolUse", "PostToolUse", "PermissionRequest", "Stop"),
+    hooks=("SessionStart", "UserPromptSubmit", "PreToolUse", "PostToolUse", "PermissionRequest", "Stop", "SessionEnd"),
     rules={
         "SessionStart": session_start,
         "UserPromptSubmit": _user_prompt,
@@ -58,6 +58,7 @@ SPEC = RuntimeSpec(
         "PostToolUse": tool_end,
         "PermissionRequest": _permission,
         "Stop": _turn_end,
+        "SessionEnd": session_end,
     },
     metadata_keys=(
         "cwd",

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/src/memgraph_context_graph/memory_backend.py

@@ -117,6 +117,7 @@ def use_native_memory(plugin: RuntimeCLIPlugin, *, home: Path | None = None) ->
     lines.append(
         f"Set memory.backend = {MEMORY_BACKEND_NATIVE}; {plugin.name}'s own memory is back from its next session."
     )
+    lines.append("Memories saved while Context Graph was your memory stay in Memgraph; they aren't copied back.")
     return lines
 
 

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/src/memgraph_context_graph/setup/engine.py

@@ -269,7 +269,11 @@ def _agent(agent: agents_mod.Agent, choices: Choices, log: Callable[[str], None]
     fix = f"memgraph-context-graph setup {agent.runtime}"
     if agent.wiring == "plugin":
         try:
-            return Outcome(agent.label, "ok", agents_mod.wire_plugin(agent, log))
+            summary = agents_mod.wire_plugin(agent, log)
+            if agent.runtime == "codex":
+                # Codex skips an untrusted plugin's hooks without a word, and `codex exec` never asks.
+                summary += "; Codex records nothing until you trust its hooks: open codex and accept them (/hooks)"
+            return Outcome(agent.label, "ok", summary)
         except agents_mod.WiringError as exc:
             return Outcome(agent.label, "fail", str(exc), fix=fix)
     if not choices.project_dirs:

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/src/memgraph_context_graph/setup/memgraph.py

@@ -180,6 +180,11 @@ def _restart(
                 f"See `docker logs {name}`. Setup never removes it, because it holds your graph.",
             )
     port = _published_port(docker, name)
+    if port is None and (current := _inspect(docker, name)) is not None and _lost_its_network(current):
+        log(f"The container {name} lost its network, so its port isn't published; reattaching it")
+        _run([docker, "network", "connect", "bridge", name])
+        _run([docker, "restart", name])
+        port = _published_port(docker, name)
     if port is None:
         raise ProvisionError(
             f"The container {name} doesn't publish Memgraph's Bolt port 7687",
@@ -191,6 +196,16 @@ def _restart(
     return Memgraph(url, version, how, container=name)
 
 
+def _lost_its_network(info: dict) -> bool:
+    """Whether a container asked for Bolt's port but is attached to no network, so Docker can't publish it.
+
+    Docker leaves a container in this state after its network is removed underneath it; the binding is still
+    in its config, and reattaching it (then a restart) makes Docker apply it. Its data is never touched.
+    """
+    bindings = info["HostConfig"].get("PortBindings") or {}
+    return "7687/tcp" in bindings and not info["NetworkSettings"].get("Networks")
+
+
 def _start(
     docker: str,
     name: str,

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/src/memgraph_context_graph/setup/tool.py

@@ -51,7 +51,8 @@ def uv_tool_environment() -> bool:
     uv = shutil.which("uv")
     if uv is None:
         return False
-    result = subprocess.run([uv, "tool", "dir"], capture_output=True, text=True, check=False)
+    # --color never: FORCE_COLOR makes uv colour even captured output, and the path would no longer compare equal.
+    result = subprocess.run([uv, "--color", "never", "tool", "dir"], capture_output=True, text=True, check=False)
     if result.returncode != 0:
         return False
     tool_dir = Path(result.stdout.strip()) / "memgraph-context-graph"
@@ -70,12 +71,12 @@ def install(*, viewer: bool, log: Callable[[str], None]) -> None:
     uv = shutil.which("uv")
     if uv is None:
         raise ToolError("uv isn't on PATH")
-    argv = [uv, "tool", "install", BASE]
+    argv = [uv, "--color", "never", "tool", "install", BASE]
     for requirement in CONNECTORS.values():
         argv += ["--with", requirement]
     if viewer or installed(VIEWER[0]):
         argv += ["--with", VIEWER[1]]
-    log(" ".join(["uv", *argv[1:]]))
+    log(" ".join(["uv", *argv[3:]]))
     process = subprocess.Popen(argv, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
     last = ""
     assert process.stdout is not None

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/tests/test_codex_adapter.py

@@ -149,6 +149,25 @@ def test_stop_ends_the_turn_with_the_reply_and_keeps_the_session_open():
     assert PLUGIN.response_for_payload(payload) == {"continue": True}
 
 
+def test_session_end_ends_the_session():
+    link = AgentLink()
+    rec = _RecordingConnector()
+    link.add_connector(rec)
+
+    # The payload Codex sends when a session really ends (no shared daemon holding it open).
+    payload = {
+        "session_id": "s1",
+        "transcript_path": "/home/u/.codex/sessions/rollout-s1.jsonl",
+        "cwd": "/work/billing",
+        "hook_event_name": "SessionEnd",
+        "reason": "other",
+    }
+    CodexHooksAdapter(link).handle_payload(payload)
+
+    (session_end,) = rec.events
+    assert session_end.event_type == EventType.SESSION_END
+
+
 def test_stop_records_the_turns_reply_before_the_turn_end():
     """The reply is the answer recall needs; the turn end that follows spawns its embedding."""
     link = AgentLink()

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/tests/test_setup.py

@@ -419,3 +419,34 @@ def test_a_second_run_starts_from_what_is_set_up(fake_agents, setup_home, projec
     choices = ask(Prompts(term), engine.default_choices(env, cwd=project), env, preselect=False)
     assert choices is not None
     assert choices.memory_backend == "native"
+
+
+def test_the_uv_tool_is_recognised_even_when_the_shell_forces_colour(monkeypatch):
+    """FORCE_COLOR makes uv colour what it prints into a pipe, which broke comparing its tool directory."""
+    import shutil
+    import subprocess
+    import sys
+    from pathlib import Path
+
+    from memgraph_context_graph.setup import tool
+
+    uv = shutil.which("uv")
+    if uv is None:
+        pytest.skip("needs uv")
+    tool_dir = subprocess.run([uv, "--color", "never", "tool", "dir"], capture_output=True, text=True, check=True)
+    monkeypatch.setenv("FORCE_COLOR", "3")
+    monkeypatch.setattr(sys, "prefix", str(Path(tool_dir.stdout.strip()) / "memgraph-context-graph"))
+
+    assert tool.uv_tool_environment()
+
+
+def test_codex_is_reported_with_the_trust_it_still_needs_and_claude_code_is_not(fake_agents, setup_home, project):
+    env = engine.look_around(cwd=project)
+    choices = engine.default_choices(env, cwd=project)
+
+    codex = engine._agent(BY_RUNTIME["codex"], choices, lambda _: None)
+    claude = engine._agent(BY_RUNTIME["claude-code"], choices, lambda _: None)
+
+    assert codex.status == "ok"
+    assert "records nothing until you trust its hooks" in codex.summary
+    assert "trust" not in claude.summary

{memgraph_context_graph-0.7.1 → memgraph_context_graph-0.7.2}/tests/test_setup_docker.py

@@ -54,7 +54,7 @@ def _query(url: str, cypher: LiteralString) -> list[dict]:
 def names():
     """Container names for this run, and the cleanup of whatever setup made under them."""
     prefix = f"cg-setup-test-{uuid.uuid4().hex[:8]}"
-    made = {"own": f"{prefix}-own", "demo": f"{prefix}-demo", "other": f"{prefix}-other"}
+    made = {"own": f"{prefix}-own", "demo": f"{prefix}-demo", "other": f"{prefix}-other", "lost": f"{prefix}-lost"}
     yield made
     for name in made.values():
         _docker("rm", "-f", name)
@@ -123,6 +123,22 @@ def test_our_stopped_container_comes_back_on_its_port_with_its_data(names, demo)
     assert _query(url, "MATCH (n:SetupProbe) RETURN count(n) AS n") == [{"n": 1}]
 
 
+def test_a_container_that_lost_its_network_is_reattached_and_published(names):
+    port = _free_port()
+    volume = f"{names['lost']}-data:/var/lib/memgraph"
+    run = ["run", "-d", "--name", names["lost"], "-v", volume, "-p", f"127.0.0.1:{port}:7687", mg.IMAGE]
+    assert _docker(*run).returncode == 0
+    # Docker keeps the binding in the config but publishes nothing once the container has no network.
+    assert _docker("network", "disconnect", "bridge", names["lost"]).returncode == 0
+    assert _docker("restart", names["lost"]).returncode == 0
+    assert _docker("port", names["lost"], "7687/tcp").stdout.strip() == ""
+
+    found = mg.find_or_start(configured_url=None, default_port=port, container=names["lost"], log=lambda _: None)
+
+    assert found.url == mg.local_url(port)
+    assert _query(found.url, "RETURN 1 AS ok") == [{"ok": 1}]
+
+
 def test_setup_then_uninstall_end_to_end(names, demo, fake_agents, setup_home, tmp_path, capsys):
     """The plain front end, the way the shim runs it without a terminal: every step, then a re-run, then undo."""
     from memgraph_context_graph.adapters._identity import load_config