feat: expose space fact listing
All checks were successful
CI / verify (push) Successful in 14s

This commit is contained in:
2026-05-14 22:36:04 +09:00
parent d056bb61a5
commit 49f75afcf4
3 changed files with 76 additions and 0 deletions

View File

@@ -47,6 +47,10 @@ export class InMemoryMemoryStore implements BoxBrainMemoryStore {
return stored;
}
async listFacts(spaceId: string): Promise<StoredFact[]> {
return [...(this.facts.get(spaceId) ?? [])].sort((a, b) => a.createdAt.localeCompare(b.createdAt));
}
async findFacts(spaceId: string, topics: string[]): Promise<StoredFact[]> {
const facts = this.facts.get(spaceId) ?? [];
if (topics.length === 0) return [...facts];
@@ -120,6 +124,17 @@ export class IdentityDbMemoryStore implements BoxBrainMemoryStore {
return this.fromIdentityFact(stored);
}
async listFacts(spaceId: string): Promise<StoredFact[]> {
const topics = await this.options.db.listTopics({ spaceName: spaceId, includeFacts: true });
const collected = new Map<string, StoredFact>();
for (const topic of topics) {
for (const fact of topic.facts) {
collected.set(fact.id, this.fromIdentityFact(fact));
}
}
return [...collected.values()].sort((a, b) => a.createdAt.localeCompare(b.createdAt));
}
async findFacts(spaceId: string, topics: string[]): Promise<StoredFact[]> {
const uniqueTopics = normalizeTopics(topics);
const collected = new Map<string, StoredFact>();

View File

@@ -167,6 +167,7 @@ export interface BoxBrainMemoryStore {
createSpace(input: { displayName: string; seedMessage: string; now: string }): Promise<MemorySpace>;
getSpace(spaceId: string): Promise<MemorySpace | null>;
addFact(spaceId: string, fact: FactDraft): Promise<StoredFact>;
listFacts(spaceId: string): Promise<StoredFact[]>;
findFacts(spaceId: string, topics: string[]): Promise<StoredFact[]>;
saveScheduleEntries(spaceId: string, entries: ScheduleEntry[]): Promise<void>;
listScheduleEntries(spaceId: string, fromInclusive: string, toExclusive: string): Promise<ScheduleEntry[]>;

60
tests/memory.test.ts Normal file
View File

@@ -0,0 +1,60 @@
import { mkdtemp, rm } from 'node:fs/promises';
import { join } from 'node:path';
import { tmpdir } from 'node:os';
import { describe, expect, it } from 'vitest';
import { createSqliteIdentityMemoryStore, InMemoryMemoryStore } from '../src';
async function seedFacts(memory: InMemoryMemoryStore, spaceId: string) {
await memory.addFact(spaceId, {
statement: 'Mina likes quiet cafes.',
topics: ['persona', 'Mina'],
source: 'test',
});
await memory.addFact(spaceId, {
statement: 'The user prefers short replies.',
topics: ['user'],
source: 'test',
});
}
describe('BoxBrain memory stores', () => {
it('lists every fact in an in-memory persona space without requiring topic guesses', async () => {
const memory = new InMemoryMemoryStore();
const space = await memory.createSpace({ displayName: 'Mina', seedMessage: 'seed', now: '2026-05-14T00:00:00.000Z' });
await seedFacts(memory, space.id);
const facts = await memory.listFacts(space.id);
expect(facts.map((fact) => fact.statement)).toEqual([
'Mina likes quiet cafes.',
'The user prefers short replies.',
]);
});
it('lists every fact in a SQLite IdentityDB-backed persona space', async () => {
const dir = await mkdtemp(join(tmpdir(), 'boxbrain-memory-'));
try {
const memory = await createSqliteIdentityMemoryStore(join(dir, 'boxbrain.sqlite'));
const space = await memory.createSpace({ displayName: 'Mina', seedMessage: 'seed', now: '2026-05-14T00:00:00.000Z' });
await memory.addFact(space.id, {
statement: 'Mina likes quiet cafes.',
topics: ['persona', 'Mina'],
source: 'test',
});
await memory.addFact(space.id, {
statement: 'The user prefers short replies.',
topics: ['user'],
source: 'test',
});
const facts = await memory.listFacts(space.id);
expect(facts.map((fact) => fact.statement)).toEqual([
'Mina likes quiet cafes.',
'The user prefers short replies.',
]);
} finally {
await rm(dir, { recursive: true, force: true });
}
});
});