@rlabs-inc/memory 0.5.8 → 0.5.10

package/README.md

@@ -424,6 +424,11 @@ This isn't just about remembering facts. It's about preserving:
 
 ## Changelog
 
+### v0.5.9
+- **Fix**: Updated Gemini CLI hooks to correctly log injected context using stderr
+- **Fix**: Removed unsupported `systemMessage` field from `BeforeAgent` hook
+- **Improvement**: Added colorful `[Memory]` logs to terminal for visibility of memory injection
+
 ### v0.5.1
 - **Improvement**: Gemini CLI hooks now show injected content to user via `systemMessage`
 - Users see exactly what memories are surfaced (session primer, retrieved memories)

package/hooks/gemini/curation.ts

@@ -53,7 +53,8 @@ async function main() {
         claude_session_id: sessionId, 
         trigger,
         cwd,
-        cli_type: 'gemini-cli'
+        cli_type: 'gemini-cli',
+        gemini_api_key: process.env.GEMINI_API_KEY,
       }),
       signal: AbortSignal.timeout(5000),
     }).catch(() => null)

package/hooks/gemini/session-start.ts

@@ -55,9 +55,10 @@ async function main() {
     const primer = result.context_text || ''
 
     if (primer) {
-      // Show user exactly what we inject - same formatted content
-      // systemMessage: shown to user in terminal
-      // additionalContext: injected into model context
+      // Log to stderr for visibility
+      console.error(`\x1b[36m[Memory] Injecting session primer (${primer.length} chars)\x1b[0m`)
+
+      // systemMessage IS supported for SessionStart
       console.log(JSON.stringify({
         systemMessage: primer,
         hookSpecificOutput: {
@@ -65,6 +66,8 @@ async function main() {
           additionalContext: primer
         }
       }))
+    } else {
+      console.log(JSON.stringify({}))
     }
   } catch (e) {
     // Fail silently, but ensure we don't output invalid JSON if we crashed mid-stream

package/hooks/gemini/user-prompt.ts

@@ -55,20 +55,18 @@ async function main() {
     const context = result.context_text || ''
 
     if (context) {
-      // Show user exactly what we inject - same formatted content
-      // systemMessage: shown to user in terminal
-      // additionalContext: injected into model context
+      // Log to stderr for visibility (BeforeAgent doesn't support systemMessage)
+      console.error(`\x1b[36m[Memory] Injecting ${context.length} chars of context\x1b[0m`)
+      
       console.log(JSON.stringify({
-        decision: "allow",
-        systemMessage: context,
         hookSpecificOutput: {
           hookEventName: "BeforeAgent",
           additionalContext: context
         }
       }))
     } else {
-      // No memories to surface - just allow without message
-      console.log(JSON.stringify({ decision: "allow" }))
+      // No memories to surface - output empty JSON (implicit allow)
+      console.log(JSON.stringify({}))
     }
   } catch {
     // Fail safe

package/package.json

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

package/src/cli/commands/serve.ts

@@ -135,6 +135,14 @@ export async function serve(options: ServeOptions) {
         console.log(c.muted(`    Use 'memory migrate --dry-run' to preview changes first`))
       }
 
+      // Note for Gemini CLI API key auth users (OAuth users don't need this)
+      if (!process.env.GEMINI_API_KEY) {
+        console.log()
+        console.log(c.muted(`  ${symbols.info} Using Gemini CLI with API key auth?`))
+        console.log(c.muted(`    Run: GEMINI_API_KEY=your-key memory serve`))
+        console.log(c.muted(`    (OAuth users can ignore this)`))
+      }
+
       console.log()
       console.log(c.muted(`  Press Ctrl+C to stop`))
       console.log()

package/src/core/curator.ts

@@ -788,6 +788,7 @@ This session has ended. Please curate the memories from this conversation accord
     sessionId: string,
     triggerType: CurationTrigger = "session_end",
     cwd?: string,
+    apiKey?: string,
   ): Promise<CurationResult> {
     const systemPrompt = this.buildCurationPrompt(triggerType);
     const userMessage =
@@ -825,6 +826,7 @@ This session has ended. Please curate the memories from this conversation accord
         ...process.env,
         MEMORY_CURATOR_ACTIVE: "1", // Prevent recursive hook triggering
         GEMINI_SYSTEM_MD: tempPromptPath, // Inject our curation prompt
+        ...(apiKey ? { GEMINI_API_KEY: apiKey } : {}),
       },
       stdout: "pipe",
       stderr: "pipe",

package/src/core/manager.ts

@@ -405,6 +405,7 @@ Please process these memories according to your management procedure. Use the ex
     sessionNumber: number,
     result: CurationResult,
     storagePaths?: StoragePaths,
+    apiKey?: string,
   ): Promise<ManagementResult> {
     // Skip if disabled via config or env var
     if (!this._config.enabled || process.env.MEMORY_MANAGER_DISABLED === "1") {
@@ -564,6 +565,7 @@ Use these tools to read existing memories, write updates, and manage the memory
         ...process.env,
         MEMORY_CURATOR_ACTIVE: "1", // Prevent recursive hook triggering
         GEMINI_SYSTEM_MD: tempPromptPath, // Inject our management prompt
+        ...(apiKey ? { GEMINI_API_KEY: apiKey } : {}),
       },
       stdout: "pipe",
       stderr: "pipe",

package/src/server/index.ts

@@ -62,6 +62,7 @@ interface CheckpointRequest {
   cwd?: string
   cli_type?: 'claude-code' | 'gemini-cli'
   project_path?: string
+  gemini_api_key?: string
 }
 
 /**
@@ -240,11 +241,15 @@ export async function createServer(config: ServerConfig = {}) {
               // Branch on CLI type - Gemini CLI vs Claude Code
               if (body.cli_type === 'gemini-cli') {
                 // Use Gemini CLI for curation (no Claude dependency)
+                // Fallback to server's GEMINI_API_KEY if hook didn't pass one
+                // (hooks spawned by Gemini CLI may not inherit env vars)
+                const geminiApiKey = body.gemini_api_key || process.env.GEMINI_API_KEY
                 logger.debug('Using Gemini CLI for curation', 'server')
                 result = await curator.curateWithGeminiCLI(
                   body.claude_session_id,
                   body.trigger,
-                  body.cwd  // Run from original project directory
+                  body.cwd,  // Run from original project directory
+                  geminiApiKey
                 )
               } else {
                 // Default: Use Claude Code (session resume or transcript parsing)
@@ -301,12 +306,15 @@ export async function createServer(config: ServerConfig = {}) {
                     let managementResult
                     if (cliType === 'gemini-cli') {
                       // Use Gemini CLI for management (no Claude dependency)
+                      // Use same API key fallback as curation (hooks don't inherit env vars)
+                      const geminiApiKey = body.gemini_api_key || process.env.GEMINI_API_KEY
                       logger.debug('Using Gemini CLI for management', 'server')
                       managementResult = await manager.manageWithGeminiCLI(
                         body.project_id,
                         sessionNumber,
                         result,
-                        storagePaths
+                        storagePaths,
+                        geminiApiKey
                       )
                     } else {
                       // Use Claude Agent SDK mode - more reliable than CLI