kartik-mem0
71a6696c87
Update git commit capture with better error handling, enhance user prompt hook, extend config parser capabilities, and refine skill docs for dream, health, mcp, pin, and stats skills.
93 lines
2.8 KiB
Bash
Executable File
93 lines
2.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Hook: PreToolUse (matcher: Bash)
|
|
#
|
|
# Detects `git commit` commands and:
|
|
# 1. Fires on_pre_commit.py in the background to capture staged changes as memory
|
|
# 2. Searches for relevant memories about the changed files and surfaces them
|
|
#
|
|
# Input: JSON on stdin with tool_name, tool_input
|
|
# Output: JSON with additionalContext (relevant memories for the commit)
|
|
|
|
set -uo pipefail
|
|
|
|
INPUT=$(cat)
|
|
|
|
COMMAND=$(echo "$INPUT" | jq -r '.tool_input.command // ""' 2>/dev/null || echo "")
|
|
|
|
if [ -z "$COMMAND" ]; then
|
|
exit 0
|
|
fi
|
|
|
|
case "$COMMAND" in
|
|
*"git commit"*|*"git merge"*|*"git rebase"*)
|
|
;;
|
|
*)
|
|
exit 0
|
|
;;
|
|
esac
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
|
|
|
if [ ! -f "$SCRIPT_DIR/on_pre_commit.py" ]; then
|
|
exit 0
|
|
fi
|
|
|
|
API_KEY="${MEM0_API_KEY:-${CLAUDE_PLUGIN_OPTION_MEM0_API_KEY:-}}"
|
|
if [ -z "$API_KEY" ]; then
|
|
exit 0
|
|
fi
|
|
|
|
# Background: capture staged changes as memory
|
|
git diff --cached --stat 2>/dev/null | python3 "$SCRIPT_DIR/on_pre_commit.py" &
|
|
|
|
# Foreground: search for relevant memories about changed files
|
|
CHANGED_FILES=$(git diff --cached --name-only 2>/dev/null | head -10 | tr '\n' ', ' | sed 's/,$//')
|
|
if [ -z "$CHANGED_FILES" ]; then
|
|
exit 0
|
|
fi
|
|
|
|
USER_ID="${MEM0_RESOLVED_USER_ID:-$USER}"
|
|
PROJECT_ID="${MEM0_PROJECT_ID:-unknown}"
|
|
|
|
CONTEXT=$(python3 -c "
|
|
import json, urllib.request, os
|
|
api_key = os.environ.get('MEM0_API_KEY', os.environ.get('CLAUDE_PLUGIN_OPTION_MEM0_API_KEY', ''))
|
|
user_id = '$USER_ID'
|
|
app_id = '$PROJECT_ID'
|
|
files = '$CHANGED_FILES'
|
|
first_file = files.split(',')[0].strip()
|
|
body = json.dumps({
|
|
'query': f'changes to {files}',
|
|
'filters': {'AND': [{'user_id': user_id}, {'app_id': app_id}]},
|
|
'top_k': 3,
|
|
}).encode()
|
|
req = urllib.request.Request(
|
|
'https://api.mem0.ai/v2/memories/search/',
|
|
data=body,
|
|
headers={'Authorization': f'Token {api_key}', 'Content-Type': 'application/json'},
|
|
method='POST',
|
|
)
|
|
try:
|
|
with urllib.request.urlopen(req, timeout=5) as r:
|
|
results = json.loads(r.read())
|
|
memories = results if isinstance(results, list) else results.get('results', [])
|
|
if memories:
|
|
lines = ['## Pre-Commit Memory Check', '', 'Relevant memories for files being committed (' + files + '):', '']
|
|
for m in memories[:3]:
|
|
mid = m.get('id', '?')[:8]
|
|
text = m.get('memory', '')[:200]
|
|
cat = (m.get('metadata') or {}).get('type', 'unknown')
|
|
lines.append(f'- [{cat}] {text} [mem0:{mid}]')
|
|
lines.append('')
|
|
lines.append('Consider: does this commit introduce a learning worth saving? If so, suggest storing it after the commit completes.')
|
|
print('\\n'.join(lines))
|
|
except Exception:
|
|
pass
|
|
" 2>/dev/null || true)
|
|
|
|
if [ -n "$CONTEXT" ]; then
|
|
jq -nc --arg ctx "$CONTEXT" '{hookSpecificOutput:{hookEventName:"PreToolUse",additionalContext:$ctx}}'
|
|
fi
|
|
|
|
exit 0
|