@rlabs-inc/memory 0.3.2 → 0.3.3

package/gemini-hooks/curation.ts

@@ -0,0 +1,73 @@
+#!/usr/bin/env bun
+// ============================================================================
+// GEMINI CURATION HOOK
+// Hook: SessionEnd / PreCompress
+//
+// Triggers memory curation.
+// ============================================================================
+
+import { styleText } from 'util'
+
+const MEMORY_API_URL = process.env.MEMORY_API_URL || 'http://localhost:8765'
+
+const info = (text: string) => styleText('cyan', text)
+const success = (text: string) => styleText('green', text)
+const warn = (text: string) => styleText('yellow', text)
+
+function getProjectId(cwd: string): string {
+  return cwd.split('/').pop() || 'default'
+}
+
+async function main() {
+  if (process.env.MEMORY_CURATOR_ACTIVE === '1') return
+
+  try {
+    const inputText = await Bun.stdin.text()
+    const input = inputText ? JSON.parse(inputText) : {}
+
+    const sessionId = input.session_id || process.env.GEMINI_SESSION_ID || 'unknown'
+    const cwd = input.cwd || process.env.GEMINI_PROJECT_DIR || process.cwd()
+    const projectId = getProjectId(cwd)
+    
+    // Gemini: PreCompress has 'trigger', SessionEnd has 'reason'
+    const eventName = input.hook_event_name || 'unknown'
+    let trigger = 'session_end'
+    
+    if (eventName === 'PreCompress') {
+        trigger = 'pre_compact'
+    }
+
+    console.error(info(`🧠 Curating memories (${eventName})...`))
+
+    const response = await fetch(`${MEMORY_API_URL}/memory/checkpoint`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({
+        session_id: sessionId,
+        project_id: projectId,
+        claude_session_id: sessionId, 
+        trigger,
+        cwd,
+        cli_type: 'gemini-cli'
+      }),
+      signal: AbortSignal.timeout(5000),
+    }).catch(() => null)
+
+    if (response?.ok) {
+      console.error(success('✨ Memory curation started'))
+      // For PreCompress, we can send a system message
+      if (eventName === 'PreCompress') {
+          console.log(JSON.stringify({
+              systemMessage: "🧠 Memories curated before compression"
+          }))
+      }
+    } else {
+      console.error(warn('⚠️ Memory server not available'))
+    }
+
+  } catch (error: any) {
+    console.error(warn(`⚠️ Hook error: ${error.message}`))
+  }
+}
+
+main()

package/gemini-hooks/session-start.ts

@@ -0,0 +1,72 @@
+#!/usr/bin/env bun
+// ============================================================================
+// GEMINI SESSION START HOOK
+// Hook: SessionStart
+//
+// Injects session primer when a new Gemini session begins.
+// ============================================================================
+
+const MEMORY_API_URL = process.env.MEMORY_API_URL || 'http://localhost:8765'
+const TIMEOUT_MS = 5000
+
+function getProjectId(cwd: string): string {
+  return cwd.split('/').pop() || 'default'
+}
+
+async function httpPost(url: string, data: object): Promise<any> {
+  try {
+    const response = await fetch(url, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify(data),
+      signal: AbortSignal.timeout(TIMEOUT_MS),
+    })
+    return response.ok ? response.json() : {}
+  } catch {
+    return {}
+  }
+}
+
+async function main() {
+  if (process.env.MEMORY_CURATOR_ACTIVE === '1') return
+
+  try {
+    const inputText = await Bun.stdin.text()
+    const input = inputText ? JSON.parse(inputText) : {}
+
+    // Gemini provides session_id in the common input fields
+    const sessionId = input.session_id || process.env.GEMINI_SESSION_ID || 'unknown'
+    const cwd = input.cwd || process.env.GEMINI_PROJECT_DIR || process.cwd()
+    const projectId = getProjectId(cwd)
+
+    const result = await httpPost(`${MEMORY_API_URL}/memory/context`, {
+      session_id: sessionId,
+      project_id: projectId,
+      current_message: '',
+      max_memories: 0,
+    })
+
+    await httpPost(`${MEMORY_API_URL}/memory/process`, {
+      session_id: sessionId,
+      project_id: projectId,
+      metadata: { event: 'session_start', platform: 'gemini' },
+    })
+
+    const primer = result.context_text || ''
+    
+    if (primer) {
+      // Gemini expects a structured JSON response for context injection
+      console.log(JSON.stringify({
+        hookSpecificOutput: {
+          hookEventName: "SessionStart",
+          additionalContext: primer
+        }
+      }))
+    }
+  } catch (e) {
+    // Fail silently, but ensure we don't output invalid JSON if we crashed mid-stream
+    process.exit(0)
+  }
+}
+
+main()

package/gemini-hooks/user-prompt.ts

@@ -0,0 +1,77 @@
+#!/usr/bin/env bun
+// ============================================================================
+// GEMINI BEFORE AGENT HOOK
+// Hook: BeforeAgent (equivalent to Claude's UserPromptSubmit)
+//
+// Intercepts user prompts to inject relevant memories.
+// ============================================================================
+
+const MEMORY_API_URL = process.env.MEMORY_API_URL || 'http://localhost:8765'
+const TIMEOUT_MS = 5000
+
+function getProjectId(cwd: string): string {
+  return cwd.split('/').pop() || 'default'
+}
+
+async function httpPost(url: string, data: object): Promise<any> {
+  try {
+    const response = await fetch(url, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify(data),
+      signal: AbortSignal.timeout(TIMEOUT_MS),
+    })
+    return response.ok ? response.json() : {}
+  } catch {
+    return {}
+  }
+}
+
+async function main() {
+  if (process.env.MEMORY_CURATOR_ACTIVE === '1') return
+
+  try {
+    const inputText = await Bun.stdin.text()
+    const input = inputText ? JSON.parse(inputText) : {}
+
+    const sessionId = input.session_id || process.env.GEMINI_SESSION_ID || 'unknown'
+    const prompt = input.prompt || '' // Gemini passes 'prompt' in BeforeAgent
+    const cwd = input.cwd || process.env.GEMINI_PROJECT_DIR || process.cwd()
+    const projectId = getProjectId(cwd)
+
+    const result = await httpPost(`${MEMORY_API_URL}/memory/context`, {
+      session_id: sessionId,
+      project_id: projectId,
+      current_message: prompt,
+      max_memories: 5,
+    })
+
+    await httpPost(`${MEMORY_API_URL}/memory/process`, {
+      session_id: sessionId,
+      project_id: projectId,
+      metadata: { platform: 'gemini' }
+    })
+
+    const context = result.context_text || ''
+    
+    if (context) {
+      // Gemini requires structured JSON output
+      console.log(JSON.stringify({
+        decision: "allow",
+        hookSpecificOutput: {
+          hookEventName: "BeforeAgent",
+          additionalContext: context
+        }
+      }))
+    } else {
+      // Must always output valid JSON or nothing? 
+      // Safest to output "allow" if no context
+      console.log(JSON.stringify({ decision: "allow" }))
+    }
+  } catch {
+    // Fail safe
+    console.log(JSON.stringify({ decision: "allow" }))
+  }
+}
+
+main()

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rlabs-inc/memory",
-  "version": "0.3.2",
+  "version": "0.3.3",
   "description": "AI Memory System - Consciousness continuity through intelligent memory curation and retrieval",
   "type": "module",
   "main": "dist/index.js",
@@ -24,7 +24,8 @@
   "files": [
     "dist",
     "src",
-    "hooks"
+    "hooks",
+    "gemini-hooks"
   ],
   "scripts": {
     "build": "bun run build:esm && bun run build:cjs",