@rlabs-inc/memory 0.5.8 → 0.5.9

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.9",
   "description": "AI Memory System - Consciousness continuity through intelligent memory curation and retrieval",
   "type": "module",
   "main": "dist/index.js",

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
 }
 
 /**
@@ -244,7 +245,8 @@ export async function createServer(config: ServerConfig = {}) {
                 result = await curator.curateWithGeminiCLI(
                   body.claude_session_id,
                   body.trigger,
-                  body.cwd  // Run from original project directory
+                  body.cwd,  // Run from original project directory
+                  body.gemini_api_key
                 )
               } else {
                 // Default: Use Claude Code (session resume or transcript parsing)
@@ -306,7 +308,8 @@ export async function createServer(config: ServerConfig = {}) {
                         body.project_id,
                         sessionNumber,
                         result,
-                        storagePaths
+                        storagePaths,
+                        body.gemini_api_key
                       )
                     } else {
                       // Use Claude Agent SDK mode - more reliable than CLI