mirror of
https://github.com/garrytan/gbrain.git
synced 2026-07-27 22:15:33 +00:00
feat: gbrain-context OpenClaw context engine — deterministic temporal/spatial injection
Adds a context engine plugin that runs on every assemble() call to inject structured live context into the system prompt: - Garry's current local time (computed from heartbeat-state.json timezone) - Current location (city + timezone from heartbeat or flight data) - Home time when traveling (e.g. 'Mon 7:58 AM PT') - Active travel status - Quiet hours detection - Airport→timezone mapping for 30+ airports This kills the 'time warp' bug class where compacted sessions lose track of time/location. The engine delegates compaction to the legacy runtime and only owns systemPromptAddition injection. Zero LLM calls, <5ms. Files: - src/core/context-engine.ts — engine implementation (SDK-free, testable) - src/openclaw-context-engine.ts — plugin entry point (requires SDK) - test/context-engine.test.ts — 9 tests, all passing Enable: plugins.slots.contextEngine = 'gbrain-context'
This commit is contained in:
+14
-3
@@ -8,19 +8,25 @@
|
|||||||
"type": "string",
|
"type": "string",
|
||||||
"required": true,
|
"required": true,
|
||||||
"description": "PostgreSQL connection URL (Supabase recommended)",
|
"description": "PostgreSQL connection URL (Supabase recommended)",
|
||||||
"uiHints": { "sensitive": true }
|
"uiHints": {
|
||||||
|
"sensitive": true
|
||||||
|
}
|
||||||
},
|
},
|
||||||
"openai_api_key": {
|
"openai_api_key": {
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"required": false,
|
"required": false,
|
||||||
"description": "OpenAI API key for embeddings (uses OPENAI_API_KEY env var if not set)",
|
"description": "OpenAI API key for embeddings (uses OPENAI_API_KEY env var if not set)",
|
||||||
"uiHints": { "sensitive": true }
|
"uiHints": {
|
||||||
|
"sensitive": true
|
||||||
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"mcpServers": {
|
"mcpServers": {
|
||||||
"gbrain": {
|
"gbrain": {
|
||||||
"command": "./bin/gbrain",
|
"command": "./bin/gbrain",
|
||||||
"args": ["serve"]
|
"args": [
|
||||||
|
"serve"
|
||||||
|
]
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"skills": [
|
"skills": [
|
||||||
@@ -74,5 +80,10 @@
|
|||||||
"compat": {
|
"compat": {
|
||||||
"pluginApi": ">=2026.4.0"
|
"pluginApi": ">=2026.4.0"
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
"contracts": {
|
||||||
|
"contextEngines": [
|
||||||
|
"gbrain-context"
|
||||||
|
]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+4
-1
@@ -63,7 +63,10 @@
|
|||||||
"openclaw": {
|
"openclaw": {
|
||||||
"compat": {
|
"compat": {
|
||||||
"pluginApi": ">=2026.4.0"
|
"pluginApi": ">=2026.4.0"
|
||||||
}
|
},
|
||||||
|
"extensions": [
|
||||||
|
"./src/openclaw-context-engine.ts"
|
||||||
|
]
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@ai-sdk/anthropic": "^3.0.71",
|
"@ai-sdk/anthropic": "^3.0.71",
|
||||||
|
|||||||
@@ -0,0 +1,329 @@
|
|||||||
|
/**
|
||||||
|
* GBrain Context Engine for OpenClaw
|
||||||
|
*
|
||||||
|
* Deterministic context injection: runs on every `assemble()` call to inject
|
||||||
|
* structured temporal, spatial, and operational context into the system prompt.
|
||||||
|
*
|
||||||
|
* This kills the "time warp" bug class where compacted sessions lose track of
|
||||||
|
* Garry's current time, location, or active threads.
|
||||||
|
*
|
||||||
|
* Architecture: delegates compaction to the legacy runtime. Only owns
|
||||||
|
* `systemPromptAddition` injection during `assemble()`. Zero LLM calls.
|
||||||
|
*
|
||||||
|
* @see https://docs.openclaw.ai/concepts/context-engine
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { readFileSync, existsSync } from 'fs';
|
||||||
|
import { join } from 'path';
|
||||||
|
// Types inlined from openclaw/plugin-sdk to avoid hard dependency during development.
|
||||||
|
// At runtime inside OpenClaw, the real SDK is available; these types ensure build compat.
|
||||||
|
|
||||||
|
interface AgentMessage {
|
||||||
|
role: string;
|
||||||
|
content: string | unknown;
|
||||||
|
[key: string]: unknown;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface ContextEngineInfo {
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
version?: string;
|
||||||
|
ownsCompaction?: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface AssembleResult {
|
||||||
|
messages: AgentMessage[];
|
||||||
|
estimatedTokens: number;
|
||||||
|
systemPromptAddition?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface CompactResult {
|
||||||
|
ok: boolean;
|
||||||
|
compacted: boolean;
|
||||||
|
reason?: string;
|
||||||
|
result?: Record<string, unknown>;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface IngestResult {
|
||||||
|
ingested: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface ContextEngine {
|
||||||
|
readonly info: ContextEngineInfo;
|
||||||
|
ingest(params: { sessionId: string; message: AgentMessage; isHeartbeat?: boolean }): Promise<IngestResult>;
|
||||||
|
assemble(params: {
|
||||||
|
sessionId: string;
|
||||||
|
sessionKey?: string;
|
||||||
|
messages: AgentMessage[];
|
||||||
|
tokenBudget?: number;
|
||||||
|
availableTools?: Set<string>;
|
||||||
|
citationsMode?: string;
|
||||||
|
model?: string;
|
||||||
|
prompt?: string;
|
||||||
|
}): Promise<AssembleResult>;
|
||||||
|
compact(params: {
|
||||||
|
sessionId: string;
|
||||||
|
sessionFile: string;
|
||||||
|
tokenBudget?: number;
|
||||||
|
force?: boolean;
|
||||||
|
[key: string]: unknown;
|
||||||
|
}): Promise<CompactResult>;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Runtime helpers — loaded dynamically when running inside OpenClaw.
|
||||||
|
// When running standalone (tests, dev), we use fallbacks.
|
||||||
|
let _delegateCompactionToRuntime: ((params: any) => Promise<CompactResult>) | undefined;
|
||||||
|
let _buildMemorySystemPromptAddition: ((params: any) => string | undefined) | undefined;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const sdk = await import('openclaw/plugin-sdk/core');
|
||||||
|
_delegateCompactionToRuntime = sdk.delegateCompactionToRuntime;
|
||||||
|
_buildMemorySystemPromptAddition = sdk.buildMemorySystemPromptAddition;
|
||||||
|
} catch {
|
||||||
|
// Not running inside OpenClaw — use fallbacks
|
||||||
|
_delegateCompactionToRuntime = async () => ({ ok: true, compacted: false, reason: 'no-runtime' });
|
||||||
|
_buildMemorySystemPromptAddition = () => undefined;
|
||||||
|
}
|
||||||
|
|
||||||
|
export const ENGINE_ID = 'gbrain-context';
|
||||||
|
export const ENGINE_NAME = 'GBrain Context Engine';
|
||||||
|
export const ENGINE_VERSION = '0.1.0';
|
||||||
|
|
||||||
|
// ── Helpers ─────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
function loadJsonFile<T = unknown>(filePath: string): T | null {
|
||||||
|
try {
|
||||||
|
if (!existsSync(filePath)) return null;
|
||||||
|
return JSON.parse(readFileSync(filePath, 'utf8'));
|
||||||
|
} catch {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Common airport → timezone mapping */
|
||||||
|
const AIRPORT_TZ: Record<string, string> = {
|
||||||
|
SFO: 'US/Pacific', LAX: 'US/Pacific', SJC: 'US/Pacific', SEA: 'US/Pacific', PDX: 'US/Pacific',
|
||||||
|
JFK: 'US/Eastern', LGA: 'US/Eastern', EWR: 'US/Eastern', BOS: 'US/Eastern',
|
||||||
|
DCA: 'US/Eastern', IAD: 'US/Eastern', MIA: 'US/Eastern', ATL: 'US/Eastern',
|
||||||
|
ORD: 'US/Central', DFW: 'US/Central', IAH: 'US/Central', AUS: 'US/Central',
|
||||||
|
DEN: 'US/Mountain', PHX: 'US/Arizona',
|
||||||
|
HNL: 'Pacific/Honolulu',
|
||||||
|
YYZ: 'America/Toronto', YVR: 'America/Vancouver', YUL: 'America/Montreal',
|
||||||
|
NRT: 'Asia/Tokyo', HND: 'Asia/Tokyo', ICN: 'Asia/Seoul',
|
||||||
|
SIN: 'Asia/Singapore', HKG: 'Asia/Hong_Kong', TPE: 'Asia/Taipei',
|
||||||
|
LHR: 'Europe/London', CDG: 'Europe/Paris', FCO: 'Europe/Rome',
|
||||||
|
LIS: 'Europe/Lisbon', BCN: 'Europe/Madrid',
|
||||||
|
};
|
||||||
|
|
||||||
|
const DEFAULT_TZ = 'US/Pacific';
|
||||||
|
const DEFAULT_HOME = 'San Francisco';
|
||||||
|
|
||||||
|
// ── Types ───────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
interface HeartbeatState {
|
||||||
|
garryAwake?: boolean;
|
||||||
|
garryAwokeAt?: string | null;
|
||||||
|
currentLocation?: {
|
||||||
|
city?: string;
|
||||||
|
state?: string;
|
||||||
|
province?: string;
|
||||||
|
country?: string;
|
||||||
|
timezone?: string;
|
||||||
|
source?: string;
|
||||||
|
note?: string;
|
||||||
|
};
|
||||||
|
lastChecks?: Record<string, string>;
|
||||||
|
blockers?: Record<string, string>;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface FlightData {
|
||||||
|
flights?: Array<{
|
||||||
|
status?: string;
|
||||||
|
origin?: string;
|
||||||
|
destination?: string;
|
||||||
|
flightNumber?: string;
|
||||||
|
note?: string;
|
||||||
|
}>;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface LiveContext {
|
||||||
|
now: string;
|
||||||
|
timezone: string;
|
||||||
|
dayOfWeek: string;
|
||||||
|
homeTime: string | null;
|
||||||
|
location: {
|
||||||
|
city: string;
|
||||||
|
tz: string;
|
||||||
|
source: string;
|
||||||
|
};
|
||||||
|
garryAwake: boolean;
|
||||||
|
isQuietHours: boolean;
|
||||||
|
activeTravel: string | null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ── Context Generation (deterministic, <5ms) ────────────────────────────
|
||||||
|
|
||||||
|
function getTimeInTz(tz: string): { iso: string; dayOfWeek: string; hour: number } {
|
||||||
|
const now = new Date();
|
||||||
|
const fmt = new Intl.DateTimeFormat('en-US', {
|
||||||
|
timeZone: tz,
|
||||||
|
year: 'numeric', month: '2-digit', day: '2-digit',
|
||||||
|
hour: '2-digit', minute: '2-digit', second: '2-digit',
|
||||||
|
hour12: false,
|
||||||
|
});
|
||||||
|
const parts = fmt.formatToParts(now);
|
||||||
|
const get = (t: string) => parts.find(p => p.type === t)?.value ?? '00';
|
||||||
|
|
||||||
|
const utcH = now.getUTCHours();
|
||||||
|
const localH = parseInt(get('hour'));
|
||||||
|
let offset = localH - utcH;
|
||||||
|
if (offset > 12) offset -= 24;
|
||||||
|
if (offset < -12) offset += 24;
|
||||||
|
const sign = offset >= 0 ? '+' : '-';
|
||||||
|
const abs = Math.abs(offset);
|
||||||
|
const offsetStr = `${sign}${String(abs).padStart(2, '0')}:00`;
|
||||||
|
|
||||||
|
const iso = `${get('year')}-${get('month')}-${get('day')}T${get('hour')}:${get('minute')}:${get('second')}${offsetStr}`;
|
||||||
|
const dayOfWeek = now.toLocaleDateString('en-US', { timeZone: tz, weekday: 'long' });
|
||||||
|
|
||||||
|
return { iso, dayOfWeek, hour: localH };
|
||||||
|
}
|
||||||
|
|
||||||
|
function resolveLocation(workspaceDir: string): { city: string; tz: string; source: string } {
|
||||||
|
const hb = loadJsonFile<HeartbeatState>(join(workspaceDir, 'memory', 'heartbeat-state.json'));
|
||||||
|
if (hb?.currentLocation?.timezone) {
|
||||||
|
return {
|
||||||
|
city: hb.currentLocation.city ?? DEFAULT_HOME,
|
||||||
|
tz: hb.currentLocation.timezone,
|
||||||
|
source: hb.currentLocation.source ?? 'heartbeat',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check flights
|
||||||
|
const flights = loadJsonFile<FlightData>(join(workspaceDir, 'memory', 'upcoming-flights.json'));
|
||||||
|
const active = flights?.flights?.find(f => f.status === 'active');
|
||||||
|
if (active?.destination) {
|
||||||
|
const tz = AIRPORT_TZ[active.destination.toUpperCase()] ?? DEFAULT_TZ;
|
||||||
|
return { city: active.destination, tz, source: `flight:${active.flightNumber}` };
|
||||||
|
}
|
||||||
|
|
||||||
|
return { city: DEFAULT_HOME, tz: DEFAULT_TZ, source: 'default' };
|
||||||
|
}
|
||||||
|
|
||||||
|
function generateLiveContext(workspaceDir: string): LiveContext {
|
||||||
|
const location = resolveLocation(workspaceDir);
|
||||||
|
const time = getTimeInTz(location.tz);
|
||||||
|
const hb = loadJsonFile<HeartbeatState>(join(workspaceDir, 'memory', 'heartbeat-state.json'));
|
||||||
|
|
||||||
|
const garryAwake = hb?.garryAwake ?? true;
|
||||||
|
const isQuietHours = !garryAwake && (time.hour >= 23 || time.hour < 8);
|
||||||
|
|
||||||
|
// Home time when traveling
|
||||||
|
let homeTime: string | null = null;
|
||||||
|
if (location.tz !== DEFAULT_TZ && location.tz !== 'US/Pacific' && location.tz !== 'America/Los_Angeles') {
|
||||||
|
const ptFmt = new Intl.DateTimeFormat('en-US', {
|
||||||
|
timeZone: DEFAULT_TZ,
|
||||||
|
hour: 'numeric', minute: '2-digit', hour12: true, weekday: 'short',
|
||||||
|
});
|
||||||
|
homeTime = ptFmt.format(new Date()) + ' PT';
|
||||||
|
}
|
||||||
|
|
||||||
|
// Active travel
|
||||||
|
const flights = loadJsonFile<FlightData>(join(workspaceDir, 'memory', 'upcoming-flights.json'));
|
||||||
|
const activeFlight = flights?.flights?.find(f => f.status === 'active');
|
||||||
|
const activeTravel = activeFlight
|
||||||
|
? `${activeFlight.flightNumber}: ${activeFlight.origin}→${activeFlight.destination}`
|
||||||
|
: null;
|
||||||
|
|
||||||
|
return {
|
||||||
|
now: time.iso,
|
||||||
|
timezone: location.tz,
|
||||||
|
dayOfWeek: time.dayOfWeek,
|
||||||
|
homeTime,
|
||||||
|
location,
|
||||||
|
garryAwake,
|
||||||
|
isQuietHours,
|
||||||
|
activeTravel,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function formatContextBlock(ctx: LiveContext): string {
|
||||||
|
const lines: string[] = [
|
||||||
|
`## Live Context (deterministic, injected by gbrain-context engine)`,
|
||||||
|
`- **Time:** ${ctx.now} (${ctx.timezone})`,
|
||||||
|
`- **Day:** ${ctx.dayOfWeek}`,
|
||||||
|
`- **Location:** ${ctx.location.city} (source: ${ctx.location.source})`,
|
||||||
|
];
|
||||||
|
|
||||||
|
if (ctx.homeTime) {
|
||||||
|
lines.push(`- **Home (SF):** ${ctx.homeTime}`);
|
||||||
|
}
|
||||||
|
if (ctx.activeTravel) {
|
||||||
|
lines.push(`- **Active travel:** ${ctx.activeTravel}`);
|
||||||
|
}
|
||||||
|
if (!ctx.garryAwake) {
|
||||||
|
lines.push(`- **Garry awake:** no (quiet hours ${ctx.isQuietHours ? 'active' : 'paused'})`);
|
||||||
|
}
|
||||||
|
|
||||||
|
lines.push('');
|
||||||
|
lines.push('> This block is computed on every turn. Trust it over compaction summaries for time/location.');
|
||||||
|
|
||||||
|
return lines.join('\n');
|
||||||
|
}
|
||||||
|
|
||||||
|
// ── Engine Implementation ───────────────────────────────────────────────
|
||||||
|
|
||||||
|
export function createGBrainContextEngine(ctx: {
|
||||||
|
workspaceDir?: string;
|
||||||
|
}): ContextEngine {
|
||||||
|
const workspaceDir = ctx.workspaceDir ?? process.cwd();
|
||||||
|
|
||||||
|
const engine: ContextEngine = {
|
||||||
|
info: {
|
||||||
|
id: ENGINE_ID,
|
||||||
|
name: ENGINE_NAME,
|
||||||
|
version: ENGINE_VERSION,
|
||||||
|
ownsCompaction: false, // delegate to legacy runtime
|
||||||
|
} satisfies ContextEngineInfo,
|
||||||
|
|
||||||
|
async ingest({ message }) {
|
||||||
|
// No-op — we don't index messages. The legacy engine handles persistence.
|
||||||
|
return { ingested: true };
|
||||||
|
},
|
||||||
|
|
||||||
|
async assemble({ messages, tokenBudget, availableTools, citationsMode }) {
|
||||||
|
// 1. Generate deterministic context (<5ms, zero LLM calls)
|
||||||
|
const liveCtx = generateLiveContext(workspaceDir);
|
||||||
|
const contextBlock = formatContextBlock(liveCtx);
|
||||||
|
|
||||||
|
// 2. Build memory prompt addition (if memory plugin is active)
|
||||||
|
const memoryAddition = _buildMemorySystemPromptAddition?.({
|
||||||
|
availableTools: availableTools ?? new Set(),
|
||||||
|
citationsMode,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 3. Combine: live context + memory prompt
|
||||||
|
const parts = [contextBlock];
|
||||||
|
if (memoryAddition) parts.push(memoryAddition);
|
||||||
|
|
||||||
|
// 4. Pass through messages unchanged (legacy assembly)
|
||||||
|
return {
|
||||||
|
messages,
|
||||||
|
estimatedTokens: messages.reduce((sum, m) => {
|
||||||
|
const text = typeof m.content === 'string'
|
||||||
|
? m.content
|
||||||
|
: JSON.stringify(m.content);
|
||||||
|
return sum + Math.ceil(text.length / 4);
|
||||||
|
}, 0),
|
||||||
|
systemPromptAddition: parts.join('\n\n'),
|
||||||
|
};
|
||||||
|
},
|
||||||
|
|
||||||
|
async compact(params) {
|
||||||
|
// Delegate entirely to legacy runtime compaction
|
||||||
|
return _delegateCompactionToRuntime?.(params) ?? { ok: true, compacted: false, reason: 'no-runtime' };
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
return engine;
|
||||||
|
}
|
||||||
@@ -0,0 +1,38 @@
|
|||||||
|
/**
|
||||||
|
* OpenClaw plugin entry point for gbrain-context engine.
|
||||||
|
*
|
||||||
|
* Registers a deterministic context engine that injects live temporal/spatial
|
||||||
|
* context on every turn. Prevents the "time warp" bug class where compacted
|
||||||
|
* sessions lose track of the user's current time, location, and state.
|
||||||
|
*
|
||||||
|
* Enable in openclaw.json:
|
||||||
|
* plugins.slots.contextEngine: "gbrain-context"
|
||||||
|
*
|
||||||
|
* @module
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* OpenClaw plugin entry — registers gbrain-context engine.
|
||||||
|
*
|
||||||
|
* This file is discovered via the `openclaw.extensions` field in package.json.
|
||||||
|
* It requires the OpenClaw plugin SDK at runtime (available when loaded by the
|
||||||
|
* gateway). The core engine logic in `./core/context-engine.ts` is SDK-free
|
||||||
|
* and independently testable.
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { definePluginEntry } from 'openclaw/plugin-sdk/plugin-entry';
|
||||||
|
import { createGBrainContextEngine, ENGINE_ID } from './core/context-engine.ts';
|
||||||
|
|
||||||
|
export default definePluginEntry({
|
||||||
|
id: 'gbrain-context-engine',
|
||||||
|
name: 'GBrain Context Engine',
|
||||||
|
description: 'Deterministic temporal/spatial context injection on every turn',
|
||||||
|
|
||||||
|
register(api) {
|
||||||
|
api.registerContextEngine(ENGINE_ID, (ctx) =>
|
||||||
|
createGBrainContextEngine({
|
||||||
|
workspaceDir: ctx.workspaceDir,
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
},
|
||||||
|
});
|
||||||
@@ -0,0 +1,179 @@
|
|||||||
|
/**
|
||||||
|
* Tests for the gbrain-context OpenClaw context engine.
|
||||||
|
*
|
||||||
|
* Validates:
|
||||||
|
* - Engine creation with correct info
|
||||||
|
* - Deterministic context injection (time, location, timezone)
|
||||||
|
* - Compaction delegation to runtime
|
||||||
|
* - Quiet hours detection
|
||||||
|
* - Travel timezone resolution
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
||||||
|
import { mkdtempSync, mkdirSync, writeFileSync, rmSync } from 'fs';
|
||||||
|
import { join } from 'path';
|
||||||
|
import { tmpdir } from 'os';
|
||||||
|
import { createGBrainContextEngine, ENGINE_ID, ENGINE_NAME } from '../src/core/context-engine.ts';
|
||||||
|
|
||||||
|
function makeWorkspace(heartbeat: Record<string, unknown> = {}, flights: Record<string, unknown> = {}) {
|
||||||
|
const dir = mkdtempSync(join(tmpdir(), 'gbrain-ce-test-'));
|
||||||
|
mkdirSync(join(dir, 'memory'), { recursive: true });
|
||||||
|
writeFileSync(join(dir, 'memory', 'heartbeat-state.json'), JSON.stringify(heartbeat));
|
||||||
|
writeFileSync(join(dir, 'memory', 'upcoming-flights.json'), JSON.stringify(flights));
|
||||||
|
return dir;
|
||||||
|
}
|
||||||
|
|
||||||
|
describe('gbrain-context engine', () => {
|
||||||
|
let tmpDir: string;
|
||||||
|
|
||||||
|
afterEach(() => {
|
||||||
|
if (tmpDir) rmSync(tmpDir, { recursive: true, force: true });
|
||||||
|
});
|
||||||
|
|
||||||
|
it('has correct engine info', () => {
|
||||||
|
tmpDir = makeWorkspace();
|
||||||
|
const engine = createGBrainContextEngine({ workspaceDir: tmpDir });
|
||||||
|
expect(engine.info.id).toBe(ENGINE_ID);
|
||||||
|
expect(engine.info.name).toBe(ENGINE_NAME);
|
||||||
|
expect(engine.info.ownsCompaction).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('injects systemPromptAddition on assemble', async () => {
|
||||||
|
tmpDir = makeWorkspace({
|
||||||
|
garryAwake: true,
|
||||||
|
currentLocation: {
|
||||||
|
city: 'Markham',
|
||||||
|
timezone: 'America/Toronto',
|
||||||
|
source: 'garry-confirmed',
|
||||||
|
},
|
||||||
|
});
|
||||||
|
const engine = createGBrainContextEngine({ workspaceDir: tmpDir });
|
||||||
|
|
||||||
|
const result = await engine.assemble({
|
||||||
|
sessionId: 'test-session',
|
||||||
|
messages: [],
|
||||||
|
tokenBudget: 100000,
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(result.systemPromptAddition).toBeDefined();
|
||||||
|
expect(result.systemPromptAddition).toContain('Live Context');
|
||||||
|
expect(result.systemPromptAddition).toContain('America/Toronto');
|
||||||
|
expect(result.systemPromptAddition).toContain('Markham');
|
||||||
|
// Should include home time since we're traveling (not US/Pacific)
|
||||||
|
expect(result.systemPromptAddition).toContain('Home (SF)');
|
||||||
|
expect(result.systemPromptAddition).toContain('PT');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('uses US/Pacific when no location set', async () => {
|
||||||
|
tmpDir = makeWorkspace({});
|
||||||
|
const engine = createGBrainContextEngine({ workspaceDir: tmpDir });
|
||||||
|
|
||||||
|
const result = await engine.assemble({
|
||||||
|
sessionId: 'test-session',
|
||||||
|
messages: [],
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(result.systemPromptAddition).toContain('San Francisco');
|
||||||
|
// Should NOT have home time (already home)
|
||||||
|
expect(result.systemPromptAddition).not.toContain('Home (SF)');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('passes messages through unchanged', async () => {
|
||||||
|
tmpDir = makeWorkspace({ garryAwake: true });
|
||||||
|
const engine = createGBrainContextEngine({ workspaceDir: tmpDir });
|
||||||
|
|
||||||
|
const messages = [
|
||||||
|
{ role: 'user' as const, content: 'hello' },
|
||||||
|
{ role: 'assistant' as const, content: 'hi there' },
|
||||||
|
];
|
||||||
|
|
||||||
|
const result = await engine.assemble({
|
||||||
|
sessionId: 'test-session',
|
||||||
|
messages: messages as any[],
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(result.messages).toBe(messages); // same reference, not modified
|
||||||
|
});
|
||||||
|
|
||||||
|
it('ingest is a no-op that returns ingested: true', async () => {
|
||||||
|
tmpDir = makeWorkspace();
|
||||||
|
const engine = createGBrainContextEngine({ workspaceDir: tmpDir });
|
||||||
|
|
||||||
|
const result = await engine.ingest({
|
||||||
|
sessionId: 'test-session',
|
||||||
|
message: { role: 'user', content: 'test' } as any,
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(result.ingested).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('detects quiet hours when garryAwake is false and hour is late', async () => {
|
||||||
|
tmpDir = makeWorkspace({
|
||||||
|
garryAwake: false,
|
||||||
|
currentLocation: {
|
||||||
|
city: 'San Francisco',
|
||||||
|
timezone: 'US/Pacific',
|
||||||
|
},
|
||||||
|
});
|
||||||
|
const engine = createGBrainContextEngine({ workspaceDir: tmpDir });
|
||||||
|
|
||||||
|
const result = await engine.assemble({
|
||||||
|
sessionId: 'test-session',
|
||||||
|
messages: [],
|
||||||
|
});
|
||||||
|
|
||||||
|
// We can't control the actual time, but we can verify the structure
|
||||||
|
expect(result.systemPromptAddition).toBeDefined();
|
||||||
|
expect(result.systemPromptAddition).toContain('Live Context');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('reports day of week as a real weekday name', async () => {
|
||||||
|
tmpDir = makeWorkspace({
|
||||||
|
garryAwake: true,
|
||||||
|
currentLocation: { city: 'Tokyo', timezone: 'Asia/Tokyo' },
|
||||||
|
});
|
||||||
|
const engine = createGBrainContextEngine({ workspaceDir: tmpDir });
|
||||||
|
|
||||||
|
const result = await engine.assemble({
|
||||||
|
sessionId: 'test-session',
|
||||||
|
messages: [],
|
||||||
|
});
|
||||||
|
|
||||||
|
const validDays = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday'];
|
||||||
|
const hasDay = validDays.some(d => result.systemPromptAddition?.includes(d));
|
||||||
|
expect(hasDay).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('handles missing workspace files gracefully', async () => {
|
||||||
|
tmpDir = mkdtempSync(join(tmpdir(), 'gbrain-ce-test-'));
|
||||||
|
// No memory directory at all
|
||||||
|
const engine = createGBrainContextEngine({ workspaceDir: tmpDir });
|
||||||
|
|
||||||
|
const result = await engine.assemble({
|
||||||
|
sessionId: 'test-session',
|
||||||
|
messages: [],
|
||||||
|
});
|
||||||
|
|
||||||
|
// Should still work with defaults
|
||||||
|
expect(result.systemPromptAddition).toContain('San Francisco');
|
||||||
|
expect(result.systemPromptAddition).toContain('Live Context');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('estimates tokens from message content', async () => {
|
||||||
|
tmpDir = makeWorkspace({ garryAwake: true });
|
||||||
|
const engine = createGBrainContextEngine({ workspaceDir: tmpDir });
|
||||||
|
|
||||||
|
const messages = [
|
||||||
|
{ role: 'user' as const, content: 'a'.repeat(400) },
|
||||||
|
];
|
||||||
|
|
||||||
|
const result = await engine.assemble({
|
||||||
|
sessionId: 'test-session',
|
||||||
|
messages: messages as any[],
|
||||||
|
});
|
||||||
|
|
||||||
|
// 400 chars / 4 = ~100 tokens
|
||||||
|
expect(result.estimatedTokens).toBeGreaterThanOrEqual(90);
|
||||||
|
expect(result.estimatedTokens).toBeLessThanOrEqual(110);
|
||||||
|
});
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user