feat: implemented supporters

src/brain/factExtractor.ts

@@ -0,0 +1,16 @@
+import { llm } from "@/openrouter";
+import { extractedFactSchema } from "@/openrouter/schema";
+import { type ExtractedFact, LlmFactExtractor } from "identitydb";
+
+export const factExtractor = new LlmFactExtractor({
+  model: {
+    async generateText({ instruction, input }) {
+      return await llm.call<ExtractedFact[]>(llm.models.identity, {
+        instruction,
+        message: input,
+        jsonSchemaName: "fact-extractor",
+        jsonSchema: extractedFactSchema,
+      });
+    },
+  },
+});

src/brain/index.ts

@@ -0,0 +1,32 @@
+import { config } from "@/config";
+import { IdentityDB, type Space } from "identitydb";
+import { llm } from "@/openrouter";
+import { brainManager, type BrainItem } from "./manager";
+
+export class Brain {
+  constructor(
+    public db: IdentityDB,
+    public space: Space,
+    public brainbase: BrainItem,
+  ) {}
+
+  static async create(
+    displayName: string,
+    seed: string,
+  ): Promise<Brain | null> {}
+
+  static async load(brainId: string): Promise<Brain | null> {
+    const brain = await brainManager.loadBrain(brainId);
+    if (!brain) return null;
+
+    const db = await IdentityDB.connect({
+      client: "sqlite",
+      filename: config.dbPath,
+    });
+
+    const space = await db.getSpaceByName(brain.spaceName);
+    if (!space) return null;
+
+    return new Brain(db, space, brain);
+  }
+}

src/brain/manager.ts

@@ -0,0 +1,55 @@
+import { config } from "@/config";
+import { readFile, writeFile } from "fs/promises";
+
+export interface BrainItem {
+  brainId: string;
+  spaceName: string;
+  displayName: string;
+  baseSystemPrompt: string;
+}
+export type BrainDB = Record<string, BrainItem>;
+
+export class BrainDBManager {
+  private get db() {
+    return readFile(config.braindbPath, { encoding: "utf-8" }).then(
+      (content) => {
+        return JSON.parse(content) as BrainDB;
+      },
+    );
+  }
+
+  private async writeDb(db: BrainDB) {
+    await writeFile(config.braindbPath, JSON.stringify(db), {
+      encoding: "utf-8",
+    });
+  }
+
+  async loadBrain(brainId: string): Promise<BrainItem | undefined> {
+    const brainOrNot = (await this.db)[brainId];
+    return brainOrNot;
+  }
+
+  async saveBrain(brainId: string, brain: BrainItem) {
+    const db = await this.db;
+    db[brainId] = brain;
+    await this.writeDb(db);
+  }
+
+  async listBrain() {
+    return Object.entries(await this.db).map(
+      ([_, { brainId, displayName }]) => ({ brainId, displayName }),
+    );
+  }
+
+  async deleteBrain(brainId: string) {
+    const db = await this.db;
+    delete db[brainId];
+    await this.writeDb(db);
+  }
+
+  async isBrainAvailable(brainId: string) {
+    return brainId in (await this.db);
+  }
+}
+
+export const brainManager = new BrainDBManager();