Files
claude-mem/docs/reports/2026-01-04--gemini-agent-failures.md
T
Alex Newman f21ea97c39 refactor: decompose monolith into modular architecture with comprehensive test suite (#538)
* fix: prevent memory_session_id from equaling content_session_id

The bug: memory_session_id was initialized to contentSessionId as a
"placeholder for FK purposes". This caused the SDK resume logic to
inject memory agent messages into the USER's Claude Code transcript,
corrupting their conversation history.

Root cause:
- SessionStore.createSDKSession initialized memory_session_id = contentSessionId
- SDKAgent checked memorySessionId !== contentSessionId but this check
  only worked if the session was fetched fresh from DB

The fix:
- SessionStore: Initialize memory_session_id as NULL, not contentSessionId
- SDKAgent: Simple truthy check !!session.memorySessionId (NULL = fresh start)
- Database migration: Ran UPDATE to set memory_session_id = NULL for 1807
  existing sessions that had the bug

Also adds [ALIGNMENT] logging across the session lifecycle to help debug
session continuity issues:
- Hook entry: contentSessionId + promptNumber
- DB lookup: contentSessionId → memorySessionId mapping proof
- Resume decision: shows which memorySessionId will be used for resume
- Capture: logs when memorySessionId is captured from first SDK response

UI: Added "Alignment" quick filter button in LogsModal to show only
alignment logs for debugging session continuity.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* refactor: improve error handling in worker-service.ts

- Fix GENERIC_CATCH anti-patterns by logging full error objects instead of just messages
- Add [ANTI-PATTERN IGNORED] markers for legitimate cases (cleanup, hot paths)
- Simplify error handling comments to be more concise
- Improve httpShutdown() error discrimination for ECONNREFUSED
- Reduce LARGE_TRY_BLOCK issues in initialization code

Part of anti-pattern cleanup plan (132 total issues)

* refactor: improve error logging in SearchManager.ts

- Pass full error objects to logger instead of just error.message
- Fixes PARTIAL_ERROR_LOGGING anti-patterns (10 instances)
- Better debugging visibility when Chroma queries fail

Part of anti-pattern cleanup (133 remaining)

* refactor: improve error logging across SessionStore and mcp-server

- SessionStore.ts: Fix error logging in column rename utility
- mcp-server.ts: Log full error objects instead of just error.message
- Improve error handling in Worker API calls and tool execution

Part of anti-pattern cleanup (133 remaining)

* Refactor hooks to streamline error handling and loading states

- Simplified error handling in useContextPreview by removing try-catch and directly checking response status.
- Refactored usePagination to eliminate try-catch, improving readability and maintaining error handling through response checks.
- Cleaned up useSSE by removing unnecessary try-catch around JSON parsing, ensuring clarity in message handling.
- Enhanced useSettings by streamlining the saving process, removing try-catch, and directly checking the result for success.

* refactor: add error handling back to SearchManager Chroma calls

- Wrap queryChroma calls in try-catch to prevent generator crashes
- Log Chroma errors as warnings and fall back gracefully
- Fixes generator failures when Chroma has issues
- Part of anti-pattern cleanup recovery

* feat: Add generator failure investigation report and observation duplication regression report

- Created a comprehensive investigation report detailing the root cause of generator failures during anti-pattern cleanup, including the impact, investigation process, and implemented fixes.
- Documented the critical regression causing observation duplication due to race conditions in the SDK agent, outlining symptoms, root cause analysis, and proposed fixes.

* fix: address PR #528 review comments - atomic cleanup and detector improvements

This commit addresses critical review feedback from PR #528:

## 1. Atomic Message Cleanup (Fix Race Condition)

**Problem**: SessionRoutes.ts generator error handler had race condition
- Queried messages then marked failed in loop
- If crash during loop → partial marking → inconsistent state

**Solution**:
- Added `markSessionMessagesFailed()` to PendingMessageStore.ts
- Single atomic UPDATE statement replaces loop
- Follows existing pattern from `resetProcessingToPending()`

**Files**:
- src/services/sqlite/PendingMessageStore.ts (new method)
- src/services/worker/http/routes/SessionRoutes.ts (use new method)

## 2. Anti-Pattern Detector Improvements

**Problem**: Detector didn't recognize logger.failure() method
- Lines 212 & 335 already included "failure"
- Lines 112-113 (PARTIAL_ERROR_LOGGING detection) did not

**Solution**: Updated regex patterns to include "failure" for consistency

**Files**:
- scripts/anti-pattern-test/detect-error-handling-antipatterns.ts

## 3. Documentation

**PR Comment**: Added clarification on memory_session_id fix location
- Points to SessionStore.ts:1155
- Explains why NULL initialization prevents message injection bug

## Review Response

Addresses "Must Address Before Merge" items from review:
 Clarified memory_session_id bug fix location (via PR comment)
 Made generator error handler message cleanup atomic
 Deferred comprehensive test suite to follow-up PR (keeps PR focused)

## Testing

- Build passes with no errors
- Anti-pattern detector runs successfully
- Atomic cleanup follows proven pattern from existing methods

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>

* fix: FOREIGN KEY constraint and missing failed_at_epoch column

Two critical bugs fixed:

1. Missing failed_at_epoch column in pending_messages table
   - Added migration 20 to create the column
   - Fixes error when trying to mark messages as failed

2. FOREIGN KEY constraint failed when storing observations
   - All three agents (SDK, Gemini, OpenRouter) were passing
     session.contentSessionId instead of session.memorySessionId
   - storeObservationsAndMarkComplete expects memorySessionId
   - Added null check and clear error message

However, observations still not saving - see investigation report.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>

* Refactor hook input parsing to improve error handling

- Added a nested try-catch block in new-hook.ts, save-hook.ts, and summary-hook.ts to handle JSON parsing errors more gracefully.
- Replaced direct error throwing with logging of the error details using logger.error.
- Ensured that the process exits cleanly after handling input in all three hooks.

* docs: add monolith refactor report with system breakdown

Comprehensive analysis of codebase identifying:
- 14 files over 500 lines requiring refactoring
- 3 critical monoliths (SessionStore, SearchManager, worker-service)
- 80% code duplication across agent files
- 5-phase refactoring roadmap with domain-based architecture

* docs: update monolith report post session-logging merge

- SessionStore grew to 2,011 lines (49 methods) - highest priority
- SearchManager reduced to 1,778 lines (improved)
- Agent files reduced by ~45 lines combined
- Added trend indicators and post-merge observations
- Core refactoring proposal remains valid

* refactor(sqlite): decompose SessionStore into modular architecture

Extract the 2011-line SessionStore.ts monolith into focused, single-responsibility
modules following grep-optimized progressive disclosure pattern:

New module structure:
- sessions/ - Session creation and retrieval (create.ts, get.ts, types.ts)
- observations/ - Observation storage and queries (store.ts, get.ts, recent.ts, files.ts, types.ts)
- summaries/ - Summary storage and queries (store.ts, get.ts, recent.ts, types.ts)
- prompts/ - User prompt management (store.ts, get.ts, types.ts)
- timeline/ - Cross-entity timeline queries (queries.ts)
- import/ - Bulk import operations (bulk.ts)
- migrations/ - Database migrations (runner.ts)

New coordinator files:
- Database.ts - ClaudeMemDatabase class with re-exports
- transactions.ts - Atomic cross-entity transactions
- Named re-export facades (Sessions.ts, Observations.ts, etc.)

Key design decisions:
- All functions take `db: Database` as first parameter (functional style)
- Named re-exports instead of index.ts for grep-friendliness
- SessionStore retained as backward-compatible wrapper
- Target file size: 50-150 lines (60% compliance)

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* refactor(agents): extract shared logic into modular architecture

Consolidate duplicate code across SDKAgent, GeminiAgent, and OpenRouterAgent
into focused utility modules. Total reduction: 500 lines (29%).

New modules in src/services/worker/agents/:
- ResponseProcessor.ts: Atomic DB transactions, Chroma sync, SSE broadcast
- ObservationBroadcaster.ts: SSE event formatting and dispatch
- SessionCleanupHelper.ts: Session state cleanup and stuck message reset
- FallbackErrorHandler.ts: Provider error detection for fallback logic
- types.ts: Shared interfaces (WorkerRef, SSE payloads, StorageResult)

Bug fix: SDKAgent was incorrectly using obs.files instead of obs.files_read
and hardcoding files_modified to empty array.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* refactor(search): extract search strategies into modular architecture

Decompose SearchManager into focused strategy pattern with:
- SearchOrchestrator: Coordinates strategy selection and fallback
- ChromaSearchStrategy: Vector semantic search via ChromaDB
- SQLiteSearchStrategy: Filter-only queries for date/project/type
- HybridSearchStrategy: Metadata filtering + semantic ranking
- ResultFormatter: Markdown table formatting for results
- TimelineBuilder: Chronological timeline construction
- Filter modules: DateFilter, ProjectFilter, TypeFilter

SearchManager now delegates to new infrastructure while maintaining
full backward compatibility with existing public API.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* refactor(context): decompose context-generator into modular architecture

Extract 660-line monolith into focused components:
- ContextBuilder: Main orchestrator (~160 lines)
- ContextConfigLoader: Configuration loading
- TokenCalculator: Token budget calculations
- ObservationCompiler: Data retrieval and query building
- MarkdownFormatter/ColorFormatter: Output formatting
- Section renderers: Header, Timeline, Summary, Footer

Maintains full backward compatibility - context-generator.ts now
delegates to new ContextBuilder while preserving public API.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* refactor(worker): decompose worker-service into modular infrastructure

Split 2000+ line monolith into focused modules:

Infrastructure:
- ProcessManager: PID files, signal handlers, child process cleanup
- HealthMonitor: Port checks, health polling, version matching
- GracefulShutdown: Coordinated cleanup on exit

Server:
- Server: Express app setup, core routes, route registration
- Middleware: Re-exports from existing middleware
- ErrorHandler: Centralized error handling with AppError class

Integrations:
- CursorHooksInstaller: Full Cursor IDE integration (registry, hooks, MCP)

WorkerService now acts as thin coordinator wiring all components together.
Maintains full backward compatibility with existing public API.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* Refactor session queue processing and database interactions

- Implement claim-and-delete pattern in SessionQueueProcessor to simplify message handling and eliminate duplicate processing.
- Update PendingMessageStore to support atomic claim-and-delete operations, removing the need for intermediate processing states.
- Introduce storeObservations method in SessionStore for simplified observation and summary storage without message tracking.
- Remove deprecated methods and clean up session state management in worker agents.
- Adjust response processing to accommodate new storage patterns, ensuring atomic transactions for observations and summaries.
- Remove unnecessary reset logic for stuck messages due to the new queue handling approach.

* Add duplicate observation cleanup script

Script to clean up duplicate observations created by the batching bug
where observations were stored once per message ID instead of once per
observation. Includes safety checks to always keep at least one copy.

Usage:
  bun scripts/cleanup-duplicates.ts           # Dry run
  bun scripts/cleanup-duplicates.ts --execute # Delete duplicates
  bun scripts/cleanup-duplicates.ts --aggressive # Ignore time window

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* test(sqlite): add comprehensive test suite for SQLite repositories

Add 44 tests across 5 test files covering:
- Sessions: CRUD operations and schema validation
- Observations: creation, retrieval, filtering, and ordering
- Prompts: persistence and association with observations
- Summaries: generation tracking and session linkage
- Transactions: context management and rollback behavior

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* test(worker): add comprehensive test suites for worker agent modules

Add test coverage for response-processor, observation-broadcaster,
session-cleanup-helper, and fallback-error-handler agents. Fix type
import issues across search module (use `import type` for type-only
imports) and update worker-service main module detection for ESM/CJS
compatibility.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* test(search): add comprehensive test suites for search module

Add test coverage for the refactored search architecture:
- SearchOrchestrator: query coordination and caching
- ResultFormatter: pagination, sorting, and field mapping
- SQLiteSearchStrategy: database search operations
- ChromaSearchStrategy: vector similarity search
- HybridSearchStrategy: combined search with score fusion

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* test(context): add comprehensive test suites for context-generator modules

Add test coverage for the modular context-generator architecture:
- context-builder.test.ts: Tests for context building and result assembly
- observation-compiler.test.ts: Tests for observation compilation with privacy tags
- token-calculator.test.ts: Tests for token budget calculations
- formatters/markdown-formatter.test.ts: Tests for markdown output formatting

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* test(infrastructure): add comprehensive test suites for worker infrastructure modules

Add test coverage for graceful-shutdown, health-monitor, and process-manager
modules extracted during the worker-service refactoring. All 32 tests pass.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* test(server): add comprehensive test suites for server modules

Add test coverage for Express server infrastructure:
- error-handler.test.ts: Tests error handling middleware including
  validation errors, database errors, and async error handling
- server.test.ts: Tests server initialization, middleware configuration,
  and route mounting for all API endpoints

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* chore(package): add test scripts for modular test suites

Add npm run scripts to simplify running tests:
- test: run all tests
- test:sqlite, test:agents, test:search, test:context, test:infra, test:server

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* build assets

* feat(tests): add detailed failure analysis reports for session ID refactor, validation, and store tests

- Created reports for session ID refactor test failures, highlighting 8 failures due to design mismatches.
- Added session ID usage validation report detailing 10 failures caused by outdated assumptions in tests.
- Documented session store test failures, focusing on foreign key constraint violations in 2 tests.
- Compiled a comprehensive test suite report summarizing overall test results, including 28 failing tests across various categories.

* fix(tests): align session ID tests with NULL-based initialization

Update test expectations to match implementation where memory_session_id
starts as NULL (not equal to contentSessionId) per architecture decision
that memory_session_id must NEVER equal contentSessionId.

Changes:
- session_id_refactor.test.ts: expect NULL initial state, add updateMemorySessionId() calls
- session_id_usage_validation.test.ts: update placeholder detection to check !== null
- session_store.test.ts: add updateMemorySessionId() before storeObservation/storeSummary

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* fix(tests): update GeminiAgent tests with correct field names and mocks

- Rename deprecated fields: claudeSessionId → contentSessionId,
  sdkSessionId → memorySessionId, pendingProcessingIds → pendingMessages
- Add missing required ActiveSession fields
- Add storeObservations mock (plural) for ResponseProcessor compatibility
- Fix settings mock to use correct CLAUDE_MEM_GEMINI_RATE_LIMITING_ENABLED key
- Add await to rejects.toThrow assertion

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* feat(tests): add logger imports and fix coverage test exclusions

Phase 3 of test suite fixes:
- Add logger imports to 34 high-priority source files (SQLite, worker, context)
- Exclude CLI-facing files from console.log check (worker-service.ts,
  integrations/*Installer.ts) as they use console.log intentionally for
  interactive user output

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* docs: update SESSION_ID_ARCHITECTURE for NULL-based initialization

Update documentation to reflect that memory_session_id starts as NULL,
not as a placeholder equal to contentSessionId. This matches the
implementation decision that memory_session_id must NEVER equal
contentSessionId to prevent injecting memory messages into user transcripts.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* chore(deps): update esbuild and MCP SDK

- esbuild: 0.25.12 → 0.27.2 (fixes minifyIdentifiers issue)
- @modelcontextprotocol/sdk: 1.20.1 → 1.25.1

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* build assets and updates

* chore: remove bun.lock and add to gitignore

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

---------

Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
2026-01-03 23:58:41 -05:00

318 lines
11 KiB
Markdown

# GeminiAgent Test Failures Analysis Report
**Date:** 2026-01-04
**Category:** GeminiAgent Tests
**Total Failures:** 6 of 6 tests
**Status:** Critical - All tests failing
---
## 1. Executive Summary
All 6 GeminiAgent tests are failing due to a combination of:
1. **Missing session data** - Test fixtures lack required `memorySessionId` field
2. **Mock module scoping issues** - `SettingsDefaultsManager` mocks not applying correctly
3. **Global fetch not being mocked** - Real API calls being made in some tests
4. **Async expectation syntax** - Incorrect usage of `rejects.toThrow()` pattern
The primary root cause is that the test session fixtures are incomplete. The `ActiveSession` type requires `memorySessionId` to be set before observations can be stored, but all test sessions set it to undefined/missing, triggering the validation error: "Cannot store observations: memorySessionId not yet captured".
---
## 2. Test Analysis
### Test 1: "should initialize with correct config"
**Status:** FAIL
**Expected Behavior:** Initialize GeminiAgent, make API call with correct URL containing model and API key
**Actual Result:** Error - "Cannot store observations: memorySessionId not yet captured"
**Root Cause:** Test session fixture missing `memorySessionId` field
### Test 2: "should handle multi-turn conversation"
**Status:** FAIL (Timeout after 5001ms)
**Expected Behavior:** Handle conversation history and send correct multi-turn format to Gemini
**Actual Result:** Test times out
**Root Cause:** Likely hanging on unresolved Promise due to mock issues. The mock fetch returns a response without valid observation XML, causing `processAgentResponse` to fail before completing.
### Test 3: "should process observations and store them"
**Status:** FAIL
**Expected Behavior:** Parse observation XML from Gemini response, call `storeObservation` and `syncObservation`
**Actual Result:** Error - "Cannot store observations: memorySessionId not yet captured"
**Root Cause:** Test session fixture missing `memorySessionId` field
### Test 4: "should fallback to Claude on rate limit error"
**Status:** FAIL
**Expected Behavior:** When Gemini returns 429, reset stuck messages and call fallback agent
**Actual Result:** Real API call made - "Gemini API error: 400 - API key not valid"
**Root Cause:**
- `mock.module()` for SettingsDefaultsManager not scoping correctly
- Real `fetch` is called instead of mock because the mock is set AFTER agent initialization
- Test mock key `'test-api-key'` is being used against real Gemini API
### Test 5: "should NOT fallback on other errors"
**Status:** FAIL (Timeout after 5001ms)
**Expected Behavior:** When Gemini returns 400, throw error without calling fallback
**Actual Result:** Times out, then throws assertion error with wrong message
**Root Cause:**
- Incorrect async expectation pattern: `expect(agent.startSession(session)).rejects.toThrow()` should be `await expect(agent.startSession(session)).rejects.toThrow()`
- The missing `await` causes the test to not wait for rejection, timing out instead
### Test 6: "should respect rate limits when billing disabled"
**Status:** FAIL
**Expected Behavior:** When `CLAUDE_MEM_GEMINI_BILLING_ENABLED` is 'false', enforce rate limiting via setTimeout
**Actual Result:** Error - "Cannot store observations: memorySessionId not yet captured"
**Root Cause:**
- Test session fixture missing `memorySessionId` field
- Rate limiting test never reaches the code path because session validation fails first
---
## 3. Current Implementation Status
### GeminiAgent.ts
- Located at: `/Users/alexnewman/Scripts/claude-mem/src/services/worker/GeminiAgent.ts`
- Uses shared `processAgentResponse()` from ResponseProcessor module
- Properly validates `memorySessionId` before storage (line 71 in ResponseProcessor.ts)
### ResponseProcessor.ts
- Located at: `/Users/alexnewman/Scripts/claude-mem/src/services/worker/agents/ResponseProcessor.ts`
- Contains strict validation at lines 70-73:
```typescript
if (!session.memorySessionId) {
throw new Error('Cannot store observations: memorySessionId not yet captured');
}
```
### FallbackErrorHandler.ts
- Contains `FALLBACK_ERROR_PATTERNS` that trigger Claude fallback: `['429', '500', '502', '503', 'ECONNREFUSED', 'ETIMEDOUT', 'fetch failed']`
- 400 errors are intentionally NOT in this list (should throw, not fallback)
---
## 4. Root Cause Analysis
### 4.1 Session Fixture Incomplete
**All test sessions are missing `memorySessionId`:**
```typescript
const session = {
sessionDbId: 1,
claudeSessionId: 'test-session', // Wrong field name
sdkSessionId: 'test-sdk', // Wrong field name
// ... other fields
} as any; // Type assertion masks the error
```
The `ActiveSession` type defines:
- `contentSessionId: string` (user's Claude Code session)
- `memorySessionId: string | null` (memory agent's session ID)
But tests use:
- `claudeSessionId` (deprecated name)
- `sdkSessionId` (deprecated name)
- No `memorySessionId` field at all
### 4.2 Mock Module Scoping
The `mock.module()` call appears before imports but may not be correctly intercepting:
```typescript
mock.module('../src/shared/SettingsDefaultsManager', () => ({...}));
```
Evidence: Test 4 makes a real API call to Gemini with the mock API key `'test-api-key'`, receiving:
```
"message": "API key not valid. Please pass a valid API key."
```
This indicates `getGeminiConfig()` is reading the mock settings, but `global.fetch` is not being mocked before the agent initialization.
### 4.3 Async Assertion Syntax Error
Test 5 uses incorrect async rejection pattern:
```typescript
// WRONG - missing await
expect(agent.startSession(session)).rejects.toThrow('Gemini API error: 400 - Invalid argument');
// CORRECT
await expect(agent.startSession(session)).rejects.toThrow('Gemini API error: 400 - Invalid argument');
```
Without `await`, the test continues and times out instead of catching the rejection.
### 4.4 Mock Ordering Issue
The `global.fetch` mock is set AFTER agent construction in `beforeEach`:
```typescript
beforeEach(() => {
// ... mock setup
agent = new GeminiAgent(mockDbManager, mockSessionManager);
originalFetch = global.fetch; // Save original
});
```
But tests set the fetch mock in the test body AFTER agent exists. While this should work for the API call, the timing may cause race conditions.
---
## 5. Recommended Fixes
### 5.1 Fix Session Fixtures (Priority: HIGH, Effort: LOW)
Add `memorySessionId` and use correct field names in all test sessions:
```typescript
const session = {
sessionDbId: 1,
contentSessionId: 'test-session', // Correct field name
memorySessionId: 'mem-session-123', // REQUIRED - add this
project: 'test-project',
userPrompt: 'test prompt',
conversationHistory: [],
lastPromptNumber: 1,
cumulativeInputTokens: 0,
cumulativeOutputTokens: 0,
pendingProcessingIds: new Set(),
pendingMessages: [], // Add missing field
abortController: new AbortController(), // Add missing field
generatorPromise: null, // Add missing field
earliestPendingTimestamp: null, // Add missing field
currentProvider: null, // Add missing field
startTime: Date.now()
} satisfies ActiveSession; // Use satisfies instead of 'as any'
```
### 5.2 Fix Mock Module Path (Priority: HIGH, Effort: LOW)
The mock path may be incorrect. Test imports use:
```typescript
import { SettingsDefaultsManager } from '../src/shared/SettingsDefaultsManager';
```
But the agent imports:
```typescript
import { SettingsDefaultsManager } from '../../shared/SettingsDefaultsManager.js';
```
Consider creating a shared test fixture or using dependency injection.
### 5.3 Fix Async Assertion (Priority: MEDIUM, Effort: LOW)
In test 5 "should NOT fallback on other errors":
```typescript
// Change from:
expect(agent.startSession(session)).rejects.toThrow('Gemini API error: 400 - Invalid argument');
// To:
await expect(agent.startSession(session)).rejects.toThrow('Gemini API error: 400');
```
### 5.4 Move Fetch Mock to beforeEach (Priority: MEDIUM, Effort: LOW)
Set default mock in beforeEach, override in specific tests:
```typescript
beforeEach(() => {
originalFetch = global.fetch;
// Default successful mock
global.fetch = mock(() => Promise.resolve(new Response(JSON.stringify({
candidates: [{ content: { parts: [{ text: '<observation><type>discovery</type><title>Test</title></observation>' }] } }],
usageMetadata: { totalTokenCount: 100 }
}))));
// ... rest of setup
});
```
### 5.5 Add Logger Mock (Priority: LOW, Effort: LOW)
The logger is trying to load settings during test execution:
```
TypeError: undefined is not an object (evaluating 'SettingsDefaultsManager.loadFromFile(settingsPath).CLAUDE_MEM_LOG_LEVEL.toUpperCase')
```
Mock the logger or extend SettingsDefaultsManager mock to handle `get()` calls:
```typescript
mock.module('../src/shared/SettingsDefaultsManager', () => ({
SettingsDefaultsManager: {
loadFromFile: () => ({
CLAUDE_MEM_GEMINI_API_KEY: 'test-api-key',
CLAUDE_MEM_GEMINI_MODEL: 'gemini-2.5-flash-lite',
CLAUDE_MEM_GEMINI_BILLING_ENABLED: billingEnabled,
CLAUDE_MEM_LOG_LEVEL: 'INFO', // Add this
CLAUDE_MEM_GEMINI_RATE_LIMITING_ENABLED: 'true' // Add this
}),
get: (key: string) => {
if (key === 'CLAUDE_MEM_LOG_LEVEL') return 'INFO';
if (key === 'CLAUDE_MEM_DATA_DIR') return '/tmp/test-claude-mem';
return '';
}
}
}));
```
---
## 6. Priority/Effort Matrix
| Fix | Priority | Effort | Impact |
|-----|----------|--------|--------|
| 5.1 Add memorySessionId to fixtures | HIGH | LOW | Fixes 4/6 tests immediately |
| 5.2 Fix mock module path | HIGH | LOW | Ensures mocks apply correctly |
| 5.3 Fix async assertion syntax | MEDIUM | LOW | Fixes test 5 timeout |
| 5.4 Move fetch mock to beforeEach | MEDIUM | LOW | Prevents race conditions |
| 5.5 Add logger mock | LOW | LOW | Cleaner test output |
### Recommended Order of Implementation:
1. **Fix session fixtures** (5.1) - This alone will likely fix tests 1, 3, and 6
2. **Fix async assertion** (5.3) - Will fix test 5 timeout
3. **Add logger mock** (5.5) - Prevents spurious errors in test output
4. **Fix mock module path** (5.2) - May fix test 4 if mocks aren't applying
5. **Move fetch mock** (5.4) - Prevents future flakiness
---
## 7. Appendix: Full Error Output
### Test 1 Error:
```
error: Cannot store observations: memorySessionId not yet captured
at processAgentResponse (ResponseProcessor.ts:72:11)
```
### Test 4 Error:
```
error: Gemini API error: 400 - {
"error": {
"code": 400,
"message": "API key not valid. Please pass a valid API key.",
"status": "INVALID_ARGUMENT"
}
}
```
### Test 5 Error:
```
error: Test "should NOT fallback on other errors" timed out after 5001ms
Expected substring: "Gemini API error: 400 - Invalid argument"
Received message: "Gemini API error: 400 - {...API key not valid...}"
```
---
## 8. Related Files
- `/Users/alexnewman/Scripts/claude-mem/tests/gemini_agent.test.ts` - Test file
- `/Users/alexnewman/Scripts/claude-mem/src/services/worker/GeminiAgent.ts` - Implementation
- `/Users/alexnewman/Scripts/claude-mem/src/services/worker/agents/ResponseProcessor.ts` - Shared processor
- `/Users/alexnewman/Scripts/claude-mem/src/services/worker/agents/FallbackErrorHandler.ts` - Fallback logic
- `/Users/alexnewman/Scripts/claude-mem/src/services/worker-types.ts` - ActiveSession type definition