Compare commits
3 Commits
05f077b798
...
b52d37170c
| Author | SHA1 | Date | |
|---|---|---|---|
| b52d37170c | |||
| 6f4f65a8ee | |||
| 488ba20eb6 |
4
bun.lock
4
bun.lock
@@ -5,7 +5,7 @@
|
|||||||
"": {
|
"": {
|
||||||
"name": "boxbrain",
|
"name": "boxbrain",
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"identitydb": "^0.4.0",
|
"identitydb": "^0.5.0",
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@types/bun": "latest",
|
"@types/bun": "latest",
|
||||||
@@ -249,7 +249,7 @@
|
|||||||
|
|
||||||
"iconv-lite": ["iconv-lite@0.7.2", "", { "dependencies": { "safer-buffer": ">= 2.1.2 < 3.0.0" } }, "sha512-im9DjEDQ55s9fL4EYzOAv0yMqmMBSZp6G0VvFyTMPKWxiSBHUj9NW/qqLmXUwXrrM7AvqSlTCfvqRb0cM8yYqw=="],
|
"iconv-lite": ["iconv-lite@0.7.2", "", { "dependencies": { "safer-buffer": ">= 2.1.2 < 3.0.0" } }, "sha512-im9DjEDQ55s9fL4EYzOAv0yMqmMBSZp6G0VvFyTMPKWxiSBHUj9NW/qqLmXUwXrrM7AvqSlTCfvqRb0cM8yYqw=="],
|
||||||
|
|
||||||
"identitydb": ["identitydb@0.4.0", "", { "dependencies": { "kysely": "^0.28.8", "mysql2": "^3.15.3", "pg": "^8.16.0" } }, "sha512-DAyipdrApjmI1HoHfhT9zuMfNLiWaYe7/k/FrUa55h7WUUASGV192AhrC6KUiMTS55dfYKDBWi0AcS7crSw+bA=="],
|
"identitydb": ["identitydb@0.5.0", "", { "dependencies": { "kysely": "^0.28.8", "mysql2": "^3.15.3", "pg": "^8.16.0" } }, "sha512-3cp14fb5nDKFakRqHdJrOBOgpDWLlvJ/K2q8405muAfqcJja9ds2tS9ksoaEiMgGU5r+mIz6lADZ3DYNpGqHVQ=="],
|
||||||
|
|
||||||
"is-property": ["is-property@1.0.2", "", {}, "sha512-Ks/IoX00TtClbGQr4TWXemAnktAQvYB7HzcCxDGqEZU6oCmb2INHuOoKxbtR+HFkmYWBKv/dOZtGRiAjDhj92g=="],
|
"is-property": ["is-property@1.0.2", "", {}, "sha512-Ks/IoX00TtClbGQr4TWXemAnktAQvYB7HzcCxDGqEZU6oCmb2INHuOoKxbtR+HFkmYWBKv/dOZtGRiAjDhj92g=="],
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "boxbrain",
|
"name": "boxbrain",
|
||||||
"version": "0.4.0",
|
"version": "0.5.0",
|
||||||
"description": "Human-like persona harness framework powered by LLMs and IdentityDB.",
|
"description": "Human-like persona harness framework powered by LLMs and IdentityDB.",
|
||||||
"license": "MIT",
|
"license": "MIT",
|
||||||
"type": "module",
|
"type": "module",
|
||||||
@@ -48,7 +48,7 @@
|
|||||||
"prepublishOnly": "bun run check && bun run test && bun run build && bun run pack:check"
|
"prepublishOnly": "bun run check && bun run test && bun run build && bun run pack:check"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"identitydb": "^0.4.0"
|
"identitydb": "^0.5.0"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@types/bun": "latest",
|
"@types/bun": "latest",
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ import {
|
|||||||
startOfUtcDay,
|
startOfUtcDay,
|
||||||
toIso,
|
toIso,
|
||||||
} from "./schedule";
|
} from "./schedule";
|
||||||
import { extractFact } from "identitydb";
|
import { ExtractedFact, extractFacts } from "identitydb";
|
||||||
import {
|
import {
|
||||||
buildMandatoryConversationContext,
|
buildMandatoryConversationContext,
|
||||||
conversationInstruction,
|
conversationInstruction,
|
||||||
@@ -331,6 +331,20 @@ export class Persona {
|
|||||||
return draft;
|
return draft;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private extractedToDraft(fact: ExtractedFact, statement: string): FactDraft {
|
||||||
|
return {
|
||||||
|
statement: fact.statement ?? statement,
|
||||||
|
topics: [...fact.topics.map((t) => t.name), "sleepMemory"],
|
||||||
|
source: fact.source ?? "boxbrain.sleepMemory",
|
||||||
|
...(typeof fact.confidence === "number"
|
||||||
|
? { confidence: fact.confidence }
|
||||||
|
: {}),
|
||||||
|
...(fact.metadata !== undefined && fact.metadata !== null
|
||||||
|
? { metadata: fact.metadata as Record<string, unknown> }
|
||||||
|
: {}),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
async sleepMemory(input: {
|
async sleepMemory(input: {
|
||||||
datetime: DateTimeInput;
|
datetime: DateTimeInput;
|
||||||
messageHistory: PersonaMessage[];
|
messageHistory: PersonaMessage[];
|
||||||
@@ -360,26 +374,17 @@ export class Persona {
|
|||||||
messages: input.messageHistory,
|
messages: input.messageHistory,
|
||||||
}),
|
}),
|
||||||
].join("\n");
|
].join("\n");
|
||||||
const extracted = await extractFact(
|
const extractedFacts = (
|
||||||
statement,
|
await extractFacts(statement, this.options.models.factExtractor)
|
||||||
this.options.models.factExtractor,
|
).map((fact) => this.extractedToDraft(fact, statement));
|
||||||
);
|
|
||||||
const draft: FactDraft = {
|
for (const fact of extractedFacts) {
|
||||||
statement: extracted.statement ?? statement,
|
await this.memory.addFact(persona.id, fact);
|
||||||
topics: [...extracted.topics.map((t) => t.name), "sleepMemory"],
|
}
|
||||||
source: extracted.source ?? "boxbrain.sleepMemory",
|
|
||||||
...(typeof extracted.confidence === 'number'
|
|
||||||
? { confidence: extracted.confidence }
|
|
||||||
: {}),
|
|
||||||
...(extracted.metadata !== undefined && extracted.metadata !== null
|
|
||||||
? { metadata: extracted.metadata as Record<string, unknown> }
|
|
||||||
: {}),
|
|
||||||
};
|
|
||||||
await this.memory.addFact(persona.id, draft);
|
|
||||||
await this.emit("persona.memory.sleep.persisted", {
|
await this.emit("persona.memory.sleep.persisted", {
|
||||||
factCount: 1,
|
factCount: extractedFacts.length,
|
||||||
});
|
});
|
||||||
return [draft];
|
return extractedFacts;
|
||||||
}
|
}
|
||||||
|
|
||||||
private async initialize(): Promise<MemorySpace> {
|
private async initialize(): Promise<MemorySpace> {
|
||||||
@@ -400,29 +405,24 @@ export class Persona {
|
|||||||
});
|
});
|
||||||
if (this.options.models?.factExtractor) {
|
if (this.options.models?.factExtractor) {
|
||||||
const statement = `Persona: ${this.mode.displayName}\nSeed: ${this.mode.seedMessage}`;
|
const statement = `Persona: ${this.mode.displayName}\nSeed: ${this.mode.seedMessage}`;
|
||||||
const extracted = await extractFact(
|
const extracteds = (
|
||||||
statement,
|
await extractFacts(statement, this.options.models.factExtractor)
|
||||||
this.options.models.factExtractor,
|
).map((fact) => this.extractedToDraft(fact, statement));
|
||||||
);
|
|
||||||
const draft: FactDraft = {
|
for (const fact of extracteds) {
|
||||||
statement: extracted.statement ?? statement,
|
await this.memory.addFact(space.id, fact);
|
||||||
topics: extracted.topics.map((t) => t.name),
|
}
|
||||||
source: extracted.source ?? "boxbrain.persona.initialization",
|
|
||||||
...(typeof extracted.confidence === 'number'
|
|
||||||
? { confidence: extracted.confidence }
|
|
||||||
: {}),
|
|
||||||
...(extracted.metadata !== undefined && extracted.metadata !== null
|
|
||||||
? { metadata: extracted.metadata as Record<string, unknown> }
|
|
||||||
: {}),
|
|
||||||
};
|
|
||||||
await this.memory.addFact(space.id, draft);
|
|
||||||
await this.emit(
|
await this.emit(
|
||||||
"persona.initialized",
|
"persona.initialized",
|
||||||
{ displayName: space.displayName, factCount: 1 },
|
{ displayName: space.displayName, factCount: extracteds.length },
|
||||||
space.id,
|
space.id,
|
||||||
);
|
);
|
||||||
} else {
|
} else {
|
||||||
const fact = defaultInitialFact(this.mode.displayName, this.mode.seedMessage);
|
const fact = defaultInitialFact(
|
||||||
|
this.mode.displayName,
|
||||||
|
this.mode.seedMessage,
|
||||||
|
);
|
||||||
await this.memory.addFact(space.id, fact);
|
await this.memory.addFact(space.id, fact);
|
||||||
await this.emit(
|
await this.emit(
|
||||||
"persona.initialized",
|
"persona.initialized",
|
||||||
|
|||||||
@@ -12,11 +12,11 @@ describe('Persona initialization', () => {
|
|||||||
models: {
|
models: {
|
||||||
factExtractor: {
|
factExtractor: {
|
||||||
async extract(input) {
|
async extract(input) {
|
||||||
return {
|
return [{
|
||||||
statement: 'Mina likes quiet cafes.',
|
statement: 'Mina likes quiet cafes.',
|
||||||
topics: [{ name: 'persona' }, { name: 'Mina' }],
|
topics: [{ name: 'persona' }, { name: 'Mina' }],
|
||||||
source: 'test',
|
source: 'test',
|
||||||
};
|
}];
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -11,15 +11,15 @@ describe('sleepMemory', () => {
|
|||||||
factExtractor: {
|
factExtractor: {
|
||||||
async extract(input) {
|
async extract(input) {
|
||||||
if (input.includes('Seed:')) {
|
if (input.includes('Seed:')) {
|
||||||
return { statement: 'Mina remembers stable details.', topics: [{ name: 'persona' }, { name: 'Mina' }] };
|
return [{ statement: 'Mina remembers stable details.', topics: [{ name: 'persona' }, { name: 'Mina' }] }];
|
||||||
}
|
}
|
||||||
expect(input).toContain('user@2026-05-01T15:00:00.000Z: 나는 타입스크립트를 2025년부터 시작했어');
|
expect(input).toContain('user@2026-05-01T15:00:00.000Z: 나는 타입스크립트를 2025년부터 시작했어');
|
||||||
expect(input).toContain('Objectivize');
|
expect(input).toContain('Objectivize');
|
||||||
return {
|
return [{
|
||||||
statement: 'The user started TypeScript in 2025.',
|
statement: 'The user started TypeScript in 2025.',
|
||||||
topics: [{ name: 'user' }, { name: 'TypeScript' }, { name: '2025' }],
|
topics: [{ name: 'user' }, { name: 'TypeScript' }, { name: '2025' }],
|
||||||
confidence: 0.9,
|
confidence: 0.9,
|
||||||
};
|
}];
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
|||||||
Reference in New Issue
Block a user