78 lines
2.5 KiB
TypeScript
78 lines
2.5 KiB
TypeScript
/** 上下文检查点（A3.8）：平台保存「发给模型的上下文」的摘要与来源，权威在平台（实施设计 1.5） */
|
||
import type { Database } from './db.js';
|
||
|
||
export interface ContextCheckpoint {
|
||
sessionId: string;
|
||
summary: string;
|
||
summarySourceIds: string[];
|
||
includedMessageIds: string[];
|
||
lastMessageId: string | null;
|
||
pinned: string[];
|
||
estimatedChars: number;
|
||
updatedAt: number;
|
||
}
|
||
|
||
interface CheckpointRow {
|
||
session_id: string;
|
||
summary: string;
|
||
summary_source_ids: string;
|
||
included_message_ids: string;
|
||
last_message_id: string | null;
|
||
pinned: string;
|
||
estimated_chars: number;
|
||
updated_at: number;
|
||
}
|
||
|
||
const parseIds = (raw: string): string[] => {
|
||
try {
|
||
const parsed = JSON.parse(raw) as unknown;
|
||
return Array.isArray(parsed) ? parsed.filter((item): item is string => typeof item === 'string') : [];
|
||
} catch {
|
||
return [];
|
||
}
|
||
};
|
||
|
||
export class ContextRepository {
|
||
constructor(private readonly db: Database) {}
|
||
|
||
save(checkpoint: Omit<ContextCheckpoint, 'updatedAt'>): void {
|
||
this.db
|
||
.prepare(
|
||
`insert into context_checkpoints (session_id, summary, summary_source_ids, included_message_ids, last_message_id, pinned, estimated_chars, updated_at)
|
||
values (?, ?, ?, ?, ?, ?, ?, ?)
|
||
on conflict(session_id) do update set
|
||
summary = excluded.summary,
|
||
summary_source_ids = excluded.summary_source_ids,
|
||
included_message_ids = excluded.included_message_ids,
|
||
last_message_id = excluded.last_message_id,
|
||
pinned = excluded.pinned,
|
||
estimated_chars = excluded.estimated_chars,
|
||
updated_at = excluded.updated_at`,
|
||
)
|
||
.run(
|
||
checkpoint.sessionId,
|
||
checkpoint.summary,
|
||
JSON.stringify(checkpoint.summarySourceIds),
|
||
JSON.stringify(checkpoint.includedMessageIds),
|
||
checkpoint.lastMessageId,
|
||
JSON.stringify(checkpoint.pinned),
|
||
checkpoint.estimatedChars,
|
||
Date.now(),
|
||
);
|
||
}
|
||
|
||
get(sessionId: string): ContextCheckpoint | undefined {
|
||
const row = this.db.prepare('select * from context_checkpoints where session_id = ?').get(sessionId) as unknown as CheckpointRow | undefined;
|
||
if (!row) return undefined;
|
||
return {
|
||
sessionId: row.session_id,
|
||
summary: row.summary,
|
||
summarySourceIds: parseIds(row.summary_source_ids),
|
||
includedMessageIds: parseIds(row.included_message_ids),
|
||
lastMessageId: row.last_message_id,
|
||
pinned: parseIds(row.pinned),
|
||
estimatedChars: row.estimated_chars,
|
||
updatedAt: row.updated_at,
|
||
};
|
||
}
|
||
} |