Add --db-path option to setup CLI and install script for consistent DB paths v0.5.1

The OpenClaw extension was spawning the MCP server without passing
AIMEMORY_DB_PATH, causing it to create a separate DB in the OpenClaw
workspace directory instead of the project's memory_db. This led to
the live viewer and MCP server reading different databases.

- Add --db-path flag to aimemory-setup (openclaw/claude subcommands)
- Add --db-path flag to scripts/install_openclaw.sh
- Inject DB_PATH constant into OpenClaw extension template at install time
- Extension spawn env now passes AIMEMORY_DB_PATH with fallback chain
- Update README with custom DB path documentation

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: 2 people

README.md

@@ -73,6 +73,32 @@ aimemory-setup claude
 
 This injects memory usage instructions into your client's configuration files (`SOUL.md`/`TOOLS.md` for OpenClaw, `CLAUDE.md` for Claude Code). Re-run anytime to update.
 
+<details>
+<summary>Custom database path</summary>
+
+By default, memories are stored in `./memory_db` (resolved to an absolute path at install time). To use a custom location:
+
+```bash
+# OpenClaw — sets the DB path in the extension and mcporter config
+aimemory-setup openclaw --db-path /path/to/my/memory_db
+
+# Claude Code
+aimemory-setup claude --db-path /path/to/my/memory_db
+
+# Shell script (OpenClaw)
+bash scripts/install_openclaw.sh --db-path /path/to/my/memory_db
+```
+
+You can also set the `AIMEMORY_DB_PATH` environment variable, which all components respect:
+
+```bash
+export AIMEMORY_DB_PATH=/path/to/my/memory_db
+aimemory-setup openclaw   # picks up the env var automatically
+```
+
+All components (MCP server, live viewer, OpenClaw extension) will use the same absolute path, ensuring data consistency.
+</details>
+
 ### 3. Connect to OpenClaw
 
 ```bash
@@ -159,7 +185,10 @@ aimemory-mcp --with-live
 # Option 2: standalone server
 aimemory-live --port 8765
 
-# Option 3: via environment variable
+# Option 3: standalone with custom DB path
+aimemory-live --db-path /path/to/memory_db
+
+# Option 4: via environment variable
 AIMEMORY_LIVE=1 aimemory-mcp
 ```
 
@@ -198,7 +227,7 @@ All settings via environment variables:
 
 | Variable | Default | Description |
 |----------|---------|-------------|
-| `AIMEMORY_DB_PATH` | `./memory_db` | ChromaDB persistence directory |
+| `AIMEMORY_DB_PATH` | `./memory_db` | ChromaDB persistence directory (use absolute path to ensure all components share the same DB) |
 | `AIMEMORY_LANGUAGE` | `ko` | Language for pattern matching (`ko` / `en`) |
 | `AIMEMORY_EMBEDDING_MODEL` | `intfloat/multilingual-e5-small` | Sentence-transformer model |
 | `AIMEMORY_LOG_LEVEL` | `INFO` | Logging level |

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "long-term-memory"
-version = "0.5.0"
+version = "0.5.1"
 description = "Long-term memory system for AI assistants — persistent, searchable, self-organizing memory powered by semantic search, knowledge graphs, and reinforcement learning"
 requires-python = ">=3.11,<3.14"
 license = "MIT"

scripts/install_openclaw.sh

@@ -2,23 +2,35 @@
 # AIMemory MCP Server — OpenClaw 자동 설치 스크립트
 #
 # Usage:
-#   bash scripts/install_openclaw.sh          # 설치
-#   bash scripts/install_openclaw.sh --remove  # 제거
+#   bash scripts/install_openclaw.sh [--db-path /absolute/path/to/db]  # 설치
+#   bash scripts/install_openclaw.sh --remove                           # 제거
 
 set -euo pipefail
 
 PROJECT_DIR="$(cd "$(dirname "$0")/.." && pwd)"
 MCPORTER_CONFIG="${HOME}/.mcporter/mcporter.json"
 TOOLS_MD="${HOME}/.openclaw/workspace/TOOLS.md"
 SERVER_NAME="aimemory"
+DB_PATH=""
 
 TOOLS_BLOCK_START="## AIMemory"
 TOOLS_BLOCK_END="<!-- \/aimemory -->"
 
 TOOLS_CONTENT_FILE="${PROJECT_DIR}/scripts/tools_content.md"
 
 install() {
+    # DB 경로 결정: --db-path > AIMEMORY_DB_PATH env > 프로젝트/memory_db
+    if [ -z "$DB_PATH" ]; then
+        DB_PATH="${AIMEMORY_DB_PATH:-${PROJECT_DIR}/memory_db}"
+    fi
+    # 상대경로면 절대경로로 변환
+    case "$DB_PATH" in
+        /*) ;; # 이미 절대경로
+        *)  DB_PATH="$(cd "$(dirname "$DB_PATH")" 2>/dev/null && pwd)/$(basename "$DB_PATH")" ;;
+    esac
+
     echo "🧠 AIMemory MCP 서버 설치 중..."
+    echo "   DB 경로: ${DB_PATH}"
 
     # 1. 의존성 확인
     if ! command -v mcporter &>/dev/null; then
@@ -57,7 +69,7 @@ install() {
         --arg python \
         --arg -m \
         --arg aimemory.mcp \
-        --env "AIMEMORY_DB_PATH=${PROJECT_DIR}/memory_db" \
+        --env "AIMEMORY_DB_PATH=${DB_PATH}" \
         --env "AIMEMORY_LANGUAGE=ko" \
         --env "AIMEMORY_EMBEDDING_MODEL=intfloat/multilingual-e5-small" \
         --env "AIMEMORY_LOG_LEVEL=INFO" \
@@ -115,11 +127,18 @@ remove() {
     echo "✅ 제거 완료"
 }
 
-case "${1:-}" in
-    --remove|--uninstall|-r)
-        remove
-        ;;
-    *)
-        install
-        ;;
-esac
+ACTION="install"
+while [ $# -gt 0 ]; do
+    case "$1" in
+        --remove|--uninstall|-r)
+            ACTION="remove"; shift ;;
+        --db-path)
+            DB_PATH="$2"; shift 2 ;;
+        --db-path=*)
+            DB_PATH="${1#*=}"; shift ;;
+        *)
+            echo "Unknown option: $1"; exit 1 ;;
+    esac
+done
+
+"$ACTION"

src/aimemory/setup_client.py

@@ -69,7 +69,21 @@ def _find_aimemory_mcp_command() -> str:
     return "aimemory-mcp"
 
 
-def _install_openclaw_extension(openclaw_dir: Path) -> None:
+def _resolve_db_path(db_path: str | None) -> str:
+    """Resolve the database path to an absolute path.
+
+    Priority:
+      1. Explicitly provided ``db_path`` argument
+      2. ``AIMEMORY_DB_PATH`` environment variable
+      3. ``./memory_db`` relative to cwd (resolved to absolute)
+    """
+    import os
+
+    path = db_path or os.environ.get("AIMEMORY_DB_PATH") or "./memory_db"
+    return str(Path(path).resolve())
+
+
+def _install_openclaw_extension(openclaw_dir: Path, *, db_path: str) -> None:
     """Install the AIMemory extension into OpenClaw's extensions directory.
 
     Creates ~/.openclaw/extensions/aimemory/ with the plugin files that
@@ -111,16 +125,18 @@ def _install_openclaw_extension(openclaw_dir: Path) -> None:
 
     # index.ts — bridge plugin
     index_ts = _load_template("openclaw_extension.ts")
-    # Inject the resolved MCP command path
     index_ts = index_ts.replace("__AIMEMORY_MCP_COMMAND__", mcp_command)
+    index_ts = index_ts.replace("__AIMEMORY_DB_PATH__", db_path)
     (ext_dir / "index.ts").write_text(index_ts, encoding="utf-8")
 
     print(f"  Installed extension to {ext_dir}")
+    print(f"  DB path: {db_path}")
 
 
-def setup_openclaw(workspace: Path) -> None:
+def setup_openclaw(workspace: Path, *, db_path: str | None = None) -> None:
     """Inject AIMemory instructions into OpenClaw workspace files."""
     workspace = workspace.expanduser()
+    resolved_db_path = _resolve_db_path(db_path)
 
     # SOUL.md — Memory Continuity section
     soul_path = workspace / "SOUL.md"
@@ -136,16 +152,18 @@ def setup_openclaw(workspace: Path) -> None:
 
     # Install OpenClaw extension
     openclaw_dir = workspace.parent
-    _install_openclaw_extension(openclaw_dir)
+    _install_openclaw_extension(openclaw_dir, db_path=resolved_db_path)
 
 
-def setup_claude(claude_dir: Path) -> None:
+def setup_claude(claude_dir: Path, *, db_path: str | None = None) -> None:
     """Inject AIMemory instructions into Claude Code's CLAUDE.md."""
     claude_dir = claude_dir.expanduser()
+    resolved_db_path = _resolve_db_path(db_path)
     claude_md = claude_dir / "CLAUDE.md"
     claude_content = _load_template("claude.md")
     _inject_block(claude_md, claude_content)
     print(f"  Updated {claude_md}")
+    print(f"  DB path: {resolved_db_path}")
 
 
 def main() -> None:
@@ -156,16 +174,25 @@ def main() -> None:
     )
     subparsers = parser.add_subparsers(dest="client")
 
+    # Common argument for db-path
+    db_path_help = (
+        "Absolute path to the ChromaDB database directory. "
+        "All clients (MCP server, live viewer, extension) will use this path. "
+        "(default: $AIMEMORY_DB_PATH or ./memory_db resolved to absolute)"
+    )
+
     # openclaw subcommand
     oc = subparsers.add_parser("openclaw", help="Setup for OpenClaw")
     oc.add_argument(
         "--workspace",
         default="~/.openclaw/workspace",
         help="OpenClaw workspace path (default: ~/.openclaw/workspace)",
     )
+    oc.add_argument("--db-path", default=None, help=db_path_help)
 
     # claude subcommand
-    subparsers.add_parser("claude", help="Setup for Claude Code")
+    cl = subparsers.add_parser("claude", help="Setup for Claude Code")
+    cl.add_argument("--db-path", default=None, help=db_path_help)
 
     args = parser.parse_args()
 
@@ -176,9 +203,9 @@ def main() -> None:
     print(f"Setting up AIMemory for {args.client}...")
 
     if args.client == "openclaw":
-        setup_openclaw(Path(args.workspace))
+        setup_openclaw(Path(args.workspace), db_path=args.db_path)
     elif args.client == "claude":
-        setup_claude(Path("~/.claude"))
+        setup_claude(Path("~/.claude"), db_path=args.db_path)
 
     print("Done!")
 

src/aimemory/setup_instructions/openclaw_extension.ts

@@ -4,6 +4,7 @@ import { createInterface } from "node:readline";
 // ── MCP stdio client ────────────────────────────────────────────────
 
 const MCP_COMMAND = "__AIMEMORY_MCP_COMMAND__";
+const DB_PATH = "__AIMEMORY_DB_PATH__";
 
 interface JsonRpcRequest {
   jsonrpc: "2.0";
@@ -33,7 +34,10 @@ function ensureMcp(): ChildProcess {
 
   mcpProcess = spawn(MCP_COMMAND, [], {
     stdio: ["pipe", "pipe", "pipe"],
-    env: { ...process.env },
+    env: {
+      ...process.env,
+      AIMEMORY_DB_PATH: process.env.AIMEMORY_DB_PATH || DB_PATH,
+    },
   });
 
   const rl = createInterface({ input: mcpProcess.stdout! });