mirror of
https://github.com/garrytan/gbrain.git
synced 2026-07-27 22:15:33 +00:00
* feat(multi-source): thread ctx.sourceId through op handlers + engine read-surface
Closes the multi-source threading gaps that the v0.31.1.1-fixwave codex
review caught. Multi-source brains were silently misrouting writes from
every CLI/MCP-driven op (put_page, add_tag, add_link, add_timeline_entry,
revert_version, put_raw_data, etc.) because the op handlers in
operations.ts ignored ctx.sourceId. Read-side ops were arbitrary-row
under same-slug-across-sources because the engine's read methods had no
source filter.
Engine layer (D12 + D16 + D21):
- engine.ts interface: getLinks/getBacklinks/getTimeline/getRawData/
getVersions/getAllSlugs/revertToVersion/putRawData all take
opts?: { sourceId?: string }.
- pglite-engine.ts + postgres-engine.ts: two-branch query for each
read method. Without opts.sourceId, NO source filter applies
(preserves pre-v0.31.8 cross-source semantics for back-link
validators and any caller that hasn't threaded sourceId yet). With
opts.sourceId, scoped to that source — the new path used by
reconcileLinks and ctx.sourceId-aware op handlers.
Op-handler layer (D7 + D16 + D20):
- operations.ts threads ctx.sourceId through 16+ handler sites:
put_page, revert_version, put_raw_data, add_tag, remove_tag,
add_link, remove_link, add_timeline_entry, create_version,
delete_page, restore_page, get_page, get_tags, get_links,
get_backlinks, get_timeline, get_versions, get_raw_data,
get_chunks, plus reconcileLinks's tx.getLinks/getBacklinks/
addLink/removeLink and engine.getAllSlugs.
- Pattern: const sourceOpts = ctx.sourceId ? { sourceId: ctx.sourceId } : {};
When ctx.sourceId is unset, engine falls through to cross-source
view (back-compat). MCP callers populate ctx.sourceId via the
transport layer.
CLI wiring (D11 + D22):
- cli.ts: makeContext is async, calls resolveSourceId() from
src/core/source-resolver.ts:58 (the canonical 6-tier chain:
--source flag → GBRAIN_SOURCE env → .gbrain-source dotfile →
path-match → brain default → 'default'). Wrapped in try/catch
so a fresh pre-init brain still returns a clean ctx with no
sourceId set.
- commands/call.ts: runCall accepts --source <id> flag. Resolves
through the same 6-tier chain and threads to handleToolCall
via the new opts.sourceId param.
- mcp/server.ts: handleToolCall accepts opts.sourceId and threads
to buildOperationContext.
Tests (D7 + D16 + D20 regression coverage):
- test/source-id-tx-regression.test.ts: 8 new op-handler-layer
cases covering add_tag/get_tags/add_link/get_links/delete_page/
put_raw_data routing under ctx.sourceId='X' vs unset, plus
D16's two-branch back-compat invariant for getLinks (cross-
source view preserved when ctx.sourceId is unset).
Closes the codex OV-1/OV-2/OV-3 findings from the v0.31.8 plan
review. Back-compat is strictly additive: callers that don't pass
opts.sourceId see the same results they did pre-v0.31.8.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
* feat(doctor): multi_source_drift check surfaces pre-v0.30.3 misroutes
Pre-v0.30.3 putPage misrouted multi-source writes from intended source X
to (default, slug). The fix-wave fixed forward-going writes but explicitly
deferred backfilling the misrouted rows. Operators have had no signal of
this silent corruption.
Adds src/core/multi-source-drift.ts exporting findMisroutedPages(engine,
sources, opts). The heuristic walks each non-default source's local_path
and surfaces slugs that exist at (default, slug) in DB but are MISSING
from (X, slug) — unambiguous evidence of the misroute shape.
Implementation notes (codex OV12 + OV13 + D17):
- FS walk handles BOTH .md and .mdx (matches src/core/sync.ts:133, which
treats both as markdown). Walks own helper instead of importing from
extract.ts so doctor doesn't crash if local_path is unreadable
(try/catch on root statSync; ENOENT/EACCES yields zero files, NOT a
thrown error that takes down doctor).
- Single batched SQL with VALUES clause: collect all candidate slugs
into one array, then ONE LEFT JOIN against pages with source_id IN
('default', X). Materialize into Map<slug, Set<source_id>>. NOT a
per-file 20K-round-trip loop.
- Bounded by limit (10K files) AND timeoutMs (5s). Bail with
walk_truncated=true rather than letting doctor hang.
- Heuristic softened per OV12: "appears misrouted to default" with TWO
possible causes flagged (pre-v0.30.3 misroute OR source X never
completed initial sync). The doctor warning suggests verification
('gbrain sources status'), not a destructive action.
Wired into runDoctor (3b-multi-source slot, after sync_failures) AND
into doctorReportRemote (D14) so thin-client operators see the check
when 'gbrain doctor' routes through the remote MCP path. Single-source
brains skip the check entirely.
Tests: test/multi-source-drift.test.ts (7 PGLite cases) covers:
- Single-source brain → skip
- Multi-source no-misroutes → ok
- Multi-source 2 misrouted slugs → warn with sample
- Healthy same-slug-across-sources NOT a false positive (the codex
OV4 redesign case — original heuristic would have false-positived)
- FS walk hits limit → walk_truncated=true
- Unreadable local_path doesn't crash
- .mdx files walked alongside .md
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
* feat(doctor): wire multi_source_drift + wedge force-retry hint (D14 + D19)
Wires the new multi_source_drift check into both runDoctor (local) and
doctorReportRemote (thin-client remote MCP path), and extends the existing
minions_migration block to detect 3-consecutive-partials wedges and emit
gbrain apply-migrations --force-retry <v> hints (D19).
Pre-v0.31.8, operators wedged on v0.29.1 (or any future migration that
hits the apply-migrations runner's 3-consecutive-partials guard) got the
generic "Run: gbrain apply-migrations --yes" hint. That command refuses
to advance past the guard — so the hint was wrong. Codex OV-11 (and the
v0.31.1.1-fixwave commit message) flagged this, but the prior plan said
to delegate to apply-migrations.ts:statusForVersion(), which would have
re-opened a separate regression: the existing forward-progress override
at doctor.ts:303 (newer completion suppresses old partials) is
cross-version and statusForVersion is per-version only.
This commit extends the existing block in place rather than replacing it:
1. Keep the forward-progress override (lines 348-356) byte-identical so
installs that moved past an old v0.11 partial don't light up with
stale wedge alerts.
2. Add a 3-consecutive-partials detector after the stuck filter. Since
`stuck` already excludes forward-progress-superseded versions, the
wedge counter only fires on actual unresolved partials.
3. Branch the message:
- wedged.length > 0 → "WEDGED MIGRATION(s): <v>. Run: gbrain
apply-migrations --force-retry <v>" (chain with && for multiple)
- else if stuck.length > 0 → existing --yes hint
- else → no message
Same shape duplicated in doctorReportRemote so thin-client operators
see the right command on the brain host.
Plus the multi_source_drift wiring (D14): same heuristic from the
new src/core/multi-source-drift.ts library, called from both local and
remote doctor paths. Single-source brains skip. Engine-null guard on
the local path (--fast and DB-down branches pass null).
Tests: test/doctor.test.ts gains 4 wedge-hint regression cases:
- Both branches present in source (forward-progress override + 3-partials
detection coexisting).
- Anti-regression guard: NO `import { statusForVersion }` from
apply-migrations.ts. The prior plan would have introduced this
import; keeping it out means doctor stays decoupled from the
migration runner's per-version semantics.
- Multiple wedged versions chain force-retry calls with `&&`.
- Both branches present in doctorReportRemote (thin-client coverage,
D14).
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
* fix(voyage): Content-Length pre-check + per-item base64 cap (D2 + D10)
The voyage compat fetch wrapper at gateway.ts:294 called
\`await resp.clone().json()\` BEFORE iterating embeddings. A
malicious or compromised Voyage endpoint of arbitrary size was
fully parsed into the JS heap before any size check could fire.
The original v0.31.8 plan put the cap on per-item base64 length,
which fires AFTER the JSON parse — defeating the OOM defense
entirely (codex OV8).
Two-layer fix sized at MAX_VOYAGE_RESPONSE_BYTES = 256 MB
("unambiguously not legit" rather than tight against typical
batches; voyage-3-large × 16K embeddings ≈ 200 MB raw fits within
the cap):
Layer 1 (PRIMARY) — Content-Length header pre-check, fires
BEFORE resp.clone().json(). Throws a descriptive error if the
header reports a length over the cap. The JSON.parse OOM vector
is now gated.
Layer 2 (defense-in-depth) — per-embedding base64 length check
inside the iteration. Catches the rare case where Layer 1 was
skipped (chunked transfer encoding has no Content-Length) AND a
single embedding string is unreasonably large. Estimates decoded
size as 0.75 × base64 length (canonical base64 → bytes ratio).
Tests: test/voyage-response-cap.test.ts — 5 structural source-pin
cases including the critical D10 invariant: "Content-Length
pre-check appears BEFORE \`const json: any = await
resp.clone().json()\` in the inbound block". A future refactor
that moves the cap below the JSON parse fails this test loudly.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
* chore: bump version and changelog (v0.31.8)
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
* fix(ci): exclude *.serial.test.ts from sharded parallel run
scripts/test-shard.sh (the GitHub Actions runner) was including
*.serial.test.ts files alongside regular tests. Serial files use
top-level mock.module(...) which leaks across files in the same Bun
process — exactly what the .serial naming convention was meant to
quarantine.
Concretely: test/eval-takes-quality-runner.serial.test.ts mocks
src/core/ai/gateway.ts with `configureGateway: () => undefined`
(no-op). Because both files landed in shard 2, the mock leaked into
test/voyage-multimodal.test.ts: when its tests called
configureVoyageMultimodal() → configureGateway(), the no-op fired and
_config stayed null. Then embedMultimodal() called requireConfig()
which threw "AI gateway is not configured" — 18 tests failed at
gateway.ts:171 with [1.00ms] each.
Local fast loop (scripts/run-unit-shard.sh) already excludes
*.serial.test.ts AND *.slow.test.ts via the same find-arg pattern.
test-shard.sh just hadn't picked up the same exclusion when it was
written. This commit:
1. Mirrors run-unit-shard.sh's exclusion pattern in test-shard.sh
(`-not -name '*.slow.test.ts' -not -name '*.serial.test.ts'`).
2. Adds a "Run *.serial.test.ts" step to .github/workflows/test.yml
on shard 1 only, calling scripts/run-serial-tests.sh
(--max-concurrency=1). Shard 1 already runs extra setup work
(`bun run verify`), so it has the natural slot for the serial
pass without slowing the parallel critical path.
Verified locally: shard 2 went from 18 voyage-multimodal failures to
0. Shard 2 file count: 81 → 78 (3 serial files removed). Total test
count after fix: 1438 (1437 pass + 1 pre-existing env-sensitive
warm-create speed gate flake — unrelated to v0.31.8 or this fix).
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
---------
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
3323 lines
140 KiB
TypeScript
3323 lines
140 KiB
TypeScript
import { PGlite } from '@electric-sql/pglite';
|
||
import { vector } from '@electric-sql/pglite/vector';
|
||
import { pg_trgm } from '@electric-sql/pglite/contrib/pg_trgm';
|
||
import type { Transaction } from '@electric-sql/pglite';
|
||
import type {
|
||
BrainEngine,
|
||
LinkBatchInput, TimelineBatchInput,
|
||
ReservedConnection,
|
||
DreamVerdict, DreamVerdictInput,
|
||
FileSpec, FileRow,
|
||
TakeBatchInput, Take, TakesListOpts, TakeHit, StaleTakeRow,
|
||
TakeResolution, SynthesisEvidenceInput,
|
||
TakesScorecard, TakesScorecardOpts, CalibrationBucket, CalibrationCurveOpts,
|
||
FactRow, FactKind, FactVisibility, FactInsertStatus,
|
||
NewFact, FactListOpts, FactsHealth,
|
||
} from './engine.ts';
|
||
import { MAX_SEARCH_LIMIT, clampSearchLimit } from './engine.ts';
|
||
import { runMigrations } from './migrate.ts';
|
||
import { PGLITE_SCHEMA_SQL, getPGLiteSchema } from './pglite-schema.ts';
|
||
import { acquireLock, releaseLock, type LockHandle } from './pglite-lock.ts';
|
||
import type {
|
||
Page, PageInput, PageFilters, PageType,
|
||
Chunk, ChunkInput, StaleChunkRow,
|
||
SearchResult, SearchOpts,
|
||
Link, GraphNode, GraphPath,
|
||
TimelineEntry, TimelineInput, TimelineOpts,
|
||
RawData,
|
||
PageVersion,
|
||
BrainStats, BrainHealth,
|
||
IngestLogEntry, IngestLogInput,
|
||
EngineConfig,
|
||
EvalCandidate, EvalCandidateInput,
|
||
EvalCaptureFailure, EvalCaptureFailureReason,
|
||
SalienceOpts, SalienceResult, AnomaliesOpts, AnomalyResult,
|
||
EmotionalWeightInputRow, EmotionalWeightWriteRow,
|
||
} from './types.ts';
|
||
import { validateSlug, contentHash, rowToPage, rowToChunk, rowToSearchResult, takeRowToTake } from './utils.ts';
|
||
import { deriveResolutionTuple, finalizeScorecard } from './takes-resolution.ts';
|
||
import { normalizeWeightForStorage } from './takes-fence.ts';
|
||
import { GBrainError, PAGE_SORT_SQL } from './types.ts';
|
||
import { computeAnomaliesFromBuckets } from './cycle/anomaly.ts';
|
||
import { resolveBoostMap, resolveHardExcludes } from './search/source-boost.ts';
|
||
import { buildSourceFactorCase, buildHardExcludeClause, buildVisibilityClause, buildRecencyComponentSql } from './search/sql-ranking.ts';
|
||
|
||
type PGLiteDB = PGlite;
|
||
|
||
// Tier 3 snapshot fast-restore. Reads a tar dump produced by
|
||
// `bun run scripts/build-pglite-snapshot.ts`. Snapshot is matched against
|
||
// the current MIGRATIONS hash via a sidecar `.version` file; on mismatch we
|
||
// silently fall through to a normal initSchema (snapshot is just an
|
||
// optimization, never authoritative).
|
||
let _snapshotWarnLogged = false;
|
||
function tryLoadSnapshot(snapshotPath: string): Blob | null {
|
||
try {
|
||
// Lazy require so production builds without these imports don't crash.
|
||
// eslint-disable-next-line @typescript-eslint/no-require-imports
|
||
const fs = require('node:fs') as typeof import('node:fs');
|
||
const crypto = require('node:crypto') as typeof import('node:crypto');
|
||
const { MIGRATIONS } = require('./migrate.ts') as typeof import('./migrate.ts');
|
||
const { PGLITE_SCHEMA_SQL } = require('./pglite-schema.ts') as typeof import('./pglite-schema.ts');
|
||
|
||
if (!fs.existsSync(snapshotPath)) {
|
||
if (!_snapshotWarnLogged) {
|
||
// eslint-disable-next-line no-console
|
||
console.warn(`[pglite] GBRAIN_PGLITE_SNAPSHOT set but file missing: ${snapshotPath} — using normal init.`);
|
||
_snapshotWarnLogged = true;
|
||
}
|
||
return null;
|
||
}
|
||
const versionPath = snapshotPath.replace(/\.tar(?:\.gz)?$/, '.version');
|
||
if (!fs.existsSync(versionPath)) {
|
||
if (!_snapshotWarnLogged) {
|
||
// eslint-disable-next-line no-console
|
||
console.warn(`[pglite] snapshot version file missing: ${versionPath} — using normal init.`);
|
||
_snapshotWarnLogged = true;
|
||
}
|
||
return null;
|
||
}
|
||
const expectedHash = computeSnapshotSchemaHash(MIGRATIONS, PGLITE_SCHEMA_SQL, crypto);
|
||
const actualHash = fs.readFileSync(versionPath, 'utf8').trim();
|
||
if (expectedHash !== actualHash) {
|
||
if (!_snapshotWarnLogged) {
|
||
// eslint-disable-next-line no-console
|
||
console.warn(`[pglite] snapshot stale (schema hash mismatch) — using normal init. Rebuild with: bun run build:pglite-snapshot`);
|
||
_snapshotWarnLogged = true;
|
||
}
|
||
return null;
|
||
}
|
||
const buf = fs.readFileSync(snapshotPath);
|
||
return new Blob([buf]);
|
||
} catch {
|
||
// Any failure -> fall through to normal init. Never block tests.
|
||
return null;
|
||
}
|
||
}
|
||
|
||
export function computeSnapshotSchemaHash(
|
||
migrations: Array<{ version: number; name: string; sql?: string; sqlFor?: { pglite?: string } }>,
|
||
schemaSQL: string,
|
||
crypto: typeof import('node:crypto'),
|
||
): string {
|
||
const hash = crypto.createHash('sha256');
|
||
hash.update('schema:');
|
||
hash.update(schemaSQL);
|
||
hash.update('\nmigrations:\n');
|
||
for (const m of migrations) {
|
||
hash.update(String(m.version));
|
||
hash.update('\t');
|
||
hash.update(m.name);
|
||
hash.update('\t');
|
||
hash.update(m.sql ?? '');
|
||
hash.update('\t');
|
||
hash.update(m.sqlFor?.pglite ?? '');
|
||
hash.update('\n');
|
||
}
|
||
return hash.digest('hex');
|
||
}
|
||
|
||
export class PGLiteEngine implements BrainEngine {
|
||
readonly kind = 'pglite' as const;
|
||
private _db: PGLiteDB | null = null;
|
||
private _lock: LockHandle | null = null;
|
||
// Tier 3: when GBRAIN_PGLITE_SNAPSHOT loaded a post-initSchema state into
|
||
// PGlite.create(loadDataDir), initSchema is a no-op (schema is already
|
||
// present + migrations already applied). Saves ~1-3s per fresh test PGLite.
|
||
private _snapshotLoaded = false;
|
||
|
||
get db(): PGLiteDB {
|
||
if (!this._db) throw new Error('PGLite not connected. Call connect() first.');
|
||
return this._db;
|
||
}
|
||
|
||
// Lifecycle
|
||
async connect(config: EngineConfig): Promise<void> {
|
||
const dataDir = config.database_path || undefined; // undefined = in-memory
|
||
|
||
// Acquire file lock to prevent concurrent PGLite access (crashes with Aborted())
|
||
this._lock = await acquireLock(dataDir);
|
||
|
||
if (!this._lock.acquired) {
|
||
throw new Error('Could not acquire PGLite lock. Another gbrain process is using the database.');
|
||
}
|
||
|
||
// Tier 3: optional snapshot fast-restore. Only applies to in-memory
|
||
// engines (no persistent dataDir). The snapshot was built from a fresh
|
||
// `initSchema()` run; if the version file matches the current MIGRATIONS
|
||
// hash, load the dump and skip the schema replay. Mismatch or missing
|
||
// file silently falls back to normal init.
|
||
let loadDataDir: Blob | undefined;
|
||
if (!dataDir && process.env.GBRAIN_PGLITE_SNAPSHOT) {
|
||
const snapshotResult = tryLoadSnapshot(process.env.GBRAIN_PGLITE_SNAPSHOT);
|
||
if (snapshotResult) {
|
||
loadDataDir = snapshotResult;
|
||
this._snapshotLoaded = true;
|
||
}
|
||
}
|
||
|
||
try {
|
||
this._db = await PGlite.create({
|
||
dataDir,
|
||
loadDataDir,
|
||
extensions: { vector, pg_trgm },
|
||
});
|
||
} catch (err) {
|
||
// v0.13.1: any PGLite.create() failure becomes actionable. Most commonly
|
||
// this is the macOS 26.3 WASM bug (#223). We deliberately do NOT suggest
|
||
// "missing migrations" as a cause — migrations run AFTER create(), so a
|
||
// create-time abort has nothing to do with them. Nest the original error
|
||
// message so debugging isn't erased.
|
||
const original = err instanceof Error ? err.message : String(err);
|
||
const wrapped = new Error(
|
||
`PGLite failed to initialize its WASM runtime.\n` +
|
||
` This is most commonly the macOS 26.3 WASM bug: https://github.com/garrytan/gbrain/issues/223\n` +
|
||
` Run \`gbrain doctor\` for a full diagnosis.\n` +
|
||
` Original error: ${original}`
|
||
);
|
||
// Release the lock so a fresh process can try again; leaking the lock
|
||
// here turns a recoverable init error into a stuck-brain state.
|
||
if (this._lock?.acquired) {
|
||
try { await releaseLock(this._lock); } catch { /* ignore cleanup error */ }
|
||
this._lock = null;
|
||
}
|
||
throw wrapped;
|
||
}
|
||
}
|
||
|
||
async disconnect(): Promise<void> {
|
||
if (this._db) {
|
||
await this._db.close();
|
||
this._db = null;
|
||
}
|
||
if (this._lock?.acquired) {
|
||
await releaseLock(this._lock);
|
||
this._lock = null;
|
||
}
|
||
}
|
||
|
||
async initSchema(): Promise<void> {
|
||
// Tier 3: snapshot was loaded into PGlite — schema + migrations already
|
||
// applied. Nothing to do. Returns immediately.
|
||
if (this._snapshotLoaded) {
|
||
return;
|
||
}
|
||
// Pre-schema bootstrap: add forward-referenced state the embedded schema
|
||
// blob requires but that older brains don't have yet (issues #366/#375/
|
||
// #378/#396 + #266/#357). Bootstrap is idempotent and a no-op on fresh
|
||
// installs and modern brains.
|
||
await this.applyForwardReferenceBootstrap();
|
||
|
||
// Resolve embedding dim/model from gateway (v0.14+). Defaults preserve v0.13.
|
||
let dims = 1536;
|
||
let model = 'text-embedding-3-large';
|
||
try {
|
||
const gw = await import('./ai/gateway.ts');
|
||
dims = gw.getEmbeddingDimensions();
|
||
model = gw.getEmbeddingModel().split(':').slice(1).join(':') || model;
|
||
} catch { /* gateway not configured — use defaults */ }
|
||
|
||
await this.db.exec(getPGLiteSchema(dims, model));
|
||
|
||
const { applied } = await runMigrations(this);
|
||
if (applied > 0) {
|
||
console.log(` ${applied} migration(s) applied`);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* Bootstrap state that PGLITE_SCHEMA_SQL forward-references but that older
|
||
* brains don't have yet. Currently covers:
|
||
*
|
||
* - `sources` table + default seed (FK target of pages.source_id) — v0.18
|
||
* - `pages.source_id` column (indexed by `idx_pages_source_id`) — v0.18
|
||
* - `links.link_source` column (indexed by `idx_links_source`) — v0.13
|
||
* - `links.origin_page_id` column (indexed by `idx_links_origin`) — v0.13
|
||
* - `content_chunks.symbol_name` column (indexed by `idx_chunks_symbol_name`) — v0.19
|
||
* - `content_chunks.language` column (indexed by `idx_chunks_language`) — v0.19
|
||
* - `content_chunks.search_vector` + `parent_symbol_path` + `doc_comment`
|
||
* + `symbol_name_qualified` columns (indexed by `idx_chunks_search_vector`
|
||
* and `idx_chunks_symbol_qualified`) — v0.20 Cathedral II
|
||
* - `pages.deleted_at` column (indexed by `pages_deleted_at_purge_idx`) — v0.26.5
|
||
* - `mcp_request_log.agent_name` + `params` + `error_message` columns
|
||
* (indexed by `idx_mcp_log_agent_time`) — v0.26.3
|
||
* - `subagent_messages.provider_id` column (indexed by
|
||
* `idx_subagent_messages_provider`) — v0.27
|
||
*
|
||
* **Maintenance contract:** when a future migration adds a column-with-index
|
||
* or new-table-with-FK referenced by PGLITE_SCHEMA_SQL, extend this method
|
||
* AND `test/schema-bootstrap-coverage.test.ts`'s `REQUIRED_BOOTSTRAP_COVERAGE`.
|
||
* The coverage test fails loudly if the bootstrap drifts behind the schema.
|
||
*/
|
||
private async applyForwardReferenceBootstrap(): Promise<void> {
|
||
// Single round-trip probe for every forward-reference target.
|
||
const { rows } = await this.db.query(`
|
||
SELECT
|
||
EXISTS (SELECT 1 FROM information_schema.tables
|
||
WHERE table_schema='public' AND table_name='pages') AS pages_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='pages' AND column_name='source_id') AS source_id_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='pages' AND column_name='deleted_at') AS deleted_at_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.tables
|
||
WHERE table_schema='public' AND table_name='links') AS links_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='links' AND column_name='link_source') AS link_source_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='links' AND column_name='origin_page_id') AS origin_page_id_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.tables
|
||
WHERE table_schema='public' AND table_name='content_chunks') AS chunks_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='content_chunks' AND column_name='symbol_name') AS symbol_name_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='content_chunks' AND column_name='language') AS language_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='content_chunks' AND column_name='search_vector') AS search_vector_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='content_chunks' AND column_name='embedding_image') AS embedding_image_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='pages' AND column_name='effective_date') AS effective_date_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.tables
|
||
WHERE table_schema='public' AND table_name='mcp_request_log') AS mcp_log_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='mcp_request_log' AND column_name='agent_name') AS agent_name_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.tables
|
||
WHERE table_schema='public' AND table_name='subagent_messages') AS subagent_messages_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='subagent_messages' AND column_name='provider_id') AS subagent_provider_id_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.tables
|
||
WHERE table_schema='public' AND table_name='ingest_log') AS ingest_log_exists,
|
||
EXISTS (SELECT 1 FROM information_schema.columns
|
||
WHERE table_schema='public' AND table_name='ingest_log' AND column_name='source_id') AS ingest_log_source_id_exists
|
||
`);
|
||
const probe = rows[0] as {
|
||
pages_exists: boolean;
|
||
source_id_exists: boolean;
|
||
deleted_at_exists: boolean;
|
||
links_exists: boolean;
|
||
link_source_exists: boolean;
|
||
origin_page_id_exists: boolean;
|
||
chunks_exists: boolean;
|
||
symbol_name_exists: boolean;
|
||
language_exists: boolean;
|
||
search_vector_exists: boolean;
|
||
embedding_image_exists: boolean;
|
||
effective_date_exists: boolean;
|
||
mcp_log_exists: boolean;
|
||
agent_name_exists: boolean;
|
||
subagent_messages_exists: boolean;
|
||
subagent_provider_id_exists: boolean;
|
||
ingest_log_exists: boolean;
|
||
ingest_log_source_id_exists: boolean;
|
||
};
|
||
|
||
const needsPagesBootstrap = probe.pages_exists && !probe.source_id_exists;
|
||
const needsLinksBootstrap = probe.links_exists
|
||
&& (!probe.link_source_exists || !probe.origin_page_id_exists);
|
||
const needsChunksBootstrap = probe.chunks_exists
|
||
&& (!probe.symbol_name_exists || !probe.language_exists || !probe.search_vector_exists);
|
||
const needsPagesDeletedAt = probe.pages_exists && !probe.deleted_at_exists;
|
||
// v0.27.1 — partial HNSW idx_chunks_embedding_image references this column.
|
||
const needsChunksEmbeddingImage = probe.chunks_exists && !probe.embedding_image_exists;
|
||
// v0.26.3 (v33): idx_mcp_log_agent_time in PGLITE_SCHEMA_SQL needs agent_name col.
|
||
const needsMcpLogBootstrap = probe.mcp_log_exists && !probe.agent_name_exists;
|
||
// v0.27 (v36): idx_subagent_messages_provider in PGLITE_SCHEMA_SQL needs
|
||
// provider_id (the SECOND column in the composite index `(job_id, provider_id)`).
|
||
const needsSubagentProviderId = probe.subagent_messages_exists && !probe.subagent_provider_id_exists;
|
||
// v0.29.1 (v40 + v41): pages_coalesce_date_idx expression index in
|
||
// PGLITE_SCHEMA_SQL references effective_date. Use effective_date_exists
|
||
// as the proxy for the five v40 + v41 pages columns.
|
||
const needsPagesRecency = probe.pages_exists && !probe.effective_date_exists;
|
||
// v0.31.2 (v50): idx_ingest_log_source_type_created in PGLITE_SCHEMA_SQL
|
||
// references source_id. Old brains have ingest_log without source_id;
|
||
// bootstrap adds the column before SCHEMA_SQL replay creates the index.
|
||
const needsIngestLogSourceId = probe.ingest_log_exists && !probe.ingest_log_source_id_exists;
|
||
|
||
// Fresh installs (no tables yet) and modern brains both no-op.
|
||
if (!needsPagesBootstrap && !needsLinksBootstrap && !needsChunksBootstrap
|
||
&& !needsPagesDeletedAt && !needsChunksEmbeddingImage
|
||
&& !needsMcpLogBootstrap && !needsSubagentProviderId
|
||
&& !needsPagesRecency && !needsIngestLogSourceId) return;
|
||
|
||
console.log(' Pre-v0.21 brain detected, applying forward-reference bootstrap');
|
||
|
||
if (needsPagesBootstrap) {
|
||
// Mirror schema-embedded.ts shape for `sources` so the subsequent
|
||
// PGLITE_SCHEMA_SQL CREATE TABLE IF NOT EXISTS is a true no-op.
|
||
await this.db.exec(`
|
||
CREATE TABLE IF NOT EXISTS sources (
|
||
id TEXT PRIMARY KEY,
|
||
name TEXT NOT NULL UNIQUE,
|
||
local_path TEXT,
|
||
last_commit TEXT,
|
||
last_sync_at TIMESTAMPTZ,
|
||
config JSONB NOT NULL DEFAULT '{}'::jsonb,
|
||
created_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
||
);
|
||
INSERT INTO sources (id, name, config)
|
||
VALUES ('default', 'default', '{"federated": true}'::jsonb)
|
||
ON CONFLICT (id) DO NOTHING;
|
||
ALTER TABLE pages ADD COLUMN IF NOT EXISTS source_id TEXT
|
||
NOT NULL DEFAULT 'default' REFERENCES sources(id) ON DELETE CASCADE;
|
||
`);
|
||
}
|
||
|
||
if (needsLinksBootstrap) {
|
||
// v11 (links_provenance_columns) is responsible for the CHECK constraint
|
||
// and backfill. The bootstrap only adds enough state for SCHEMA_SQL's
|
||
// `CREATE INDEX idx_links_source/origin` not to crash. v11 runs later
|
||
// via runMigrations and is idempotent (`IF NOT EXISTS` everywhere).
|
||
await this.db.exec(`
|
||
ALTER TABLE links ADD COLUMN IF NOT EXISTS link_source TEXT;
|
||
ALTER TABLE links ADD COLUMN IF NOT EXISTS origin_page_id INTEGER
|
||
REFERENCES pages(id) ON DELETE SET NULL;
|
||
`);
|
||
}
|
||
|
||
if (needsChunksBootstrap) {
|
||
// v26 (content_chunks_code_metadata) adds symbol_name + language; v27
|
||
// (Cathedral II) adds parent_symbol_path + doc_comment +
|
||
// symbol_name_qualified + search_vector. PGLITE_SCHEMA_SQL has indexes
|
||
// (idx_chunks_search_vector, idx_chunks_symbol_qualified) that need the
|
||
// v27 columns to exist before they run. v26 + v27 run later via
|
||
// runMigrations and are idempotent.
|
||
await this.db.exec(`
|
||
ALTER TABLE content_chunks ADD COLUMN IF NOT EXISTS language TEXT;
|
||
ALTER TABLE content_chunks ADD COLUMN IF NOT EXISTS symbol_name TEXT;
|
||
ALTER TABLE content_chunks ADD COLUMN IF NOT EXISTS parent_symbol_path TEXT[];
|
||
ALTER TABLE content_chunks ADD COLUMN IF NOT EXISTS doc_comment TEXT;
|
||
ALTER TABLE content_chunks ADD COLUMN IF NOT EXISTS symbol_name_qualified TEXT;
|
||
ALTER TABLE content_chunks ADD COLUMN IF NOT EXISTS search_vector TSVECTOR;
|
||
`);
|
||
}
|
||
|
||
if (needsPagesDeletedAt) {
|
||
// v34 (destructive_guard_columns) adds the column + sources columns +
|
||
// partial purge index. Bootstrap only adds enough for PGLITE_SCHEMA_SQL's
|
||
// `CREATE INDEX pages_deleted_at_purge_idx ... WHERE deleted_at IS NOT NULL`
|
||
// not to crash. v34 runs later via runMigrations and is idempotent.
|
||
await this.db.exec(`
|
||
ALTER TABLE pages ADD COLUMN IF NOT EXISTS deleted_at TIMESTAMPTZ;
|
||
`);
|
||
}
|
||
|
||
if (needsChunksEmbeddingImage) {
|
||
// v39 (multimodal_dual_column_v0_27_1) adds modality + embedding_image
|
||
// columns to content_chunks plus the partial HNSW index that references
|
||
// the column. Bootstrap mirrors enough for PGLITE_SCHEMA_SQL's
|
||
// `CREATE INDEX idx_chunks_embedding_image ... WHERE embedding_image IS NOT NULL`
|
||
// not to crash. v39 runs later via runMigrations and is idempotent.
|
||
await this.db.exec(`
|
||
ALTER TABLE content_chunks ADD COLUMN IF NOT EXISTS modality TEXT NOT NULL DEFAULT 'text';
|
||
ALTER TABLE content_chunks ADD COLUMN IF NOT EXISTS embedding_image vector(1024);
|
||
`);
|
||
}
|
||
|
||
if (needsMcpLogBootstrap) {
|
||
// v33 (admin_dashboard_columns_v0_26_3) adds agent_name + params +
|
||
// error_message to mcp_request_log. PGLITE_SCHEMA_SQL's
|
||
// `CREATE INDEX idx_mcp_log_agent_time ON mcp_request_log(agent_name,...)`
|
||
// crashes without agent_name. v33 runs later via runMigrations and is
|
||
// idempotent (and also handles backfill).
|
||
await this.db.exec(`
|
||
ALTER TABLE mcp_request_log ADD COLUMN IF NOT EXISTS agent_name TEXT;
|
||
ALTER TABLE mcp_request_log ADD COLUMN IF NOT EXISTS params JSONB;
|
||
ALTER TABLE mcp_request_log ADD COLUMN IF NOT EXISTS error_message TEXT;
|
||
`);
|
||
}
|
||
|
||
if (needsSubagentProviderId) {
|
||
// v36 (subagent_provider_neutral_persistence_v0_27) adds provider_id +
|
||
// schema_version on subagent_messages and subagent_tool_executions.
|
||
// PGLITE_SCHEMA_SQL's `CREATE INDEX idx_subagent_messages_provider ON
|
||
// subagent_messages (job_id, provider_id)` crashes without provider_id
|
||
// (composite-index second column). v36 runs later via runMigrations and
|
||
// is idempotent.
|
||
await this.db.exec(`
|
||
ALTER TABLE subagent_messages ADD COLUMN IF NOT EXISTS provider_id TEXT;
|
||
`);
|
||
}
|
||
|
||
if (needsPagesRecency) {
|
||
// v40 (pages_emotional_weight) adds emotional_weight; v41
|
||
// (pages_recency_columns) adds effective_date + effective_date_source +
|
||
// import_filename + salience_touched_at and the
|
||
// `pages_coalesce_date_idx ON pages ((COALESCE(effective_date, updated_at)))`
|
||
// expression index. PGLITE_SCHEMA_SQL's CREATE INDEX for that expression
|
||
// crashes before v41 runs. Bootstrap adds all five additive columns;
|
||
// v40 + v41 run later via runMigrations and are idempotent.
|
||
await this.db.exec(`
|
||
ALTER TABLE pages ADD COLUMN IF NOT EXISTS emotional_weight REAL NOT NULL DEFAULT 0.0;
|
||
ALTER TABLE pages ADD COLUMN IF NOT EXISTS effective_date TIMESTAMPTZ;
|
||
ALTER TABLE pages ADD COLUMN IF NOT EXISTS effective_date_source TEXT;
|
||
ALTER TABLE pages ADD COLUMN IF NOT EXISTS import_filename TEXT;
|
||
ALTER TABLE pages ADD COLUMN IF NOT EXISTS salience_touched_at TIMESTAMPTZ;
|
||
`);
|
||
}
|
||
|
||
if (needsIngestLogSourceId) {
|
||
// v50 (ingest_log_source_id) adds source_id + the
|
||
// idx_ingest_log_source_type_created composite index.
|
||
// PGLITE_SCHEMA_SQL's CREATE INDEX (source_id, source_type, created_at)
|
||
// crashes without source_id. Bootstrap adds the column with NOT NULL
|
||
// DEFAULT 'default' so the index can build cleanly.
|
||
await this.db.exec(`
|
||
ALTER TABLE ingest_log ADD COLUMN IF NOT EXISTS source_id TEXT NOT NULL DEFAULT 'default';
|
||
`);
|
||
}
|
||
}
|
||
|
||
async withReservedConnection<T>(fn: (conn: ReservedConnection) => Promise<T>): Promise<T> {
|
||
// PGLite has no connection pool. The single backing connection is
|
||
// always effectively reserved — pass it through.
|
||
const db = this.db;
|
||
const conn: ReservedConnection = {
|
||
async executeRaw<R = Record<string, unknown>>(sql: string, params?: unknown[]): Promise<R[]> {
|
||
const { rows } = await db.query(sql, params);
|
||
return rows as R[];
|
||
},
|
||
};
|
||
return fn(conn);
|
||
}
|
||
|
||
async transaction<T>(fn: (engine: BrainEngine) => Promise<T>): Promise<T> {
|
||
return this.db.transaction(async (tx) => {
|
||
const txEngine = Object.create(this) as PGLiteEngine;
|
||
Object.defineProperty(txEngine, 'db', { get: () => tx });
|
||
return fn(txEngine);
|
||
});
|
||
}
|
||
|
||
// Pages CRUD
|
||
async getPage(slug: string, opts?: { sourceId?: string; includeDeleted?: boolean }): Promise<Page | null> {
|
||
// v0.26.5: hide soft-deleted by default; opt-in via opts.includeDeleted.
|
||
const includeDeleted = opts?.includeDeleted === true;
|
||
const sourceId = opts?.sourceId;
|
||
const where: string[] = ['slug = $1'];
|
||
const params: unknown[] = [slug];
|
||
if (sourceId) {
|
||
params.push(sourceId);
|
||
where.push(`source_id = $${params.length}`);
|
||
}
|
||
if (!includeDeleted) {
|
||
where.push('deleted_at IS NULL');
|
||
}
|
||
const { rows } = await this.db.query(
|
||
`SELECT id, slug, type, title, compiled_truth, timeline, frontmatter, content_hash, created_at, updated_at, deleted_at
|
||
FROM pages WHERE ${where.join(' AND ')} LIMIT 1`,
|
||
params
|
||
);
|
||
if (rows.length === 0) return null;
|
||
return rowToPage(rows[0] as Record<string, unknown>);
|
||
}
|
||
|
||
async putPage(slug: string, page: PageInput, opts?: { sourceId?: string }): Promise<Page> {
|
||
slug = validateSlug(slug);
|
||
const hash = page.content_hash || contentHash(page);
|
||
const frontmatter = page.frontmatter || {};
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
|
||
// v0.18.0 Step 5+: source_id is now in the INSERT column list so multi-
|
||
// source callers land on the intended (source_id, slug) row. Omitting it
|
||
// let the schema DEFAULT 'default' apply, fabricating duplicate slugs that
|
||
// later made bare-slug subqueries return multiple rows.
|
||
// ON CONFLICT target is (source_id, slug); global UNIQUE(slug) dropped in v17.
|
||
const pageKind = page.page_kind || 'markdown';
|
||
// v0.29.1 — additive opt-in columns. COALESCE(EXCLUDED.x, pages.x)
|
||
// preserves existing values when caller omits them (auto-link path,
|
||
// code reindex, etc.). Mirrors postgres-engine.ts.
|
||
const effectiveDate = page.effective_date instanceof Date
|
||
? page.effective_date.toISOString()
|
||
: (page.effective_date ?? null);
|
||
const effectiveDateSource = page.effective_date_source ?? null;
|
||
const importFilename = page.import_filename ?? null;
|
||
const { rows } = await this.db.query(
|
||
`INSERT INTO pages (source_id, slug, type, page_kind, title, compiled_truth, timeline, frontmatter, content_hash, updated_at, effective_date, effective_date_source, import_filename)
|
||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8::jsonb, $9, now(), $10::timestamptz, $11, $12)
|
||
ON CONFLICT (source_id, slug) DO UPDATE SET
|
||
type = EXCLUDED.type,
|
||
page_kind = EXCLUDED.page_kind,
|
||
title = EXCLUDED.title,
|
||
compiled_truth = EXCLUDED.compiled_truth,
|
||
timeline = EXCLUDED.timeline,
|
||
frontmatter = EXCLUDED.frontmatter,
|
||
content_hash = EXCLUDED.content_hash,
|
||
updated_at = now(),
|
||
effective_date = COALESCE(EXCLUDED.effective_date, pages.effective_date),
|
||
effective_date_source = COALESCE(EXCLUDED.effective_date_source, pages.effective_date_source),
|
||
import_filename = COALESCE(EXCLUDED.import_filename, pages.import_filename)
|
||
RETURNING id, slug, type, title, compiled_truth, timeline, frontmatter, content_hash, created_at, updated_at, effective_date, effective_date_source, import_filename`,
|
||
[sourceId, slug, page.type, pageKind, page.title, page.compiled_truth, page.timeline || '', JSON.stringify(frontmatter), hash, effectiveDate, effectiveDateSource, importFilename]
|
||
);
|
||
return rowToPage(rows[0] as Record<string, unknown>);
|
||
}
|
||
|
||
async deletePage(slug: string, opts?: { sourceId?: string }): Promise<void> {
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
await this.db.query(
|
||
'DELETE FROM pages WHERE slug = $1 AND source_id = $2',
|
||
[slug, sourceId]
|
||
);
|
||
}
|
||
|
||
async softDeletePage(slug: string, opts?: { sourceId?: string }): Promise<{ slug: string } | null> {
|
||
// Idempotent-as-null: only flip rows currently active. Source filter is
|
||
// optional; without it the first matching row across sources gets soft-deleted.
|
||
const sourceId = opts?.sourceId;
|
||
const where: string[] = ['slug = $1', 'deleted_at IS NULL'];
|
||
const params: unknown[] = [slug];
|
||
if (sourceId) {
|
||
params.push(sourceId);
|
||
where.push(`source_id = $${params.length}`);
|
||
}
|
||
const { rows } = await this.db.query(
|
||
`UPDATE pages SET deleted_at = now() WHERE ${where.join(' AND ')} RETURNING slug`,
|
||
params
|
||
);
|
||
if (rows.length === 0) return null;
|
||
return { slug: (rows[0] as { slug: string }).slug };
|
||
}
|
||
|
||
async restorePage(slug: string, opts?: { sourceId?: string }): Promise<boolean> {
|
||
const sourceId = opts?.sourceId;
|
||
const where: string[] = ['slug = $1', 'deleted_at IS NOT NULL'];
|
||
const params: unknown[] = [slug];
|
||
if (sourceId) {
|
||
params.push(sourceId);
|
||
where.push(`source_id = $${params.length}`);
|
||
}
|
||
const { rows } = await this.db.query(
|
||
`UPDATE pages SET deleted_at = NULL WHERE ${where.join(' AND ')} RETURNING slug`,
|
||
params
|
||
);
|
||
return rows.length > 0;
|
||
}
|
||
|
||
async purgeDeletedPages(olderThanHours: number): Promise<{ slugs: string[]; count: number }> {
|
||
// Clamp to non-negative integer; cascade through FKs (content_chunks,
|
||
// page_links, chunk_relations) on DELETE.
|
||
const hours = Math.max(0, Math.floor(olderThanHours));
|
||
const { rows } = await this.db.query(
|
||
`DELETE FROM pages
|
||
WHERE deleted_at IS NOT NULL
|
||
AND deleted_at < now() - ($1 || ' hours')::interval
|
||
RETURNING slug`,
|
||
[hours]
|
||
);
|
||
const slugs = (rows as { slug: string }[]).map((r) => r.slug);
|
||
return { slugs, count: slugs.length };
|
||
}
|
||
|
||
async listPages(filters?: PageFilters): Promise<Page[]> {
|
||
const limit = filters?.limit || 100;
|
||
const offset = filters?.offset || 0;
|
||
|
||
const where: string[] = [];
|
||
const params: unknown[] = [];
|
||
const tagJoin = filters?.tag ? 'JOIN tags t ON t.page_id = p.id' : '';
|
||
|
||
if (filters?.type) {
|
||
params.push(filters.type);
|
||
where.push(`p.type = $${params.length}`);
|
||
}
|
||
if (filters?.tag) {
|
||
params.push(filters.tag);
|
||
where.push(`t.tag = $${params.length}`);
|
||
}
|
||
if (filters?.updated_after) {
|
||
params.push(filters.updated_after);
|
||
where.push(`p.updated_at > $${params.length}::timestamptz`);
|
||
}
|
||
// slugPrefix uses the (source_id, slug) UNIQUE btree for index range scans.
|
||
// Escape LIKE metacharacters so the user prefix is treated as a literal.
|
||
if (filters?.slugPrefix) {
|
||
const escaped = filters.slugPrefix.replace(/[\\%_]/g, (c) => '\\' + c) + '%';
|
||
params.push(escaped);
|
||
where.push(`p.slug LIKE $${params.length} ESCAPE '\\'`);
|
||
}
|
||
// v0.26.5: hide soft-deleted by default; opt in via filters.includeDeleted.
|
||
if (filters?.includeDeleted !== true) {
|
||
where.push('p.deleted_at IS NULL');
|
||
}
|
||
|
||
const whereSql = where.length > 0 ? `WHERE ${where.join(' AND ')}` : '';
|
||
params.push(limit, offset);
|
||
const limitSql = `LIMIT $${params.length - 1} OFFSET $${params.length}`;
|
||
|
||
// v0.29: ORDER BY threading via PAGE_SORT_SQL whitelist (no SQL injection).
|
||
const sortKey = filters?.sort && PAGE_SORT_SQL[filters.sort] ? filters.sort : 'updated_desc';
|
||
const orderBy = PAGE_SORT_SQL[sortKey];
|
||
|
||
const { rows } = await this.db.query(
|
||
`SELECT p.* FROM pages p ${tagJoin} ${whereSql}
|
||
ORDER BY ${orderBy} ${limitSql}`,
|
||
params
|
||
);
|
||
|
||
return (rows as Record<string, unknown>[]).map(rowToPage);
|
||
}
|
||
|
||
async getAllSlugs(opts?: { sourceId?: string }): Promise<Set<string>> {
|
||
// v0.31.8 (D12): when opts.sourceId is set, return only that source's
|
||
// slugs (used by reconcileLinks so wikilink resolution doesn't span
|
||
// unrelated sources). Without opts, returns the union across sources
|
||
// (pre-v0.31.8 behavior — preserved for callers that still expect the
|
||
// brain-wide slug index, e.g. extract.ts's link resolver).
|
||
if (opts?.sourceId) {
|
||
const { rows } = await this.db.query(
|
||
'SELECT slug FROM pages WHERE source_id = $1',
|
||
[opts.sourceId]
|
||
);
|
||
return new Set((rows as { slug: string }[]).map(r => r.slug));
|
||
}
|
||
const { rows } = await this.db.query('SELECT slug FROM pages');
|
||
return new Set((rows as { slug: string }[]).map(r => r.slug));
|
||
}
|
||
|
||
async resolveSlugs(partial: string): Promise<string[]> {
|
||
// Try exact match first
|
||
const exact = await this.db.query('SELECT slug FROM pages WHERE slug = $1', [partial]);
|
||
if (exact.rows.length > 0) return [(exact.rows[0] as { slug: string }).slug];
|
||
|
||
// Fuzzy match via pg_trgm
|
||
const { rows } = await this.db.query(
|
||
`SELECT slug, similarity(title, $1) AS sim
|
||
FROM pages
|
||
WHERE title % $1 OR slug ILIKE $2
|
||
ORDER BY sim DESC
|
||
LIMIT 5`,
|
||
[partial, '%' + partial + '%']
|
||
);
|
||
return (rows as { slug: string }[]).map(r => r.slug);
|
||
}
|
||
|
||
// Search
|
||
//
|
||
// v0.20.0 Cathedral II Layer 3 (1b): keyword search now ranks at
|
||
// chunk-grain internally using content_chunks.search_vector, then dedups
|
||
// to best-chunk-per-page on the way out. External shape (page-grain,
|
||
// one row per matched page, best chunk selected) is identical to
|
||
// v0.19.0 — backlinks, enrichment-service.countMentions, list_pages,
|
||
// etc. all see the same contract. A2 two-pass (Layer 7) consumes
|
||
// searchKeywordChunks for raw chunk-grain results without the dedup.
|
||
//
|
||
// The DISTINCT ON pattern is translated into a two-stage query because
|
||
// PGLite's query planner handles CTEs-with-DISTINCT-ON less optimally
|
||
// than direct window function + GROUP BY. Fetch more chunks than the
|
||
// page limit (3x) to ensure N dedup'd pages survive; bounded and fast.
|
||
async searchKeyword(query: string, opts?: SearchOpts): Promise<SearchResult[]> {
|
||
const limit = clampSearchLimit(opts?.limit);
|
||
const offset = opts?.offset || 0;
|
||
const detailFilter = opts?.detail === 'low' ? `AND cc.chunk_source = 'compiled_truth'` : '';
|
||
|
||
if (opts?.limit && opts.limit > MAX_SEARCH_LIMIT) {
|
||
console.warn(`[gbrain] Warning: search limit clamped from ${opts.limit} to ${MAX_SEARCH_LIMIT}`);
|
||
}
|
||
|
||
// Fetch 3x to give dedup headroom, then page-dedup + re-limit.
|
||
const innerLimit = Math.min(limit * 3, MAX_SEARCH_LIMIT * 3);
|
||
|
||
// Source-aware ranking (v0.22): see postgres-engine.ts for rationale.
|
||
const boostMap = resolveBoostMap();
|
||
const sourceFactorCase = buildSourceFactorCase('p.slug', boostMap, opts?.detail);
|
||
const hardExcludePrefixes = resolveHardExcludes(opts?.exclude_slug_prefixes, opts?.include_slug_prefixes);
|
||
const hardExcludeClause = buildHardExcludeClause('p.slug', hardExcludePrefixes);
|
||
|
||
// v0.20.0 Cathedral II Layer 10 C1/C2: language + symbol-kind filters.
|
||
const params: unknown[] = [query, innerLimit, limit, offset];
|
||
let extraFilter = '';
|
||
if (opts?.language) {
|
||
params.push(opts.language);
|
||
extraFilter += ` AND cc.language = $${params.length}`;
|
||
}
|
||
if (opts?.symbolKind) {
|
||
params.push(opts.symbolKind);
|
||
extraFilter += ` AND cc.symbol_type = $${params.length}`;
|
||
}
|
||
// v0.29.1 — since/until date filter (Postgres parity, codex pass-1 #10).
|
||
// Reads against COALESCE(effective_date, updated_at) so date filtering
|
||
// matches user intent (a meeting was on its event_date, not when it
|
||
// got reimported). Same param shape as Postgres engine.
|
||
if (opts?.afterDate) {
|
||
params.push(opts.afterDate);
|
||
extraFilter += ` AND COALESCE(p.effective_date, p.updated_at, p.created_at) > $${params.length}::timestamptz`;
|
||
}
|
||
if (opts?.beforeDate) {
|
||
params.push(opts.beforeDate);
|
||
extraFilter += ` AND COALESCE(p.effective_date, p.updated_at, p.created_at) < $${params.length}::timestamptz`;
|
||
}
|
||
|
||
// v0.26.5: visibility filter (soft-deleted + archived-source).
|
||
const visibilityClause = buildVisibilityClause('p', 's');
|
||
|
||
const { rows } = await this.db.query(
|
||
`WITH ranked AS (
|
||
SELECT
|
||
p.slug, p.id as page_id, p.title, p.type, p.source_id,
|
||
cc.id as chunk_id, cc.chunk_index, cc.chunk_text, cc.chunk_source,
|
||
ts_rank(cc.search_vector, websearch_to_tsquery('english', $1)) * ${sourceFactorCase} AS score,
|
||
CASE WHEN p.updated_at < (
|
||
SELECT MAX(te.created_at) FROM timeline_entries te WHERE te.page_id = p.id
|
||
) THEN true ELSE false END AS stale
|
||
FROM content_chunks cc
|
||
JOIN pages p ON p.id = cc.page_id
|
||
JOIN sources s ON s.id = p.source_id
|
||
WHERE cc.search_vector @@ websearch_to_tsquery('english', $1) ${detailFilter}${extraFilter} ${hardExcludeClause} ${visibilityClause}
|
||
-- v0.27.1: hide image rows from default text-keyword search so
|
||
-- OCR text doesn't drown text-page hits. Image-similarity queries
|
||
-- run a separate vector path on embedding_image.
|
||
AND cc.modality = 'text'
|
||
ORDER BY score DESC
|
||
LIMIT $2
|
||
),
|
||
best_per_page AS (
|
||
SELECT DISTINCT ON (slug) *
|
||
FROM ranked
|
||
ORDER BY slug, score DESC
|
||
)
|
||
SELECT * FROM best_per_page
|
||
ORDER BY score DESC
|
||
LIMIT $3 OFFSET $4`,
|
||
params
|
||
);
|
||
|
||
return (rows as Record<string, unknown>[]).map(rowToSearchResult);
|
||
}
|
||
|
||
/**
|
||
* v0.20.0 Cathedral II Layer 3 (1b) chunk-grain keyword search.
|
||
*
|
||
* Ranks at chunk grain via content_chunks.search_vector WITHOUT the
|
||
* dedup-to-page pass that searchKeyword applies on return. Used by
|
||
* A2 two-pass retrieval (Layer 7) as the anchor-discovery primitive:
|
||
* two-pass wants the top-N chunks (regardless of page), not the
|
||
* best chunk per top-N pages.
|
||
*
|
||
* Most callers should prefer searchKeyword (external page-grain
|
||
* contract). This method is intentionally a narrow internal knob.
|
||
*/
|
||
async searchKeywordChunks(query: string, opts?: SearchOpts): Promise<SearchResult[]> {
|
||
const limit = clampSearchLimit(opts?.limit);
|
||
const offset = opts?.offset || 0;
|
||
const detailFilter = opts?.detail === 'low' ? `AND cc.chunk_source = 'compiled_truth'` : '';
|
||
|
||
if (opts?.limit && opts.limit > MAX_SEARCH_LIMIT) {
|
||
console.warn(`[gbrain] Warning: search limit clamped from ${opts.limit} to ${MAX_SEARCH_LIMIT}`);
|
||
}
|
||
|
||
// Source-aware ranking applied here too — searchKeywordChunks is the
|
||
// chunk-grain anchor primitive that two-pass retrieval (Layer 7) uses.
|
||
const boostMap = resolveBoostMap();
|
||
const sourceFactorCase = buildSourceFactorCase('p.slug', boostMap, opts?.detail);
|
||
const hardExcludePrefixes = resolveHardExcludes(opts?.exclude_slug_prefixes, opts?.include_slug_prefixes);
|
||
const hardExcludeClause = buildHardExcludeClause('p.slug', hardExcludePrefixes);
|
||
|
||
const params: unknown[] = [query, limit, offset];
|
||
let extraFilter = '';
|
||
if (opts?.language) {
|
||
params.push(opts.language);
|
||
extraFilter += ` AND cc.language = $${params.length}`;
|
||
}
|
||
if (opts?.symbolKind) {
|
||
params.push(opts.symbolKind);
|
||
extraFilter += ` AND cc.symbol_type = $${params.length}`;
|
||
}
|
||
// v0.29.1 since/until parity (codex pass-1 #10).
|
||
if (opts?.afterDate) {
|
||
params.push(opts.afterDate);
|
||
extraFilter += ` AND COALESCE(p.effective_date, p.updated_at, p.created_at) > $${params.length}::timestamptz`;
|
||
}
|
||
if (opts?.beforeDate) {
|
||
params.push(opts.beforeDate);
|
||
extraFilter += ` AND COALESCE(p.effective_date, p.updated_at, p.created_at) < $${params.length}::timestamptz`;
|
||
}
|
||
|
||
// v0.26.5: visibility filter for the chunk-grain anchor primitive.
|
||
const visibilityClause = buildVisibilityClause('p', 's');
|
||
|
||
const { rows } = await this.db.query(
|
||
`SELECT
|
||
p.slug, p.id as page_id, p.title, p.type, p.source_id,
|
||
cc.id as chunk_id, cc.chunk_index, cc.chunk_text, cc.chunk_source,
|
||
ts_rank(cc.search_vector, websearch_to_tsquery('english', $1)) * ${sourceFactorCase} AS score,
|
||
CASE WHEN p.updated_at < (
|
||
SELECT MAX(te.created_at) FROM timeline_entries te WHERE te.page_id = p.id
|
||
) THEN true ELSE false END AS stale
|
||
FROM content_chunks cc
|
||
JOIN pages p ON p.id = cc.page_id
|
||
JOIN sources s ON s.id = p.source_id
|
||
WHERE cc.search_vector @@ websearch_to_tsquery('english', $1) ${detailFilter}${extraFilter} ${hardExcludeClause} ${visibilityClause}
|
||
ORDER BY score DESC
|
||
LIMIT $2 OFFSET $3`,
|
||
params
|
||
);
|
||
|
||
return (rows as Record<string, unknown>[]).map(rowToSearchResult);
|
||
}
|
||
|
||
async searchVector(embedding: Float32Array, opts?: SearchOpts): Promise<SearchResult[]> {
|
||
const limit = clampSearchLimit(opts?.limit);
|
||
const offset = opts?.offset || 0;
|
||
const vecStr = '[' + Array.from(embedding).join(',') + ']';
|
||
const detailFilter = opts?.detail === 'low' ? `AND cc.chunk_source = 'compiled_truth'` : '';
|
||
|
||
if (opts?.limit && opts.limit > MAX_SEARCH_LIMIT) {
|
||
console.warn(`[gbrain] Warning: search limit clamped from ${opts.limit} to ${MAX_SEARCH_LIMIT}`);
|
||
}
|
||
|
||
// Two-stage CTE (v0.22): pure-distance ORDER BY in inner CTE preserves
|
||
// HNSW; outer SELECT re-ranks by raw_score * source_factor over the
|
||
// narrow candidate pool. innerLimit scales with offset to preserve the
|
||
// pagination contract. See postgres-engine.ts searchVector for rationale.
|
||
const boostMap = resolveBoostMap();
|
||
// Outer SELECT references the aliased CTE column. Aliasing the CTE as `hc`
|
||
// disambiguates the correlated subquery (`te.page_id = hc.page_id`) from
|
||
// the inner column. Without the alias, an unqualified `page_id` in the
|
||
// subquery's WHERE would lexically resolve back to `te.page_id` itself
|
||
// and degrade to `te.page_id = te.page_id` (always true), making every
|
||
// result stale=true. Codex caught this in adversarial review.
|
||
const sourceFactorCaseOnSlug = buildSourceFactorCase('hc.slug', boostMap, opts?.detail);
|
||
const hardExcludePrefixes = resolveHardExcludes(opts?.exclude_slug_prefixes, opts?.include_slug_prefixes);
|
||
const hardExcludeClause = buildHardExcludeClause('p.slug', hardExcludePrefixes);
|
||
const innerLimit = offset + Math.max(limit * 5, 100);
|
||
|
||
const params: unknown[] = [vecStr, innerLimit, limit, offset];
|
||
let extraFilter = '';
|
||
if (opts?.language) {
|
||
params.push(opts.language);
|
||
extraFilter += ` AND cc.language = $${params.length}`;
|
||
}
|
||
if (opts?.symbolKind) {
|
||
params.push(opts.symbolKind);
|
||
extraFilter += ` AND cc.symbol_type = $${params.length}`;
|
||
}
|
||
// v0.29.1 since/until parity (codex pass-1 #10). Filter applied INSIDE
|
||
// the inner CTE so HNSW's candidate pool already excludes out-of-range
|
||
// pages — preserves pagination contract.
|
||
if (opts?.afterDate) {
|
||
params.push(opts.afterDate);
|
||
extraFilter += ` AND COALESCE(p.effective_date, p.updated_at, p.created_at) > $${params.length}::timestamptz`;
|
||
}
|
||
if (opts?.beforeDate) {
|
||
params.push(opts.beforeDate);
|
||
extraFilter += ` AND COALESCE(p.effective_date, p.updated_at, p.created_at) < $${params.length}::timestamptz`;
|
||
}
|
||
|
||
// v0.26.5: visibility filter applied in the inner CTE so HNSW sees the
|
||
// same candidate count it always did. See postgres-engine.ts for rationale.
|
||
const visibilityClause = buildVisibilityClause('p', 's');
|
||
|
||
// v0.27.1: column routing. Default 'embedding' targets the brain's
|
||
// primary text-embedding column; 'embedding_image' targets the
|
||
// multimodal column populated by importImageFile. Image-similarity
|
||
// queries pass embeddingColumn='embedding_image' AND a 1024-dim vector
|
||
// produced by gateway.embedMultimodal — must match the column dim.
|
||
const col = opts?.embeddingColumn === 'embedding_image' ? 'embedding_image' : 'embedding';
|
||
// Image rows live in modality='image'; text/code in 'text'. Restrict
|
||
// to the modality matching the column to avoid cross-mode dim leaks.
|
||
const modalityFilter = col === 'embedding_image' ? `AND cc.modality = 'image'` : `AND cc.modality = 'text'`;
|
||
|
||
const { rows } = await this.db.query(
|
||
`WITH hnsw_candidates AS (
|
||
SELECT
|
||
p.slug, p.id as page_id, p.title, p.type, p.source_id, p.updated_at,
|
||
cc.id as chunk_id, cc.chunk_index, cc.chunk_text, cc.chunk_source,
|
||
1 - (cc.${col} <=> $1::vector) AS raw_score
|
||
FROM content_chunks cc
|
||
JOIN pages p ON p.id = cc.page_id
|
||
JOIN sources s ON s.id = p.source_id
|
||
WHERE cc.${col} IS NOT NULL ${modalityFilter} ${detailFilter}${extraFilter} ${hardExcludeClause} ${visibilityClause}
|
||
ORDER BY cc.${col} <=> $1::vector
|
||
LIMIT $2
|
||
)
|
||
SELECT
|
||
hc.slug, hc.page_id, hc.title, hc.type, hc.source_id,
|
||
hc.chunk_id, hc.chunk_index, hc.chunk_text, hc.chunk_source,
|
||
hc.raw_score * ${sourceFactorCaseOnSlug} AS score,
|
||
CASE WHEN hc.updated_at < (
|
||
SELECT MAX(te.created_at) FROM timeline_entries te WHERE te.page_id = hc.page_id
|
||
) THEN true ELSE false END AS stale
|
||
FROM hnsw_candidates hc
|
||
ORDER BY score DESC
|
||
LIMIT $3
|
||
OFFSET $4`,
|
||
params
|
||
);
|
||
|
||
return (rows as Record<string, unknown>[]).map(rowToSearchResult);
|
||
}
|
||
|
||
async getEmbeddingsByChunkIds(ids: number[]): Promise<Map<number, Float32Array>> {
|
||
if (ids.length === 0) return new Map();
|
||
const { rows } = await this.db.query(
|
||
`SELECT id, embedding FROM content_chunks WHERE id = ANY($1::int[]) AND embedding IS NOT NULL`,
|
||
[ids]
|
||
);
|
||
const result = new Map<number, Float32Array>();
|
||
for (const row of rows as Record<string, unknown>[]) {
|
||
if (row.embedding) {
|
||
const emb = typeof row.embedding === 'string'
|
||
? new Float32Array(JSON.parse(row.embedding))
|
||
: row.embedding as Float32Array;
|
||
result.set(row.id as number, emb);
|
||
}
|
||
}
|
||
return result;
|
||
}
|
||
|
||
// Chunks
|
||
async upsertChunks(slug: string, chunks: ChunkInput[], opts?: { sourceId?: string }): Promise<void> {
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
|
||
// Source-scope the page-id lookup so duplicate slugs in different sources
|
||
// do not return multiple rows or target the wrong page.
|
||
const pageResult = await this.db.query(
|
||
'SELECT id FROM pages WHERE slug = $1 AND source_id = $2',
|
||
[slug, sourceId]
|
||
);
|
||
if (pageResult.rows.length === 0) throw new Error(`Page not found: ${slug} (source=${sourceId})`);
|
||
const pageId = (pageResult.rows[0] as { id: number }).id;
|
||
|
||
// Remove chunks that no longer exist
|
||
const newIndices = chunks.map(c => c.chunk_index);
|
||
if (newIndices.length > 0) {
|
||
// PGLite doesn't auto-serialize arrays, so use ANY with explicit array cast
|
||
await this.db.query(
|
||
`DELETE FROM content_chunks WHERE page_id = $1 AND chunk_index != ALL($2::int[])`,
|
||
[pageId, newIndices]
|
||
);
|
||
} else {
|
||
await this.db.query('DELETE FROM content_chunks WHERE page_id = $1', [pageId]);
|
||
return;
|
||
}
|
||
|
||
// Batch upsert: build dynamic multi-row INSERT.
|
||
// v0.19.0: includes language/symbol_name/symbol_type/start_line/end_line
|
||
// so code chunks carry their tree-sitter metadata into the DB. Markdown
|
||
// chunks pass NULL for all five. Order must match the column list.
|
||
// v0.20.0 Cathedral II Layer 6: adds parent_symbol_path / doc_comment /
|
||
// symbol_name_qualified so nested-chunk emission (A3) and eventual A1
|
||
// edge resolution can round-trip metadata through upserts.
|
||
// v0.27.1 (Phase 8): added `modality` + `embedding_image` to the column
|
||
// list. Image chunks pass embedding=null + embedding_image=Float32Array
|
||
// (1024-dim Voyage). Text/code chunks pass embedding=Float32Array +
|
||
// embedding_image=null. Default modality='text' when omitted.
|
||
const cols = '(page_id, chunk_index, chunk_text, chunk_source, embedding, model, token_count, embedded_at, language, symbol_name, symbol_type, start_line, end_line, parent_symbol_path, doc_comment, symbol_name_qualified, modality, embedding_image)';
|
||
const rowParts: string[] = [];
|
||
const params: unknown[] = [];
|
||
let paramIdx = 1;
|
||
|
||
for (const chunk of chunks) {
|
||
const embeddingStr = chunk.embedding
|
||
? '[' + Array.from(chunk.embedding).join(',') + ']'
|
||
: null;
|
||
const embeddingImageStr = chunk.embedding_image
|
||
? '[' + Array.from(chunk.embedding_image).join(',') + ']'
|
||
: null;
|
||
const parentPath = chunk.parent_symbol_path && chunk.parent_symbol_path.length > 0
|
||
? chunk.parent_symbol_path
|
||
: null;
|
||
const modality = chunk.modality ?? 'text';
|
||
|
||
// Inline ::vector NULL literals to avoid a per-branch placeholder.
|
||
const embeddingPh = embeddingStr ? `$${paramIdx++}::vector` : 'NULL';
|
||
const embeddedAtPh = embeddingStr ? 'now()' : 'NULL';
|
||
const embeddingImagePh = embeddingImageStr ? `$${paramIdx++}::vector` : 'NULL';
|
||
|
||
rowParts.push(
|
||
`($${paramIdx++}, $${paramIdx++}, $${paramIdx++}, $${paramIdx++}, ` +
|
||
`${embeddingPh}, $${paramIdx++}, $${paramIdx++}, ${embeddedAtPh}, ` +
|
||
`$${paramIdx++}, $${paramIdx++}, $${paramIdx++}, $${paramIdx++}, $${paramIdx++}, ` +
|
||
`$${paramIdx++}::text[], $${paramIdx++}, $${paramIdx++}, ` +
|
||
`$${paramIdx++}, ${embeddingImagePh})`,
|
||
);
|
||
|
||
// Param push order MUST match placeholder allocation order. Both
|
||
// embedding placeholders (when present) are allocated BEFORE the
|
||
// bulk row placeholders, so their values must be pushed first.
|
||
if (embeddingStr) params.push(embeddingStr);
|
||
if (embeddingImageStr) params.push(embeddingImageStr);
|
||
params.push(
|
||
pageId, chunk.chunk_index, chunk.chunk_text, chunk.chunk_source,
|
||
chunk.model || 'text-embedding-3-large', chunk.token_count || null,
|
||
chunk.language || null, chunk.symbol_name || null, chunk.symbol_type || null,
|
||
chunk.start_line ?? null, chunk.end_line ?? null,
|
||
parentPath, chunk.doc_comment || null, chunk.symbol_name_qualified || null,
|
||
modality,
|
||
);
|
||
}
|
||
|
||
// CONSISTENCY: when chunk_text changes and no new embedding is supplied, BOTH embedding AND
|
||
// embedded_at must reset to NULL so `embed --stale` correctly picks up the row for re-embedding.
|
||
// See postgres-engine.ts upsertChunks for the full rationale — pglite mirrors it for parity.
|
||
await this.db.query(
|
||
`INSERT INTO content_chunks ${cols} VALUES ${rowParts.join(', ')}
|
||
ON CONFLICT (page_id, chunk_index) DO UPDATE SET
|
||
chunk_text = EXCLUDED.chunk_text,
|
||
chunk_source = EXCLUDED.chunk_source,
|
||
embedding = CASE WHEN EXCLUDED.chunk_text != content_chunks.chunk_text THEN EXCLUDED.embedding ELSE COALESCE(EXCLUDED.embedding, content_chunks.embedding) END,
|
||
model = COALESCE(EXCLUDED.model, content_chunks.model),
|
||
token_count = EXCLUDED.token_count,
|
||
embedded_at = CASE
|
||
WHEN EXCLUDED.chunk_text != content_chunks.chunk_text AND EXCLUDED.embedding IS NULL THEN NULL
|
||
ELSE COALESCE(EXCLUDED.embedded_at, content_chunks.embedded_at)
|
||
END,
|
||
language = EXCLUDED.language,
|
||
symbol_name = EXCLUDED.symbol_name,
|
||
symbol_type = EXCLUDED.symbol_type,
|
||
start_line = EXCLUDED.start_line,
|
||
end_line = EXCLUDED.end_line,
|
||
parent_symbol_path = EXCLUDED.parent_symbol_path,
|
||
doc_comment = EXCLUDED.doc_comment,
|
||
symbol_name_qualified = EXCLUDED.symbol_name_qualified,
|
||
modality = EXCLUDED.modality,
|
||
embedding_image = COALESCE(EXCLUDED.embedding_image, content_chunks.embedding_image)`,
|
||
params
|
||
);
|
||
}
|
||
|
||
async getChunks(slug: string, opts?: { sourceId?: string }): Promise<Chunk[]> {
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
const { rows } = await this.db.query(
|
||
`SELECT cc.* FROM content_chunks cc
|
||
JOIN pages p ON p.id = cc.page_id
|
||
WHERE p.slug = $1 AND p.source_id = $2
|
||
ORDER BY cc.chunk_index`,
|
||
[slug, sourceId]
|
||
);
|
||
return (rows as Record<string, unknown>[]).map(r => rowToChunk(r));
|
||
}
|
||
|
||
async countStaleChunks(): Promise<number> {
|
||
const { rows } = await this.db.query(
|
||
`SELECT count(*)::int AS count
|
||
FROM content_chunks
|
||
WHERE embedding IS NULL`,
|
||
);
|
||
const count = (rows[0] as { count: number } | undefined)?.count ?? 0;
|
||
return Number(count);
|
||
}
|
||
|
||
async listStaleChunks(): Promise<StaleChunkRow[]> {
|
||
const { rows } = await this.db.query(
|
||
`SELECT p.slug, cc.chunk_index, cc.chunk_text, cc.chunk_source,
|
||
cc.model, cc.token_count
|
||
FROM content_chunks cc
|
||
JOIN pages p ON p.id = cc.page_id
|
||
WHERE cc.embedding IS NULL
|
||
ORDER BY p.id, cc.chunk_index
|
||
LIMIT 100000`,
|
||
);
|
||
return rows as unknown as StaleChunkRow[];
|
||
}
|
||
|
||
async deleteChunks(slug: string, opts?: { sourceId?: string }): Promise<void> {
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
// Source-qualify the page-id subquery; slugs are only unique per source.
|
||
await this.db.query(
|
||
`DELETE FROM content_chunks
|
||
WHERE page_id = (SELECT id FROM pages WHERE slug = $1 AND source_id = $2)`,
|
||
[slug, sourceId]
|
||
);
|
||
}
|
||
|
||
// Links
|
||
async addLink(
|
||
from: string,
|
||
to: string,
|
||
context?: string,
|
||
linkType?: string,
|
||
linkSource?: string,
|
||
originSlug?: string,
|
||
originField?: string,
|
||
opts?: { fromSourceId?: string; toSourceId?: string; originSourceId?: string },
|
||
): Promise<void> {
|
||
const fromSrc = opts?.fromSourceId ?? 'default';
|
||
const toSrc = opts?.toSourceId ?? 'default';
|
||
const originSrc = opts?.originSourceId ?? 'default';
|
||
|
||
// Source-qualified pre-check gives a clean missing-page error before the
|
||
// INSERT SELECT path can silently return zero rows.
|
||
const exists = await this.db.query(
|
||
`SELECT 1 FROM pages WHERE slug = $1 AND source_id = $2
|
||
INTERSECT
|
||
SELECT 1 FROM pages WHERE slug = $3 AND source_id = $4`,
|
||
[from, fromSrc, to, toSrc]
|
||
);
|
||
if (exists.rows.length === 0) {
|
||
throw new Error(`addLink failed: page "${from}" (source=${fromSrc}) or "${to}" (source=${toSrc}) not found`);
|
||
}
|
||
const src = linkSource ?? 'markdown';
|
||
// Mirror addLinksBatch's VALUES + composite JOIN shape. The old cross-
|
||
// product over pages f/t fanned out across sources containing the slugs.
|
||
await this.db.query(
|
||
`INSERT INTO links (from_page_id, to_page_id, link_type, context, link_source, origin_page_id, origin_field)
|
||
SELECT f.id, t.id, v.link_type, v.context, v.link_source, o.id, v.origin_field
|
||
FROM (VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10))
|
||
AS v(from_slug, to_slug, link_type, context, link_source, origin_slug, origin_field, from_source_id, to_source_id, origin_source_id)
|
||
JOIN pages f ON f.slug = v.from_slug AND f.source_id = v.from_source_id
|
||
JOIN pages t ON t.slug = v.to_slug AND t.source_id = v.to_source_id
|
||
LEFT JOIN pages o ON o.slug = v.origin_slug AND o.source_id = v.origin_source_id
|
||
ON CONFLICT (from_page_id, to_page_id, link_type, link_source, origin_page_id) DO UPDATE SET
|
||
context = EXCLUDED.context,
|
||
origin_field = EXCLUDED.origin_field`,
|
||
[from, to, linkType || '', context || '', src, originSlug ?? null, originField ?? null, fromSrc, toSrc, originSrc]
|
||
);
|
||
}
|
||
|
||
async addLinksBatch(links: LinkBatchInput[]): Promise<number> {
|
||
if (links.length === 0) return 0;
|
||
// unnest() pattern: 10 array-typed bound parameters regardless of batch
|
||
// size. Same shape as PostgresEngine (v0.18). Avoids the 65535-parameter
|
||
// cap.
|
||
//
|
||
// v0.18.0: every JOIN composite-keys on (slug, source_id) so the batch
|
||
// can't fan out across sources when the same slug exists in multiple
|
||
// sources. Origin JOIN uses LEFT JOIN on a composite key — NULL
|
||
// origin_slug leaves origin_page_id NULL, same as pre-v0.18.
|
||
const fromSlugs = links.map(l => l.from_slug);
|
||
const toSlugs = links.map(l => l.to_slug);
|
||
const linkTypes = links.map(l => l.link_type || '');
|
||
const contexts = links.map(l => l.context || '');
|
||
const linkSources = links.map(l => l.link_source || 'markdown');
|
||
const originSlugs = links.map(l => l.origin_slug || null);
|
||
const originFields = links.map(l => l.origin_field || null);
|
||
const fromSourceIds = links.map(l => l.from_source_id || 'default');
|
||
const toSourceIds = links.map(l => l.to_source_id || 'default');
|
||
const originSourceIds = links.map(l => l.origin_source_id || 'default');
|
||
const result = await this.db.query(
|
||
`INSERT INTO links (from_page_id, to_page_id, link_type, context, link_source, origin_page_id, origin_field)
|
||
SELECT f.id, t.id, v.link_type, v.context, v.link_source, o.id, v.origin_field
|
||
FROM unnest($1::text[], $2::text[], $3::text[], $4::text[], $5::text[], $6::text[], $7::text[], $8::text[], $9::text[], $10::text[])
|
||
AS v(from_slug, to_slug, link_type, context, link_source, origin_slug, origin_field, from_source_id, to_source_id, origin_source_id)
|
||
JOIN pages f ON f.slug = v.from_slug AND f.source_id = v.from_source_id
|
||
JOIN pages t ON t.slug = v.to_slug AND t.source_id = v.to_source_id
|
||
LEFT JOIN pages o ON o.slug = v.origin_slug AND o.source_id = v.origin_source_id
|
||
ON CONFLICT (from_page_id, to_page_id, link_type, link_source, origin_page_id) DO NOTHING
|
||
RETURNING 1`,
|
||
[fromSlugs, toSlugs, linkTypes, contexts, linkSources, originSlugs, originFields, fromSourceIds, toSourceIds, originSourceIds]
|
||
);
|
||
return result.rows.length;
|
||
}
|
||
|
||
async removeLink(
|
||
from: string,
|
||
to: string,
|
||
linkType?: string,
|
||
linkSource?: string,
|
||
opts?: { fromSourceId?: string; toSourceId?: string },
|
||
): Promise<void> {
|
||
const fromSrc = opts?.fromSourceId ?? 'default';
|
||
const toSrc = opts?.toSourceId ?? 'default';
|
||
// Each branch source-qualifies page-id subqueries so a delete only targets
|
||
// the intended edge between per-source slug rows.
|
||
if (linkType !== undefined && linkSource !== undefined) {
|
||
await this.db.query(
|
||
`DELETE FROM links
|
||
WHERE from_page_id = (SELECT id FROM pages WHERE slug = $1 AND source_id = $2)
|
||
AND to_page_id = (SELECT id FROM pages WHERE slug = $3 AND source_id = $4)
|
||
AND link_type = $5
|
||
AND link_source IS NOT DISTINCT FROM $6`,
|
||
[from, fromSrc, to, toSrc, linkType, linkSource]
|
||
);
|
||
} else if (linkType !== undefined) {
|
||
await this.db.query(
|
||
`DELETE FROM links
|
||
WHERE from_page_id = (SELECT id FROM pages WHERE slug = $1 AND source_id = $2)
|
||
AND to_page_id = (SELECT id FROM pages WHERE slug = $3 AND source_id = $4)
|
||
AND link_type = $5`,
|
||
[from, fromSrc, to, toSrc, linkType]
|
||
);
|
||
} else if (linkSource !== undefined) {
|
||
await this.db.query(
|
||
`DELETE FROM links
|
||
WHERE from_page_id = (SELECT id FROM pages WHERE slug = $1 AND source_id = $2)
|
||
AND to_page_id = (SELECT id FROM pages WHERE slug = $3 AND source_id = $4)
|
||
AND link_source IS NOT DISTINCT FROM $5`,
|
||
[from, fromSrc, to, toSrc, linkSource]
|
||
);
|
||
} else {
|
||
await this.db.query(
|
||
`DELETE FROM links
|
||
WHERE from_page_id = (SELECT id FROM pages WHERE slug = $1 AND source_id = $2)
|
||
AND to_page_id = (SELECT id FROM pages WHERE slug = $3 AND source_id = $4)`,
|
||
[from, fromSrc, to, toSrc]
|
||
);
|
||
}
|
||
}
|
||
|
||
async getLinks(slug: string, opts?: { sourceId?: string }): Promise<Link[]> {
|
||
// v0.31.8 (D16): two-branch query. Without opts.sourceId, no source filter
|
||
// (preserves pre-v0.31.8 cross-source semantics for back-link validators
|
||
// and read-side op handlers that haven't threaded sourceId yet). With
|
||
// opts.sourceId, scope to that source — used by reconcileLinks and any
|
||
// ctx.sourceId-aware read op (D20).
|
||
if (opts?.sourceId) {
|
||
const { rows } = await this.db.query(
|
||
`SELECT f.slug as from_slug, t.slug as to_slug,
|
||
l.link_type, l.context, l.link_source,
|
||
o.slug as origin_slug, l.origin_field
|
||
FROM links l
|
||
JOIN pages f ON f.id = l.from_page_id
|
||
JOIN pages t ON t.id = l.to_page_id
|
||
LEFT JOIN pages o ON o.id = l.origin_page_id
|
||
WHERE f.slug = $1 AND f.source_id = $2`,
|
||
[slug, opts.sourceId]
|
||
);
|
||
return rows as unknown as Link[];
|
||
}
|
||
const { rows } = await this.db.query(
|
||
`SELECT f.slug as from_slug, t.slug as to_slug,
|
||
l.link_type, l.context, l.link_source,
|
||
o.slug as origin_slug, l.origin_field
|
||
FROM links l
|
||
JOIN pages f ON f.id = l.from_page_id
|
||
JOIN pages t ON t.id = l.to_page_id
|
||
LEFT JOIN pages o ON o.id = l.origin_page_id
|
||
WHERE f.slug = $1`,
|
||
[slug]
|
||
);
|
||
return rows as unknown as Link[];
|
||
}
|
||
|
||
async getBacklinks(slug: string, opts?: { sourceId?: string }): Promise<Link[]> {
|
||
// v0.31.8 (D16): two-branch query. See getLinks() comment.
|
||
if (opts?.sourceId) {
|
||
const { rows } = await this.db.query(
|
||
`SELECT f.slug as from_slug, t.slug as to_slug,
|
||
l.link_type, l.context, l.link_source,
|
||
o.slug as origin_slug, l.origin_field
|
||
FROM links l
|
||
JOIN pages f ON f.id = l.from_page_id
|
||
JOIN pages t ON t.id = l.to_page_id
|
||
LEFT JOIN pages o ON o.id = l.origin_page_id
|
||
WHERE t.slug = $1 AND t.source_id = $2`,
|
||
[slug, opts.sourceId]
|
||
);
|
||
return rows as unknown as Link[];
|
||
}
|
||
const { rows } = await this.db.query(
|
||
`SELECT f.slug as from_slug, t.slug as to_slug,
|
||
l.link_type, l.context, l.link_source,
|
||
o.slug as origin_slug, l.origin_field
|
||
FROM links l
|
||
JOIN pages f ON f.id = l.from_page_id
|
||
JOIN pages t ON t.id = l.to_page_id
|
||
LEFT JOIN pages o ON o.id = l.origin_page_id
|
||
WHERE t.slug = $1`,
|
||
[slug]
|
||
);
|
||
return rows as unknown as Link[];
|
||
}
|
||
|
||
async findByTitleFuzzy(
|
||
name: string,
|
||
dirPrefix?: string,
|
||
minSimilarity: number = 0.55,
|
||
): Promise<{ slug: string; similarity: number } | null> {
|
||
// Inline threshold comparison instead of `SET LOCAL pg_trgm.similarity_threshold`.
|
||
// The GUC only scopes to the current transaction and pglite auto-commits each
|
||
// .query() call, so the SET LOCAL would be a no-op. Using similarity() >= $N
|
||
// directly gives predictable behavior. Tie-breaker: sort by slug so re-runs
|
||
// pick the same winner.
|
||
const prefixPattern = dirPrefix ? `${dirPrefix}/%` : '%';
|
||
const { rows } = await this.db.query(
|
||
`SELECT slug, similarity(title, $1) AS sim
|
||
FROM pages
|
||
WHERE similarity(title, $1) >= $3
|
||
AND slug LIKE $2
|
||
ORDER BY sim DESC, slug ASC
|
||
LIMIT 1`,
|
||
[name, prefixPattern, minSimilarity]
|
||
);
|
||
if (rows.length === 0) return null;
|
||
const row = rows[0] as { slug: string; sim: number };
|
||
return { slug: row.slug, similarity: row.sim };
|
||
}
|
||
|
||
async traverseGraph(slug: string, depth: number = 5): Promise<GraphNode[]> {
|
||
// Cycle prevention: visited array tracks page IDs already in the path.
|
||
// Prevents exponential blowup on cyclic subgraphs (e.g., A->B->A).
|
||
const { rows } = await this.db.query(
|
||
`WITH RECURSIVE graph AS (
|
||
SELECT p.id, p.slug, p.title, p.type, 0 as depth, ARRAY[p.id] as visited
|
||
FROM pages p WHERE p.slug = $1
|
||
|
||
UNION ALL
|
||
|
||
SELECT p2.id, p2.slug, p2.title, p2.type, g.depth + 1, g.visited || p2.id
|
||
FROM graph g
|
||
JOIN links l ON l.from_page_id = g.id
|
||
JOIN pages p2 ON p2.id = l.to_page_id
|
||
WHERE g.depth < $2
|
||
AND NOT (p2.id = ANY(g.visited))
|
||
)
|
||
SELECT DISTINCT g.slug, g.title, g.type, g.depth,
|
||
coalesce(
|
||
-- jsonb_agg(DISTINCT ...) collapses duplicate (to_slug, link_type)
|
||
-- edges that originate from different provenance (markdown body
|
||
-- vs frontmatter vs auto-extracted). Presentation-only dedup;
|
||
-- the links table still preserves every provenance row. See
|
||
-- plan Bug 6/10.
|
||
(SELECT jsonb_agg(DISTINCT jsonb_build_object('to_slug', p3.slug, 'link_type', l2.link_type))
|
||
FROM links l2
|
||
JOIN pages p3 ON p3.id = l2.to_page_id
|
||
WHERE l2.from_page_id = g.id),
|
||
'[]'::jsonb
|
||
) as links
|
||
FROM graph g
|
||
ORDER BY g.depth, g.slug`,
|
||
[slug, depth]
|
||
);
|
||
|
||
return (rows as Record<string, unknown>[]).map(r => ({
|
||
slug: r.slug as string,
|
||
title: r.title as string,
|
||
type: r.type as PageType,
|
||
depth: r.depth as number,
|
||
links: (typeof r.links === 'string' ? JSON.parse(r.links) : r.links) as { to_slug: string; link_type: string }[],
|
||
}));
|
||
}
|
||
|
||
async traversePaths(
|
||
slug: string,
|
||
opts?: { depth?: number; linkType?: string; direction?: 'in' | 'out' | 'both' },
|
||
): Promise<GraphPath[]> {
|
||
const depth = opts?.depth ?? 5;
|
||
const direction = opts?.direction ?? 'out';
|
||
const linkType = opts?.linkType ?? null;
|
||
const linkTypeWhere = linkType !== null ? 'AND l.link_type = $3' : '';
|
||
const params: unknown[] = [slug, depth];
|
||
if (linkType !== null) params.push(linkType);
|
||
|
||
let sql: string;
|
||
if (direction === 'out') {
|
||
sql = `
|
||
WITH RECURSIVE walk AS (
|
||
SELECT p.id, p.slug, 0::int AS depth, ARRAY[p.id] AS visited
|
||
FROM pages p WHERE p.slug = $1
|
||
UNION ALL
|
||
SELECT p2.id, p2.slug, w.depth + 1, w.visited || p2.id
|
||
FROM walk w
|
||
JOIN links l ON l.from_page_id = w.id
|
||
JOIN pages p2 ON p2.id = l.to_page_id
|
||
WHERE w.depth < $2
|
||
AND NOT (p2.id = ANY(w.visited))
|
||
${linkTypeWhere}
|
||
)
|
||
SELECT w.slug AS from_slug, p2.slug AS to_slug,
|
||
l.link_type, l.context, w.depth + 1 AS depth
|
||
FROM walk w
|
||
JOIN links l ON l.from_page_id = w.id
|
||
JOIN pages p2 ON p2.id = l.to_page_id
|
||
WHERE w.depth < $2
|
||
${linkTypeWhere}
|
||
ORDER BY depth, from_slug, to_slug
|
||
`;
|
||
} else if (direction === 'in') {
|
||
sql = `
|
||
WITH RECURSIVE walk AS (
|
||
SELECT p.id, p.slug, 0::int AS depth, ARRAY[p.id] AS visited
|
||
FROM pages p WHERE p.slug = $1
|
||
UNION ALL
|
||
SELECT p2.id, p2.slug, w.depth + 1, w.visited || p2.id
|
||
FROM walk w
|
||
JOIN links l ON l.to_page_id = w.id
|
||
JOIN pages p2 ON p2.id = l.from_page_id
|
||
WHERE w.depth < $2
|
||
AND NOT (p2.id = ANY(w.visited))
|
||
${linkTypeWhere}
|
||
)
|
||
SELECT p2.slug AS from_slug, w.slug AS to_slug,
|
||
l.link_type, l.context, w.depth + 1 AS depth
|
||
FROM walk w
|
||
JOIN links l ON l.to_page_id = w.id
|
||
JOIN pages p2 ON p2.id = l.from_page_id
|
||
WHERE w.depth < $2
|
||
${linkTypeWhere}
|
||
ORDER BY depth, from_slug, to_slug
|
||
`;
|
||
} else {
|
||
// both: walk in both directions, emit every traversed edge (preserving its
|
||
// natural from->to direction from the links table).
|
||
sql = `
|
||
WITH RECURSIVE walk AS (
|
||
SELECT p.id, 0::int AS depth, ARRAY[p.id] AS visited
|
||
FROM pages p WHERE p.slug = $1
|
||
UNION ALL
|
||
SELECT p2.id, w.depth + 1, w.visited || p2.id
|
||
FROM walk w
|
||
JOIN links l ON (l.from_page_id = w.id OR l.to_page_id = w.id)
|
||
JOIN pages p2 ON p2.id = CASE WHEN l.from_page_id = w.id THEN l.to_page_id ELSE l.from_page_id END
|
||
WHERE w.depth < $2
|
||
AND NOT (p2.id = ANY(w.visited))
|
||
${linkTypeWhere}
|
||
)
|
||
SELECT pf.slug AS from_slug, pt.slug AS to_slug,
|
||
l.link_type, l.context, w.depth + 1 AS depth
|
||
FROM walk w
|
||
JOIN links l ON (l.from_page_id = w.id OR l.to_page_id = w.id)
|
||
JOIN pages pf ON pf.id = l.from_page_id
|
||
JOIN pages pt ON pt.id = l.to_page_id
|
||
WHERE w.depth < $2
|
||
${linkTypeWhere}
|
||
ORDER BY depth, from_slug, to_slug
|
||
`;
|
||
}
|
||
|
||
const { rows } = await this.db.query(sql, params);
|
||
// Dedup edges (same from/to/type/depth can appear via multiple visited paths).
|
||
const seen = new Set<string>();
|
||
const result: GraphPath[] = [];
|
||
for (const r of rows as Record<string, unknown>[]) {
|
||
const key = `${r.from_slug}|${r.to_slug}|${r.link_type}|${r.depth}`;
|
||
if (seen.has(key)) continue;
|
||
seen.add(key);
|
||
result.push({
|
||
from_slug: r.from_slug as string,
|
||
to_slug: r.to_slug as string,
|
||
link_type: r.link_type as string,
|
||
context: (r.context as string) || '',
|
||
depth: r.depth as number,
|
||
});
|
||
}
|
||
return result;
|
||
}
|
||
|
||
async getBacklinkCounts(slugs: string[]): Promise<Map<string, number>> {
|
||
const result = new Map<string, number>();
|
||
if (slugs.length === 0) return result;
|
||
// Initialize all slugs to 0 so callers get a consistent map.
|
||
for (const s of slugs) result.set(s, 0);
|
||
|
||
// PGLite needs explicit cast for array binding (does not auto-serialize JS arrays).
|
||
const { rows } = await this.db.query(
|
||
`SELECT p.slug AS slug, COUNT(l.id)::int AS cnt
|
||
FROM pages p
|
||
LEFT JOIN links l ON l.to_page_id = p.id
|
||
WHERE p.slug = ANY($1::text[])
|
||
GROUP BY p.slug`,
|
||
[slugs]
|
||
);
|
||
for (const r of rows as { slug: string; cnt: number }[]) {
|
||
result.set(r.slug, Number(r.cnt));
|
||
}
|
||
return result;
|
||
}
|
||
|
||
async getPageTimestamps(slugs: string[]): Promise<Map<string, Date>> {
|
||
if (slugs.length === 0) return new Map();
|
||
const { rows } = await this.db.query(
|
||
`SELECT slug, COALESCE(updated_at, created_at) as ts
|
||
FROM pages WHERE slug = ANY($1::text[])`,
|
||
[slugs]
|
||
);
|
||
return new Map(rows.map((r: any) => [r.slug as string, new Date(r.ts as string)]));
|
||
}
|
||
|
||
async getEffectiveDates(refs: Array<{slug: string; source_id: string}>): Promise<Map<string, Date>> {
|
||
if (refs.length === 0) return new Map();
|
||
const slugs = refs.map(r => r.slug);
|
||
const sourceIds = refs.map(r => r.source_id);
|
||
const { rows } = await this.db.query(
|
||
`SELECT p.slug, p.source_id, COALESCE(p.effective_date, p.updated_at, p.created_at) AS ts
|
||
FROM pages p
|
||
JOIN unnest($1::text[], $2::text[]) AS u(slug, source_id)
|
||
ON p.slug = u.slug AND p.source_id = u.source_id`,
|
||
[slugs, sourceIds],
|
||
);
|
||
const out = new Map<string, Date>();
|
||
for (const r of rows as Array<{slug: string; source_id: string; ts: string | Date}>) {
|
||
const key = `${r.source_id}::${r.slug}`;
|
||
out.set(key, r.ts instanceof Date ? r.ts : new Date(r.ts));
|
||
}
|
||
return out;
|
||
}
|
||
|
||
async getSalienceScores(refs: Array<{slug: string; source_id: string}>): Promise<Map<string, number>> {
|
||
if (refs.length === 0) return new Map();
|
||
const slugs = refs.map(r => r.slug);
|
||
const sourceIds = refs.map(r => r.source_id);
|
||
const { rows } = await this.db.query(
|
||
`SELECT p.slug, p.source_id,
|
||
(COALESCE(p.emotional_weight, 0) * 5
|
||
+ ln(1 + COUNT(DISTINCT t.id))) AS score
|
||
FROM pages p
|
||
JOIN unnest($1::text[], $2::text[]) AS u(slug, source_id)
|
||
ON p.slug = u.slug AND p.source_id = u.source_id
|
||
LEFT JOIN takes t ON t.page_id = p.id AND t.active = TRUE
|
||
GROUP BY p.id`,
|
||
[slugs, sourceIds],
|
||
);
|
||
const out = new Map<string, number>();
|
||
for (const r of rows as Array<{slug: string; source_id: string; score: number | string}>) {
|
||
const key = `${r.source_id}::${r.slug}`;
|
||
out.set(key, Number(r.score));
|
||
}
|
||
return out;
|
||
}
|
||
|
||
async findOrphanPages(): Promise<Array<{ slug: string; title: string; domain: string | null }>> {
|
||
const { rows } = await this.db.query(
|
||
`SELECT
|
||
p.slug,
|
||
COALESCE(p.title, p.slug) AS title,
|
||
p.frontmatter->>'domain' AS domain
|
||
FROM pages p
|
||
WHERE NOT EXISTS (
|
||
SELECT 1 FROM links l WHERE l.to_page_id = p.id
|
||
)
|
||
ORDER BY p.slug`
|
||
);
|
||
return rows as Array<{ slug: string; title: string; domain: string | null }>;
|
||
}
|
||
|
||
// Tags
|
||
async addTag(slug: string, tag: string, opts?: { sourceId?: string }): Promise<void> {
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
// Pre-check source-scoped page existence; ON CONFLICT only handles the
|
||
// already-tagged case, not missing pages.
|
||
const page = await this.db.query(
|
||
'SELECT id FROM pages WHERE slug = $1 AND source_id = $2',
|
||
[slug, sourceId]
|
||
);
|
||
if (page.rows.length === 0) throw new Error(`addTag failed: page "${slug}" (source=${sourceId}) not found`);
|
||
await this.db.query(
|
||
`INSERT INTO tags (page_id, tag)
|
||
VALUES ($1, $2)
|
||
ON CONFLICT (page_id, tag) DO NOTHING`,
|
||
[(page.rows[0] as { id: number }).id, tag]
|
||
);
|
||
}
|
||
|
||
async removeTag(slug: string, tag: string, opts?: { sourceId?: string }): Promise<void> {
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
// Source-qualify the page-id subquery; slugs are only unique per source.
|
||
await this.db.query(
|
||
`DELETE FROM tags
|
||
WHERE page_id = (SELECT id FROM pages WHERE slug = $1 AND source_id = $2)
|
||
AND tag = $3`,
|
||
[slug, sourceId, tag]
|
||
);
|
||
}
|
||
|
||
async getTags(slug: string, opts?: { sourceId?: string }): Promise<string[]> {
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
// Source-qualify the page-id subquery; slugs are only unique per source.
|
||
const { rows } = await this.db.query(
|
||
`SELECT tag FROM tags
|
||
WHERE page_id = (SELECT id FROM pages WHERE slug = $1 AND source_id = $2)
|
||
ORDER BY tag`,
|
||
[slug, sourceId]
|
||
);
|
||
return (rows as { tag: string }[]).map(r => r.tag);
|
||
}
|
||
|
||
// Timeline
|
||
async addTimelineEntry(
|
||
slug: string,
|
||
entry: TimelineInput,
|
||
opts?: { skipExistenceCheck?: boolean; sourceId?: string },
|
||
): Promise<void> {
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
if (!opts?.skipExistenceCheck) {
|
||
const { rows } = await this.db.query(
|
||
'SELECT 1 FROM pages WHERE slug = $1 AND source_id = $2',
|
||
[slug, sourceId]
|
||
);
|
||
if (rows.length === 0) {
|
||
throw new Error(`addTimelineEntry failed: page "${slug}" (source=${sourceId}) not found`);
|
||
}
|
||
}
|
||
// ON CONFLICT DO NOTHING via the (page_id, date, summary) unique index.
|
||
// Source-qualify the page-id lookup so multi-source brains don't fan
|
||
// timeline rows out across every source containing the slug.
|
||
await this.db.query(
|
||
`INSERT INTO timeline_entries (page_id, date, source, summary, detail)
|
||
SELECT id, $2::date, $3, $4, $5
|
||
FROM pages WHERE slug = $1 AND source_id = $6
|
||
ON CONFLICT (page_id, date, summary) DO NOTHING`,
|
||
[slug, entry.date, entry.source || '', entry.summary, entry.detail || '', sourceId]
|
||
);
|
||
}
|
||
|
||
async addTimelineEntriesBatch(entries: TimelineBatchInput[]): Promise<number> {
|
||
if (entries.length === 0) return 0;
|
||
const slugs = entries.map(e => e.slug);
|
||
const dates = entries.map(e => e.date);
|
||
const sources = entries.map(e => e.source || '');
|
||
const summaries = entries.map(e => e.summary);
|
||
const details = entries.map(e => e.detail || '');
|
||
const sourceIds = entries.map(e => e.source_id || 'default');
|
||
const result = await this.db.query(
|
||
`INSERT INTO timeline_entries (page_id, date, source, summary, detail)
|
||
SELECT p.id, v.date::date, v.source, v.summary, v.detail
|
||
FROM unnest($1::text[], $2::text[], $3::text[], $4::text[], $5::text[], $6::text[])
|
||
AS v(slug, date, source, summary, detail, source_id)
|
||
JOIN pages p ON p.slug = v.slug AND p.source_id = v.source_id
|
||
ON CONFLICT (page_id, date, summary) DO NOTHING
|
||
RETURNING 1`,
|
||
[slugs, dates, sources, summaries, details, sourceIds]
|
||
);
|
||
return result.rows.length;
|
||
}
|
||
|
||
async getTimeline(slug: string, opts?: TimelineOpts): Promise<TimelineEntry[]> {
|
||
// v0.31.8 (D16): build WHERE clause dynamically so opts.sourceId composes
|
||
// cleanly with the existing after/before filters. Without sourceId, no
|
||
// source filter applies (preserves pre-v0.31.8 cross-source semantics).
|
||
const limit = opts?.limit || 100;
|
||
const where: string[] = ['p.slug = $1'];
|
||
const params: unknown[] = [slug];
|
||
if (opts?.after) {
|
||
params.push(opts.after);
|
||
where.push(`te.date >= $${params.length}::date`);
|
||
}
|
||
if (opts?.before) {
|
||
params.push(opts.before);
|
||
where.push(`te.date <= $${params.length}::date`);
|
||
}
|
||
if (opts?.sourceId) {
|
||
params.push(opts.sourceId);
|
||
where.push(`p.source_id = $${params.length}`);
|
||
}
|
||
params.push(limit);
|
||
const result = await this.db.query(
|
||
`SELECT te.* FROM timeline_entries te
|
||
JOIN pages p ON p.id = te.page_id
|
||
WHERE ${where.join(' AND ')}
|
||
ORDER BY te.date DESC LIMIT $${params.length}`,
|
||
params
|
||
);
|
||
return result.rows as unknown as TimelineEntry[];
|
||
}
|
||
|
||
// Raw data
|
||
async putRawData(
|
||
slug: string,
|
||
source: string,
|
||
data: object,
|
||
opts?: { sourceId?: string },
|
||
): Promise<void> {
|
||
// v0.31.8 (D21): two-branch INSERT-SELECT. Without opts.sourceId, the
|
||
// page-id lookup matches every same-slug page (pre-v0.31.8 behavior; can
|
||
// still trip Postgres 21000 on multi-source brains — caller's choice).
|
||
// With opts.sourceId, the lookup is source-scoped so the right row
|
||
// gets the raw_data attached.
|
||
if (opts?.sourceId) {
|
||
await this.db.query(
|
||
`INSERT INTO raw_data (page_id, source, data)
|
||
SELECT id, $2, $3::jsonb
|
||
FROM pages WHERE slug = $1 AND source_id = $4
|
||
ON CONFLICT (page_id, source) DO UPDATE SET
|
||
data = EXCLUDED.data,
|
||
fetched_at = now()`,
|
||
[slug, source, JSON.stringify(data), opts.sourceId]
|
||
);
|
||
return;
|
||
}
|
||
await this.db.query(
|
||
`INSERT INTO raw_data (page_id, source, data)
|
||
SELECT id, $2, $3::jsonb
|
||
FROM pages WHERE slug = $1
|
||
ON CONFLICT (page_id, source) DO UPDATE SET
|
||
data = EXCLUDED.data,
|
||
fetched_at = now()`,
|
||
[slug, source, JSON.stringify(data)]
|
||
);
|
||
}
|
||
|
||
async getRawData(
|
||
slug: string,
|
||
source?: string,
|
||
opts?: { sourceId?: string },
|
||
): Promise<RawData[]> {
|
||
// v0.31.8 (D21): build WHERE clause dynamically. Without opts.sourceId,
|
||
// no source filter (preserves pre-v0.31.8 cross-source read).
|
||
const where: string[] = ['p.slug = $1'];
|
||
const params: unknown[] = [slug];
|
||
if (source) {
|
||
params.push(source);
|
||
where.push(`rd.source = $${params.length}`);
|
||
}
|
||
if (opts?.sourceId) {
|
||
params.push(opts.sourceId);
|
||
where.push(`p.source_id = $${params.length}`);
|
||
}
|
||
const result = await this.db.query(
|
||
`SELECT rd.source, rd.data, rd.fetched_at FROM raw_data rd
|
||
JOIN pages p ON p.id = rd.page_id
|
||
WHERE ${where.join(' AND ')}`,
|
||
params
|
||
);
|
||
return result.rows as unknown as RawData[];
|
||
}
|
||
|
||
// Files (v0.27.1): see PostgresEngine.upsertFile for the same contract.
|
||
async upsertFile(spec: FileSpec): Promise<{ id: number; created: boolean }> {
|
||
const sourceId = spec.source_id ?? 'default';
|
||
const result = await this.db.query<{ id: number; created: boolean }>(
|
||
`INSERT INTO files (source_id, page_slug, page_id, filename, storage_path, mime_type, size_bytes, content_hash, metadata)
|
||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9::jsonb)
|
||
ON CONFLICT (storage_path) DO UPDATE SET
|
||
page_slug = EXCLUDED.page_slug,
|
||
page_id = EXCLUDED.page_id,
|
||
filename = EXCLUDED.filename,
|
||
mime_type = EXCLUDED.mime_type,
|
||
size_bytes = EXCLUDED.size_bytes,
|
||
content_hash = EXCLUDED.content_hash,
|
||
metadata = EXCLUDED.metadata
|
||
RETURNING id, (xmax = 0) AS created`,
|
||
[
|
||
sourceId,
|
||
spec.page_slug ?? null,
|
||
spec.page_id ?? null,
|
||
spec.filename,
|
||
spec.storage_path,
|
||
spec.mime_type ?? null,
|
||
spec.size_bytes ?? null,
|
||
spec.content_hash,
|
||
JSON.stringify(spec.metadata ?? {}),
|
||
]
|
||
);
|
||
if (result.rows.length === 0) {
|
||
throw new Error(`upsertFile returned no rows for ${spec.storage_path}`);
|
||
}
|
||
return { id: result.rows[0].id, created: !!result.rows[0].created };
|
||
}
|
||
|
||
async getFile(sourceId: string, storagePath: string): Promise<FileRow | null> {
|
||
const result = await this.db.query<FileRow>(
|
||
`SELECT id, source_id, page_slug, page_id, filename, storage_path, mime_type, size_bytes, content_hash, metadata, created_at
|
||
FROM files
|
||
WHERE source_id = $1 AND storage_path = $2
|
||
LIMIT 1`,
|
||
[sourceId, storagePath]
|
||
);
|
||
return result.rows.length > 0 ? (result.rows[0] as FileRow) : null;
|
||
}
|
||
|
||
async listFilesForPage(pageId: number): Promise<FileRow[]> {
|
||
const result = await this.db.query<FileRow>(
|
||
`SELECT id, source_id, page_slug, page_id, filename, storage_path, mime_type, size_bytes, content_hash, metadata, created_at
|
||
FROM files
|
||
WHERE page_id = $1
|
||
ORDER BY created_at ASC`,
|
||
[pageId]
|
||
);
|
||
return result.rows as FileRow[];
|
||
}
|
||
|
||
// Dream-cycle significance verdict cache (v0.23).
|
||
async getDreamVerdict(filePath: string, contentHash: string): Promise<DreamVerdict | null> {
|
||
const result = await this.db.query<{
|
||
worth_processing: boolean;
|
||
reasons: string[] | null;
|
||
judged_at: Date | string;
|
||
}>(
|
||
`SELECT worth_processing, reasons, judged_at
|
||
FROM dream_verdicts
|
||
WHERE file_path = $1 AND content_hash = $2`,
|
||
[filePath, contentHash]
|
||
);
|
||
if (result.rows.length === 0) return null;
|
||
const r = result.rows[0];
|
||
return {
|
||
worth_processing: r.worth_processing,
|
||
reasons: r.reasons ?? [],
|
||
judged_at: r.judged_at instanceof Date ? r.judged_at.toISOString() : String(r.judged_at),
|
||
};
|
||
}
|
||
|
||
async putDreamVerdict(filePath: string, contentHash: string, verdict: DreamVerdictInput): Promise<void> {
|
||
await this.db.query(
|
||
`INSERT INTO dream_verdicts (file_path, content_hash, worth_processing, reasons)
|
||
VALUES ($1, $2, $3, $4::jsonb)
|
||
ON CONFLICT (file_path, content_hash) DO UPDATE SET
|
||
worth_processing = EXCLUDED.worth_processing,
|
||
reasons = EXCLUDED.reasons,
|
||
judged_at = now()`,
|
||
[filePath, contentHash, verdict.worth_processing, JSON.stringify(verdict.reasons)]
|
||
);
|
||
}
|
||
|
||
// ============================================================
|
||
// v0.31: Hot memory — facts table operations
|
||
// ============================================================
|
||
|
||
async insertFact(
|
||
input: NewFact,
|
||
ctx: { source_id: string; supersedeId?: number },
|
||
): Promise<{ id: number; status: FactInsertStatus }> {
|
||
const validFrom = input.valid_from ?? new Date();
|
||
const validUntil = input.valid_until ?? null;
|
||
const kind = input.kind ?? 'fact';
|
||
const visibility = input.visibility ?? 'private';
|
||
const notability = input.notability ?? 'medium';
|
||
const confidence = input.confidence ?? 1.0;
|
||
const entitySlug = input.entity_slug ?? null;
|
||
const context = input.context ?? null;
|
||
const sourceSession = input.source_session ?? null;
|
||
const embedding = input.embedding ?? null;
|
||
const embeddedAt = embedding ? new Date() : null;
|
||
const embedStr = embedding ? toPgVectorLiteral(embedding) : null;
|
||
|
||
if (ctx.supersedeId !== undefined) {
|
||
// Supersede flow: insert new + expire old in one txn so observers never
|
||
// see both rows active simultaneously.
|
||
const result = await this.db.transaction(async (tx) => {
|
||
const ins = await tx.query<{ id: number }>(
|
||
`INSERT INTO facts (
|
||
source_id, entity_slug, fact, kind, visibility, notability, context,
|
||
valid_from, valid_until, source, source_session, confidence,
|
||
embedding, embedded_at
|
||
) VALUES (
|
||
$1,$2,$3,$4,$5,$6,$7,$8,$9,$10,$11,$12, ${embedStr === null ? 'NULL' : `$13::vector`}, ${embedStr === null ? 'NULL' : '$14'}
|
||
) RETURNING id`,
|
||
embedStr === null
|
||
? [ctx.source_id, entitySlug, input.fact, kind, visibility, notability, context, validFrom, validUntil, input.source, sourceSession, confidence]
|
||
: [ctx.source_id, entitySlug, input.fact, kind, visibility, notability, context, validFrom, validUntil, input.source, sourceSession, confidence, embedStr, embeddedAt],
|
||
);
|
||
const newId = ins.rows[0].id;
|
||
await tx.query(
|
||
`UPDATE facts SET expired_at = now(), superseded_by = $1
|
||
WHERE id = $2 AND expired_at IS NULL`,
|
||
[newId, ctx.supersedeId],
|
||
);
|
||
return newId;
|
||
});
|
||
return { id: result, status: 'superseded' };
|
||
}
|
||
|
||
const ins = await this.db.query<{ id: number }>(
|
||
`INSERT INTO facts (
|
||
source_id, entity_slug, fact, kind, visibility, notability, context,
|
||
valid_from, valid_until, source, source_session, confidence,
|
||
embedding, embedded_at
|
||
) VALUES (
|
||
$1,$2,$3,$4,$5,$6,$7,$8,$9,$10,$11,$12, ${embedStr === null ? 'NULL' : `$13::vector`}, ${embedStr === null ? 'NULL' : '$14'}
|
||
) RETURNING id`,
|
||
embedStr === null
|
||
? [ctx.source_id, entitySlug, input.fact, kind, visibility, notability, context, validFrom, validUntil, input.source, sourceSession, confidence]
|
||
: [ctx.source_id, entitySlug, input.fact, kind, visibility, notability, context, validFrom, validUntil, input.source, sourceSession, confidence, embedStr, embeddedAt],
|
||
);
|
||
return { id: ins.rows[0].id, status: 'inserted' };
|
||
}
|
||
|
||
async expireFact(id: number, opts?: { supersededBy?: number; at?: Date }): Promise<boolean> {
|
||
const at = opts?.at ?? new Date();
|
||
const result = await this.db.query(
|
||
`UPDATE facts SET expired_at = $1, superseded_by = COALESCE($2, superseded_by)
|
||
WHERE id = $3 AND expired_at IS NULL`,
|
||
[at, opts?.supersededBy ?? null, id],
|
||
);
|
||
return (result.affectedRows ?? 0) > 0;
|
||
}
|
||
|
||
async listFactsByEntity(
|
||
source_id: string,
|
||
entitySlug: string,
|
||
opts?: FactListOpts,
|
||
): Promise<FactRow[]> {
|
||
return this._listFacts(source_id, {
|
||
...opts,
|
||
whereClauses: [`entity_slug = $entitySlug`],
|
||
whereParams: { entitySlug },
|
||
order: 'valid_from DESC, id DESC',
|
||
});
|
||
}
|
||
|
||
async listFactsSince(
|
||
source_id: string,
|
||
since: Date,
|
||
opts?: FactListOpts & { entitySlug?: string },
|
||
): Promise<FactRow[]> {
|
||
const where: string[] = [`created_at >= $since`];
|
||
const params: Record<string, unknown> = { since };
|
||
if (opts?.entitySlug) {
|
||
where.push(`entity_slug = $entitySlug`);
|
||
params.entitySlug = opts.entitySlug;
|
||
}
|
||
return this._listFacts(source_id, {
|
||
...opts,
|
||
whereClauses: where,
|
||
whereParams: params,
|
||
order: 'created_at DESC, id DESC',
|
||
});
|
||
}
|
||
|
||
async listFactsBySession(
|
||
source_id: string,
|
||
sessionId: string,
|
||
opts?: FactListOpts,
|
||
): Promise<FactRow[]> {
|
||
return this._listFacts(source_id, {
|
||
...opts,
|
||
whereClauses: [`source_session = $sessionId`],
|
||
whereParams: { sessionId },
|
||
order: 'created_at DESC, id DESC',
|
||
});
|
||
}
|
||
|
||
async listSupersessions(
|
||
source_id: string,
|
||
opts?: { since?: Date; limit?: number },
|
||
): Promise<FactRow[]> {
|
||
const where: string[] = [`expired_at IS NOT NULL`, `superseded_by IS NOT NULL`];
|
||
const params: Record<string, unknown> = {};
|
||
if (opts?.since) {
|
||
where.push(`expired_at >= $since`);
|
||
params.since = opts.since;
|
||
}
|
||
return this._listFacts(source_id, {
|
||
activeOnly: false,
|
||
limit: opts?.limit,
|
||
whereClauses: where,
|
||
whereParams: params,
|
||
order: 'expired_at DESC, id DESC',
|
||
});
|
||
}
|
||
|
||
async findCandidateDuplicates(
|
||
source_id: string,
|
||
entitySlug: string,
|
||
factText: string,
|
||
opts?: { k?: number; embedding?: Float32Array },
|
||
): Promise<FactRow[]> {
|
||
const k = Math.min(Math.max(opts?.k ?? 5, 1), 20);
|
||
if (opts?.embedding) {
|
||
// Embedding-cosine ordered candidates within the entity bucket.
|
||
const vec = toPgVectorLiteral(opts.embedding);
|
||
const result = await this.db.query<FactRowSqlShape>(
|
||
`SELECT * FROM facts
|
||
WHERE source_id = $1
|
||
AND entity_slug = $2
|
||
AND expired_at IS NULL
|
||
AND embedding IS NOT NULL
|
||
ORDER BY embedding <=> $3::vector
|
||
LIMIT $4`,
|
||
[source_id, entitySlug, vec, k],
|
||
);
|
||
return result.rows.map(rowToFact);
|
||
}
|
||
// Recency fallback when no embedding.
|
||
const result = await this.db.query<FactRowSqlShape>(
|
||
`SELECT * FROM facts
|
||
WHERE source_id = $1
|
||
AND entity_slug = $2
|
||
AND expired_at IS NULL
|
||
ORDER BY created_at DESC, id DESC
|
||
LIMIT $3`,
|
||
[source_id, entitySlug, k],
|
||
);
|
||
return result.rows.map(rowToFact);
|
||
}
|
||
|
||
async consolidateFact(id: number, takeId: number): Promise<void> {
|
||
await this.db.query(
|
||
`UPDATE facts SET consolidated_at = now(), consolidated_into = $1 WHERE id = $2`,
|
||
[takeId, id],
|
||
);
|
||
}
|
||
|
||
async getFactsHealth(source_id: string): Promise<FactsHealth> {
|
||
const total = await this.db.query<{
|
||
total_active: number; total_today: number; total_week: number;
|
||
total_expired: number; total_consolidated: number;
|
||
}>(
|
||
`SELECT
|
||
COUNT(*) FILTER (WHERE expired_at IS NULL) AS total_active,
|
||
COUNT(*) FILTER (WHERE expired_at IS NULL AND created_at > now() - interval '24 hours') AS total_today,
|
||
COUNT(*) FILTER (WHERE expired_at IS NULL AND created_at > now() - interval '7 days') AS total_week,
|
||
COUNT(*) FILTER (WHERE expired_at IS NOT NULL) AS total_expired,
|
||
COUNT(*) FILTER (WHERE consolidated_at IS NOT NULL) AS total_consolidated
|
||
FROM facts WHERE source_id = $1`,
|
||
[source_id],
|
||
);
|
||
const top = await this.db.query<{ entity_slug: string; count: number }>(
|
||
`SELECT entity_slug, COUNT(*)::int AS count
|
||
FROM facts
|
||
WHERE source_id = $1 AND expired_at IS NULL AND entity_slug IS NOT NULL
|
||
GROUP BY entity_slug
|
||
ORDER BY count DESC, entity_slug ASC
|
||
LIMIT 5`,
|
||
[source_id],
|
||
);
|
||
const r = total.rows[0] ?? {
|
||
total_active: 0, total_today: 0, total_week: 0, total_expired: 0, total_consolidated: 0,
|
||
};
|
||
return {
|
||
source_id,
|
||
total_active: Number(r.total_active),
|
||
total_today: Number(r.total_today),
|
||
total_week: Number(r.total_week),
|
||
total_expired: Number(r.total_expired),
|
||
total_consolidated: Number(r.total_consolidated),
|
||
top_entities: top.rows.map(t => ({ entity_slug: t.entity_slug, count: Number(t.count) })),
|
||
};
|
||
}
|
||
|
||
/**
|
||
* Internal helper: shared list-facts query builder.
|
||
* Supports source_id always, plus arbitrary additional WHERE clauses.
|
||
*/
|
||
private async _listFacts(
|
||
source_id: string,
|
||
opts: FactListOpts & {
|
||
whereClauses?: string[];
|
||
whereParams?: Record<string, unknown>;
|
||
order: string;
|
||
},
|
||
): Promise<FactRow[]> {
|
||
const limit = clampSearchLimit(opts.limit, 50, MAX_SEARCH_LIMIT);
|
||
const offset = Math.max(0, opts.offset ?? 0);
|
||
const whereParts: string[] = [`source_id = $source_id`];
|
||
const params: Record<string, unknown> = { source_id };
|
||
if (opts.activeOnly !== false) {
|
||
whereParts.push(`expired_at IS NULL`);
|
||
}
|
||
if (opts.kinds && opts.kinds.length > 0) {
|
||
whereParts.push(`kind = ANY($kinds)`);
|
||
params.kinds = opts.kinds;
|
||
}
|
||
if (opts.visibility && opts.visibility.length > 0) {
|
||
whereParts.push(`visibility = ANY($visibility)`);
|
||
params.visibility = opts.visibility;
|
||
}
|
||
for (const c of opts.whereClauses ?? []) whereParts.push(c);
|
||
Object.assign(params, opts.whereParams ?? {});
|
||
|
||
// Convert $name placeholders to numbered $1, $2, ... for PGLite.
|
||
const orderedKeys = Object.keys(params);
|
||
const indexFor = (name: string): number => orderedKeys.indexOf(name) + 1;
|
||
const sql = `SELECT * FROM facts
|
||
WHERE ${whereParts.join(' AND ').replace(/\$(\w+)/g, (_m, k) => `$${indexFor(k)}`)}
|
||
ORDER BY ${opts.order}
|
||
LIMIT ${limit} OFFSET ${offset}`;
|
||
const result = await this.db.query<FactRowSqlShape>(sql, orderedKeys.map(k => params[k]));
|
||
return result.rows.map(rowToFact);
|
||
}
|
||
|
||
// ============================================================
|
||
// v0.28: Takes (typed/weighted/attributed claims) + synthesis_evidence
|
||
// ============================================================
|
||
|
||
async addTakesBatch(rowsIn: TakeBatchInput[]): Promise<number> {
|
||
if (rowsIn.length === 0) return 0;
|
||
let weightClamped = 0;
|
||
const pageIds = rowsIn.map(r => r.page_id);
|
||
const rowNums = rowsIn.map(r => r.row_num);
|
||
const claims = rowsIn.map(r => r.claim);
|
||
const kinds = rowsIn.map(r => r.kind);
|
||
const holders = rowsIn.map(r => r.holder);
|
||
const weights = rowsIn.map(r => {
|
||
const { weight, clamped } = normalizeWeightForStorage(r.weight);
|
||
if (clamped) weightClamped++;
|
||
return weight;
|
||
});
|
||
const sinces = rowsIn.map(r => r.since_date ?? null);
|
||
const untils = rowsIn.map(r => r.until_date ?? null);
|
||
const sources = rowsIn.map(r => r.source ?? null);
|
||
const supersededBys = rowsIn.map(r => r.superseded_by ?? null);
|
||
const actives = rowsIn.map(r => r.active ?? true);
|
||
if (weightClamped > 0) {
|
||
process.stderr.write(`[takes] TAKES_WEIGHT_CLAMPED: ${weightClamped} row(s) had weight outside [0,1]; clamped\n`);
|
||
}
|
||
const result = await this.db.query(
|
||
`INSERT INTO takes (page_id, row_num, claim, kind, holder, weight, since_date, until_date, source, superseded_by, active)
|
||
SELECT v.page_id::int, v.row_num::int, v.claim, v.kind, v.holder, v.weight::real,
|
||
v.since_date::text, v.until_date::text, v.source, v.superseded_by::int, v.active::boolean
|
||
FROM unnest($1::int[], $2::int[], $3::text[], $4::text[], $5::text[], $6::real[],
|
||
$7::text[], $8::text[], $9::text[], $10::int[], $11::boolean[])
|
||
AS v(page_id, row_num, claim, kind, holder, weight, since_date, until_date, source, superseded_by, active)
|
||
ON CONFLICT (page_id, row_num) DO UPDATE SET
|
||
claim = EXCLUDED.claim,
|
||
kind = EXCLUDED.kind,
|
||
holder = EXCLUDED.holder,
|
||
weight = EXCLUDED.weight,
|
||
since_date = EXCLUDED.since_date,
|
||
until_date = EXCLUDED.until_date,
|
||
source = EXCLUDED.source,
|
||
superseded_by = EXCLUDED.superseded_by,
|
||
active = EXCLUDED.active,
|
||
updated_at = now()
|
||
RETURNING 1`,
|
||
[pageIds, rowNums, claims, kinds, holders, weights, sinces, untils, sources, supersededBys, actives]
|
||
);
|
||
return result.rows.length;
|
||
}
|
||
|
||
async listTakes(opts: TakesListOpts = {}): Promise<Take[]> {
|
||
const limit = clampSearchLimit(opts.limit, 100, 500);
|
||
const offset = Math.max(0, Math.floor(opts.offset ?? 0));
|
||
const active = opts.active ?? true;
|
||
const sortBy = opts.sortBy ?? 'created_at';
|
||
const { rows } = await this.db.query(
|
||
`SELECT t.*, p.slug AS page_slug
|
||
FROM takes t
|
||
JOIN pages p ON p.id = t.page_id
|
||
WHERE 1=1
|
||
AND ($1::int IS NULL OR t.page_id = $1::int)
|
||
AND ($2::text IS NULL OR p.slug = $2::text)
|
||
AND ($3::text IS NULL OR t.holder = $3::text)
|
||
AND ($4::text IS NULL OR t.kind = $4::text)
|
||
AND ($5::boolean IS NULL OR t.active = $5::boolean)
|
||
AND (
|
||
$6::boolean IS NULL
|
||
OR ($6::boolean = true AND t.resolved_at IS NOT NULL)
|
||
OR ($6::boolean = false AND t.resolved_at IS NULL)
|
||
)
|
||
AND ($7::text[] IS NULL OR t.holder = ANY($7::text[]))
|
||
ORDER BY
|
||
CASE WHEN $8 = 'weight' THEN t.weight END DESC NULLS LAST,
|
||
CASE WHEN $8 = 'since_date' THEN t.since_date END DESC NULLS LAST,
|
||
CASE WHEN $8 = 'created_at' THEN t.created_at END DESC NULLS LAST
|
||
LIMIT $9 OFFSET $10`,
|
||
[
|
||
opts.page_id ?? null,
|
||
opts.page_slug ?? null,
|
||
opts.holder ?? null,
|
||
opts.kind ?? null,
|
||
active,
|
||
opts.resolved === undefined ? null : opts.resolved,
|
||
opts.takesHoldersAllowList ?? null,
|
||
sortBy,
|
||
limit,
|
||
offset,
|
||
]
|
||
);
|
||
return rows.map((r) => takeRowToTake(r as Record<string, unknown>));
|
||
}
|
||
|
||
async searchTakes(
|
||
query: string,
|
||
opts: { limit?: number; takesHoldersAllowList?: string[] } = {},
|
||
): Promise<TakeHit[]> {
|
||
const limit = clampSearchLimit(opts.limit, 30, 100);
|
||
const { rows } = await this.db.query(
|
||
`SELECT t.id AS take_id, t.page_id, p.slug AS page_slug, t.row_num,
|
||
t.claim, t.kind, t.holder, t.weight,
|
||
similarity(t.claim, $1)::real AS score
|
||
FROM takes t
|
||
JOIN pages p ON p.id = t.page_id
|
||
WHERE t.active
|
||
AND t.claim % $1
|
||
AND ($2::text[] IS NULL OR t.holder = ANY($2::text[]))
|
||
ORDER BY score DESC, t.weight DESC
|
||
LIMIT $3`,
|
||
[query, opts.takesHoldersAllowList ?? null, limit]
|
||
);
|
||
return rows as unknown as TakeHit[];
|
||
}
|
||
|
||
async searchTakesVector(
|
||
embedding: Float32Array,
|
||
opts: { limit?: number; takesHoldersAllowList?: string[] } = {},
|
||
): Promise<TakeHit[]> {
|
||
const limit = clampSearchLimit(opts.limit, 30, 100);
|
||
const vec = `[${Array.from(embedding).join(',')}]`;
|
||
const { rows } = await this.db.query(
|
||
`SELECT t.id AS take_id, t.page_id, p.slug AS page_slug, t.row_num,
|
||
t.claim, t.kind, t.holder, t.weight,
|
||
(1 - (t.embedding <=> $1::vector))::real AS score
|
||
FROM takes t
|
||
JOIN pages p ON p.id = t.page_id
|
||
WHERE t.active
|
||
AND t.embedding IS NOT NULL
|
||
AND ($2::text[] IS NULL OR t.holder = ANY($2::text[]))
|
||
ORDER BY t.embedding <=> $1::vector
|
||
LIMIT $3`,
|
||
[vec, opts.takesHoldersAllowList ?? null, limit]
|
||
);
|
||
return rows as unknown as TakeHit[];
|
||
}
|
||
|
||
async getTakeEmbeddings(ids: number[]): Promise<Map<number, Float32Array>> {
|
||
if (ids.length === 0) return new Map();
|
||
const { rows } = await this.db.query(
|
||
`SELECT id, embedding FROM takes WHERE id = ANY($1::bigint[]) AND embedding IS NOT NULL`,
|
||
[ids]
|
||
);
|
||
const out = new Map<number, Float32Array>();
|
||
for (const r of rows as Array<{ id: number; embedding: unknown }>) {
|
||
const v = r.embedding;
|
||
if (typeof v === 'string') {
|
||
const trimmed = v.replace(/^\[|\]$/g, '');
|
||
const arr = trimmed.split(',').map(parseFloat).filter(n => !Number.isNaN(n));
|
||
out.set(Number(r.id), new Float32Array(arr));
|
||
} else if (Array.isArray(v)) {
|
||
out.set(Number(r.id), new Float32Array(v as number[]));
|
||
}
|
||
}
|
||
return out;
|
||
}
|
||
|
||
async countStaleTakes(): Promise<number> {
|
||
const { rows } = await this.db.query(
|
||
`SELECT count(*)::int AS count FROM takes WHERE active AND embedding IS NULL`
|
||
);
|
||
return Number((rows[0] as { count?: number } | undefined)?.count ?? 0);
|
||
}
|
||
|
||
async listStaleTakes(): Promise<StaleTakeRow[]> {
|
||
const { rows } = await this.db.query(
|
||
`SELECT t.id AS take_id, p.slug AS page_slug, t.row_num, t.claim
|
||
FROM takes t
|
||
JOIN pages p ON p.id = t.page_id
|
||
WHERE t.active AND t.embedding IS NULL
|
||
ORDER BY t.id
|
||
LIMIT 100000`
|
||
);
|
||
return rows as unknown as StaleTakeRow[];
|
||
}
|
||
|
||
async updateTake(
|
||
pageId: number,
|
||
rowNum: number,
|
||
fields: { weight?: number; since_date?: string; source?: string },
|
||
): Promise<void> {
|
||
let weight = fields.weight;
|
||
if (weight !== undefined) {
|
||
const norm = normalizeWeightForStorage(weight);
|
||
if (norm.clamped) {
|
||
process.stderr.write(`[takes] TAKES_WEIGHT_CLAMPED: updateTake clamped weight ${weight} → ${norm.weight}\n`);
|
||
}
|
||
weight = norm.weight;
|
||
}
|
||
const result = await this.db.query(
|
||
`UPDATE takes SET
|
||
weight = COALESCE($3::real, weight),
|
||
since_date = COALESCE($4::text, since_date),
|
||
source = COALESCE($5::text, source),
|
||
updated_at = now()
|
||
WHERE page_id = $1 AND row_num = $2
|
||
RETURNING 1`,
|
||
[pageId, rowNum, weight ?? null, fields.since_date ?? null, fields.source ?? null]
|
||
);
|
||
if (result.rows.length === 0) {
|
||
throw new GBrainError(
|
||
'TAKE_ROW_NOT_FOUND',
|
||
`take not found at page_id=${pageId} row=${rowNum}`,
|
||
'list takes for this page with `gbrain takes <slug>` to see valid row numbers',
|
||
);
|
||
}
|
||
}
|
||
|
||
async supersedeTake(
|
||
pageId: number,
|
||
oldRow: number,
|
||
newRow: Omit<TakeBatchInput, 'page_id' | 'row_num' | 'superseded_by'>,
|
||
): Promise<{ oldRow: number; newRow: number }> {
|
||
return await this.db.transaction(async (tx) => {
|
||
const existingRes = await tx.query(
|
||
`SELECT resolved_at FROM takes WHERE page_id = $1 AND row_num = $2`,
|
||
[pageId, oldRow]
|
||
);
|
||
const existing = existingRes.rows[0] as { resolved_at?: unknown } | undefined;
|
||
if (!existing) {
|
||
throw new GBrainError('TAKE_ROW_NOT_FOUND', `take not found at page_id=${pageId} row=${oldRow}`, 'list takes with `gbrain takes <slug>`');
|
||
}
|
||
if (existing.resolved_at) {
|
||
throw new GBrainError('TAKE_RESOLVED_IMMUTABLE', `take ${pageId}#${oldRow} is resolved`, 'resolved bets are immutable; add a new take instead');
|
||
}
|
||
const maxRowRes = await tx.query(
|
||
`SELECT COALESCE(MAX(row_num), 0) + 1 AS next FROM takes WHERE page_id = $1`,
|
||
[pageId]
|
||
);
|
||
const newRowNum = Number((maxRowRes.rows[0] as { next?: number })?.next ?? 1);
|
||
const w = Math.max(0, Math.min(1, newRow.weight ?? 0.5));
|
||
await tx.query(
|
||
`INSERT INTO takes (page_id, row_num, claim, kind, holder, weight, since_date, until_date, source, active)
|
||
VALUES ($1, $2, $3, $4, $5, $6, $7::text, $8::text, $9, $10)`,
|
||
[
|
||
pageId, newRowNum, newRow.claim, newRow.kind, newRow.holder, w,
|
||
newRow.since_date ?? null, newRow.until_date ?? null, newRow.source ?? null,
|
||
newRow.active ?? true,
|
||
]
|
||
);
|
||
await tx.query(
|
||
`UPDATE takes SET active = false, superseded_by = $3, updated_at = now()
|
||
WHERE page_id = $1 AND row_num = $2`,
|
||
[pageId, oldRow, newRowNum]
|
||
);
|
||
return { oldRow, newRow: newRowNum };
|
||
});
|
||
}
|
||
|
||
async resolveTake(pageId: number, rowNum: number, resolution: TakeResolution): Promise<void> {
|
||
const existingRes = await this.db.query(
|
||
`SELECT resolved_at FROM takes WHERE page_id = $1 AND row_num = $2`,
|
||
[pageId, rowNum]
|
||
);
|
||
const existing = existingRes.rows[0] as { resolved_at?: unknown } | undefined;
|
||
if (!existing) {
|
||
throw new GBrainError('TAKE_ROW_NOT_FOUND', `take not found at page_id=${pageId} row=${rowNum}`, 'list takes with `gbrain takes <slug>`');
|
||
}
|
||
if (existing.resolved_at) {
|
||
throw new GBrainError('TAKE_ALREADY_RESOLVED', `take ${pageId}#${rowNum} already resolved`, 'resolution is immutable; add a new take to record a new outcome');
|
||
}
|
||
// v0.30.0: derive (quality, outcome) tuple. quality wins when both set.
|
||
const { quality, outcome } = deriveResolutionTuple(resolution);
|
||
await this.db.query(
|
||
`UPDATE takes SET
|
||
resolved_at = now(),
|
||
resolved_quality = $3::text,
|
||
resolved_outcome = $4,
|
||
resolved_value = $5::real,
|
||
resolved_unit = $6::text,
|
||
resolved_source = $7::text,
|
||
resolved_by = $8,
|
||
updated_at = now()
|
||
WHERE page_id = $1 AND row_num = $2`,
|
||
[
|
||
pageId, rowNum,
|
||
quality,
|
||
outcome,
|
||
resolution.value ?? null,
|
||
resolution.unit ?? null,
|
||
resolution.source ?? null,
|
||
resolution.resolvedBy,
|
||
]
|
||
);
|
||
}
|
||
|
||
/**
|
||
* v0.30.0: aggregate scorecard. SQL-level allow-list filter (D4 fail-closed).
|
||
* Hidden-holder rows contribute zero to aggregates.
|
||
*/
|
||
async getScorecard(opts: TakesScorecardOpts, allowList: string[] | undefined): Promise<TakesScorecard> {
|
||
// Build the WHERE clause with positional params. PGLite (postgres-via-WASM)
|
||
// shares the SQL dialect with real Postgres so the math expressions match.
|
||
const params: unknown[] = [];
|
||
const clauses: string[] = [];
|
||
if (opts.holder !== undefined) { params.push(opts.holder); clauses.push(`AND holder = $${params.length}`); }
|
||
if (opts.domainPrefix !== undefined) {
|
||
params.push(opts.domainPrefix + '%');
|
||
clauses.push(`AND EXISTS (SELECT 1 FROM pages p WHERE p.id = takes.page_id AND p.slug LIKE $${params.length})`);
|
||
}
|
||
if (opts.since !== undefined) { params.push(opts.since); clauses.push(`AND since_date >= $${params.length}`); }
|
||
if (opts.until !== undefined) { params.push(opts.until); clauses.push(`AND since_date <= $${params.length}`); }
|
||
if (allowList !== undefined) { params.push(allowList); clauses.push(`AND holder = ANY($${params.length}::text[])`); }
|
||
const where = clauses.join(' ');
|
||
const res = await this.db.query(
|
||
`SELECT
|
||
COUNT(*) FILTER (WHERE kind = 'bet')::int AS total_bets,
|
||
COUNT(*) FILTER (WHERE resolved_quality IS NOT NULL)::int AS resolved,
|
||
COUNT(*) FILTER (WHERE resolved_quality = 'correct')::int AS correct,
|
||
COUNT(*) FILTER (WHERE resolved_quality = 'incorrect')::int AS incorrect,
|
||
COUNT(*) FILTER (WHERE resolved_quality = 'partial')::int AS partial,
|
||
AVG(
|
||
CASE WHEN resolved_quality IN ('correct','incorrect')
|
||
THEN POWER(weight - (CASE resolved_quality WHEN 'correct' THEN 1 ELSE 0 END), 2)
|
||
END
|
||
)::float AS brier
|
||
FROM takes
|
||
WHERE 1=1 ${where}`,
|
||
params,
|
||
);
|
||
const r = res.rows[0] as { total_bets: number; resolved: number; correct: number; incorrect: number; partial: number; brier: number | null };
|
||
return finalizeScorecard(r);
|
||
}
|
||
|
||
/**
|
||
* v0.30.0: calibration curve. Bins resolved correct/incorrect bets by stated weight.
|
||
*/
|
||
async getCalibrationCurve(opts: CalibrationCurveOpts, allowList: string[] | undefined): Promise<CalibrationBucket[]> {
|
||
const bucketSize = opts.bucketSize && opts.bucketSize > 0 && opts.bucketSize <= 1 ? opts.bucketSize : 0.1;
|
||
const maxIdx = Math.floor(1 / bucketSize) - 1;
|
||
const params: unknown[] = [bucketSize, maxIdx];
|
||
const clauses: string[] = [];
|
||
if (opts.holder !== undefined) { params.push(opts.holder); clauses.push(`AND holder = $${params.length}`); }
|
||
if (allowList !== undefined) { params.push(allowList); clauses.push(`AND holder = ANY($${params.length}::text[])`); }
|
||
const where = clauses.join(' ');
|
||
// NUMERIC casts for exact decimal arithmetic — keeps PGLite + Postgres
|
||
// bucket boundaries identical at FP-edge weights (e.g. 0.7/0.1).
|
||
// See parity test in test/e2e/takes-scorecard-parity.test.ts.
|
||
const res = await this.db.query(
|
||
`WITH binned AS (
|
||
SELECT
|
||
LEAST(FLOOR(weight::numeric / $1::numeric)::int, $2::int)::int AS bucket_idx,
|
||
weight,
|
||
(resolved_quality = 'correct')::int AS hit
|
||
FROM takes
|
||
WHERE resolved_quality IN ('correct','incorrect')
|
||
${where}
|
||
)
|
||
SELECT
|
||
(bucket_idx::numeric * $1::numeric)::float AS bucket_lo,
|
||
((bucket_idx + 1)::numeric * $1::numeric)::float AS bucket_hi,
|
||
COUNT(*)::int AS n,
|
||
AVG(hit)::float AS observed,
|
||
AVG(weight)::float AS predicted
|
||
FROM binned
|
||
GROUP BY bucket_idx
|
||
ORDER BY bucket_idx`,
|
||
params,
|
||
);
|
||
return (res.rows as { bucket_lo: number; bucket_hi: number; n: number; observed: number | null; predicted: number | null }[]).map(r => ({
|
||
bucket_lo: r.bucket_lo,
|
||
bucket_hi: r.bucket_hi,
|
||
n: r.n,
|
||
observed: r.n > 0 ? r.observed : null,
|
||
predicted: r.n > 0 ? r.predicted : null,
|
||
}));
|
||
}
|
||
|
||
async addSynthesisEvidence(rowsIn: SynthesisEvidenceInput[]): Promise<number> {
|
||
if (rowsIn.length === 0) return 0;
|
||
const synthesisIds = rowsIn.map(r => r.synthesis_page_id);
|
||
const takePageIds = rowsIn.map(r => r.take_page_id);
|
||
const takeRowNums = rowsIn.map(r => r.take_row_num);
|
||
const citationIxs = rowsIn.map(r => r.citation_index);
|
||
const result = await this.db.query(
|
||
`INSERT INTO synthesis_evidence (synthesis_page_id, take_page_id, take_row_num, citation_index)
|
||
SELECT v.synthesis_page_id::int, v.take_page_id::int, v.take_row_num::int, v.citation_index::int
|
||
FROM unnest($1::int[], $2::int[], $3::int[], $4::int[])
|
||
AS v(synthesis_page_id, take_page_id, take_row_num, citation_index)
|
||
ON CONFLICT (synthesis_page_id, take_page_id, take_row_num) DO NOTHING
|
||
RETURNING 1`,
|
||
[synthesisIds, takePageIds, takeRowNums, citationIxs]
|
||
);
|
||
return result.rows.length;
|
||
}
|
||
|
||
// Versions
|
||
async createVersion(slug: string, opts?: { sourceId?: string }): Promise<PageVersion> {
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
const { rows } = await this.db.query(
|
||
`INSERT INTO page_versions (page_id, compiled_truth, frontmatter)
|
||
SELECT id, compiled_truth, frontmatter
|
||
FROM pages WHERE slug = $1 AND source_id = $2
|
||
RETURNING *`,
|
||
[slug, sourceId]
|
||
);
|
||
if (rows.length === 0) throw new Error(`createVersion failed: page "${slug}" (source=${sourceId}) not found`);
|
||
return rows[0] as unknown as PageVersion;
|
||
}
|
||
|
||
async getVersions(slug: string, opts?: { sourceId?: string }): Promise<PageVersion[]> {
|
||
// v0.31.8 (D16): two-branch. Without opts.sourceId, joins return versions
|
||
// for every same-slug page (preserves pre-v0.31.8 cross-source view).
|
||
if (opts?.sourceId) {
|
||
const { rows } = await this.db.query(
|
||
`SELECT pv.* FROM page_versions pv
|
||
JOIN pages p ON p.id = pv.page_id
|
||
WHERE p.slug = $1 AND p.source_id = $2
|
||
ORDER BY pv.snapshot_at DESC`,
|
||
[slug, opts.sourceId]
|
||
);
|
||
return rows as unknown as PageVersion[];
|
||
}
|
||
const { rows } = await this.db.query(
|
||
`SELECT pv.* FROM page_versions pv
|
||
JOIN pages p ON p.id = pv.page_id
|
||
WHERE p.slug = $1
|
||
ORDER BY pv.snapshot_at DESC`,
|
||
[slug]
|
||
);
|
||
return rows as unknown as PageVersion[];
|
||
}
|
||
|
||
async revertToVersion(
|
||
slug: string,
|
||
versionId: number,
|
||
opts?: { sourceId?: string },
|
||
): Promise<void> {
|
||
// v0.31.8 (D12): when opts.sourceId is set, scope BOTH the page lookup
|
||
// and the version row reference. Without it, multi-source brains can
|
||
// revert the wrong same-slug page (the one Postgres returns first).
|
||
if (opts?.sourceId) {
|
||
await this.db.query(
|
||
`UPDATE pages SET
|
||
compiled_truth = pv.compiled_truth,
|
||
frontmatter = pv.frontmatter,
|
||
updated_at = now()
|
||
FROM page_versions pv
|
||
WHERE pages.slug = $1 AND pages.source_id = $3
|
||
AND pv.id = $2 AND pv.page_id = pages.id`,
|
||
[slug, versionId, opts.sourceId]
|
||
);
|
||
return;
|
||
}
|
||
await this.db.query(
|
||
`UPDATE pages SET
|
||
compiled_truth = pv.compiled_truth,
|
||
frontmatter = pv.frontmatter,
|
||
updated_at = now()
|
||
FROM page_versions pv
|
||
WHERE pages.slug = $1 AND pv.id = $2 AND pv.page_id = pages.id`,
|
||
[slug, versionId]
|
||
);
|
||
}
|
||
|
||
// Stats + health
|
||
async getStats(): Promise<BrainStats> {
|
||
const { rows: [stats] } = await this.db.query(`
|
||
SELECT
|
||
-- v0.26.5: exclude soft-deleted from page_count (mirrors postgres-engine).
|
||
(SELECT count(*) FROM pages WHERE deleted_at IS NULL) as page_count,
|
||
(SELECT count(*) FROM content_chunks) as chunk_count,
|
||
(SELECT count(*) FROM content_chunks WHERE embedded_at IS NOT NULL) as embedded_count,
|
||
(SELECT count(*) FROM links) as link_count,
|
||
(SELECT count(DISTINCT tag) FROM tags) as tag_count,
|
||
(SELECT count(*) FROM timeline_entries) as timeline_entry_count
|
||
`);
|
||
|
||
const { rows: types } = await this.db.query(
|
||
`SELECT type, count(*)::int as count FROM pages GROUP BY type ORDER BY count DESC`
|
||
);
|
||
const pages_by_type: Record<string, number> = {};
|
||
for (const t of types as { type: string; count: number }[]) {
|
||
pages_by_type[t.type] = t.count;
|
||
}
|
||
|
||
const s = stats as Record<string, unknown>;
|
||
return {
|
||
page_count: Number(s.page_count),
|
||
chunk_count: Number(s.chunk_count),
|
||
embedded_count: Number(s.embedded_count),
|
||
link_count: Number(s.link_count),
|
||
tag_count: Number(s.tag_count),
|
||
timeline_entry_count: Number(s.timeline_entry_count),
|
||
pages_by_type,
|
||
};
|
||
}
|
||
|
||
async getHealth(): Promise<BrainHealth> {
|
||
// Combined metrics from master (brain_score components: dead_links, link_count,
|
||
// pages_with_timeline) and v0.10.3 graph layer (link_coverage, timeline_coverage,
|
||
// most_connected). Both coexist: master's brain_score is the composite
|
||
// dashboard, v0.10.3 metrics give entity-page-level granularity.
|
||
const { rows: [h] } = await this.db.query(`
|
||
WITH entity_pages AS (
|
||
SELECT id, slug FROM pages WHERE type IN ('person', 'company')
|
||
)
|
||
SELECT
|
||
(SELECT count(*) FROM pages) as page_count,
|
||
(SELECT count(*) FROM content_chunks WHERE embedded_at IS NOT NULL)::float /
|
||
GREATEST((SELECT count(*) FROM content_chunks), 1)::float as embed_coverage,
|
||
(SELECT count(*) FROM pages p
|
||
WHERE p.updated_at < (SELECT MAX(te.created_at) FROM timeline_entries te WHERE te.page_id = p.id)
|
||
) as stale_pages,
|
||
-- Bug 11 — orphan = islanded (no inbound AND no outbound).
|
||
-- See BrainHealth.orphan_pages docstring; docs updated to match this.
|
||
(SELECT count(*) FROM pages p
|
||
WHERE NOT EXISTS (SELECT 1 FROM links l WHERE l.to_page_id = p.id)
|
||
AND NOT EXISTS (SELECT 1 FROM links l WHERE l.from_page_id = p.id)
|
||
) as orphan_pages,
|
||
(SELECT count(*) FROM links l
|
||
WHERE NOT EXISTS (SELECT 1 FROM pages p WHERE p.id = l.to_page_id)
|
||
) as dead_links,
|
||
(SELECT count(*) FROM content_chunks WHERE embedded_at IS NULL) as missing_embeddings,
|
||
(SELECT count(*) FROM links) as link_count,
|
||
(SELECT count(DISTINCT page_id) FROM timeline_entries) as pages_with_timeline,
|
||
(SELECT count(*) FROM entity_pages e
|
||
WHERE EXISTS (SELECT 1 FROM links l WHERE l.to_page_id = e.id))::float /
|
||
GREATEST((SELECT count(*) FROM entity_pages), 1)::float as link_coverage,
|
||
(SELECT count(*) FROM entity_pages e
|
||
WHERE EXISTS (SELECT 1 FROM timeline_entries te WHERE te.page_id = e.id))::float /
|
||
GREATEST((SELECT count(*) FROM entity_pages), 1)::float as timeline_coverage
|
||
`);
|
||
|
||
// Top 5 most connected entities by total link count (in + out).
|
||
const { rows: connected } = await this.db.query(`
|
||
SELECT p.slug,
|
||
(SELECT count(*) FROM links l WHERE l.from_page_id = p.id OR l.to_page_id = p.id)::int as link_count
|
||
FROM pages p
|
||
WHERE p.type IN ('person', 'company')
|
||
ORDER BY link_count DESC
|
||
LIMIT 5
|
||
`);
|
||
|
||
const r = h as Record<string, unknown>;
|
||
const pageCount = Number(r.page_count);
|
||
const embedCoverage = Number(r.embed_coverage);
|
||
const orphanPages = Number(r.orphan_pages);
|
||
const deadLinks = Number(r.dead_links);
|
||
const linkCount = Number(r.link_count);
|
||
const pagesWithTimeline = Number(r.pages_with_timeline);
|
||
|
||
const linkDensity = pageCount > 0 ? Math.min(linkCount / pageCount, 1) : 0;
|
||
const timelineCoverageDensity = pageCount > 0 ? Math.min(pagesWithTimeline / pageCount, 1) : 0;
|
||
const noOrphans = pageCount > 0 ? 1 - (orphanPages / pageCount) : 1;
|
||
const noDeadLinks = pageCount > 0 ? 1 - Math.min(deadLinks / pageCount, 1) : 1;
|
||
// Bug 11 — per-component points. Sum equals brainScore by construction
|
||
// so `doctor` can render a breakdown that adds up to the total.
|
||
const embedCoverageScore = pageCount === 0 ? 0 : Math.round(embedCoverage * 35);
|
||
const linkDensityScore = pageCount === 0 ? 0 : Math.round(linkDensity * 25);
|
||
const timelineCoverageScore = pageCount === 0 ? 0 : Math.round(timelineCoverageDensity * 15);
|
||
const noOrphansScore = pageCount === 0 ? 0 : Math.round(noOrphans * 15);
|
||
const noDeadLinksScore = pageCount === 0 ? 0 : Math.round(noDeadLinks * 10);
|
||
const brainScore = embedCoverageScore + linkDensityScore + timelineCoverageScore + noOrphansScore + noDeadLinksScore;
|
||
|
||
return {
|
||
page_count: pageCount,
|
||
embed_coverage: embedCoverage,
|
||
stale_pages: Number(r.stale_pages),
|
||
orphan_pages: orphanPages,
|
||
missing_embeddings: Number(r.missing_embeddings),
|
||
brain_score: brainScore,
|
||
dead_links: deadLinks,
|
||
link_coverage: Number(r.link_coverage),
|
||
timeline_coverage: Number(r.timeline_coverage),
|
||
most_connected: (connected as { slug: string; link_count: number }[]).map(c => ({
|
||
slug: c.slug,
|
||
link_count: Number(c.link_count),
|
||
})),
|
||
embed_coverage_score: embedCoverageScore,
|
||
link_density_score: linkDensityScore,
|
||
timeline_coverage_score: timelineCoverageScore,
|
||
no_orphans_score: noOrphansScore,
|
||
no_dead_links_score: noDeadLinksScore,
|
||
};
|
||
}
|
||
|
||
// Ingest log
|
||
async logIngest(entry: IngestLogInput): Promise<void> {
|
||
// v0.31.2 (codex P1 #3): source_id threaded so multi-source brains can
|
||
// scope ingest_log queries. Default 'default' matches the column DEFAULT.
|
||
const sourceId = entry.source_id ?? 'default';
|
||
await this.db.query(
|
||
`INSERT INTO ingest_log (source_id, source_type, source_ref, pages_updated, summary)
|
||
VALUES ($1, $2, $3, $4::jsonb, $5)`,
|
||
[sourceId, entry.source_type, entry.source_ref, JSON.stringify(entry.pages_updated), entry.summary]
|
||
);
|
||
}
|
||
|
||
async getIngestLog(opts?: { limit?: number }): Promise<IngestLogEntry[]> {
|
||
const limit = opts?.limit || 50;
|
||
const { rows } = await this.db.query(
|
||
`SELECT * FROM ingest_log ORDER BY created_at DESC LIMIT $1`,
|
||
[limit]
|
||
);
|
||
// Belt-and-suspenders source_id fallback for any pre-v50 row that
|
||
// somehow survived without the backfill.
|
||
return (rows as unknown as IngestLogEntry[]).map(r => ({
|
||
...r,
|
||
source_id: r.source_id ?? 'default',
|
||
}));
|
||
}
|
||
|
||
// Sync
|
||
async updateSlug(oldSlug: string, newSlug: string, opts?: { sourceId?: string }): Promise<void> {
|
||
newSlug = validateSlug(newSlug);
|
||
const sourceId = opts?.sourceId ?? 'default';
|
||
// Source-qualify so a rename in source A doesn't sweep up same-slug rows
|
||
// in sources B/C/D (mirrors postgres-engine.ts).
|
||
await this.db.query(
|
||
`UPDATE pages SET slug = $1, updated_at = now() WHERE slug = $2 AND source_id = $3`,
|
||
[newSlug, oldSlug, sourceId]
|
||
);
|
||
}
|
||
|
||
async rewriteLinks(_oldSlug: string, _newSlug: string): Promise<void> {
|
||
// Stub: links use integer page_id FKs, already correct after updateSlug.
|
||
}
|
||
|
||
// Config
|
||
async getConfig(key: string): Promise<string | null> {
|
||
const { rows } = await this.db.query('SELECT value FROM config WHERE key = $1', [key]);
|
||
return rows.length > 0 ? (rows[0] as { value: string }).value : null;
|
||
}
|
||
|
||
async setConfig(key: string, value: string): Promise<void> {
|
||
await this.db.query(
|
||
`INSERT INTO config (key, value) VALUES ($1, $2)
|
||
ON CONFLICT (key) DO UPDATE SET value = EXCLUDED.value`,
|
||
[key, value]
|
||
);
|
||
}
|
||
|
||
// Migration support
|
||
async runMigration(_version: number, sql: string): Promise<void> {
|
||
await this.db.exec(sql);
|
||
}
|
||
|
||
async getChunksWithEmbeddings(slug: string): Promise<Chunk[]> {
|
||
const { rows } = await this.db.query(
|
||
`SELECT cc.* FROM content_chunks cc
|
||
JOIN pages p ON p.id = cc.page_id
|
||
WHERE p.slug = $1
|
||
ORDER BY cc.chunk_index`,
|
||
[slug]
|
||
);
|
||
return (rows as Record<string, unknown>[]).map(r => rowToChunk(r, true));
|
||
}
|
||
|
||
async executeRaw<T = Record<string, unknown>>(sql: string, params?: unknown[]): Promise<T[]> {
|
||
const { rows } = await this.db.query(sql, params);
|
||
return rows as T[];
|
||
}
|
||
|
||
// ============================================================
|
||
// v0.20.0 Cathedral II: code edges (Layer 1 stubs — filled by Layer 5)
|
||
// ============================================================
|
||
// Declared here so the interface contract is satisfied and consumers can
|
||
// import against them. Implementations throw until the edge extractor +
|
||
// per-lang tree-sitter queries land in Layer 5/6.
|
||
// ============================================================
|
||
|
||
async addCodeEdges(edges: import('./types.ts').CodeEdgeInput[]): Promise<number> {
|
||
if (edges.length === 0) return 0;
|
||
let inserted = 0;
|
||
// Split into resolved vs unresolved. Resolved rows carry to_chunk_id
|
||
// (known target chunk); unresolved rows only know the qualified name.
|
||
const resolved = edges.filter(e => e.to_chunk_id != null);
|
||
const unresolved = edges.filter(e => e.to_chunk_id == null);
|
||
|
||
if (resolved.length > 0) {
|
||
const rowParts: string[] = [];
|
||
const params: unknown[] = [];
|
||
let p = 1;
|
||
for (const e of resolved) {
|
||
rowParts.push(`($${p++}, $${p++}, $${p++}, $${p++}, $${p++}, $${p++}::jsonb, $${p++})`);
|
||
params.push(
|
||
e.from_chunk_id, e.to_chunk_id, e.from_symbol_qualified,
|
||
e.to_symbol_qualified, e.edge_type,
|
||
JSON.stringify(e.edge_metadata ?? {}),
|
||
e.source_id ?? null,
|
||
);
|
||
}
|
||
const res = await this.db.query(
|
||
`INSERT INTO code_edges_chunk
|
||
(from_chunk_id, to_chunk_id, from_symbol_qualified, to_symbol_qualified, edge_type, edge_metadata, source_id)
|
||
VALUES ${rowParts.join(', ')}
|
||
ON CONFLICT (from_chunk_id, to_chunk_id, edge_type) DO NOTHING`,
|
||
params,
|
||
);
|
||
inserted += res.affectedRows ?? 0;
|
||
}
|
||
if (unresolved.length > 0) {
|
||
const rowParts: string[] = [];
|
||
const params: unknown[] = [];
|
||
let p = 1;
|
||
for (const e of unresolved) {
|
||
rowParts.push(`($${p++}, $${p++}, $${p++}, $${p++}, $${p++}::jsonb, $${p++})`);
|
||
params.push(
|
||
e.from_chunk_id, e.from_symbol_qualified, e.to_symbol_qualified, e.edge_type,
|
||
JSON.stringify(e.edge_metadata ?? {}),
|
||
e.source_id ?? null,
|
||
);
|
||
}
|
||
const res = await this.db.query(
|
||
`INSERT INTO code_edges_symbol
|
||
(from_chunk_id, from_symbol_qualified, to_symbol_qualified, edge_type, edge_metadata, source_id)
|
||
VALUES ${rowParts.join(', ')}
|
||
ON CONFLICT (from_chunk_id, to_symbol_qualified, edge_type) DO NOTHING`,
|
||
params,
|
||
);
|
||
inserted += res.affectedRows ?? 0;
|
||
}
|
||
return inserted;
|
||
}
|
||
|
||
async deleteCodeEdgesForChunks(chunkIds: number[]): Promise<void> {
|
||
if (chunkIds.length === 0) return;
|
||
// Both directions on code_edges_chunk; from-only on code_edges_symbol
|
||
// (unresolved edges don't have a to_chunk_id to match against).
|
||
await this.db.query(
|
||
`DELETE FROM code_edges_chunk WHERE from_chunk_id = ANY($1::int[]) OR to_chunk_id = ANY($1::int[])`,
|
||
[chunkIds],
|
||
);
|
||
await this.db.query(
|
||
`DELETE FROM code_edges_symbol WHERE from_chunk_id = ANY($1::int[])`,
|
||
[chunkIds],
|
||
);
|
||
}
|
||
|
||
async getCallersOf(
|
||
qualifiedName: string,
|
||
opts?: { sourceId?: string; allSources?: boolean; limit?: number },
|
||
): Promise<import('./types.ts').CodeEdgeResult[]> {
|
||
const limit = Math.min(opts?.limit ?? 100, 500);
|
||
const sourceClause = opts?.allSources || !opts?.sourceId
|
||
? ''
|
||
: `AND source_id = '${opts.sourceId.replace(/'/g, "''")}'`;
|
||
const { rows } = await this.db.query(
|
||
`SELECT id, from_chunk_id, to_chunk_id, from_symbol_qualified, to_symbol_qualified,
|
||
edge_type, edge_metadata, source_id, true as resolved
|
||
FROM code_edges_chunk
|
||
WHERE to_symbol_qualified = $1 ${sourceClause}
|
||
UNION ALL
|
||
SELECT id, from_chunk_id, NULL as to_chunk_id, from_symbol_qualified, to_symbol_qualified,
|
||
edge_type, edge_metadata, source_id, false as resolved
|
||
FROM code_edges_symbol
|
||
WHERE to_symbol_qualified = $1 ${sourceClause}
|
||
LIMIT $2`,
|
||
[qualifiedName, limit],
|
||
);
|
||
return (rows as Record<string, unknown>[]).map(rowToCodeEdge);
|
||
}
|
||
|
||
async getCalleesOf(
|
||
qualifiedName: string,
|
||
opts?: { sourceId?: string; allSources?: boolean; limit?: number },
|
||
): Promise<import('./types.ts').CodeEdgeResult[]> {
|
||
const limit = Math.min(opts?.limit ?? 100, 500);
|
||
const sourceClause = opts?.allSources || !opts?.sourceId
|
||
? ''
|
||
: `AND source_id = '${opts.sourceId.replace(/'/g, "''")}'`;
|
||
const { rows } = await this.db.query(
|
||
`SELECT id, from_chunk_id, to_chunk_id, from_symbol_qualified, to_symbol_qualified,
|
||
edge_type, edge_metadata, source_id, true as resolved
|
||
FROM code_edges_chunk
|
||
WHERE from_symbol_qualified = $1 ${sourceClause}
|
||
UNION ALL
|
||
SELECT id, from_chunk_id, NULL as to_chunk_id, from_symbol_qualified, to_symbol_qualified,
|
||
edge_type, edge_metadata, source_id, false as resolved
|
||
FROM code_edges_symbol
|
||
WHERE from_symbol_qualified = $1 ${sourceClause}
|
||
LIMIT $2`,
|
||
[qualifiedName, limit],
|
||
);
|
||
return (rows as Record<string, unknown>[]).map(rowToCodeEdge);
|
||
}
|
||
|
||
async getEdgesByChunk(
|
||
chunkId: number,
|
||
opts?: { direction?: 'in' | 'out' | 'both'; edgeType?: string; limit?: number },
|
||
): Promise<import('./types.ts').CodeEdgeResult[]> {
|
||
const direction = opts?.direction ?? 'both';
|
||
const limit = Math.min(opts?.limit ?? 50, 200);
|
||
const edgeTypeClause = opts?.edgeType ? `AND edge_type = '${opts.edgeType.replace(/'/g, "''")}'` : '';
|
||
// Build the chunk-table filter based on direction. Unresolved edges
|
||
// (code_edges_symbol) only carry from_chunk_id — there's no inbound
|
||
// direction into them from a chunk ID, so we include them only when
|
||
// direction is 'out' or 'both'.
|
||
let chunkFilter = '';
|
||
if (direction === 'in') chunkFilter = `WHERE to_chunk_id = $1`;
|
||
else if (direction === 'out') chunkFilter = `WHERE from_chunk_id = $1`;
|
||
else chunkFilter = `WHERE from_chunk_id = $1 OR to_chunk_id = $1`;
|
||
|
||
let symbolFilter = '';
|
||
if (direction === 'out' || direction === 'both') {
|
||
symbolFilter = `WHERE from_chunk_id = $1`;
|
||
}
|
||
|
||
const unionClause = symbolFilter ? `
|
||
UNION ALL
|
||
SELECT id, from_chunk_id, NULL as to_chunk_id, from_symbol_qualified, to_symbol_qualified,
|
||
edge_type, edge_metadata, source_id, false as resolved
|
||
FROM code_edges_symbol
|
||
${symbolFilter} ${edgeTypeClause}
|
||
` : '';
|
||
|
||
const { rows } = await this.db.query(
|
||
`SELECT id, from_chunk_id, to_chunk_id, from_symbol_qualified, to_symbol_qualified,
|
||
edge_type, edge_metadata, source_id, true as resolved
|
||
FROM code_edges_chunk
|
||
${chunkFilter} ${edgeTypeClause}
|
||
${unionClause}
|
||
LIMIT $2`,
|
||
[chunkId, limit],
|
||
);
|
||
return (rows as Record<string, unknown>[]).map(rowToCodeEdge);
|
||
}
|
||
|
||
// Eval capture (v0.25.0). See BrainEngine interface docs.
|
||
async logEvalCandidate(input: EvalCandidateInput): Promise<number> {
|
||
const { rows } = await this.db.query<{ id: number }>(
|
||
`INSERT INTO eval_candidates (
|
||
tool_name, query, retrieved_slugs, retrieved_chunk_ids, source_ids,
|
||
expand_enabled, detail, detail_resolved, vector_enabled, expansion_applied,
|
||
latency_ms, remote, job_id, subagent_id
|
||
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14)
|
||
RETURNING id`,
|
||
[
|
||
input.tool_name,
|
||
input.query,
|
||
input.retrieved_slugs,
|
||
input.retrieved_chunk_ids,
|
||
input.source_ids,
|
||
input.expand_enabled,
|
||
input.detail,
|
||
input.detail_resolved,
|
||
input.vector_enabled,
|
||
input.expansion_applied,
|
||
input.latency_ms,
|
||
input.remote,
|
||
input.job_id,
|
||
input.subagent_id,
|
||
]
|
||
);
|
||
return rows[0]!.id;
|
||
}
|
||
|
||
async listEvalCandidates(filter?: { since?: Date; limit?: number; tool?: 'query' | 'search' }): Promise<EvalCandidate[]> {
|
||
const raw = filter?.limit;
|
||
const limit = (raw === undefined || raw === null || !Number.isFinite(raw) || raw <= 0)
|
||
? 1000
|
||
: Math.min(Math.floor(raw), 100000);
|
||
const since = filter?.since ?? new Date(0);
|
||
const tool = filter?.tool ?? null;
|
||
// id DESC tiebreaker — see postgres-engine for rationale.
|
||
const { rows } = tool
|
||
? await this.db.query(
|
||
`SELECT * FROM eval_candidates
|
||
WHERE created_at >= $1 AND tool_name = $2
|
||
ORDER BY created_at DESC, id DESC LIMIT $3`,
|
||
[since, tool, limit]
|
||
)
|
||
: await this.db.query(
|
||
`SELECT * FROM eval_candidates
|
||
WHERE created_at >= $1
|
||
ORDER BY created_at DESC, id DESC LIMIT $2`,
|
||
[since, limit]
|
||
);
|
||
return rows as unknown as EvalCandidate[];
|
||
}
|
||
|
||
async deleteEvalCandidatesBefore(date: Date): Promise<number> {
|
||
const { rows } = await this.db.query(
|
||
`DELETE FROM eval_candidates WHERE created_at < $1 RETURNING id`,
|
||
[date]
|
||
);
|
||
return rows.length;
|
||
}
|
||
|
||
async logEvalCaptureFailure(reason: EvalCaptureFailureReason): Promise<void> {
|
||
await this.db.query(
|
||
`INSERT INTO eval_capture_failures (reason) VALUES ($1)`,
|
||
[reason]
|
||
);
|
||
}
|
||
|
||
async listEvalCaptureFailures(filter?: { since?: Date }): Promise<EvalCaptureFailure[]> {
|
||
const since = filter?.since ?? new Date(0);
|
||
const { rows } = await this.db.query(
|
||
`SELECT * FROM eval_capture_failures WHERE ts >= $1 ORDER BY ts DESC`,
|
||
[since]
|
||
);
|
||
return rows as unknown as EvalCaptureFailure[];
|
||
}
|
||
|
||
// ============================================================
|
||
// v0.29 — Salience + Anomaly Detection
|
||
// ============================================================
|
||
|
||
async batchLoadEmotionalInputs(slugs?: string[]): Promise<EmotionalWeightInputRow[]> {
|
||
// Two CTEs avoid the N×M cartesian product (codex C4#4).
|
||
const baseSql = `
|
||
WITH page_tags AS (
|
||
SELECT page_id, array_agg(DISTINCT tag) AS tags
|
||
FROM tags GROUP BY page_id
|
||
),
|
||
page_takes AS (
|
||
SELECT page_id, json_agg(json_build_object(
|
||
'holder', holder, 'weight', weight, 'kind', kind, 'active', active
|
||
)) AS takes
|
||
FROM takes WHERE active = TRUE GROUP BY page_id
|
||
)
|
||
SELECT p.slug, p.source_id,
|
||
COALESCE(pt.tags, ARRAY[]::text[]) AS tags,
|
||
COALESCE(pk.takes, '[]'::json) AS takes
|
||
FROM pages p
|
||
LEFT JOIN page_tags pt ON pt.page_id = p.id
|
||
LEFT JOIN page_takes pk ON pk.page_id = p.id
|
||
`;
|
||
const { rows } = slugs
|
||
? await this.db.query(`${baseSql} WHERE p.slug = ANY($1::text[])`, [slugs])
|
||
: await this.db.query(baseSql);
|
||
return (rows as Record<string, unknown>[]).map(r => ({
|
||
slug: String(r.slug),
|
||
source_id: String(r.source_id),
|
||
tags: (r.tags as string[]) ?? [],
|
||
takes: (r.takes as EmotionalWeightInputRow['takes']) ?? [],
|
||
}));
|
||
}
|
||
|
||
async setEmotionalWeightBatch(rows: EmotionalWeightWriteRow[]): Promise<number> {
|
||
if (rows.length === 0) return 0;
|
||
const slugs = rows.map(r => r.slug);
|
||
const sourceIds = rows.map(r => r.source_id);
|
||
const weights = rows.map(r => r.weight);
|
||
// Composite-keyed UPDATE FROM unnest (codex C4#3).
|
||
// v0.29.1: bump salience_touched_at when emotional_weight actually changes
|
||
// so the salience query window picks up newly-salient old pages. Mirror
|
||
// of postgres-engine.ts.
|
||
const result = await this.db.query(
|
||
`UPDATE pages
|
||
SET emotional_weight = u.weight,
|
||
salience_touched_at = CASE
|
||
WHEN pages.emotional_weight IS DISTINCT FROM u.weight THEN now()
|
||
ELSE pages.salience_touched_at
|
||
END
|
||
FROM unnest($1::text[], $2::text[], $3::real[])
|
||
AS u(slug, source_id, weight)
|
||
WHERE pages.slug = u.slug AND pages.source_id = u.source_id
|
||
RETURNING 1`,
|
||
[slugs, sourceIds, weights]
|
||
);
|
||
return result.rows.length;
|
||
}
|
||
|
||
async getRecentSalience(opts: SalienceOpts): Promise<SalienceResult[]> {
|
||
const days = Math.max(0, opts.days ?? 14);
|
||
const limit = clampSearchLimit(opts.limit, 20, 100);
|
||
const slugPrefix = opts.slugPrefix;
|
||
const boundaryIso = new Date(Date.now() - days * 86400000).toISOString();
|
||
|
||
const params: unknown[] = [boundaryIso];
|
||
let prefixCondition = '';
|
||
if (slugPrefix) {
|
||
const escaped = slugPrefix.replace(/[\\%_]/g, (c) => '\\' + c) + '%';
|
||
params.push(escaped);
|
||
prefixCondition = `AND p.slug LIKE $${params.length} ESCAPE '\\'`;
|
||
}
|
||
params.push(limit);
|
||
const limitParam = `$${params.length}`;
|
||
|
||
// v0.29.1: third score term via buildRecencyComponentSql. Default
|
||
// 'flat' = v0.29.0 behavior. 'on' opts into per-prefix decay.
|
||
const recencyBias = opts.recency_bias ?? 'flat';
|
||
let recencySql: string;
|
||
if (recencyBias === 'on') {
|
||
const { resolveRecencyDecayMap, DEFAULT_FALLBACK } = await import('./search/recency-decay.ts');
|
||
recencySql = buildRecencyComponentSql({
|
||
slugColumn: 'p.slug',
|
||
dateExpr: 'COALESCE(p.effective_date, p.updated_at)',
|
||
decayMap: resolveRecencyDecayMap(),
|
||
fallback: DEFAULT_FALLBACK,
|
||
});
|
||
} else {
|
||
recencySql = buildRecencyComponentSql({
|
||
slugColumn: 'p.slug',
|
||
dateExpr: 'p.updated_at',
|
||
decayMap: {},
|
||
fallback: { halflifeDays: 1, coefficient: 1.0 },
|
||
});
|
||
}
|
||
const { rows } = await this.db.query(
|
||
`SELECT p.slug, p.source_id, p.title, p.type, p.updated_at, p.emotional_weight,
|
||
COUNT(DISTINCT t.id) AS take_count,
|
||
COALESCE(AVG(t.weight), 0) AS take_avg_weight,
|
||
(p.emotional_weight * 5)
|
||
+ ln(1 + COUNT(DISTINCT t.id))
|
||
+ ${recencySql}
|
||
AS score
|
||
FROM pages p
|
||
LEFT JOIN takes t ON t.page_id = p.id AND t.active = TRUE
|
||
WHERE GREATEST(p.updated_at, COALESCE(p.salience_touched_at, p.updated_at)) >= $1::timestamptz
|
||
${prefixCondition}
|
||
GROUP BY p.id
|
||
ORDER BY score DESC
|
||
LIMIT ${limitParam}`,
|
||
params
|
||
);
|
||
return (rows as Record<string, unknown>[]).map(r => ({
|
||
slug: String(r.slug),
|
||
source_id: String(r.source_id),
|
||
title: String(r.title ?? ''),
|
||
type: r.type as SalienceResult['type'],
|
||
updated_at: r.updated_at as Date,
|
||
emotional_weight: Number(r.emotional_weight ?? 0),
|
||
take_count: Number(r.take_count ?? 0),
|
||
take_avg_weight: Number(r.take_avg_weight ?? 0),
|
||
score: Number(r.score ?? 0),
|
||
}));
|
||
}
|
||
|
||
async findAnomalies(opts: AnomaliesOpts): Promise<AnomalyResult[]> {
|
||
const sigma = opts.sigma ?? 3.0;
|
||
const lookbackDays = Math.max(1, opts.lookback_days ?? 30);
|
||
const sinceIso = (opts.since ?? new Date().toISOString().slice(0, 10));
|
||
const sinceDate = new Date(sinceIso + 'T00:00:00Z');
|
||
const sinceEnd = new Date(sinceDate.getTime() + 86400000);
|
||
const baselineStart = new Date(sinceDate.getTime() - lookbackDays * 86400000);
|
||
|
||
const tagBaselineRes = await this.db.query(
|
||
`WITH days AS (
|
||
SELECT day::date FROM generate_series(
|
||
$1::date, $2::date - 1, '1 day'::interval
|
||
) AS day
|
||
),
|
||
cohort_keys AS (
|
||
SELECT DISTINCT t.tag FROM tags t JOIN pages p ON p.id = t.page_id
|
||
WHERE p.updated_at >= $1::timestamptz AND p.updated_at < $2::timestamptz
|
||
),
|
||
touched AS (
|
||
SELECT t.tag,
|
||
date_trunc('day', p.updated_at)::date AS day,
|
||
COUNT(DISTINCT p.id) AS cnt
|
||
FROM tags t JOIN pages p ON p.id = t.page_id
|
||
WHERE p.updated_at >= $1::timestamptz AND p.updated_at < $2::timestamptz
|
||
GROUP BY 1, 2
|
||
)
|
||
SELECT cd.tag AS cohort_value, d.day::text AS day, COALESCE(t.cnt, 0)::int AS count
|
||
FROM cohort_keys cd CROSS JOIN days d
|
||
LEFT JOIN touched t ON t.tag = cd.tag AND t.day = d.day`,
|
||
[baselineStart.toISOString(), sinceDate.toISOString()]
|
||
);
|
||
|
||
const typeBaselineRes = await this.db.query(
|
||
`WITH days AS (
|
||
SELECT day::date FROM generate_series(
|
||
$1::date, $2::date - 1, '1 day'::interval
|
||
) AS day
|
||
),
|
||
cohort_keys AS (
|
||
SELECT DISTINCT p.type FROM pages p
|
||
WHERE p.updated_at >= $1::timestamptz AND p.updated_at < $2::timestamptz
|
||
),
|
||
touched AS (
|
||
SELECT p.type,
|
||
date_trunc('day', p.updated_at)::date AS day,
|
||
COUNT(DISTINCT p.id) AS cnt
|
||
FROM pages p
|
||
WHERE p.updated_at >= $1::timestamptz AND p.updated_at < $2::timestamptz
|
||
GROUP BY 1, 2
|
||
)
|
||
SELECT cd.type AS cohort_value, d.day::text AS day, COALESCE(t.cnt, 0)::int AS count
|
||
FROM cohort_keys cd CROSS JOIN days d
|
||
LEFT JOIN touched t ON t.type = cd.type AND t.day = d.day`,
|
||
[baselineStart.toISOString(), sinceDate.toISOString()]
|
||
);
|
||
|
||
const tagTodayRes = await this.db.query(
|
||
`SELECT t.tag AS cohort_value,
|
||
COUNT(DISTINCT p.id)::int AS count,
|
||
array_agg(DISTINCT p.slug) AS slugs
|
||
FROM tags t JOIN pages p ON p.id = t.page_id
|
||
WHERE p.updated_at >= $1::timestamptz AND p.updated_at < $2::timestamptz
|
||
GROUP BY 1`,
|
||
[sinceIso, sinceEnd.toISOString()]
|
||
);
|
||
|
||
const typeTodayRes = await this.db.query(
|
||
`SELECT p.type AS cohort_value,
|
||
COUNT(DISTINCT p.id)::int AS count,
|
||
array_agg(DISTINCT p.slug) AS slugs
|
||
FROM pages p
|
||
WHERE p.updated_at >= $1::timestamptz AND p.updated_at < $2::timestamptz
|
||
GROUP BY 1`,
|
||
[sinceIso, sinceEnd.toISOString()]
|
||
);
|
||
|
||
const baseline = [
|
||
...(tagBaselineRes.rows as Record<string, unknown>[]).map(r => ({
|
||
cohort_kind: 'tag' as const,
|
||
cohort_value: String(r.cohort_value),
|
||
day: String(r.day),
|
||
count: Number(r.count),
|
||
})),
|
||
...(typeBaselineRes.rows as Record<string, unknown>[]).map(r => ({
|
||
cohort_kind: 'type' as const,
|
||
cohort_value: String(r.cohort_value),
|
||
day: String(r.day),
|
||
count: Number(r.count),
|
||
})),
|
||
];
|
||
const today = [
|
||
...(tagTodayRes.rows as Record<string, unknown>[]).map(r => ({
|
||
cohort_kind: 'tag' as const,
|
||
cohort_value: String(r.cohort_value),
|
||
count: Number(r.count),
|
||
page_slugs: (r.slugs as string[]) ?? [],
|
||
})),
|
||
...(typeTodayRes.rows as Record<string, unknown>[]).map(r => ({
|
||
cohort_kind: 'type' as const,
|
||
cohort_value: String(r.cohort_value),
|
||
count: Number(r.count),
|
||
page_slugs: (r.slugs as string[]) ?? [],
|
||
})),
|
||
];
|
||
|
||
return computeAnomaliesFromBuckets(baseline, today, sigma);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* Raw row shape returned from `SELECT * FROM facts`. The `embedding`
|
||
* column comes back as a string (`[0.1,0.2,...]`) on PGLite when
|
||
* postgres-style types aren't auto-decoded; we parse on the way out.
|
||
*/
|
||
interface FactRowSqlShape {
|
||
id: number;
|
||
source_id: string;
|
||
entity_slug: string | null;
|
||
fact: string;
|
||
kind: FactKind;
|
||
visibility: FactVisibility;
|
||
notability: 'high' | 'medium' | 'low';
|
||
context: string | null;
|
||
valid_from: Date | string;
|
||
valid_until: Date | string | null;
|
||
expired_at: Date | string | null;
|
||
superseded_by: number | null;
|
||
consolidated_at: Date | string | null;
|
||
consolidated_into: number | null;
|
||
source: string;
|
||
source_session: string | null;
|
||
confidence: number;
|
||
embedding: string | number[] | Float32Array | null;
|
||
embedded_at: Date | string | null;
|
||
created_at: Date | string;
|
||
}
|
||
|
||
function toDate(v: Date | string | null): Date | null {
|
||
if (v == null) return null;
|
||
if (v instanceof Date) return v;
|
||
return new Date(v);
|
||
}
|
||
|
||
function rowToFact(row: FactRowSqlShape): FactRow {
|
||
let embedding: Float32Array | null = null;
|
||
if (row.embedding != null) {
|
||
if (row.embedding instanceof Float32Array) embedding = row.embedding;
|
||
else if (Array.isArray(row.embedding)) embedding = new Float32Array(row.embedding);
|
||
else if (typeof row.embedding === 'string') {
|
||
// pgvector text format: "[0.1,0.2,...]"
|
||
const trimmed = row.embedding.trim();
|
||
const inner = trimmed.startsWith('[') ? trimmed.slice(1, -1) : trimmed;
|
||
const parts = inner.split(',').map(p => parseFloat(p.trim())).filter(Number.isFinite);
|
||
embedding = parts.length > 0 ? new Float32Array(parts) : null;
|
||
}
|
||
}
|
||
return {
|
||
id: Number(row.id),
|
||
source_id: row.source_id,
|
||
entity_slug: row.entity_slug,
|
||
fact: row.fact,
|
||
kind: row.kind,
|
||
visibility: row.visibility,
|
||
// v0.31.2: notability column added by migration v46. Same fallback
|
||
// as Postgres (belt-and-suspenders with the NOT NULL DEFAULT).
|
||
notability: row.notability ?? 'medium',
|
||
context: row.context,
|
||
valid_from: toDate(row.valid_from)!,
|
||
valid_until: toDate(row.valid_until),
|
||
expired_at: toDate(row.expired_at),
|
||
superseded_by: row.superseded_by == null ? null : Number(row.superseded_by),
|
||
consolidated_at: toDate(row.consolidated_at),
|
||
consolidated_into: row.consolidated_into == null ? null : Number(row.consolidated_into),
|
||
source: row.source,
|
||
source_session: row.source_session,
|
||
confidence: Number(row.confidence),
|
||
embedding,
|
||
embedded_at: toDate(row.embedded_at),
|
||
created_at: toDate(row.created_at)!,
|
||
};
|
||
}
|
||
|
||
/**
|
||
* Encode a Float32Array as the pgvector text-form literal `[0.1,0.2,...]`.
|
||
* Both PGLite and Postgres accept this when the parameter is cast to ::vector.
|
||
*/
|
||
function toPgVectorLiteral(v: Float32Array | number[]): string {
|
||
if (v instanceof Float32Array) return '[' + Array.from(v).join(',') + ']';
|
||
return '[' + v.join(',') + ']';
|
||
}
|
||
|
||
function rowToCodeEdge(row: Record<string, unknown>): import('./types.ts').CodeEdgeResult {
|
||
return {
|
||
id: row.id as number,
|
||
from_chunk_id: row.from_chunk_id as number,
|
||
to_chunk_id: row.to_chunk_id == null ? null : (row.to_chunk_id as number),
|
||
from_symbol_qualified: (row.from_symbol_qualified as string) ?? '',
|
||
to_symbol_qualified: (row.to_symbol_qualified as string) ?? '',
|
||
edge_type: (row.edge_type as string) ?? '',
|
||
edge_metadata: (row.edge_metadata as Record<string, unknown>) ?? {},
|
||
source_id: row.source_id == null ? null : (row.source_id as string),
|
||
resolved: Boolean(row.resolved),
|
||
};
|
||
}
|