mirror of
https://github.com/catlog22/Claude-Code-Workflow.git
synced 2026-03-03 15:43:11 +08:00
feat(memorycore): add tags system, session summaries, hook injection, tag filtering, and solidify compress mode
Implement 5 interconnected memorycore enhancements: 1. Tags backend: add tags TEXT column to memories table with migration, JSON array storage, full CRUD support via upsertMemory/getMemory/getMemories 2. LLM auto-tag extraction: extend extraction prompt to produce tags, parse and validate in pipeline, create CMEM from extraction results 3. Session summary API: expose rollout_summary via new REST endpoints GET /api/core-memory/sessions/summaries and sessions/:id/summary 4. Hook injection: increase SESSION_START_LIMIT to 1500, add Component 5 (Recent Sessions) to UnifiedContextBuilder with 300-char budget 5. Tag filtering: add getMemoriesByTags() with json_each() for safe SQL matching, wire through MCP tool, CLI --tags flag, REST ?tags= param 6. Solidify compress mode: add --type compress to solidify.md with getRecentMemories(), archiveMemories(), buildCompressionMetadata() Security fixes: safeParseTags() for corrupt DB data, json_each() instead of LIKE injection, ESCAPE clause for searchSessionsByKeyword, singleton store in unified-context-builder.
This commit is contained in:
catlog22
@@ -10,6 +10,16 @@ import { StoragePaths, ensureStorageDir } from '../config/storage-paths.js';
|
||||
import { UnifiedVectorIndex, isUnifiedEmbedderAvailable } from './unified-vector-index.js';
|
||||
import type { ChunkMetadata } from './unified-vector-index.js';
|
||||
|
||||
// Helpers
|
||||
function safeParseTags(raw: string | null | undefined): string[] {
|
||||
try {
|
||||
const parsed = JSON.parse(raw || '[]');
|
||||
return Array.isArray(parsed) ? parsed : [];
|
||||
} catch {
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
// Types
|
||||
export interface CoreMemory {
|
||||
id: string; // Format: CMEM-YYYYMMDD-HHMMSS
|
||||
@@ -20,6 +30,7 @@ export interface CoreMemory {
|
||||
updated_at: string;
|
||||
archived: boolean;
|
||||
metadata?: string; // JSON string
|
||||
tags?: string[]; // JSON array stored as TEXT
|
||||
}
|
||||
|
||||
export interface SessionCluster {
|
||||
@@ -137,7 +148,8 @@ export class CoreMemoryStore {
|
||||
created_at TEXT NOT NULL,
|
||||
updated_at TEXT NOT NULL,
|
||||
archived INTEGER DEFAULT 0,
|
||||
metadata TEXT
|
||||
metadata TEXT,
|
||||
tags TEXT DEFAULT '[]'
|
||||
);
|
||||
|
||||
-- Session clusters table
|
||||
@@ -313,6 +325,16 @@ export class CoreMemoryStore {
|
||||
|
||||
// Re-enable foreign key constraints
|
||||
this.db.pragma('foreign_keys = ON');
|
||||
|
||||
// Add tags column to existing memories table
|
||||
try {
|
||||
this.db.exec(`ALTER TABLE memories ADD COLUMN tags TEXT DEFAULT '[]'`);
|
||||
} catch (e) {
|
||||
const msg = (e as Error).message || '';
|
||||
if (!msg.includes('duplicate column name')) {
|
||||
throw e; // Re-throw unexpected errors
|
||||
}
|
||||
}
|
||||
} catch (e) {
|
||||
// If migration fails, continue - tables may not exist
|
||||
try {
|
||||
@@ -408,10 +430,11 @@ export class CoreMemoryStore {
|
||||
// Update existing memory
|
||||
const stmt = this.db.prepare(`
|
||||
UPDATE memories
|
||||
SET content = ?, summary = ?, raw_output = ?, updated_at = ?, archived = ?, metadata = ?
|
||||
SET content = ?, summary = ?, raw_output = ?, updated_at = ?, archived = ?, metadata = ?, tags = ?
|
||||
WHERE id = ?
|
||||
`);
|
||||
|
||||
const tags = memory.tags ?? existingMemory.tags ?? [];
|
||||
stmt.run(
|
||||
memory.content,
|
||||
memory.summary || existingMemory.summary,
|
||||
@@ -419,6 +442,7 @@ export class CoreMemoryStore {
|
||||
now,
|
||||
memory.archived !== undefined ? (memory.archived ? 1 : 0) : existingMemory.archived ? 1 : 0,
|
||||
memory.metadata || existingMemory.metadata,
|
||||
JSON.stringify(tags),
|
||||
id
|
||||
);
|
||||
|
||||
@@ -429,8 +453,8 @@ export class CoreMemoryStore {
|
||||
} else {
|
||||
// Insert new memory
|
||||
const stmt = this.db.prepare(`
|
||||
INSERT INTO memories (id, content, summary, raw_output, created_at, updated_at, archived, metadata)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
INSERT INTO memories (id, content, summary, raw_output, created_at, updated_at, archived, metadata, tags)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
`);
|
||||
|
||||
stmt.run(
|
||||
@@ -441,7 +465,8 @@ export class CoreMemoryStore {
|
||||
now,
|
||||
now,
|
||||
memory.archived ? 1 : 0,
|
||||
memory.metadata || null
|
||||
memory.metadata || null,
|
||||
JSON.stringify(memory.tags || [])
|
||||
);
|
||||
|
||||
// Sync new content to vector index
|
||||
@@ -467,12 +492,13 @@ export class CoreMemoryStore {
|
||||
created_at: row.created_at,
|
||||
updated_at: row.updated_at,
|
||||
archived: Boolean(row.archived),
|
||||
metadata: row.metadata
|
||||
metadata: row.metadata,
|
||||
tags: safeParseTags(row.tags)
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Get all memories
|
||||
* Get memories with optional filtering by archived status
|
||||
*/
|
||||
getMemories(options: { archived?: boolean; limit?: number; offset?: number } = {}): CoreMemory[] {
|
||||
const { archived, limit = 50, offset = 0 } = options;
|
||||
@@ -506,7 +532,52 @@ export class CoreMemoryStore {
|
||||
created_at: row.created_at,
|
||||
updated_at: row.updated_at,
|
||||
archived: Boolean(row.archived),
|
||||
metadata: row.metadata
|
||||
metadata: row.metadata,
|
||||
tags: safeParseTags(row.tags)
|
||||
}));
|
||||
}
|
||||
|
||||
/**
|
||||
* Get memories filtered by tags (AND logic - must contain ALL specified tags)
|
||||
*/
|
||||
getMemoriesByTags(tags: string[], options: { archived?: boolean; limit?: number; offset?: number } = {}): CoreMemory[] {
|
||||
const { archived, limit = 50, offset = 0 } = options;
|
||||
|
||||
if (tags.length === 0) {
|
||||
return this.getMemories({ archived, limit, offset });
|
||||
}
|
||||
|
||||
// Use json_each for proper structured matching (safe from injection)
|
||||
const conditions = tags.map(() => `EXISTS (SELECT 1 FROM json_each(memories.tags) WHERE json_each.value = ?)`).join(' AND ');
|
||||
const params: (string | number)[] = [...tags];
|
||||
|
||||
let archiveClause = '';
|
||||
if (archived !== undefined) {
|
||||
archiveClause = ' AND archived = ?';
|
||||
params.push(archived ? 1 : 0);
|
||||
}
|
||||
|
||||
const query = `
|
||||
SELECT * FROM memories
|
||||
WHERE ${conditions}${archiveClause}
|
||||
ORDER BY updated_at DESC
|
||||
LIMIT ? OFFSET ?
|
||||
`;
|
||||
params.push(limit, offset);
|
||||
|
||||
const stmt = this.db.prepare(query);
|
||||
const rows = stmt.all(...params) as any[];
|
||||
|
||||
return rows.map(row => ({
|
||||
id: row.id,
|
||||
content: row.content,
|
||||
summary: row.summary,
|
||||
raw_output: row.raw_output,
|
||||
created_at: row.created_at,
|
||||
updated_at: row.updated_at,
|
||||
archived: Boolean(row.archived),
|
||||
metadata: row.metadata,
|
||||
tags: safeParseTags(row.tags)
|
||||
}));
|
||||
}
|
||||
|
||||
@@ -534,6 +605,60 @@ export class CoreMemoryStore {
|
||||
stmt.run(new Date().toISOString(), id);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get recent memories ordered by creation time (newest first)
|
||||
* Used by compression flow to select candidates for consolidation
|
||||
*/
|
||||
getRecentMemories(limit: number = 20, excludeArchived: boolean = true): CoreMemory[] {
|
||||
const query = excludeArchived
|
||||
? `SELECT * FROM memories WHERE archived = 0 ORDER BY created_at DESC LIMIT ?`
|
||||
: `SELECT * FROM memories ORDER BY created_at DESC LIMIT ?`;
|
||||
|
||||
const stmt = this.db.prepare(query);
|
||||
const rows = stmt.all(limit) as any[];
|
||||
|
||||
return rows.map(row => ({
|
||||
id: row.id,
|
||||
content: row.content,
|
||||
summary: row.summary,
|
||||
raw_output: row.raw_output,
|
||||
created_at: row.created_at,
|
||||
updated_at: row.updated_at,
|
||||
archived: Boolean(row.archived),
|
||||
metadata: row.metadata,
|
||||
tags: safeParseTags(row.tags)
|
||||
}));
|
||||
}
|
||||
|
||||
/**
|
||||
* Archive multiple memories in a single transaction
|
||||
* Used after successful compression to archive source memories
|
||||
*/
|
||||
archiveMemories(ids: string[]): void {
|
||||
if (ids.length === 0) return;
|
||||
|
||||
const now = new Date().toISOString();
|
||||
const placeholders = ids.map(() => '?').join(', ');
|
||||
const stmt = this.db.prepare(`
|
||||
UPDATE memories
|
||||
SET archived = 1, updated_at = ?
|
||||
WHERE id IN (${placeholders})
|
||||
`);
|
||||
stmt.run(now, ...ids);
|
||||
}
|
||||
|
||||
/**
|
||||
* Build metadata JSON for a compressed memory
|
||||
* Tracks source memory IDs, compression ratio, and timestamp
|
||||
*/
|
||||
buildCompressionMetadata(sourceIds: string[], originalSize: number, compressedSize: number): string {
|
||||
return JSON.stringify({
|
||||
compressed_from: sourceIds,
|
||||
compression_ratio: compressedSize / originalSize,
|
||||
compressed_at: new Date().toISOString()
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete a memory
|
||||
*/
|
||||
@@ -969,11 +1094,12 @@ ${memory.content}
|
||||
searchSessionsByKeyword(keyword: string): SessionMetadataCache[] {
|
||||
const stmt = this.db.prepare(`
|
||||
SELECT * FROM session_metadata_cache
|
||||
WHERE title LIKE ? OR summary LIKE ? OR keywords LIKE ?
|
||||
WHERE title LIKE ? ESCAPE '\\' OR summary LIKE ? ESCAPE '\\' OR keywords LIKE ? ESCAPE '\\'
|
||||
ORDER BY access_count DESC, last_accessed DESC
|
||||
`);
|
||||
|
||||
const pattern = `%${keyword}%`;
|
||||
const escaped = keyword.replace(/[%_\\]/g, c => '\\' + c);
|
||||
const pattern = `%${escaped}%`;
|
||||
const rows = stmt.all(pattern, pattern, pattern) as any[];
|
||||
|
||||
return rows.map(row => ({
|
||||
@@ -1413,6 +1539,39 @@ ${memory.content}
|
||||
}));
|
||||
}
|
||||
|
||||
/**
|
||||
* Get session summaries from stage1_outputs, ordered by generated_at descending.
|
||||
* Returns lightweight objects with thread_id, rollout_summary, and generated_at.
|
||||
*/
|
||||
getSessionSummaries(limit: number = 20): Array<{ thread_id: string; rollout_summary: string; generated_at: number }> {
|
||||
const stmt = this.db.prepare(
|
||||
`SELECT thread_id, rollout_summary, generated_at FROM stage1_outputs ORDER BY generated_at DESC LIMIT ?`
|
||||
);
|
||||
const rows = stmt.all(limit) as any[];
|
||||
return rows.map(row => ({
|
||||
thread_id: row.thread_id,
|
||||
rollout_summary: row.rollout_summary,
|
||||
generated_at: row.generated_at,
|
||||
}));
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a single session summary by thread_id.
|
||||
* Returns null if no extraction output exists for the given thread.
|
||||
*/
|
||||
getSessionSummary(threadId: string): { thread_id: string; rollout_summary: string; generated_at: number } | null {
|
||||
const stmt = this.db.prepare(
|
||||
`SELECT thread_id, rollout_summary, generated_at FROM stage1_outputs WHERE thread_id = ?`
|
||||
);
|
||||
const row = stmt.get(threadId) as any;
|
||||
if (!row) return null;
|
||||
return {
|
||||
thread_id: row.thread_id,
|
||||
rollout_summary: row.rollout_summary,
|
||||
generated_at: row.generated_at,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Count Phase 1 outputs
|
||||
*/
|
||||
@@ -1581,7 +1740,8 @@ export function getMemoriesFromProject(projectId: string): CoreMemory[] {
|
||||
created_at: row.created_at,
|
||||
updated_at: row.updated_at,
|
||||
archived: Boolean(row.archived),
|
||||
metadata: row.metadata
|
||||
metadata: row.metadata,
|
||||
tags: safeParseTags(row.tags)
|
||||
}));
|
||||
}
|
||||
|
||||
@@ -1621,7 +1781,8 @@ export function findMemoryAcrossProjects(memoryId: string): { memory: CoreMemory
|
||||
created_at: row.created_at,
|
||||
updated_at: row.updated_at,
|
||||
archived: Boolean(row.archived),
|
||||
metadata: row.metadata
|
||||
metadata: row.metadata,
|
||||
tags: safeParseTags(row.tags)
|
||||
},
|
||||
projectId
|
||||
};
|
||||
|
||||
@@ -40,6 +40,7 @@ export interface ExtractionInput {
|
||||
export interface ExtractionOutput {
|
||||
raw_memory: string;
|
||||
rollout_summary: string;
|
||||
tags: string[];
|
||||
}
|
||||
|
||||
export interface TranscriptFilterOptions {
|
||||
@@ -289,10 +290,10 @@ export class MemoryExtractionPipeline {
|
||||
* Applies secret redaction and size limit enforcement.
|
||||
*
|
||||
* @param llmOutput - Raw text output from the LLM
|
||||
* @returns Validated ExtractionOutput with raw_memory and rollout_summary
|
||||
* @returns Validated ExtractionOutput with raw_memory, rollout_summary, and tags
|
||||
*/
|
||||
postProcess(llmOutput: string): ExtractionOutput {
|
||||
let parsed: { raw_memory?: string; rollout_summary?: string } | null = null;
|
||||
let parsed: { raw_memory?: string; rollout_summary?: string; tags?: string[] } | null = null;
|
||||
|
||||
// Mode 1: Pure JSON
|
||||
try {
|
||||
@@ -333,7 +334,17 @@ export class MemoryExtractionPipeline {
|
||||
rolloutSummary = rolloutSummary.substring(0, MAX_SUMMARY_CHARS);
|
||||
}
|
||||
|
||||
return { raw_memory: rawMemory, rollout_summary: rolloutSummary };
|
||||
// Extract and validate tags (fallback to empty array)
|
||||
let tags: string[] = [];
|
||||
if (parsed.tags && Array.isArray(parsed.tags)) {
|
||||
tags = parsed.tags
|
||||
.filter((t: unknown) => typeof t === 'string')
|
||||
.map((t: string) => t.toLowerCase().trim())
|
||||
.filter((t: string) => t.length > 0)
|
||||
.slice(0, 8);
|
||||
}
|
||||
|
||||
return { raw_memory: rawMemory, rollout_summary: rolloutSummary, tags };
|
||||
}
|
||||
|
||||
// ========================================================================
|
||||
@@ -388,6 +399,14 @@ export class MemoryExtractionPipeline {
|
||||
const store = getCoreMemoryStore(this.projectPath);
|
||||
store.upsertStage1Output(output);
|
||||
|
||||
// Create/update a core memory (CMEM) from extraction results with tags
|
||||
store.upsertMemory({
|
||||
id: `CMEM-EXT-${sessionId}`,
|
||||
content: extracted.raw_memory,
|
||||
summary: extracted.rollout_summary,
|
||||
tags: extracted.tags,
|
||||
});
|
||||
|
||||
// Sync extracted content to vector index (fire-and-forget)
|
||||
this.syncExtractionToVectorIndex(output);
|
||||
|
||||
|
||||
@@ -3,7 +3,7 @@
|
||||
*
|
||||
* Provides system and user prompt templates for extracting structured memory
|
||||
* from CLI session transcripts. The LLM output must conform to a JSON schema
|
||||
* with raw_memory and rollout_summary fields.
|
||||
* with raw_memory, rollout_summary, and tags fields.
|
||||
*
|
||||
* Design spec section 4.4: Prompt structure with outcome triage rules.
|
||||
*/
|
||||
@@ -12,20 +12,29 @@
|
||||
* System prompt for the extraction LLM call.
|
||||
*
|
||||
* Instructs the model to:
|
||||
* - Produce a JSON object with raw_memory and rollout_summary
|
||||
* - Produce a JSON object with raw_memory, rollout_summary, and tags
|
||||
* - Follow structure markers in raw_memory (# summary, Memory context, etc.)
|
||||
* - Apply outcome triage rules for categorizing task results
|
||||
* - Keep rollout_summary concise (1-2 sentences)
|
||||
* - Generate 3-8 lowercase tags capturing topic, action, and technology
|
||||
*/
|
||||
export const EXTRACTION_SYSTEM_PROMPT = `You are a memory extraction agent. Your job is to read a CLI session transcript and produce structured memory output.
|
||||
|
||||
You MUST respond with a valid JSON object containing exactly two fields:
|
||||
You MUST respond with a valid JSON object containing exactly three fields:
|
||||
|
||||
{
|
||||
"raw_memory": "<structured memory text>",
|
||||
"rollout_summary": "<1-2 sentence summary>"
|
||||
"rollout_summary": "<1-2 sentence summary>",
|
||||
"tags": ["<tag1>", "<tag2>", ...]
|
||||
}
|
||||
|
||||
## tags format
|
||||
|
||||
An array of 3-8 short lowercase tags (1-3 words each) capturing:
|
||||
- Main topic or domain (e.g., "authentication", "database migration")
|
||||
- Action type (e.g., "bug fix", "refactoring", "new feature")
|
||||
- Key technology (e.g., "react", "typescript", "sqlite")
|
||||
|
||||
## raw_memory format
|
||||
|
||||
The raw_memory field must follow this structure:
|
||||
@@ -87,5 +96,5 @@ Session ID: ${sessionId}
|
||||
${transcript}
|
||||
--- END TRANSCRIPT ---
|
||||
|
||||
Respond with a JSON object containing "raw_memory" and "rollout_summary" fields.`;
|
||||
Respond with a JSON object containing "raw_memory", "rollout_summary", and "tags" fields.`;
|
||||
}
|
||||
|
||||
@@ -38,10 +38,21 @@ export async function handleCoreMemoryRoutes(ctx: RouteContext): Promise<boolean
|
||||
const archived = archivedParam === null ? undefined : archivedParam === 'true';
|
||||
const limit = parseInt(url.searchParams.get('limit') || '100', 10);
|
||||
const offset = parseInt(url.searchParams.get('offset') || '0', 10);
|
||||
const tagsParam = url.searchParams.get('tags');
|
||||
|
||||
try {
|
||||
const store = getCoreMemoryStore(projectPath);
|
||||
const memories = store.getMemories({ archived, limit, offset });
|
||||
|
||||
// Use tag filter if tags query parameter is provided
|
||||
let memories;
|
||||
if (tagsParam) {
|
||||
const tags = tagsParam.split(',').map(t => t.trim()).filter(Boolean);
|
||||
memories = tags.length > 0
|
||||
? store.getMemoriesByTags(tags, { archived, limit, offset })
|
||||
: store.getMemories({ archived, limit, offset });
|
||||
} else {
|
||||
memories = store.getMemories({ archived, limit, offset });
|
||||
}
|
||||
|
||||
res.writeHead(200, { 'Content-Type': 'application/json' });
|
||||
res.end(JSON.stringify({ success: true, memories }));
|
||||
@@ -78,7 +89,7 @@ export async function handleCoreMemoryRoutes(ctx: RouteContext): Promise<boolean
|
||||
// API: Core Memory - Create or update memory
|
||||
if (pathname === '/api/core-memory/memories' && req.method === 'POST') {
|
||||
handlePostRequest(req, res, async (body) => {
|
||||
const { content, summary, raw_output, id, archived, metadata, path: projectPath } = body;
|
||||
const { content, summary, raw_output, id, archived, metadata, tags, path: projectPath } = body;
|
||||
|
||||
if (!content) {
|
||||
return { error: 'content is required', status: 400 };
|
||||
@@ -94,7 +105,8 @@ export async function handleCoreMemoryRoutes(ctx: RouteContext): Promise<boolean
|
||||
summary,
|
||||
raw_output,
|
||||
archived,
|
||||
metadata: metadata ? JSON.stringify(metadata) : undefined
|
||||
metadata: metadata ? JSON.stringify(metadata) : undefined,
|
||||
tags
|
||||
});
|
||||
|
||||
// Broadcast update event
|
||||
@@ -828,5 +840,48 @@ export async function handleCoreMemoryRoutes(ctx: RouteContext): Promise<boolean
|
||||
return true;
|
||||
}
|
||||
|
||||
// API: Get session summaries (list)
|
||||
if (pathname === '/api/core-memory/sessions/summaries' && req.method === 'GET') {
|
||||
const projectPath = url.searchParams.get('path') || initialPath;
|
||||
const limit = parseInt(url.searchParams.get('limit') || '20', 10);
|
||||
|
||||
try {
|
||||
const store = getCoreMemoryStore(projectPath);
|
||||
const summaries = store.getSessionSummaries(limit);
|
||||
|
||||
res.writeHead(200, { 'Content-Type': 'application/json' });
|
||||
res.end(JSON.stringify({ success: true, summaries }));
|
||||
} catch (error: unknown) {
|
||||
res.writeHead(500, { 'Content-Type': 'application/json' });
|
||||
res.end(JSON.stringify({ error: (error as Error).message }));
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
// API: Get single session summary by thread ID
|
||||
if (pathname.match(/^\/api\/core-memory\/sessions\/[^\/]+\/summary$/) && req.method === 'GET') {
|
||||
const parts = pathname.split('/');
|
||||
const threadId = parts[4]; // /api/core-memory/sessions/:id/summary
|
||||
const projectPath = url.searchParams.get('path') || initialPath;
|
||||
|
||||
try {
|
||||
const store = getCoreMemoryStore(projectPath);
|
||||
const summary = store.getSessionSummary(threadId);
|
||||
|
||||
if (!summary) {
|
||||
res.writeHead(404, { 'Content-Type': 'application/json' });
|
||||
res.end(JSON.stringify({ error: 'Session summary not found' }));
|
||||
return true;
|
||||
}
|
||||
|
||||
res.writeHead(200, { 'Content-Type': 'application/json' });
|
||||
res.end(JSON.stringify({ success: true, ...summary }));
|
||||
} catch (error: unknown) {
|
||||
res.writeHead(500, { 'Content-Type': 'application/json' });
|
||||
res.end(JSON.stringify({ error: (error as Error).message }));
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
@@ -91,10 +91,21 @@ export async function handleMemoryRoutes(ctx: RouteContext): Promise<boolean> {
|
||||
// API: Memory Module - Get all memories (core memory list)
|
||||
if (pathname === '/api/memory' && req.method === 'GET') {
|
||||
const projectPath = url.searchParams.get('path') || initialPath;
|
||||
const tagsParam = url.searchParams.get('tags');
|
||||
|
||||
try {
|
||||
const store = getCoreMemoryStore(projectPath);
|
||||
const memories = store.getMemories({ archived: false, limit: 100 });
|
||||
|
||||
// Use tag filter if tags query parameter is provided
|
||||
let memories;
|
||||
if (tagsParam) {
|
||||
const tags = tagsParam.split(',').map(t => t.trim()).filter(Boolean);
|
||||
memories = tags.length > 0
|
||||
? store.getMemoriesByTags(tags, { archived: false, limit: 100 })
|
||||
: store.getMemories({ archived: false, limit: 100 });
|
||||
} else {
|
||||
memories = store.getMemories({ archived: false, limit: 100 });
|
||||
}
|
||||
|
||||
// Calculate total size
|
||||
const totalSize = memories.reduce((sum, m) => sum + (m.content?.length || 0), 0);
|
||||
@@ -109,7 +120,7 @@ export async function handleMemoryRoutes(ctx: RouteContext): Promise<boolean> {
|
||||
createdAt: m.created_at,
|
||||
updatedAt: m.updated_at,
|
||||
source: m.metadata || undefined,
|
||||
tags: [], // TODO: Extract tags from metadata if available
|
||||
tags: m.tags || [],
|
||||
size: m.content?.length || 0
|
||||
}));
|
||||
|
||||
@@ -139,7 +150,7 @@ export async function handleMemoryRoutes(ctx: RouteContext): Promise<boolean> {
|
||||
|
||||
try {
|
||||
const store = getCoreMemoryStore(basePath);
|
||||
const memory = store.upsertMemory({ content });
|
||||
const memory = store.upsertMemory({ content, tags });
|
||||
|
||||
// Broadcast update event
|
||||
broadcastToClients({
|
||||
@@ -156,7 +167,7 @@ export async function handleMemoryRoutes(ctx: RouteContext): Promise<boolean> {
|
||||
createdAt: memory.created_at,
|
||||
updatedAt: memory.updated_at,
|
||||
source: memory.metadata || undefined,
|
||||
tags: tags || [],
|
||||
tags: memory.tags || [],
|
||||
size: memory.content?.length || 0
|
||||
};
|
||||
} catch (error: unknown) {
|
||||
@@ -175,7 +186,7 @@ export async function handleMemoryRoutes(ctx: RouteContext): Promise<boolean> {
|
||||
|
||||
try {
|
||||
const store = getCoreMemoryStore(basePath);
|
||||
const memory = store.upsertMemory({ id: memoryId, content });
|
||||
const memory = store.upsertMemory({ id: memoryId, content, tags });
|
||||
|
||||
// Broadcast update event
|
||||
broadcastToClients({
|
||||
@@ -192,7 +203,7 @@ export async function handleMemoryRoutes(ctx: RouteContext): Promise<boolean> {
|
||||
createdAt: memory.created_at,
|
||||
updatedAt: memory.updated_at,
|
||||
source: memory.metadata || undefined,
|
||||
tags: tags || [],
|
||||
tags: memory.tags || [],
|
||||
size: memory.content?.length || 0
|
||||
};
|
||||
} catch (error: unknown) {
|
||||
|
||||
@@ -7,7 +7,7 @@
|
||||
* - session-end: task generation for async background processing
|
||||
*
|
||||
* Character limits:
|
||||
* - session-start: <= 1000 chars
|
||||
* - session-start: <= 1500 chars
|
||||
* - per-prompt: <= 500 chars
|
||||
*/
|
||||
|
||||
@@ -19,7 +19,7 @@ import { SessionStateService, type SessionState } from './session-state-service.
|
||||
// =============================================================================
|
||||
|
||||
/** Maximum character count for session-start context */
|
||||
const SESSION_START_LIMIT = 1000;
|
||||
const SESSION_START_LIMIT = 1500;
|
||||
|
||||
/** Maximum character count for per-prompt context */
|
||||
const PER_PROMPT_LIMIT = 500;
|
||||
|
||||
@@ -7,7 +7,7 @@
|
||||
* - session-end: incremental embedding + clustering + heat score update tasks
|
||||
*
|
||||
* Character limits:
|
||||
* - session-start: <= 1000 chars
|
||||
* - session-start: <= 1500 chars
|
||||
* - per-prompt: <= 500 chars
|
||||
*/
|
||||
|
||||
@@ -29,7 +29,7 @@ import { SessionClusteringService } from './session-clustering-service.js';
|
||||
// =============================================================================
|
||||
|
||||
/** Maximum character count for session-start context */
|
||||
const SESSION_START_LIMIT = 1000;
|
||||
const SESSION_START_LIMIT = 1500;
|
||||
|
||||
/** Maximum character count for per-prompt context */
|
||||
const PER_PROMPT_LIMIT = 500;
|
||||
@@ -52,6 +52,12 @@ const VECTOR_TOP_K = 8;
|
||||
/** Minimum vector similarity score */
|
||||
const VECTOR_MIN_SCORE = 0.3;
|
||||
|
||||
/** Maximum characters for the recent sessions component */
|
||||
const RECENT_SESSIONS_LIMIT = 300;
|
||||
|
||||
/** Number of recent sessions to show */
|
||||
const RECENT_SESSIONS_COUNT = 5;
|
||||
|
||||
// =============================================================================
|
||||
// Types
|
||||
// =============================================================================
|
||||
@@ -84,11 +90,12 @@ export class UnifiedContextBuilder {
|
||||
/**
|
||||
* Build context for session-start hook injection.
|
||||
*
|
||||
* Components (assembled in order, truncated to <= 1000 chars total):
|
||||
* Components (assembled in order, truncated to <= 1500 chars total):
|
||||
* 1. MEMORY.md summary (up to 500 chars)
|
||||
* 2. Cluster overview (top 3 active clusters)
|
||||
* 3. Hot entities (top 5 within last 7 days)
|
||||
* 4. Solidified patterns (skills/*.md file list)
|
||||
* 5. Recent sessions (last 5 session summaries)
|
||||
*/
|
||||
async buildSessionStartContext(): Promise<string> {
|
||||
const sections: string[] = [];
|
||||
@@ -117,6 +124,12 @@ export class UnifiedContextBuilder {
|
||||
sections.push(patterns);
|
||||
}
|
||||
|
||||
// Component 5: Recent sessions
|
||||
const recentSessions = await this.buildRecentSessions();
|
||||
if (recentSessions) {
|
||||
sections.push(recentSessions);
|
||||
}
|
||||
|
||||
if (sections.length === 0) {
|
||||
return '';
|
||||
}
|
||||
@@ -285,8 +298,8 @@ export class UnifiedContextBuilder {
|
||||
*/
|
||||
private async buildClusterOverview(): Promise<string> {
|
||||
try {
|
||||
const { CoreMemoryStore } = await import('./core-memory-store.js');
|
||||
const store = new CoreMemoryStore(this.projectPath);
|
||||
const { getCoreMemoryStore } = await import('./core-memory-store.js');
|
||||
const store = getCoreMemoryStore(this.projectPath);
|
||||
const clusters = store.listClusters('active');
|
||||
|
||||
if (clusters.length === 0) {
|
||||
@@ -380,6 +393,39 @@ export class UnifiedContextBuilder {
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Build recent sessions component.
|
||||
* Shows last N session summaries from CoreMemoryStore.
|
||||
*/
|
||||
private async buildRecentSessions(): Promise<string> {
|
||||
try {
|
||||
const { getCoreMemoryStore } = await import('./core-memory-store.js');
|
||||
const store = getCoreMemoryStore(this.projectPath);
|
||||
const summaries = store.getSessionSummaries(RECENT_SESSIONS_COUNT);
|
||||
|
||||
if (summaries.length === 0) {
|
||||
return '';
|
||||
}
|
||||
|
||||
let output = '## Recent Sessions\n';
|
||||
for (const s of summaries) {
|
||||
const shortId = s.thread_id.substring(0, 12);
|
||||
const summaryText = s.rollout_summary.length > 60
|
||||
? s.rollout_summary.substring(0, 60) + '...'
|
||||
: s.rollout_summary;
|
||||
output += `- ${shortId}: ${summaryText}\n`;
|
||||
}
|
||||
|
||||
if (output.length > RECENT_SESSIONS_LIMIT) {
|
||||
output = output.substring(0, RECENT_SESSIONS_LIMIT);
|
||||
}
|
||||
|
||||
return output;
|
||||
} catch {
|
||||
return '';
|
||||
}
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------
|
||||
// Private: Formatting helpers
|
||||
// ---------------------------------------------------------------------------
|
||||
|
||||
Reference in New Issue
Block a user