@shadowforge0/aquifer-memory 1.7.0 → 1.8.1

package/consumers/shared/config.js

@@ -8,6 +8,17 @@ const path = require('path');
 // ---------------------------------------------------------------------------
 
 const DEFAULTS = {
+  storage: {
+    backend: 'postgres',
+    postgres: {
+      url: null,
+      max: null,
+      idleTimeoutMs: null,
+    },
+    local: {
+      path: '.aquifer/aquifer.local.json',
+    },
+  },
   db: { url: null, max: 10, idleTimeoutMs: 30000 },
   schema: 'aquifer',
   tenantId: 'default',
@@ -45,6 +56,16 @@ const DEFAULTS = {
     activeScopeKey: null,
     activeScopePath: null,
   },
+  codex: {
+    checkpoint: {
+      checkIntervalMs: null,
+      checkIntervalMinutes: 10,
+      everyMessages: 20,
+      everyUserMessages: null,
+      quietMs: 3000,
+      claimTtlMs: 60000,
+    },
+  },
   rerank: {
     enabled: false,
     provider: null,    // 'tei' | 'jina' | 'openrouter' | 'custom'
@@ -69,6 +90,10 @@ const DEFAULTS = {
 // ---------------------------------------------------------------------------
 
 const ENV_MAP = [
+  ['AQUIFER_BACKEND',             'storage.backend'],
+  ['AQUIFER_STORAGE_BACKEND',     'storage.backend'],
+  ['AQUIFER_POSTGRES_URL',        'storage.postgres.url'],
+  ['AQUIFER_LOCAL_PATH',          'storage.local.path'],
   ['DATABASE_URL',              'db.url'],
   ['AQUIFER_DB_URL',            'db.url'],
   ['AQUIFER_DB_MAX',            'db.max',            Number],
@@ -95,6 +120,12 @@ const ENV_MAP = [
   ['AQUIFER_MEMORY_SERVING_MODE', 'memory.servingMode'],
   ['AQUIFER_MEMORY_ACTIVE_SCOPE_KEY', 'memory.activeScopeKey'],
   ['AQUIFER_MEMORY_ACTIVE_SCOPE_PATH', 'memory.activeScopePath'],
+  ['AQUIFER_CODEX_CHECKPOINT_CHECK_INTERVAL_MS', 'codex.checkpoint.checkIntervalMs', Number],
+  ['AQUIFER_CODEX_CHECKPOINT_CHECK_INTERVAL_MINUTES', 'codex.checkpoint.checkIntervalMinutes', Number],
+  ['AQUIFER_CODEX_CHECKPOINT_EVERY_MESSAGES', 'codex.checkpoint.everyMessages', Number],
+  ['AQUIFER_CODEX_CHECKPOINT_EVERY_USER_MESSAGES', 'codex.checkpoint.everyUserMessages', Number],
+  ['AQUIFER_CODEX_CHECKPOINT_QUIET_MS', 'codex.checkpoint.quietMs', Number],
+  ['AQUIFER_CODEX_CHECKPOINT_CLAIM_TTL_MS', 'codex.checkpoint.claimTtlMs', Number],
   ['AQUIFER_RERANK_ENABLED',   'rerank.enabled',    Boolean],
   ['AQUIFER_RERANK_PROVIDER',  'rerank.provider'],
   ['AQUIFER_RERANK_BASE_URL',  'rerank.baseUrl'],
@@ -141,6 +172,34 @@ function coerceEnvValue(raw, type) {
   return raw;
 }
 
+function normalizeStorageConfig(config) {
+  const storage = config.storage && typeof config.storage === 'object' ? config.storage : {};
+  const backend = String(storage.backend || 'postgres').trim().toLowerCase();
+  if (backend !== 'postgres' && backend !== 'local') {
+    throw new Error(`Invalid Aquifer backend: "${storage.backend}". Must be one of: postgres, local`);
+  }
+  const postgres = {
+    ...DEFAULTS.storage.postgres,
+    ...(storage.postgres || {}),
+  };
+  const local = {
+    ...DEFAULTS.storage.local,
+    ...(storage.local || {}),
+  };
+
+  if (postgres.url) config.db.url = postgres.url;
+  if (config.db.url && !postgres.url) postgres.url = config.db.url;
+  if (postgres.max !== null && postgres.max !== undefined) config.db.max = postgres.max;
+  if (postgres.idleTimeoutMs !== null && postgres.idleTimeoutMs !== undefined) {
+    config.db.idleTimeoutMs = postgres.idleTimeoutMs;
+  }
+  postgres.max = config.db.max;
+  postgres.idleTimeoutMs = config.db.idleTimeoutMs;
+
+  config.storage = { backend, postgres, local };
+  return config;
+}
+
 // ---------------------------------------------------------------------------
 // loadConfig
 // ---------------------------------------------------------------------------
@@ -193,7 +252,7 @@ function loadConfig(opts = {}) {
     };
   }
 
-  return config;
+  return normalizeStorageConfig(config);
 }
 
 module.exports = { loadConfig, DEFAULTS };

package/consumers/shared/factory.js

@@ -1,7 +1,6 @@
 'use strict';
 
-const { Pool } = require('pg');
-const { createAquifer, createEmbedder } = require('../../index');
+const { createLocalAquifer } = require('../../core/backends/local');
 const { loadConfig } = require('./config');
 const { createLlmFn } = require('./llm');
 
@@ -11,12 +10,18 @@ const { createLlmFn } = require('./llm');
 
 function createAquiferFromConfig(overrides) {
   const config = loadConfig({ overrides });
+  const backendKind = config.storage?.backend || 'postgres';
+
+  if (backendKind === 'local') {
+    return createLocalAquifer(config);
+  }
 
   if (!config.db.url) {
-    throw new Error('Database URL is required (set DATABASE_URL or AQUIFER_DB_URL)');
+    throw new Error('Database URL is required for PostgreSQL backend (set DATABASE_URL, AQUIFER_DB_URL, or AQUIFER_POSTGRES_URL)');
   }
 
   // Pool
+  const { Pool } = require('pg');
   const pool = new Pool({
     connectionString: config.db.url,
     max: config.db.max || 10,
@@ -26,6 +31,7 @@ function createAquiferFromConfig(overrides) {
   // Embed function (optional — lazy validation in core)
   let embedFn = null;
   if (config.embed && config.embed.baseUrl && config.embed.model) {
+    const { createEmbedder } = require('../../index');
     // Detect provider from baseUrl
     const isOllama = config.embed.baseUrl.includes('11434') || config.embed.baseUrl.includes('ollama');
     const embedder = isOllama
@@ -80,6 +86,7 @@ function createAquiferFromConfig(overrides) {
     rerankOpts = rerankConfig;
   }
 
+  const { createAquifer } = require('../../index');
   const aquifer = createAquifer({
     db: pool,
     ownsPool: true,