feat(mem0-plugin): improve auto-triggering — pre-fetch, dedup, skill enforcement v0.2.3 (#5237)

2026-05-23 17:20:54 +05:30
parent 16a7702d09
commit 99beb007ab
54 changed files with 1817 additions and 1162 deletions
@@ -46,10 +46,6 @@ fi
 SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
 . "$SCRIPT_DIR/_identity.sh" 2>/dev/null || true

-if [ -z "${MEM0_API_KEY:-}" ]; then
-  exit 0
-fi
-
 # Extract error class/message (first matching line)
 ERROR_LINE=$(echo "$TOOL_RESULT" | grep -iE '(Error:|Exception:|panic:|FAIL:|fatal:)' | head -1 | sed 's/^[[:space:]]*//' | cut -c1-120)

@@ -62,45 +58,60 @@ if [ -n "$TRACE_FILES" ]; then
  FILE_DISPLAY=$(echo "$TRACE_FILES" | sed 's/^/  - /')
 fi

-USER_ID="$MEM0_RESOLVED_USER_ID"
+USER_ID="${MEM0_RESOLVED_USER_ID:-${USER:-default}}"

-cat <<EOF
+# Extract query (first 80 chars of error line)
+ERROR_QUERY=$(echo "$ERROR_LINE" | cut -c1-80)

-## Error detected in command output
-
-\`$COMMAND\` produced an error:
-> $ERROR_LINE
-
-EOF
-
-if [ -n "$FILE_DISPLAY" ]; then
-  cat <<EOF
-**Files in stack trace:**
-$FILE_DISPLAY
-
-EOF
-fi
-
-cat <<EOF
-Search mem0 for prior occurrences — this error may have been seen before:
- \`search_memories(query="$(echo "$ERROR_LINE" | cut -c1-60)", filters={"AND": [{"user_id": "$USER_ID"}, {"app_id": "$MEM0_PROJECT_ID"}, {"metadata": {"type": "anti_pattern"}}]})\`
- \`search_memories(query="$(echo "$ERROR_LINE" | cut -c1-60)", filters={"AND": [{"user_id": "$USER_ID"}, {"app_id": "$MEM0_PROJECT_ID"}, {"metadata": {"type": "bug_fix"}}]})\`
-EOF
-
-if [ -n "$TRACE_FILES" ]; then
-  FIRST_FILE=$(echo "$TRACE_FILES" | head -1 | sed 's/:[0-9]*//')
-  cat <<EOF
- \`search_memories(query="$FIRST_FILE", filters={"AND": [{"user_id": "$USER_ID"}, {"app_id": "$MEM0_PROJECT_ID"}]})\`
-EOF
-fi
-
-cat <<EOF
-
-If mem0 returns relevant context, use it to debug faster.
-If you solve this, store the fix as an \`anti_pattern\` or \`bug_fix\` memory for next time.
-EOF
-
-# Telemetry
+# Telemetry (fire regardless of API key)
 python3 "$SCRIPT_DIR/telemetry.py" bash_error --error_detected 2>/dev/null &

+# No API key — skip output entirely
+if [ -z "${MEM0_API_KEY:-}" ]; then
+  exit 0
+fi
+
+# Pre-fetch memories: anti_pattern and bug_fix searches
+RESULTS=$(PYTHONPATH="$SCRIPT_DIR" MEM0_SEARCH_QUERY="$ERROR_QUERY" MEM0_SEARCH_USER="$USER_ID" \
+  MEM0_API_KEY="${MEM0_API_KEY}" MEM0_PROJECT_ID="${MEM0_PROJECT_ID:-unknown}" \
+  python3 -c "
+import os, sys
+sys.path.insert(0, os.environ.get('PYTHONPATH', '.'))
+from _search import search_memories, format_results_for_context
+
+api_key = os.environ.get('MEM0_API_KEY', '')
+user_id = os.environ.get('MEM0_SEARCH_USER', 'default')
+project_id = os.environ.get('MEM0_PROJECT_ID', 'unknown')
+query = os.environ.get('MEM0_SEARCH_QUERY', '')
+
+r1 = search_memories(api_key, user_id, project_id, query, metadata_type='anti_pattern', top_k=3)
+r2 = search_memories(api_key, user_id, project_id, query, metadata_type='bug_fix', top_k=3)
+
+seen = set()
+combined = []
+for m in r1 + r2:
+    mid = m.get('id', '')
+    if mid not in seen:
+        seen.add(mid)
+        combined.append(m)
+
+print(format_results_for_context(combined, heading='Prior error memories'), end='')
+" 2>/dev/null || echo "")
+
+# Output error header
+printf '\n## Error detected in command output\n\n'
+printf '`%s` produced an error:\n> %s\n\n' "$COMMAND" "$ERROR_LINE"
+
+if [ -n "$FILE_DISPLAY" ]; then
+  printf '**Files in stack trace:**\n%s\n\n' "$FILE_DISPLAY"
+fi
+
+if [ -n "$RESULTS" ]; then
+  printf '%s\n' "$RESULTS"
+else
+  printf 'No prior memories found for this error.\n\n'
+fi
+
+printf 'If you solve this, store the fix as an `anti_pattern` or `bug_fix` memory for next time.\n'
+
 exit 0