dsh-diary-memory 0.0.0-stage → 0.1.0

package/lib/retrieval-service.d.ts

@@ -0,0 +1,55 @@
+import { Service } from '@deepseek-ai/cordis';
+import type { Context } from '@deepseek-ai/cordis';
+import type { Session } from '@deepseek-ai/dsh-session';
+import { BailianClient } from './bailian-client.js';
+import { MemoryStore } from './memory-store.js';
+import type { Diary } from './diary.js';
+import type { ResolvedConfig } from './config.js';
+import type { DiaryProjectionState } from './projection.js';
+declare module '@deepseek-ai/cordis' {
+    interface Context {
+        diaryMemory: RetrievalService;
+    }
+}
+declare module '@deepseek-ai/dsh-llm' {
+    interface MessageSourceMap {
+        'diary-memory': {
+            kind: 'diary-memory';
+            form: 'recall';
+        };
+    }
+}
+export interface RecallOptions {
+    limit?: number;
+    sessionId?: string;
+    signal?: AbortSignal;
+}
+export interface RecallHit {
+    id: string;
+    sessionId: string;
+    recordedAt: string;
+    diary: Diary;
+    similarity: number;
+    score: number;
+}
+export declare class RetrievalService extends Service {
+    readonly config: ResolvedConfig;
+    readonly client: BailianClient;
+    readonly store: MemoryStore;
+    readonly lifetime: AbortController;
+    private readonly jobs;
+    private readonly recovery;
+    constructor(ctx: Context, config: ResolvedConfig, dependencies?: {
+        store?: MemoryStore;
+        client?: BailianClient;
+    });
+    initialize(): Promise<void>;
+    stateOf(session: Session): DiaryProjectionState;
+    track<T>(operation: () => Promise<T>): Promise<T>;
+    dispose(): Promise<void>;
+    recall(query: string, options?: RecallOptions): Promise<RecallHit[]>;
+    /** Session log is authoritative. Only a matching flushed replacement can publish memory. */
+    reconcile(session: Session): Promise<void>;
+    private reconcileOnce;
+    continuityContext(): string;
+}

package/lib/retrieval-service.js

@@ -0,0 +1,123 @@
+import { Service } from '@deepseek-ai/cordis';
+import { isDeepStrictEqual } from 'node:util';
+import { BailianClient } from './bailian-client.js';
+import { MemoryStore } from './memory-store.js';
+import { renderDiary } from './diary.js';
+import { diaryProjection } from './projection.js';
+export class RetrievalService extends Service {
+    config;
+    client;
+    store;
+    lifetime = new AbortController();
+    jobs = new Set();
+    recovery = new WeakMap();
+    constructor(ctx, config, dependencies = {}) {
+        super(ctx, 'diaryMemory');
+        this.config = config;
+        this.store = dependencies.store ?? new MemoryStore(config.databasePath);
+        this.client = dependencies.client ?? new BailianClient(config, { signal: this.lifetime.signal });
+        ctx.effect(() => () => this.dispose(), 'diaryMemory.lifecycle()');
+    }
+    async initialize() {
+        this.ctx.sessionProjections.register(diaryProjection);
+        this.ctx.on('session/created', session => {
+            void this.track(() => this.reconcile(session)).catch(() => this.ctx.logger.warn('日记恢复未完成；保留待提交记录，下次使用时重试'));
+        });
+        for (const session of this.ctx.sessions.list())
+            await this.track(() => this.reconcile(session));
+    }
+    stateOf(session) {
+        const state = this.ctx.sessionProjections.stateOf(session, diaryProjection.key);
+        if (!state)
+            throw new Error('日记轮次 projection 未注册');
+        return state;
+    }
+    track(operation) {
+        this.lifetime.signal.throwIfAborted();
+        // Preserve native synchronous lock/maintenance admission before the first await.
+        let promise;
+        try {
+            promise = operation();
+        }
+        catch (error) {
+            promise = Promise.reject(error);
+        }
+        this.jobs.add(promise);
+        void promise.then(() => this.jobs.delete(promise), () => this.jobs.delete(promise));
+        return promise;
+    }
+    async dispose() {
+        this.lifetime.abort(new Error('日记记忆插件已卸载'));
+        await Promise.allSettled([...this.jobs]);
+        this.store.close();
+    }
+    recall(query, options = {}) {
+        return this.track(async () => {
+            const text = query.trim();
+            if (!text || text.length > 8000)
+                throw new TypeError('recall query 必须为 1–8000 字符');
+            const limit = options.limit ?? this.config.recallCount;
+            if (!Number.isInteger(limit) || limit < 1 || limit > this.config.candidateCount)
+                throw new TypeError('recall limit 超出候选数量');
+            const signal = AbortSignal.any([this.lifetime.signal, ...options.signal ? [options.signal] : []]);
+            signal.throwIfAborted();
+            const namespace = this.config.namespace;
+            if (!this.store.count(namespace, this.client.embeddingKey)) {
+                if (this.store.count(namespace))
+                    throw new Error('已有日记使用不同的 embedding 模型、维度或端点，请使用原向量配置或重新索引');
+                return [];
+            }
+            const vector = await this.client.embed(text, { textType: 'query', signal });
+            const candidates = this.store.search(vector, { namespace, embeddingKey: this.client.embeddingKey, limit: this.config.candidateCount, sessionId: options.sessionId });
+            if (!candidates.length)
+                return [];
+            const ranking = await this.client.rerank(text, candidates.map(item => item.search_text), { topN: Math.min(limit, candidates.length), signal });
+            signal.throwIfAborted();
+            return ranking.map(item => {
+                const match = candidates[item.index];
+                return { id: match.id, sessionId: match.session_id, recordedAt: match.diary.recordedAt, diary: match.diary, similarity: match.similarity, score: item.relevance_score };
+            });
+        });
+    }
+    /** Session log is authoritative. Only a matching flushed replacement can publish memory. */
+    reconcile(session) {
+        const existing = this.recovery.get(session);
+        if (existing)
+            return existing;
+        const promise = this.reconcileOnce(session).finally(() => this.recovery.delete(session));
+        this.recovery.set(session, promise);
+        return promise;
+    }
+    async reconcileOnce(session) {
+        const pending = this.store.pending(session.id);
+        if (!pending.length)
+            return;
+        const events = session.snapshotEvents();
+        const commits = [];
+        for (const record of pending) {
+            const checkpoint = events.find((event, index) => {
+                if (event.type !== 'user/message' || event.data.source.kind !== 'compact-checkpoint' || event.data.source.compactionId !== record.compaction_id || typeof event.surfaceOp !== 'object' || event.surfaceOp.op !== 'replace')
+                    return false;
+                const summary = events[index - 1];
+                return summary?.type === 'compaction/summary'
+                    && summary.data.compactionId === record.compaction_id
+                    && isDeepStrictEqual(summary.data.shadowedSeqs, record.sourceSeqs)
+                    && isDeepStrictEqual(summary.data.summary, [{ type: 'text', text: renderDiary(record.diary) }]);
+            });
+            if (checkpoint)
+                commits.push({ id: record.id, seq: checkpoint.seq });
+            else if (events.some(event => event.type === 'compaction/end' && event.data.compactionId === record.compaction_id && event.data.error !== undefined))
+                this.store.abort(record.id);
+        }
+        if (!commits.length)
+            return;
+        if (!await this.ctx.sessions.flush(session))
+            throw new Error('没有会话持久化提供方；日记仍为 prepared，不对检索可见');
+        for (const item of commits)
+            this.store.commit(item.id, item.seq);
+    }
+    continuityContext() {
+        return this.store.recent(this.config.namespace, this.config.recentDiaryCount).reverse().map(record => renderDiary(record.diary)).join('\n\n---\n\n');
+    }
+}
+//# sourceMappingURL=retrieval-service.js.map

package/lib/retrieval-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieval-service.js","sourceRoot":"","sources":["../src/retrieval-service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAA;AAG7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAA;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAA;AACnD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAA;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AAGxC,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAA;AAqBjD,MAAM,OAAO,gBAAiB,SAAQ,OAAO;IAClC,MAAM,CAAgB;IACtB,MAAM,CAAe;IACrB,KAAK,CAAa;IAClB,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAA;IACxB,IAAI,GAAG,IAAI,GAAG,EAAoB,CAAA;IAClC,QAAQ,GAAG,IAAI,OAAO,EAA0B,CAAA;IAEjE,YAAY,GAAY,EAAE,MAAsB,EAAE,eAAgE,EAAE;QAClH,KAAK,CAAC,GAAG,EAAE,aAAa,CAAC,CAAA;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;QACpB,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC,KAAK,IAAI,IAAI,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAA;QACvE,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,MAAM,IAAI,IAAI,aAAa,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAA;QAChG,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,yBAAyB,CAAC,CAAA;IACnE,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAA;QACrD,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,iBAAiB,EAAE,OAAO,CAAC,EAAE;YACvC,KAAK,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAA;QAC7G,CAAC,CAAC,CAAA;QACF,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE;YAAE,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;IACjG,CAAC;IAED,OAAO,CAAC,OAAgB;QACtB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,OAAO,CAAC,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC,CAAA;QAC/E,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAA;QAClD,OAAO,KAAK,CAAA;IACd,CAAC;IAED,KAAK,CAAI,SAA2B;QAClC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,cAAc,EAAE,CAAA;QACrC,iFAAiF;QACjF,IAAI,OAAmB,CAAA;QACvB,IAAI,CAAC;YAAC,OAAO,GAAG,SAAS,EAAE,CAAA;QAAC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAAC,CAAC;QAC/E,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QACtB,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;QACnF,OAAO,OAAO,CAAA;IAChB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAA;QAC3C,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QACxC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAA;IACpB,CAAC;IAED,MAAM,CAAC,KAAa,EAAE,UAAyB,EAAE;QAC/C,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE;YAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAA;YACzB,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,IAAI;gBAAE,MAAM,IAAI,SAAS,CAAC,4BAA4B,CAAC,CAAA;YAClF,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,CAAA;YACtD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc;gBAAE,MAAM,IAAI,SAAS,CAAC,qBAAqB,CAAC,CAAA;YAC3H,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;YACjG,MAAM,CAAC,cAAc,EAAE,CAAA;YACvB,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAA;YACvC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;gBAC3D,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC;oBAAE,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAA;gBAC9F,OAAO,EAAE,CAAA;YACX,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAA;YAC3E,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAA;YACpK,IAAI,CAAC,UAAU,CAAC,MAAM;gBAAE,OAAO,EAAE,CAAA;YACjC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAA;YAC9I,MAAM,CAAC,cAAc,EAAE,CAAA;YACvB,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;gBACxB,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;gBACpC,OAAO,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,UAAU,EAAE,UAAU,EAAE,KAAK,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,CAAC,eAAe,EAAE,CAAA;YACzK,CAAC,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,4FAA4F;IAC5F,SAAS,CAAC,OAAgB;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QAC3C,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAA;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;QACxF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QACnC,OAAO,OAAO,CAAA;IAChB,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,OAAgB;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;QAC9C,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAM;QAC3B,MAAM,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAA;QACvC,MAAM,OAAO,GAAkC,EAAE,CAAA;QACjD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;gBAC9C,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,oBAAoB,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,KAAK,MAAM,CAAC,aAAa,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,IAAI,KAAK,CAAC,SAAS,CAAC,EAAE,KAAK,SAAS;oBAAE,OAAO,KAAK,CAAA;gBACxO,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAA;gBACjC,OAAO,OAAO,EAAE,IAAI,KAAK,oBAAoB;uBACxC,OAAO,CAAC,IAAI,CAAC,YAAY,KAAK,MAAM,CAAC,aAAa;uBAClD,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC;uBAC/D,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAA;YACnG,CAAC,CAAC,CAAA;YACF,IAAI,UAAU;gBAAE,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,GAAG,EAAE,UAAU,CAAC,GAAG,EAAE,CAAC,CAAA;iBAC/D,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,IAAI,CAAC,YAAY,KAAK,MAAM,CAAC,aAAa,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,KAAK,SAAS,CAAC;gBAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;QACnL,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAM;QAC3B,IAAI,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAA;QAC/F,KAAK,MAAM,IAAI,IAAI,OAAO;YAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,GAAG,CAAC,CAAA;IAClE,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACtJ,CAAC;CACF"}

package/lib/tools.d.ts

@@ -0,0 +1,5 @@
+import type { Context } from '@deepseek-ai/cordis';
+export declare const name = "diary-recall";
+export declare const inject: string[];
+/** Model-visible read-only recall, scoped by the companion preset and namespace. */
+export declare function apply(ctx: Context): void;

package/lib/tools.js

@@ -0,0 +1,18 @@
+export const name = 'diary-recall';
+export const inject = ['tools', 'diaryMemory'];
+/** Model-visible read-only recall, scoped by the companion preset and namespace. */
+export function apply(ctx) {
+    ctx.tools.register({
+        name: 'diary_recall',
+        description: '想起我们之前的经历、称呼、偏好、情绪变化、约定或我自己的反思。输入自然语言，会检索历史日记并按相关性重排。日记中的主观感受和不确定解释不等于已确认事实。',
+        parameters: { type: 'object', properties: { query: { type: 'string', minLength: 1, maxLength: 8000 } }, required: ['query'], additionalProperties: false },
+        output: {
+            schema: { type: 'object', properties: { memories: { type: 'array', items: { type: 'object', additionalProperties: true } } }, required: ['memories'], additionalProperties: false },
+            render: (_args, value) => [{ type: 'text', text: JSON.stringify(value) }],
+        },
+        async execute(args, execution) {
+            return { memories: await ctx.diaryMemory.recall(args.query, { signal: execution.signal }) };
+        },
+    });
+}
+//# sourceMappingURL=tools.js.map

package/lib/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAGA,MAAM,CAAC,MAAM,IAAI,GAAG,cAAc,CAAA;AAClC,MAAM,CAAC,MAAM,MAAM,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAA;AAE9C,oFAAoF;AACpF,MAAM,UAAU,KAAK,CAAC,GAAY;IAChC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC;QACjB,IAAI,EAAE,cAAc;QACpB,WAAW,EAAE,8EAA8E;QAC3F,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,oBAAoB,EAAE,KAAK,EAAE;QAC1J,MAAM,EAAE;YACN,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,UAAU,CAAC,EAAE,oBAAoB,EAAE,KAAK,EAAE;YACnL,MAAM,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;SAC1E;QACD,KAAK,CAAC,OAAO,CAAC,IAAuB,EAAE,SAAS;YAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,GAAG,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,CAAA;QAC7F,CAAC;KACF,CAAC,CAAA;AACJ,CAAC"}

package/locale/compaction/en.json

@@ -0,0 +1 @@
+{"meta":{"title":"Diary compaction","description":"Writes and indexes a diary before native DSH compaction, preserving tool call boundaries and recoverable commits."}}

package/locale/compaction/zh.json

@@ -0,0 +1 @@
+{"meta":{"title":"日记压缩编排","description":"先写日记并持久化向量，再通过 DSH 原生压缩提交；保护工具调用边界。"}}

package/locale/en.json

@@ -0,0 +1 @@
+{"meta":{"title":"Companion diary memory","description":"Bailian embedding and reranking for a personal companion's diary, continuity, reflection, and safe context compaction."}}

package/locale/tools/en.json

@@ -0,0 +1 @@
+{"meta":{"title":"Recall diary","description":"Recall shared experiences, preferences, promises, and reflections through diary_recall."}}

package/locale/tools/zh.json

@@ -0,0 +1 @@
+{"meta":{"title":"回想日记","description":"通过 diary_recall 检索往日经历、偏好、约定和自我反思。"}}

package/locale/zh.json

@@ -0,0 +1 @@
+{"meta":{"title":"陪伴日记记忆","description":"使用百炼向量与重排保存、回想共同经历；安全压缩会话，延续情感与自我反思。"}}

package/package.json

@@ -1,6 +1,73 @@
 {
   "name": "dsh-diary-memory",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "version": "0.1.0",
+  "description": "为 DSH 的陪伴与个人数字分身提供有温度、可演化的日记记忆、百炼检索和安全压缩。",
+  "type": "module",
+  "main": "./lib/index.js",
+  "types": "./lib/index.d.ts",
+  "license": "MIT",
+  "engines": { "node": ">=22.15", "dsh": "0.2.1-rc.3" },
+  "files": ["lib", "src", "locale", "icon.svg", "cordis.patch.yml", "README.md", "LICENSE"],
+  "exports": {
+    ".": { "types": "./lib/index.d.ts", "default": "./lib/index.js" },
+    "./compaction": "./lib/orchestrator.js",
+    "./compaction/locale/*.json": "./locale/compaction/*.json",
+    "./compaction/icon": "./icon.svg",
+    "./tools": "./lib/tools.js",
+    "./tools/locale/*.json": "./locale/tools/*.json",
+    "./tools/icon": "./icon.svg",
+    "./bailian-client": "./lib/bailian-client.js",
+    "./memory-store": "./lib/memory-store.js",
+    "./locale/*.json": "./locale/*.json",
+    "./icon": "./icon.svg",
+    "./package.json": "./package.json"
+  },
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "check": "tsc -p tsconfig.json --noEmit",
+    "test": "tsc -p tsconfig.json && node --test test/*.test.mjs",
+    "prepack": "npm run build"
+  },
+  "dependencies": { "better-sqlite3": "^12.4.1", "zod": "4.4.3" },
+  "peerDependencies": {
+    "@deepseek-ai/cordis": "~4.0.5",
+    "@deepseek-ai/schemastery": "~3.18.5",
+    "@deepseek-ai/dsh-agent": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-compaction": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-compaction-basic": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-llm": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-session": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-session-projection": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-tools": "0.2.1-rc.3"
+  },
+  "peerDependenciesMeta": {
+    "@deepseek-ai/cordis": { "optional": true },
+    "@deepseek-ai/schemastery": { "optional": true },
+    "@deepseek-ai/dsh-agent": { "optional": true },
+    "@deepseek-ai/dsh-compaction": { "optional": true },
+    "@deepseek-ai/dsh-compaction-basic": { "optional": true },
+    "@deepseek-ai/dsh-llm": { "optional": true },
+    "@deepseek-ai/dsh-session": { "optional": true },
+    "@deepseek-ai/dsh-session-projection": { "optional": true },
+    "@deepseek-ai/dsh-tools": { "optional": true }
+  },
+  "devDependencies": {
+    "@deepseek-ai/cordis": "4.0.5",
+    "@deepseek-ai/schemastery": "3.18.5",
+    "@deepseek-ai/dsh-agent": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-compaction": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-compaction-basic": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-llm": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-session": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-session-projection": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-token-meter": "0.2.1-rc.3",
+    "@deepseek-ai/dsh-tools": "0.2.1-rc.3",
+    "@types/better-sqlite3": "^7.6.13",
+    "@types/node": "^24.0.0",
+    "typescript": "^5.9.3",
+    "yaml": "^2.8.1"
+  },
+  "dsh": {
+    "bundle": { "patch": "./cordis.patch.yml" }
+  }
+}

package/src/bailian-client.ts

@@ -0,0 +1,121 @@
+import { setTimeout as delay } from 'node:timers/promises'
+import type { ResolvedConfig } from './config.js'
+
+interface BailianFailure {
+  status?: number
+  requestId?: string
+  retryable?: boolean
+}
+export interface RerankResult { index: number; relevance_score: number }
+
+export class BailianError extends Error {
+  readonly status?: number
+  readonly requestId?: string
+  readonly retryable: boolean
+  constructor(message: string, { status, requestId, retryable = false }: BailianFailure = {}) {
+    super(message)
+    this.name = 'BailianError'
+    this.status = status
+    this.requestId = requestId
+    this.retryable = retryable
+  }
+}
+
+export function validateVector(vector: unknown, dimensions: number): number[] {
+  if (!Array.isArray(vector) || vector.length !== dimensions || !vector.every(n => typeof n === 'number' && Number.isFinite(n))) throw new BailianError('百炼返回的向量维度或数值无效')
+  if (!vector.some(n => n !== 0)) throw new BailianError('百炼返回了零向量')
+  return vector
+}
+
+/** Native DashScope APIs; no OpenAI SDK or compatibility endpoint is involved. */
+export class BailianClient {
+  readonly config: ResolvedConfig
+  readonly fetch: typeof globalThis.fetch
+  readonly signal?: AbortSignal
+  readonly embeddingKey: string
+  constructor(config: ResolvedConfig, { fetch: fetchImpl = globalThis.fetch, signal }: { fetch?: typeof globalThis.fetch; signal?: AbortSignal } = {}) {
+    this.config = config
+    this.fetch = fetchImpl
+    this.signal = signal
+    // Model names alone do not identify the vector space across endpoints/dimensions.
+    this.embeddingKey = JSON.stringify([config.baseUrl, config.embeddingModel, config.embeddingDimensions])
+  }
+
+  async request(path: string, body: unknown, callerSignal?: AbortSignal): Promise<unknown> {
+    const { config } = this
+    if (!config.apiKey) throw new BailianError('请先配置百炼 apiKey 或 DSH_DIARY_BAILIAN_API_KEY')
+    const signals = [this.signal, callerSignal].filter((item): item is AbortSignal => item !== undefined)
+    const parent = signals.length ? AbortSignal.any(signals) : undefined
+    for (let attempt = 0; ; attempt++) {
+      parent?.throwIfAborted()
+      const signal = AbortSignal.any([...(parent ? [parent] : []), AbortSignal.timeout(config.timeoutMs)])
+      let backoff = Math.min(500 * 2 ** attempt, 10000)
+      try {
+        const response = await this.fetch(new URL(path, config.baseUrl), {
+          method: 'POST',
+          headers: { Authorization: `Bearer ${config.apiKey}`, 'Content-Type': 'application/json' },
+          body: JSON.stringify(body),
+          signal,
+          redirect: 'error',
+        })
+        // Keep the timeout active while the body is being read, too.
+        if (!response.ok) {
+          const retryAfter = response.headers.get('retry-after')
+          if (retryAfter) {
+            const ms = /^\d+(\.\d+)?$/.test(retryAfter) ? Number(retryAfter) * 1000 : Date.parse(retryAfter) - Date.now()
+            if (Number.isFinite(ms)) backoff = Math.min(30000, Math.max(backoff, ms))
+          }
+          await response.body?.cancel()
+          throw new BailianError(`百炼请求失败（HTTP ${response.status}）`, {
+            status: response.status,
+            requestId: response.headers.get('x-request-id') ?? undefined,
+            retryable: response.status === 429 || response.status >= 500,
+          })
+        }
+        const data: unknown = await response.json()
+        if (typeof data !== 'object' || data === null || !('output' in data) || !data.output || ('code' in data && data.code)) throw new BailianError('百炼返回了 API 错误或缺少 output')
+        parent?.throwIfAborted()
+        return data.output
+      } catch (error) {
+        parent?.throwIfAborted()
+        const failure = error instanceof BailianError ? error : new BailianError(
+          signal.aborted ? '百炼请求超时' : '百炼网络请求失败或响应不是有效 JSON',
+          { retryable: error instanceof TypeError || signal.aborted },
+        )
+        if (!failure.retryable || attempt >= config.retries) throw failure
+      }
+      await delay(backoff, undefined, parent ? { signal: parent } : {})
+    }
+  }
+
+  async embed(text: string, { textType = 'document', signal }: { textType?: 'document' | 'query'; signal?: AbortSignal } = {}): Promise<number[]> {
+    if (typeof text !== 'string' || !text.trim()) throw new TypeError('embedding 文本不能为空')
+    if (!['document', 'query'].includes(textType)) throw new TypeError('无效的 embedding textType')
+    const { embeddingModel, embeddingDimensions } = this.config
+    const output = await this.request('/api/v1/services/embeddings/text-embedding/text-embedding', {
+      model: embeddingModel,
+      input: { texts: [text] },
+      parameters: { text_type: textType, dimension: embeddingDimensions },
+    }, signal) as { embeddings?: { text_index?: unknown; embedding?: unknown }[] }
+    if (output.embeddings?.length !== 1 || output.embeddings[0].text_index !== 0) throw new BailianError('百炼 embedding 的 text_index 无效')
+    return validateVector(output.embeddings[0].embedding, embeddingDimensions)
+  }
+
+  async rerank(query: string, documents: string[], { topN = documents.length, signal }: { topN?: number; signal?: AbortSignal } = {}): Promise<RerankResult[]> {
+    if (!documents.length) return []
+    if (typeof query !== 'string' || !query.trim() || documents.some(text => typeof text !== 'string' || !text.trim())) throw new TypeError('rerank 的查询和文档不能为空')
+    if (!Number.isInteger(topN) || topN < 1 || topN > documents.length) throw new TypeError('无效的 rerank topN')
+    const output = await this.request('/api/v1/services/rerank/text-rerank/text-rerank', {
+      model: this.config.rerankModel,
+      input: { query, documents },
+      parameters: { top_n: topN, return_documents: false },
+    }, signal) as { results?: RerankResult[] }
+    if (!Array.isArray(output.results) || output.results.length !== topN) throw new BailianError('百炼 rerank 结果数量无效')
+    const seen = new Set()
+    for (const result of output.results) {
+      if (!Number.isInteger(result.index) || result.index < 0 || result.index >= documents.length || seen.has(result.index) || !Number.isFinite(result.relevance_score) || result.relevance_score < 0 || result.relevance_score > 1) throw new BailianError('百炼 rerank 返回了无效或重复的文档索引/分数')
+      seen.add(result.index)
+    }
+    return [...output.results].sort((a, b) => b.relevance_score - a.relevance_score || a.index - b.index)
+  }
+}

package/src/config.ts

@@ -0,0 +1,107 @@
+import z from '@deepseek-ai/schemastery'
+import { homedir } from 'node:os'
+import { join, resolve } from 'node:path'
+
+export interface DiaryConfig {
+  apiKey: string
+  baseUrl: string
+  embeddingModel: string
+  embeddingDimensions: number
+  rerankModel: string
+  compressionRounds: number
+  retainRounds: number
+  namespace: string
+  databasePath: string
+  candidateCount: number
+  recallCount: number
+  timeoutMs: number
+  retries: number
+  diaryProvider: string
+  diaryModel: string
+  diaryMaxTokens: number
+  diaryInputMaxChars: number
+  recentDiaryCount: number
+  autoRecall: boolean
+}
+export type ResolvedConfig = Readonly<DiaryConfig>
+
+export const defaults = Object.freeze({
+  apiKey: '',
+  baseUrl: 'https://dashscope.aliyuncs.com',
+  embeddingModel: 'text-embedding-v4',
+  embeddingDimensions: 1024,
+  rerankModel: 'gte-rerank-v2',
+  compressionRounds: 20,
+  retainRounds: 2,
+  namespace: 'companion',
+  databasePath: '',
+  candidateCount: 30,
+  recallCount: 6,
+  timeoutMs: 30000,
+  retries: 2,
+  diaryProvider: '',
+  diaryModel: '',
+  diaryMaxTokens: 3000,
+  diaryInputMaxChars: 200000,
+  recentDiaryCount: 3,
+  autoRecall: true,
+})
+
+// DSH reads this schema for its native plugin configuration form.
+export const Config: z<DiaryConfig> = z.object({
+  apiKey: z.string().role('secret').default('').description('百炼 API Key；留空读取 DSH_DIARY_BAILIAN_API_KEY 或 DASHSCOPE_API_KEY。'),
+  baseUrl: z.string().default(defaults.baseUrl).description('百炼地域端点；须与 API Key 的地域一致。'),
+  embeddingModel: z.string().default(defaults.embeddingModel),
+  embeddingDimensions: z.number().step(1).min(1).max(4096).default(defaults.embeddingDimensions),
+  rerankModel: z.string().default(defaults.rerankModel),
+  compressionRounds: z.number().step(1).min(2).default(defaults.compressionRounds).description('累计完成多少个会话轮次后，在下一次安全的 pre-step 压缩。工具调用不额外计轮次。'),
+  retainRounds: z.number().step(1).min(1).default(defaults.retainRounds).description('至少保留最近多少个用户输入及其后续消息。'),
+  namespace: z.string().default(defaults.namespace).description('数字分身的记忆空间。只有使用同一空间的会话共享日记与自我演化。'),
+  databasePath: z.string().default('').description('SQLite 文件；默认位于当前 DSH profile 的 diary-memory/ 中。'),
+  candidateCount: z.number().step(1).min(1).max(100).default(defaults.candidateCount),
+  recallCount: z.number().step(1).min(1).max(100).default(defaults.recallCount),
+  timeoutMs: z.number().step(1).min(100).max(300000).default(defaults.timeoutMs),
+  retries: z.number().step(1).min(0).max(5).default(defaults.retries),
+  diaryProvider: z.string().default('').description('日记生成使用 DSH 的 LLM 路由；两个字段均留空则继承会话模型。'),
+  diaryModel: z.string().default(''),
+  diaryMaxTokens: z.number().step(1).min(256).max(16000).default(defaults.diaryMaxTokens),
+  diaryInputMaxChars: z.number().step(1).min(1000).default(defaults.diaryInputMaxChars),
+  recentDiaryCount: z.number().step(1).min(0).max(10).default(defaults.recentDiaryCount),
+  autoRecall: z.boolean().default(true).description('每个新用户输入自动检索日记，并通过 agent.inject 写入可重放的上下文。'),
+})
+
+export function resolveConfig(input: Partial<DiaryConfig> = {}, env: NodeJS.ProcessEnv = process.env): ResolvedConfig {
+  for (const key of Object.keys(input)) {
+    if (!(key in defaults)) throw new Error(`未知日记配置项：${key}`)
+  }
+  const config = Config({ ...defaults, ...input })
+  const ranges: [keyof Pick<DiaryConfig, 'embeddingDimensions' | 'compressionRounds' | 'retainRounds' | 'candidateCount' | 'recallCount' | 'timeoutMs' | 'retries' | 'diaryMaxTokens' | 'diaryInputMaxChars' | 'recentDiaryCount'>, number, number][] = [
+    ['embeddingDimensions', 1, 4096], ['compressionRounds', 2, 1000000],
+    ['retainRounds', 1, 1000000], ['candidateCount', 1, 100], ['recallCount', 1, 100],
+    ['timeoutMs', 100, 300000], ['retries', 0, 5], ['diaryMaxTokens', 256, 16000],
+    ['diaryInputMaxChars', 1000, 10000000], ['recentDiaryCount', 0, 10],
+  ]
+  for (const [key, min, max] of ranges) {
+    if (!Number.isSafeInteger(config[key]) || config[key] < min || config[key] > max) {
+      throw new Error(`${key} 必须为 ${min}–${max} 的整数`)
+    }
+  }
+  if (config.compressionRounds <= config.retainRounds) throw new Error('compressionRounds 必须大于 retainRounds')
+  if (config.recallCount > config.candidateCount) throw new Error('recallCount 不能大于 candidateCount')
+  config.diaryProvider = config.diaryProvider.trim()
+  config.diaryModel = config.diaryModel.trim()
+  if (Boolean(config.diaryProvider) !== Boolean(config.diaryModel)) throw new Error('diaryProvider 和 diaryModel 必须同时设置')
+  for (const key of ['namespace', 'embeddingModel', 'rerankModel'] as const) {
+    config[key] = config[key].trim()
+    if (!config[key] || config[key].length > 200) throw new Error(`${key} 不能为空且不能超过 200 字符`)
+  }
+  const endpoint = new URL(config.baseUrl)
+  if (endpoint.username || endpoint.password || endpoint.search || endpoint.hash || endpoint.pathname !== '/') throw new Error('baseUrl 必须是无路径、凭据和查询参数的百炼 origin')
+  if (endpoint.protocol !== 'https:' && !(endpoint.protocol === 'http:' && ['127.0.0.1', '[::1]', 'localhost'].includes(endpoint.hostname))) throw new Error('baseUrl 需要 HTTPS；本地测试可以使用 loopback HTTP')
+  config.baseUrl = endpoint.origin
+  config.apiKey = config.apiKey.trim() || env.DSH_DIARY_BAILIAN_API_KEY?.trim() || env.DASHSCOPE_API_KEY?.trim() || ''
+  const profileDir = env.DSH_PROFILE_DIR?.trim() || join(env.DSH_HOME?.trim() || join(homedir(), '.dsh'), 'profiles', env.DSH_PROFILE?.trim() || 'default')
+  config.databasePath = config.databasePath || join(profileDir, 'diary-memory', 'memories.sqlite3')
+  if (config.databasePath !== ':memory:') config.databasePath = resolve(config.databasePath)
+  return Object.freeze(config)
+}