@rlabs-inc/memory 0.3.1 → 0.3.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rlabs-inc/memory",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "description": "AI Memory System - Consciousness continuity through intelligent memory curation and retrieval",
   "type": "module",
   "main": "dist/index.js",

package/src/cli/commands/install-gemini.ts

@@ -0,0 +1,146 @@
+// ============================================================================
+// INSTALL GEMINI COMMAND - Set up Gemini CLI hooks
+// ============================================================================
+
+import { homedir } from 'os'
+import { join } from 'path'
+import { existsSync, mkdirSync } from 'fs'
+import { c, symbols, fmt } from '../colors.ts'
+
+interface InstallOptions {
+  verbose?: boolean
+  force?: boolean
+}
+
+export async function installGemini(options: InstallOptions) {
+  console.log()
+  console.log(c.header(`${symbols.brain} Memory - Install Gemini Hooks`))
+  console.log()
+
+  const geminiDir = join(homedir(), '.gemini')
+  const settingsPath = join(geminiDir, 'settings.json')
+
+  // Find the hooks directory (relative to this CLI)
+  const cliPath = import.meta.dir
+  const packageRoot = join(cliPath, '..', '..', '..')
+  const hooksDir = join(packageRoot, 'gemini-hooks')
+
+  console.log(`  ${fmt.kv('Gemini config', geminiDir)}`)
+  console.log(`  ${fmt.kv('Hooks source', hooksDir)}`)
+  console.log()
+
+  // Check if hooks directory exists
+  if (!existsSync(hooksDir)) {
+    console.log(c.error(`  ${symbols.cross} Hooks directory not found at ${hooksDir}`))
+    process.exit(1)
+  }
+
+  // Ensure .gemini directory exists
+  if (!existsSync(geminiDir)) {
+    try {
+        mkdirSync(geminiDir, { recursive: true })
+        console.log(`  ${c.success(symbols.tick)} Created ${geminiDir}`)
+    } catch {
+        console.log(`  ${c.warn(symbols.warning)} Could not create ${geminiDir} (sandbox restriction?)`)
+        console.log(`  ${c.muted('Skipping config write, printing manual instructions instead.')}`)
+    }
+  }
+
+  // Read existing settings or create new
+  let settings: any = {}
+  if (existsSync(settingsPath)) {
+    try {
+      const content = await Bun.file(settingsPath).text()
+      settings = JSON.parse(content)
+      console.log(`  ${c.success(symbols.tick)} Found existing settings.json`)
+    } catch {
+      console.log(`  ${c.warn(symbols.warning)} Could not parse settings.json`)
+    }
+  }
+
+  // Build hooks configuration
+  const sessionStartHook = join(hooksDir, 'session-start.ts')
+  const userPromptHook = join(hooksDir, 'user-prompt.ts')
+  const curationHook = join(hooksDir, 'curation.ts')
+
+  // Based on Gemini CLI documentation
+  const hooksConfig = {
+    SessionStart: [
+      {
+        matcher: 'startup|resume',
+        hooks: [
+          {
+            type: 'command',
+            command: `bun "${sessionStartHook}"`,
+            timeout: 10000
+          }
+        ]
+      }
+    ],
+    BeforeAgent: [
+      {
+        hooks: [
+          {
+            type: 'command',
+            command: `bun "${userPromptHook}"`,
+            timeout: 10000
+          }
+        ]
+      }
+    ],
+    PreCompress: [
+      {
+        matcher: 'auto|manual',
+        hooks: [
+          {
+            type: 'command',
+            command: `bun "${curationHook}"`,
+            timeout: 120000
+          }
+        ]
+      }
+    ],
+    SessionEnd: [
+      {
+        hooks: [
+          {
+            type: 'command',
+            command: `bun "${curationHook}"`,
+            timeout: 120000
+          }
+        ]
+      }
+    ]
+  }
+
+  // Merge hooks
+  if (!settings.hooks) {
+      settings.hooks = {}
+  }
+  
+  settings.hooks = {
+    ...settings.hooks,
+    ...hooksConfig
+  }
+
+  // Write settings
+  try {
+    if (existsSync(geminiDir)) {
+        await Bun.write(settingsPath, JSON.stringify(settings, null, 2))
+        console.log(`  ${c.success(symbols.tick)} Updated ${settingsPath}`)
+    } else {
+        throw new Error("Gemini directory does not exist")
+    }
+  } catch (error: any) {
+    console.log(c.error(`  ${symbols.cross} Failed to write settings: ${error.message}`))
+    console.log()
+    console.log(c.bold('Manual Installation Instructions:'))
+    console.log('Add the following to your ~/.gemini/settings.json:')
+    console.log()
+    console.log(JSON.stringify({ hooks: hooksConfig }, null, 2))
+    console.log()
+  }
+
+  console.log()
+  console.log(c.success(`${symbols.sparkles} Gemini hooks configured!`))
+}

package/src/cli/index.ts

@@ -23,6 +23,7 @@ ${c.bold('Commands:')}
   ${c.command('serve')}      Start the memory server ${c.muted('(default)')}
   ${c.command('stats')}      Show memory statistics
   ${c.command('install')}    Set up Claude Code hooks
+  ${c.command('install-gemini')} Set up Gemini CLI hooks
   ${c.command('doctor')}     Check system health
   ${c.command('help')}       Show this help message
 
@@ -103,6 +104,12 @@ async function main() {
       break
     }
 
+    case 'install-gemini': {
+      const { installGemini } = await import('./commands/install-gemini.ts')
+      await installGemini(values)
+      break
+    }
+
     case 'doctor':
     case 'check': {
       const { doctor } = await import('./commands/doctor.ts')

package/src/core/curator.ts

@@ -296,20 +296,23 @@ Return ONLY this JSON structure:
 
   /**
    * Curate using CLI subprocess (for hook mode)
-   * Resumes a Claude Code session and asks it to curate
+   * Resumes a session and asks it to curate
    */
   async curateWithCLI(
     sessionId: string,
     triggerType: CurationTrigger = 'session_end',
-    cwd?: string
+    cwd?: string,
+    cliTypeOverride?: 'claude-code' | 'gemini-cli'
   ): Promise<CurationResult> {
+    const type = cliTypeOverride ?? this._config.cliType
     const systemPrompt = this.buildCurationPrompt(triggerType)
     const userMessage = 'This session has ended. Please curate the memories from our conversation according to the instructions in your system prompt. Return ONLY the JSON structure.'
 
     // Build CLI command based on type
     const args: string[] = []
+    let command = this._config.cliCommand
 
-    if (this._config.cliType === 'claude-code') {
+    if (type === 'claude-code') {
       args.push(
         '--resume', sessionId,
         '-p', userMessage,
@@ -318,7 +321,8 @@ Return ONLY this JSON structure:
         '--max-turns', '1'
       )
     } else {
-      // gemini-cli (doesn't support --append-system-prompt)
+      // gemini-cli
+      command = 'gemini' // Default to 'gemini' in PATH for gemini-cli
       args.push(
         '--resume', sessionId,
         '-p', `${systemPrompt}\n\n${userMessage}`,
@@ -327,7 +331,7 @@ Return ONLY this JSON structure:
     }
 
     // Execute CLI
-    const proc = Bun.spawn([this._config.cliCommand, ...args], {
+    const proc = Bun.spawn([command, ...args], {
       cwd,
       env: {
         ...process.env,

package/src/server/index.ts

@@ -166,7 +166,8 @@ export async function createServer(config: ServerConfig = {}) {
               const result = await curator.curateWithCLI(
                 body.claude_session_id,
                 body.trigger,
-                body.cwd
+                body.cwd,
+                body.cli_type
               )
 
               if (result.memories.length > 0) {