diff --git a/src/tools/memories.ts b/src/tools/memories.ts index 84930d0..f1c069e 100644 --- a/src/tools/memories.ts +++ b/src/tools/memories.ts @@ -45,7 +45,18 @@ interface MemoryListArgs { * Add a new memory/learning (enhanced with session_id and task_id) */ export async function memoryAdd(args: MemoryAddArgs): Promise { - const { category, title, content, context, project, session_id, task_id } = args; + let { category, title, content, context, project, session_id, task_id } = args; + + // Validate session_id exists if provided (graceful degradation) + if (session_id) { + const sessionExists = await queryOne<{ exists: boolean }>( + 'SELECT EXISTS(SELECT 1 FROM sessions WHERE id = $1)', + [session_id] + ); + if (!sessionExists?.exists) { + session_id = undefined; // Set to null if session doesn't exist + } + } // Generate embedding for semantic search const embedText = `${title}. ${content}`;