cal/cognitive-memory
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: per-project graph routing via COGNITIVE_MEMORY_GRAPH env var

Browse Source 
Allows projects to automatically route all memory operations to a
specific graph without passing graph= on every MCP call. Set the env
var in a project's .claude/settings.json to target a named graph.

Resolution order: explicit graph param > env var > "default".

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Cal Corum 2026-02-28 22:48:19 -06:00
parent 1fefb1d5e2
commit 082ef9c8a7
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mcp_server.py
View File

@ -44,10 +44,17 @@ _clients: Dict[str, CognitiveMemoryClient] = {}
def get_client(graph: Optional[str] = None) -> CognitiveMemoryClient:
"""Get or create a CognitiveMemoryClient for the given graph."""
key = graph or "default"
"""Get or create a CognitiveMemoryClient for the given graph.
Resolution order: explicit graph param > COGNITIVE_MEMORY_GRAPH env var > "default".
Set COGNITIVE_MEMORY_GRAPH per-project via Claude Code's .claude/settings.json env overrides.
"""
import os
effective_graph = graph or os.environ.get("COGNITIVE_MEMORY_GRAPH") or None
key = effective_graph or "default"
if key not in _clients:
path = resolve_graph_path(graph)
path = resolve_graph_path(effective_graph)
_clients[key] = CognitiveMemoryClient(memory_dir=path)
return _clients[key]