@dsh-plugin/dsh-synapse 0.0.0-stage → 0.1.0-dev.41044228975

package/lib/host/index.js

@@ -0,0 +1,192 @@
+import { mkdir, readFile } from "node:fs/promises";
+import path from "node:path";
+import { template, validateConfig } from "./config.js";
+import { pathsFor, SkillStore, safePath } from "./files.js";
+import { Inference } from "./runtime.js";
+import { Memory, parseHermes } from "./memory.js";
+import { McpManager } from "./mcp.js";
+import { Agent } from "./agent.js";
+import { Gateway } from "./gateway.js";
+import { Bot } from "./bot.js";
+import { Reviews } from "./github.js";
+import { json, body, browserAuthorized } from "./http.js";
+export const name = "@dsh-plugin/dsh-synapse";
+export const inject = ["dshLoader"];
+export { template };
+export async function createService(ctx, c, onError) {
+    validateConfig(c);
+    const paths = pathsFor(ctx, c);
+    await safePath(paths.profile, "data", "dsh-synapse");
+    await mkdir(paths.data, { recursive: true });
+    const config = () => c, inference = new Inference(paths.data), skills = new SkillStore(paths);
+    const memory = new Memory(path.join(paths.data, "memory.sqlite"), inference, config), mcp = new McpManager(config);
+    const agent = new Agent({ config, paths, memory, skills, mcp, onError }), gateway = new Gateway(agent, config, onError);
+    const bot = new Bot(paths, config, gateway, onError), reviews = new Reviews(memory.db, agent, config, bot, onError);
+    const close = async () => {
+        await Promise.allSettled([
+            reviews.close(),
+            bot.stop(),
+            gateway.close(),
+            mcp.close(),
+            inference.close(),
+        ]);
+        memory.close();
+    };
+    try {
+        await gateway.start();
+        await bot.start();
+        void reviews.drain();
+        if (c.enabled && c.ragEnabled)
+            void inference.embed("初始化记忆模型").catch(onError);
+    }
+    catch (e) {
+        await close();
+        throw e;
+    }
+    return {
+        paths,
+        config,
+        inference,
+        skills,
+        memory,
+        mcp,
+        agent,
+        gateway,
+        bot,
+        reviews,
+        close,
+    };
+}
+export async function apply(ctx) {
+    const handle = await ctx.dshLoader.config.register(ctx, template);
+    let service, disposed = false, lastError = "", queue = Promise.resolve();
+    const onError = (e) => {
+        lastError = e.message;
+        ctx.logger?.warn?.(`[synapse] ${e.message}`);
+    };
+    const enqueue = (fn) => {
+        const result = queue.then(fn);
+        queue = result.catch(onError);
+        return result;
+    };
+    const reconfigure = (value) => enqueue(async () => {
+        if (disposed)
+            return;
+        validateConfig(value);
+        pathsFor(ctx, value);
+        if (service) {
+            const old = service;
+            service = null;
+            await old.close();
+        }
+        service = await createService(ctx, value, onError);
+        lastError = "";
+    });
+    const off = handle.watch(({ value }) => void reconfigure(value).catch(() => { }));
+    const manage = async (req, res) => {
+        try {
+            if (!browserAuthorized(ctx, req, res))
+                return;
+            const url = new URL(req.url, "http://localhost");
+            const action = url.pathname.slice("/api/dsh-synapse/".length);
+            if (action === "status")
+                return json(res, {
+                    running: Boolean(service),
+                    bot: service?.bot.state || "stopped",
+                    lastError,
+                });
+            if (!service)
+                return json(res, { error: lastError || "服务尚未就绪" }, 503);
+            // Serialize management with configuration changes; no writes into a closing DB.
+            await enqueue(async () => {
+                if (!service)
+                    throw new Error("服务正在重载");
+                if (req.method === "GET" && action === "skills")
+                    return json(res, await service.skills.list());
+                if (req.method === "GET" && action === "reviews")
+                    return json(res, service.reviews.list());
+                if (req.method === "GET" && action === "memory")
+                    return json(res, service.memory.list(url.searchParams.get("channel") || "chat"));
+                if (req.method === "GET" && action === "features") {
+                    const live = await service.bot.liveFeatures();
+                    return json(res, service.config().features.map((f) => ({
+                        id: f.id,
+                        value: {
+                            ...JSON.parse(f.json),
+                            ...(live.find((x) => x.id === f.id)?.value || {}),
+                        },
+                    })));
+                }
+                if (req.method === "POST") {
+                    const data = JSON.parse((await body(req)).toString() || "{}");
+                    if (action === "skills/save")
+                        return json(res, await service.skills.save(data.name, data.content, data.revision));
+                    if (action === "skills/delete") {
+                        await service.skills.remove(data.name, data.revision);
+                        return json(res, { ok: true });
+                    }
+                    if (action === "mcp/tools")
+                        return json(res, await service.mcp.list(data.name));
+                    if (action === "memory/import") {
+                        if (!service.config().ragEnabled)
+                            throw new Error("请先启用 RAG");
+                        const items = parseHermes(data.markdown, data.channel);
+                        return json(res, data.preview ? { items } : await service.memory.addMany(items));
+                    }
+                    if (action === "memory/search")
+                        return json(res, await service.memory.search(String(data.query), data.channel));
+                    if (action === "reviews/retry") {
+                        service.reviews.retry(data.id);
+                        return json(res, { ok: true });
+                    }
+                }
+                json(res, { error: "Not found" }, 404);
+            });
+        }
+        catch (e) {
+            if (!res.headersSent)
+                json(res, { error: e.message }, e.status || 400);
+        }
+    };
+    const routes = [];
+    ctx.effect(() => () => {
+        disposed = true;
+        off();
+        for (const release of routes)
+            release();
+        void enqueue(async () => {
+            await service?.close();
+            service = null;
+        });
+    });
+    routes.push(ctx.dshLoader.web.register("/api/dsh-synapse/", manage));
+    routes.push(ctx.dshLoader.web.post("/hooks/dsh-synapse/github", (req, res) => {
+        void enqueue(async () => {
+            if (!service)
+                return json(res, { error: "服务尚未就绪" }, 503);
+            await service.reviews.webhook(req, res);
+        }).catch((e) => {
+            if (!res.headersSent)
+                json(res, { error: e.message }, e.status || 400);
+        });
+    }));
+    // A same-origin standalone panel is also available for older DSH clients.
+    routes.push(ctx.dshLoader.web.get("/synapse", async (req, res) => {
+        if (!browserAuthorized(ctx, req, res))
+            return;
+        const html = await readFile(new URL("./panel.html", import.meta.url), "utf8");
+        res.writeHead(200, {
+            "Content-Type": "text/html; charset=utf-8",
+            "Content-Security-Policy": "default-src 'self'; script-src 'self'; style-src 'self' 'unsafe-inline'; frame-ancestors 'self'",
+        });
+        res.end(html);
+    }));
+    routes.push(ctx.dshLoader.web.get("/synapse-panel.js", async (req, res) => {
+        if (!browserAuthorized(ctx, req, res))
+            return;
+        res.writeHead(200, { "Content-Type": "text/javascript; charset=utf-8" });
+        res.end(await readFile(new URL("./panel.js", import.meta.url), "utf8"));
+    }));
+    await reconfigure(handle.get().value).catch(() => { });
+}
+//# sourceMappingURL=index.js.map

package/lib/host/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AACtC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAC/B,OAAO,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AACtC,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAE1D,MAAM,CAAC,MAAM,IAAI,GAAG,yBAAyB,CAAC;AAC9C,MAAM,CAAC,MAAM,MAAM,GAAG,CAAC,WAAW,CAAC,CAAC;AACpC,OAAO,EAAE,QAAQ,EAAE,CAAC;AACpB,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,GAAmB,EACnB,CAAgB,EAChB,OAAkB;IAElB,cAAc,CAAC,CAAC,CAAC,CAAC;IAClB,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IAC/B,MAAM,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;IACrD,MAAM,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7C,MAAM,MAAM,GAAG,GAAG,EAAE,CAAC,CAAC,EACpB,SAAS,GAAG,IAAI,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,EACrC,MAAM,GAAG,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC;IACjC,MAAM,MAAM,GAAG,IAAI,MAAM,CACrB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,eAAe,CAAC,EACtC,SAAS,EACT,MAAM,CACP,EACD,GAAG,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;IAC/B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,EACtE,OAAO,GAAG,IAAI,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAChD,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,EAClD,OAAO,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAChE,MAAM,KAAK,GAAG,KAAK,IAAI,EAAE;QACvB,MAAM,OAAO,CAAC,UAAU,CAAC;YACvB,OAAO,CAAC,KAAK,EAAE;YACf,GAAG,CAAC,IAAI,EAAE;YACV,OAAO,CAAC,KAAK,EAAE;YACf,GAAG,CAAC,KAAK,EAAE;YACX,SAAS,CAAC,KAAK,EAAE;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC,CAAC;IACF,IAAI,CAAC;QACH,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;QACtB,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC;QAClB,KAAK,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,UAAU;YAC3B,KAAK,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACnD,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,KAAK,EAAE,CAAC;QACd,MAAM,CAAC,CAAC;IACV,CAAC;IACD,OAAO;QACL,KAAK;QACL,MAAM;QACN,SAAS;QACT,MAAM;QACN,MAAM;QACN,GAAG;QACH,KAAK;QACL,OAAO;QACP,GAAG;QACH,OAAO;QACP,KAAK;KACN,CAAC;AACJ,CAAC;AACD,MAAM,CAAC,KAAK,UAAU,KAAK,CAAC,GAAmB;IAC7C,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAChD,GAAG,EACH,QAAQ,CACT,CAAC;IACF,IAAI,OAAO,EACT,QAAQ,GAAG,KAAK,EAChB,SAAS,GAAG,EAAE,EACd,KAAK,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAC5B,MAAM,OAAO,GAAG,CAAC,CAAC,EAAE,EAAE;QACpB,SAAS,GAAG,CAAC,CAAC,OAAO,CAAC;QACtB,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;IAC/C,CAAC,CAAC;IACF,MAAM,OAAO,GAAG,CAAC,EAAE,EAAE,EAAE;QACrB,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC9B,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC9B,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IACF,MAAM,WAAW,GAAG,CAAC,KAAK,EAAE,EAAE,CAC5B,OAAO,CAAC,KAAK,IAAI,EAAE;QACjB,IAAI,QAAQ;YAAE,OAAO;QACrB,cAAc,CAAC,KAAK,CAAC,CAAC;QACtB,QAAQ,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACrB,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,GAAG,GAAG,OAAO,CAAC;YACpB,OAAO,GAAG,IAAI,CAAC;YACf,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;QACD,OAAO,GAAG,MAAM,aAAa,CAAC,GAAG,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACnD,SAAS,GAAG,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IACL,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CACtB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,WAAW,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CACvD,CAAC;IACF,MAAM,MAAM,GAAG,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QAChC,IAAI,CAAC;YACH,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBAAE,OAAO;YAC9C,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAC9D,IAAI,MAAM,KAAK,QAAQ;gBACrB,OAAO,IAAI,CAAC,GAAG,EAAE;oBACf,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC;oBACzB,GAAG,EAAE,OAAO,EAAE,GAAG,CAAC,KAAK,IAAI,SAAS;oBACpC,SAAS;iBACV,CAAC,CAAC;YACL,IAAI,CAAC,OAAO;gBACV,OAAO,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,SAAS,IAAI,QAAQ,EAAE,EAAE,GAAG,CAAC,CAAC;YAC1D,gFAAgF;YAChF,MAAM,OAAO,CAAC,KAAK,IAAI,EAAE;gBACvB,IAAI,CAAC,OAAO;oBAAE,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACxC,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,QAAQ;oBAC7C,OAAO,IAAI,CAAC,GAAG,EAAE,MAAM,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;gBAChD,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,SAAS;oBAC9C,OAAO,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC3C,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,QAAQ;oBAC7C,OAAO,IAAI,CACT,GAAG,EACH,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,MAAM,CAAC,CAC/D,CAAC;gBACJ,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,UAAU,EAAE,CAAC;oBAClD,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC;oBAC9C,OAAO,IAAI,CACT,GAAG,EACH,OAAO,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBACpC,EAAE,EAAE,CAAC,CAAC,EAAE;wBACR,KAAK,EAAE;4BACL,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;4BACrB,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,IAAI,EAAE,CAAC;yBAClD;qBACF,CAAC,CAAC,CACJ,CAAC;gBACJ,CAAC;gBACD,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;oBAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC,CAAC;oBAC9D,IAAI,MAAM,KAAK,aAAa;wBAC1B,OAAO,IAAI,CACT,GAAG,EACH,MAAM,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,CAClE,CAAC;oBACJ,IAAI,MAAM,KAAK,eAAe,EAAE,CAAC;wBAC/B,MAAM,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;wBACtD,OAAO,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;oBACjC,CAAC;oBACD,IAAI,MAAM,KAAK,WAAW;wBACxB,OAAO,IAAI,CAAC,GAAG,EAAE,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;oBACtD,IAAI,MAAM,KAAK,eAAe,EAAE,CAAC;wBAC/B,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,UAAU;4BAAE,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;wBAC9D,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;wBACvD,OAAO,IAAI,CACT,GAAG,EACH,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAC/D,CAAC;oBACJ,CAAC;oBACD,IAAI,MAAM,KAAK,eAAe;wBAC5B,OAAO,IAAI,CACT,GAAG,EACH,MAAM,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,CAC9D,CAAC;oBACJ,IAAI,MAAM,KAAK,eAAe,EAAE,CAAC;wBAC/B,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBAC/B,OAAO,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;oBACjC,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,GAAG,CAAC,CAAC;YACzC,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,GAAG,CAAC,WAAW;gBAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC;QACzE,CAAC;IACH,CAAC,CAAC;IACF,MAAM,MAAM,GAAG,EAAE,CAAC;IAClB,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE;QACpB,QAAQ,GAAG,IAAI,CAAC;QAChB,GAAG,EAAE,CAAC;QACN,KAAK,MAAM,OAAO,IAAI,MAAM;YAAE,OAAO,EAAE,CAAC;QACxC,KAAK,OAAO,CAAC,KAAK,IAAI,EAAE;YACtB,MAAM,OAAO,EAAE,KAAK,EAAE,CAAC;YACvB,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IACH,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,mBAAmB,EAAE,MAAM,CAAC,CAAC,CAAC;IACrE,MAAM,CAAC,IAAI,CACT,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,2BAA2B,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;QAC/D,KAAK,OAAO,CAAC,KAAK,IAAI,EAAE;YACtB,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,GAAG,CAAC,CAAC;YACzD,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;YACb,IAAI,CAAC,GAAG,CAAC,WAAW;gBAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CACH,CAAC;IACF,0EAA0E;IAC1E,MAAM,CAAC,IAAI,CACT,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QACnD,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;YAAE,OAAO;QAC9C,MAAM,IAAI,GAAG,MAAM,QAAQ,CACzB,IAAI,GAAG,CAAC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EACxC,MAAM,CACP,CAAC;QACF,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE;YACjB,cAAc,EAAE,0BAA0B;YAC1C,yBAAyB,EACvB,iGAAiG;SACpG,CAAC,CAAC;QACH,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC,CAAC,CACH,CAAC;IACF,MAAM,CAAC,IAAI,CACT,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,mBAAmB,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QAC5D,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;YAAE,OAAO;QAC9C,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,gCAAgC,EAAE,CAAC,CAAC;QACzE,GAAG,CAAC,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,GAAG,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC,CACH,CAAC;IACF,MAAM,WAAW,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;AACxD,CAAC"}

package/lib/host/inference-worker.d.ts

@@ -0,0 +1 @@
+export {};

package/lib/host/inference-worker.js

@@ -0,0 +1,72 @@
+import { parentPort, workerData } from "node:worker_threads";
+import { createRequire } from "node:module";
+import { pathToFileURL } from "node:url";
+const require = createRequire(workerData.manifest);
+const runtime = await import(pathToFileURL(require.resolve("@huggingface/transformers")).href);
+const { pipeline, env, AutoTokenizer, AutoModelForSequenceClassification } = runtime.default ?? runtime;
+env.cacheDir = workerData.cache;
+env.allowLocalModels = false;
+let embedder, tokenizer, ranker;
+async function load(factory) {
+    try {
+        env.remoteHost = "https://huggingface.co/";
+        return await factory();
+    }
+    catch (direct) {
+        env.remoteHost = "https://hf-mirror.com/";
+        try {
+            return await factory();
+        }
+        catch (mirror) {
+            throw new Error(`模型直连和镜像均失败：${direct.message}; ${mirror.message}`);
+        }
+    }
+}
+let queue = Promise.resolve();
+parentPort.on("message", (message) => {
+    queue = queue
+        .then(async () => {
+        const { id, method, args } = message;
+        try {
+            let value;
+            if (method === "embed") {
+                embedder ??= await load(() => pipeline("feature-extraction", "Xenova/bge-small-zh-v1.5", {
+                    dtype: "q8",
+                    device: "cpu",
+                }));
+                const output = await embedder((args.query ? "为这个句子生成表示以用于检索相关文章：" : "") +
+                    args.text, {
+                    pooling: "cls",
+                    normalize: true,
+                    truncation: true,
+                    max_length: 512,
+                });
+                value = Array.from(output.data);
+            }
+            else if (method === "rerank") {
+                tokenizer ??= await load(() => AutoTokenizer.from_pretrained("Xenova/bge-reranker-base"));
+                ranker ??= await load(() => AutoModelForSequenceClassification.from_pretrained("Xenova/bge-reranker-base", { dtype: "q8", device: "cpu" }));
+                value = [];
+                for (let i = 0; i < args.documents.length; i += 4) {
+                    const docs = args.documents.slice(i, i + 4);
+                    const inputs = tokenizer(docs.map(() => args.query), {
+                        text_pair: docs,
+                        padding: true,
+                        truncation: true,
+                        max_length: 512,
+                    });
+                    const { logits } = await ranker(inputs);
+                    value.push(...Array.from(logits.data).map((n) => 1 / (1 + Math.exp(-n))));
+                }
+            }
+            else
+                throw new Error("未知推理方法");
+            parentPort.postMessage({ id, value });
+        }
+        catch (e) {
+            parentPort.postMessage({ id, error: e.message });
+        }
+    })
+        .catch((e) => parentPort.postMessage({ id: message.id, error: e.message }));
+});
+//# sourceMappingURL=inference-worker.js.map

package/lib/host/inference-worker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"inference-worker.js","sourceRoot":"","sources":["../../src/inference-worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,MAAM,OAAO,GAAG,aAAa,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;AACnD,MAAM,OAAO,GAAG,MAAM,MAAM,CAC1B,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC,CAAC,IAAI,CACjE,CAAC;AACF,MAAM,EAAE,QAAQ,EAAE,GAAG,EAAE,aAAa,EAAE,kCAAkC,EAAE,GACxE,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC;AAC7B,GAAG,CAAC,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC;AAChC,GAAG,CAAC,gBAAgB,GAAG,KAAK,CAAC;AAC7B,IAAI,QAAQ,EAAE,SAAS,EAAE,MAAM,CAAC;AAChC,KAAK,UAAU,IAAI,CAAC,OAAO;IACzB,IAAI,CAAC;QACH,GAAG,CAAC,UAAU,GAAG,yBAAyB,CAAC;QAC3C,OAAO,MAAM,OAAO,EAAE,CAAC;IACzB,CAAC;IAAC,OAAO,MAAM,EAAE,CAAC;QAChB,GAAG,CAAC,UAAU,GAAG,wBAAwB,CAAC;QAC1C,IAAI,CAAC;YACH,OAAO,MAAM,OAAO,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CACb,cAAc,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,EAAE,CAClD,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC;AACD,IAAI,KAAK,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;AAC9B,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;IACnC,KAAK,GAAG,KAAK;SACV,IAAI,CAAC,KAAK,IAAI,EAAE;QACf,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC;QACrC,IAAI,CAAC;YACH,IAAI,KAAK,CAAC;YACV,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;gBACvB,QAAQ,KAAK,MAAM,IAAI,CAAC,GAAG,EAAE,CAC3B,QAAQ,CAAC,oBAAoB,EAAE,0BAA0B,EAAE;oBACzD,KAAK,EAAE,IAAI;oBACX,MAAM,EAAE,KAAK;iBACd,CAAC,CACH,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAC3B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,EAAE,CAAC;oBACvC,IAAI,CAAC,IAAI,EACX;oBACE,OAAO,EAAE,KAAK;oBACd,SAAS,EAAE,IAAI;oBACf,UAAU,EAAE,IAAI;oBAChB,UAAU,EAAE,GAAG;iBAChB,CACF,CAAC;gBACF,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAClC,CAAC;iBAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,SAAS,KAAK,MAAM,IAAI,CAAC,GAAG,EAAE,CAC5B,aAAa,CAAC,eAAe,CAAC,0BAA0B,CAAC,CAC1D,CAAC;gBACF,MAAM,KAAK,MAAM,IAAI,CAAC,GAAG,EAAE,CACzB,kCAAkC,CAAC,eAAe,CAChD,0BAA0B,EAC1B,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAC/B,CACF,CAAC;gBACF,KAAK,GAAG,EAAE,CAAC;gBACX,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;oBAClD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;oBAC5C,MAAM,MAAM,GAAG,SAAS,CACtB,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAC1B;wBACE,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,IAAI;wBACb,UAAU,EAAE,IAAI;wBAChB,UAAU,EAAE,GAAG;qBAChB,CACF,CAAC;oBACF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,CAAC;oBACxC,KAAK,CAAC,IAAI,CACR,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAoB,CAAC,CAAC,GAAG,CAC5C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAC9B,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;;gBAAM,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC;YACjC,UAAU,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACxC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,UAAU,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QACnD,CAAC;IACH,CAAC,CAAC;SACD,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;AAChF,CAAC,CAAC,CAAC"}

package/lib/host/mcp.d.ts

@@ -0,0 +1,138 @@
+import type { ConfigSource, AvailableTool } from "./types.js";
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+export declare class McpManager {
+    config: ConfigSource;
+    connections: Map<string, Promise<Client>>;
+    closed: boolean;
+    constructor(config: ConfigSource);
+    connect(name: any): Promise<Client<{
+        method: string;
+        params?: {
+            [x: string]: unknown;
+            _meta?: {
+                [x: string]: unknown;
+                progressToken?: string | number;
+                "io.modelcontextprotocol/related-task"?: {
+                    taskId: string;
+                };
+            };
+        };
+    }, {
+        method: string;
+        params?: {
+            [x: string]: unknown;
+            _meta?: {
+                [x: string]: unknown;
+                progressToken?: string | number;
+                "io.modelcontextprotocol/related-task"?: {
+                    taskId: string;
+                };
+            };
+        };
+    }, {
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            progressToken?: string | number;
+            "io.modelcontextprotocol/related-task"?: {
+                taskId: string;
+            };
+        };
+    }>>;
+    list(name: any): Promise<any[]>;
+    available(): Promise<AvailableTool[]>;
+    call(tool: any, args: any): Promise<{
+        [x: string]: unknown;
+        content: ({
+            type: "text";
+            text: string;
+            annotations?: {
+                audience?: ("user" | "assistant")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: Record<string, unknown> | undefined;
+        } | {
+            type: "image";
+            data: string;
+            mimeType: string;
+            annotations?: {
+                audience?: ("user" | "assistant")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: Record<string, unknown> | undefined;
+        } | {
+            type: "audio";
+            data: string;
+            mimeType: string;
+            annotations?: {
+                audience?: ("user" | "assistant")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: Record<string, unknown> | undefined;
+        } | {
+            type: "resource";
+            resource: {
+                uri: string;
+                text: string;
+                mimeType?: string | undefined;
+                _meta?: Record<string, unknown> | undefined;
+            } | {
+                uri: string;
+                blob: string;
+                mimeType?: string | undefined;
+                _meta?: Record<string, unknown> | undefined;
+            };
+            annotations?: {
+                audience?: ("user" | "assistant")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: Record<string, unknown> | undefined;
+        } | {
+            uri: string;
+            name: string;
+            type: "resource_link";
+            description?: string | undefined;
+            mimeType?: string | undefined;
+            size?: number | undefined;
+            annotations?: {
+                audience?: ("user" | "assistant")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            icons?: {
+                src: string;
+                mimeType?: string | undefined;
+                sizes?: string[] | undefined;
+                theme?: "light" | "dark" | undefined;
+            }[] | undefined;
+            title?: string | undefined;
+        })[];
+        _meta?: {
+            [x: string]: unknown;
+            progressToken?: string | number | undefined;
+            "io.modelcontextprotocol/related-task"?: {
+                taskId: string;
+            } | undefined;
+        } | undefined;
+        structuredContent?: Record<string, unknown> | undefined;
+        isError?: boolean | undefined;
+    } | {
+        [x: string]: unknown;
+        toolResult: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            progressToken?: string | number | undefined;
+            "io.modelcontextprotocol/related-task"?: {
+                taskId: string;
+            } | undefined;
+        } | undefined;
+    }>;
+    close(): Promise<void>;
+}

package/lib/host/mcp.js

@@ -0,0 +1,104 @@
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
+export class McpManager {
+    config;
+    connections;
+    closed;
+    constructor(config) {
+        this.config = config;
+        this.connections = new Map();
+        this.closed = false;
+    }
+    async connect(name) {
+        if (this.closed)
+            throw new Error("MCP 已关闭");
+        const spec = this.config().mcp.find((x) => x.name === name && x.enabled);
+        if (!spec)
+            throw new Error("MCP 服务未启用");
+        if (!this.connections.has(name))
+            this.connections.set(name, (async () => {
+                const client = new Client({ name: "dsh-synapse", version: "0.1.0" });
+                const transport = spec.transport === "stdio"
+                    ? new StdioClientTransport({
+                        command: spec.command,
+                        args: JSON.parse(spec.argsJson),
+                        env: {
+                            PATH: process.env.PATH ?? "",
+                            ...(process.env.SystemRoot
+                                ? { SystemRoot: process.env.SystemRoot }
+                                : {}),
+                            ...JSON.parse(spec.envJson),
+                        },
+                        stderr: "ignore",
+                    })
+                    : spec.transport === "sse"
+                        ? new SSEClientTransport(new URL(spec.url), {
+                            requestInit: { headers: JSON.parse(spec.headersJson) },
+                        })
+                        : new StreamableHTTPClientTransport(new URL(spec.url), {
+                            requestInit: { headers: JSON.parse(spec.headersJson) },
+                        });
+                try {
+                    await client.connect(transport, { timeout: 15000 });
+                    if (this.closed) {
+                        await client.close();
+                        throw new Error("MCP 已关闭");
+                    }
+                    return client;
+                }
+                catch (e) {
+                    await transport.close().catch(() => { });
+                    this.connections.delete(name);
+                    throw e;
+                }
+            })());
+        return this.connections.get(name);
+    }
+    async list(name) {
+        const c = await this.connect(name);
+        let cursor;
+        const tools = [];
+        do {
+            const result = await c.listTools(cursor ? { cursor } : undefined, {
+                timeout: 15000,
+            });
+            tools.push(...result.tools);
+            cursor = result.nextCursor;
+            if (tools.length > 2000)
+                throw new Error("MCP 工具数量超限");
+        } while (cursor);
+        return tools;
+    }
+    async available() {
+        const result = [];
+        for (const spec of this.config().mcp.filter((s) => s.enabled)) {
+            const allowed = new Set(spec.allowedTools.split(/\s+/).filter(Boolean));
+            if (!allowed.size)
+                continue;
+            for (const tool of await this.list(spec.name))
+                if (allowed.has(tool.name))
+                    result.push({
+                        server: spec.name,
+                        rawName: tool.name,
+                        name: `mcp_${result.length}`,
+                        description: `${spec.name}/${tool.name}: ${tool.description || ""}`,
+                        parameters: tool.inputSchema,
+                    });
+        }
+        return result;
+    }
+    async call(tool, args) {
+        const spec = this.config().mcp.find((s) => s.name === tool.server && s.enabled);
+        if (!spec?.allowedTools.split(/\s+/).includes(tool.rawName))
+            throw new Error("工具不在允许列表");
+        return (await this.connect(tool.server)).callTool({ name: tool.rawName, arguments: args }, undefined, { timeout: 60000 });
+    }
+    async close() {
+        this.closed = true;
+        await Promise.allSettled([...this.connections.values()].map(async (p) => (await p).close()));
+        this.connections.clear();
+    }
+}
+//# sourceMappingURL=mcp.js.map

package/lib/host/mcp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../../src/mcp.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAC;AAC7E,MAAM,OAAO,UAAU;IACrB,MAAM,CAAe;IACrB,WAAW,CAA+B;IAC1C,MAAM,CAAU;IAEhB,YAAY,MAAoB;QAC9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;IACtB,CAAC;IACD,KAAK,CAAC,OAAO,CAAC,IAAI;QAChB,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC;QACzE,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;QACxC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC;YAC7B,IAAI,CAAC,WAAW,CAAC,GAAG,CAClB,IAAI,EACJ,CAAC,KAAK,IAAI,EAAE;gBACV,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBACrE,MAAM,SAAS,GACb,IAAI,CAAC,SAAS,KAAK,OAAO;oBACxB,CAAC,CAAC,IAAI,oBAAoB,CAAC;wBACvB,OAAO,EAAE,IAAI,CAAC,OAAO;wBACrB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;wBAC/B,GAAG,EAAE;4BACH,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE;4BAC5B,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU;gCACxB,CAAC,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE;gCACxC,CAAC,CAAC,EAAE,CAAC;4BACP,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;yBAC5B;wBACD,MAAM,EAAE,QAAQ;qBACjB,CAAC;oBACJ,CAAC,CAAC,IAAI,CAAC,SAAS,KAAK,KAAK;wBACxB,CAAC,CAAC,IAAI,kBAAkB,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;4BACxC,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE;yBACvD,CAAC;wBACJ,CAAC,CAAC,IAAI,6BAA6B,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;4BACnD,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE;yBACvD,CAAC,CAAC;gBACX,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;oBACpD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;wBAChB,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;wBACrB,MAAM,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;oBAC7B,CAAC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;oBACxC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;oBAC9B,MAAM,CAAC,CAAC;gBACV,CAAC;YACH,CAAC,CAAC,EAAE,CACL,CAAC;QACJ,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,IAAI;QACb,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,MAAM,CAAC;QACX,MAAM,KAAK,GAAG,EAAE,CAAC;QACjB,GAAG,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE;gBAChE,OAAO,EAAE,KAAK;aACf,CAAC,CAAC;YACH,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5B,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC;YAC3B,IAAI,KAAK,CAAC,MAAM,GAAG,IAAI;gBAAE,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QACzD,CAAC,QAAQ,MAAM,EAAE;QACjB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,KAAK,CAAC,SAAS;QACb,MAAM,MAAM,GAAoB,EAAE,CAAC;QACnC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9D,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;YACxE,IAAI,CAAC,OAAO,CAAC,IAAI;gBAAE,SAAS;YAC5B,KAAK,MAAM,IAAI,IAAI,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC3C,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;oBACxB,MAAM,CAAC,IAAI,CAAC;wBACV,MAAM,EAAE,IAAI,CAAC,IAAI;wBACjB,OAAO,EAAE,IAAI,CAAC,IAAI;wBAClB,IAAI,EAAE,OAAO,MAAM,CAAC,MAAM,EAAE;wBAC5B,WAAW,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,WAAW,IAAI,EAAE,EAAE;wBACnE,UAAU,EAAE,IAAI,CAAC,WAAW;qBAC7B,CAAC,CAAC;QACT,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI;QACnB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,IAAI,CACjC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,MAAM,IAAI,CAAC,CAAC,OAAO,CAC3C,CAAC;QACF,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC;YACzD,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;QAC9B,OAAO,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAC/C,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,EACvC,SAAS,EACT,EAAE,OAAO,EAAE,KAAK,EAAE,CACnB,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,OAAO,CAAC,UAAU,CACtB,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CACnE,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;CACF"}

package/lib/host/memory.d.ts

@@ -0,0 +1,26 @@
+import type { ConfigSource, Embeddings, MemoryItem } from "./types.js";
+import Database from "better-sqlite3";
+export declare const tokenize: (text: string) => string;
+export declare function parseHermes(markdown: any, channel?: string): any[];
+export declare class Memory {
+    inference: Embeddings;
+    config: ConfigSource;
+    db: Database.Database;
+    constructor(file: string, inference: Embeddings, config: ConfigSource);
+    addMany(items: MemoryItem[]): Promise<{
+        inserted: number;
+        duplicates: number;
+    }>;
+    remove(id: any): void;
+    search(query: any, channel: any): Promise<{
+        score: any;
+        id: number;
+        hash: string;
+        created: string;
+        channel: import("./types.js").Channel;
+        source: string;
+        text: string;
+    }[]>;
+    list(channel: any, limit?: number): unknown[];
+    close(): void;
+}