squish-memory 1.0.2 → 1.2.0

package/dist/core/mcp/tools.d.ts

@@ -2,8 +2,6 @@ import { MCPToolDefinition } from './types.js';
 export declare const squishSearchTool: MCPToolDefinition;
 export declare const squishRememberTool: MCPToolDefinition;
 export declare const squishRecallTool: MCPToolDefinition;
-export declare const squishEmbedTool: MCPToolDefinition;
-export declare const squishQMDSearchTool: MCPToolDefinition;
 export declare const squishHealthTool: MCPToolDefinition;
 export declare function getAllSquishTools(): MCPToolDefinition[];
 //# sourceMappingURL=tools.d.ts.map

package/dist/core/mcp/tools.js

@@ -1,6 +1,5 @@
 import { hybridSearch } from '../memory/hybrid-retrieval.js';
-import { rememberMemory, getMemoryById } from '../memory/memories.js';
-import { getEmbedding } from '../embeddings.js';
+import { rememberMemory, getMemory } from '../memory/memories.js';
 import { getQMDClient } from '../embeddings/qmd-client.js';
 import { logger } from '../logger.js';
 function textResult(text) {
@@ -83,13 +82,39 @@ export const squishRememberTool = {
                     type: 'string',
                     description: 'Project path',
                 },
+                // Rich context fields (Agent 4 feedback)
+                source: {
+                    type: 'string',
+                    description: 'Source of this memory (e.g., "voice", "chat", "document")',
+                },
+                reasoning: {
+                    type: 'string',
+                    description: 'Why this memory is important',
+                },
+                context: {
+                    type: 'string',
+                    description: 'What triggered this memory',
+                },
+                examples: {
+                    type: 'string',
+                    description: 'When to apply this knowledge',
+                },
+                exceptions: {
+                    type: 'string',
+                    description: 'When NOT to apply this',
+                },
+                tier: {
+                    type: 'string',
+                    description: 'Memory tier: hot (active) or cold (archived)',
+                    enum: ['hot', 'cold'],
+                },
             },
             required: ['content'],
         },
     },
     handler: async (args) => {
         try {
-            const { content, type = 'observation', tags = [], project } = args;
+            const { content, type = 'observation', tags = [], project, source, reasoning, context, examples, exceptions, tier = 'hot' } = args;
             if (!content) {
                 return errorResult('Content is required');
             }
@@ -98,6 +123,12 @@ export const squishRememberTool = {
                 type,
                 tags,
                 project,
+                source,
+                reasoning,
+                memoryContext: context,
+                examples,
+                exceptions,
+                tier,
             });
             return textResult(`Memory stored: ${memory.id}`);
         }
@@ -128,7 +159,7 @@ export const squishRecallTool = {
             if (!memoryId) {
                 return errorResult('Memory ID is required');
             }
-            const memory = await getMemoryById(memoryId);
+            const memory = await getMemory(memoryId);
             if (!memory) {
                 return errorResult('Memory not found');
             }
@@ -140,90 +171,6 @@ export const squishRecallTool = {
         }
     },
 };
-export const squishEmbedTool = {
-    tool: {
-        name: 'squish_embed',
-        description: 'Generate embeddings for text using configured provider (supports multimodal)',
-        inputSchema: {
-            type: 'object',
-            properties: {
-                text: {
-                    type: 'string',
-                    description: 'Text to embed',
-                },
-            },
-            required: ['text'],
-        },
-    },
-    handler: async (args) => {
-        try {
-            const { text } = args;
-            if (!text) {
-                return errorResult('Text is required');
-            }
-            const embedding = await getEmbedding(text);
-            if (!embedding) {
-                return errorResult('Failed to generate embedding');
-            }
-            return textResult(JSON.stringify({
-                dimensions: embedding.length,
-                preview: embedding.slice(0, 5),
-            }, null, 2));
-        }
-        catch (error) {
-            logger.error('Embed error:', error);
-            return errorResult(error instanceof Error ? error.message : 'Failed to generate embedding');
-        }
-    },
-};
-export const squishQMDSearchTool = {
-    tool: {
-        name: 'squish_qmd_search',
-        description: 'Search markdown files using QMD (local, fast BM25 + vector)',
-        inputSchema: {
-            type: 'object',
-            properties: {
-                query: {
-                    type: 'string',
-                    description: 'Search query',
-                },
-                collection: {
-                    type: 'string',
-                    description: 'QMD collection name (optional)',
-                },
-                limit: {
-                    type: 'number',
-                    description: 'Maximum results (default: 10)',
-                },
-            },
-            required: ['query'],
-        },
-    },
-    handler: async (args) => {
-        try {
-            const { query, collection, limit = 10 } = args;
-            if (!query) {
-                return errorResult('Query is required');
-            }
-            const client = await getQMDClient();
-            const available = await client.isAvailable();
-            if (!available) {
-                return errorResult('QMD not available');
-            }
-            const results = await client.search({
-                query,
-                collection,
-                limit: limit || 10
-            });
-            const formatted = results.map((r, i) => `${i + 1}. ${r.path || r.file} (score: ${r.score?.toFixed(2)})\n   ${r.content?.substring(0, 150)}...`).join('\n\n');
-            return textResult(`QMD found ${results.length} results:\n\n${formatted}`);
-        }
-        catch (error) {
-            logger.error('QMD search error:', error);
-            return errorResult(error instanceof Error ? error.message : 'QMD search failed');
-        }
-    },
-};
 export const squishHealthTool = {
     tool: {
         name: 'squish_health',
@@ -323,8 +270,6 @@ export function getAllSquishTools() {
         squishSearchTool,
         squishRememberTool,
         squishRecallTool,
-        squishEmbedTool,
-        squishQMDSearchTool,
         squishHealthTool,
         squishGetSearchTracesTool,
         squishGetTraceByIdTool,

package/dist/core/mcp/types.d.ts

@@ -3,302 +3,74 @@ export declare const MCPToolSchema: z.ZodObject<{
     name: z.ZodString;
     description: z.ZodString;
     inputSchema: z.ZodRecord<z.ZodString, z.ZodAny>;
-}, "strip", z.ZodTypeAny, {
-    name: string;
-    description: string;
-    inputSchema: Record<string, any>;
-}, {
-    name: string;
-    description: string;
-    inputSchema: Record<string, any>;
-}>;
+}, z.core.$strip>;
 export declare const MCPToolCallSchema: z.ZodObject<{
     name: z.ZodString;
     arguments: z.ZodRecord<z.ZodString, z.ZodAny>;
-}, "strip", z.ZodTypeAny, {
-    name: string;
-    arguments: Record<string, any>;
-}, {
-    name: string;
-    arguments: Record<string, any>;
-}>;
+}, z.core.$strip>;
 export declare const MCPToolResultSchema: z.ZodObject<{
     content: z.ZodArray<z.ZodObject<{
-        type: z.ZodEnum<["text", "image", "resource"]>;
+        type: z.ZodEnum<{
+            text: "text";
+            image: "image";
+            resource: "resource";
+        }>;
         text: z.ZodOptional<z.ZodString>;
         data: z.ZodOptional<z.ZodString>;
         mimeType: z.ZodOptional<z.ZodString>;
-    }, "strip", z.ZodTypeAny, {
-        type: "text" | "image" | "resource";
-        text?: string | undefined;
-        data?: string | undefined;
-        mimeType?: string | undefined;
-    }, {
-        type: "text" | "image" | "resource";
-        text?: string | undefined;
-        data?: string | undefined;
-        mimeType?: string | undefined;
-    }>, "many">;
+    }, z.core.$strip>>;
     isError: z.ZodOptional<z.ZodBoolean>;
-}, "strip", z.ZodTypeAny, {
-    content: {
-        type: "text" | "image" | "resource";
-        text?: string | undefined;
-        data?: string | undefined;
-        mimeType?: string | undefined;
-    }[];
-    isError?: boolean | undefined;
-}, {
-    content: {
-        type: "text" | "image" | "resource";
-        text?: string | undefined;
-        data?: string | undefined;
-        mimeType?: string | undefined;
-    }[];
-    isError?: boolean | undefined;
-}>;
+}, z.core.$strip>;
 export declare const MCPCapabilitiesSchema: z.ZodObject<{
     tools: z.ZodOptional<z.ZodObject<{
         listChanged: z.ZodOptional<z.ZodBoolean>;
-    }, "strip", z.ZodTypeAny, {
-        listChanged?: boolean | undefined;
-    }, {
-        listChanged?: boolean | undefined;
-    }>>;
+    }, z.core.$strip>>;
     resources: z.ZodOptional<z.ZodObject<{
         subscribe: z.ZodOptional<z.ZodBoolean>;
         listChanged: z.ZodOptional<z.ZodBoolean>;
-    }, "strip", z.ZodTypeAny, {
-        listChanged?: boolean | undefined;
-        subscribe?: boolean | undefined;
-    }, {
-        listChanged?: boolean | undefined;
-        subscribe?: boolean | undefined;
-    }>>;
+    }, z.core.$strip>>;
     prompts: z.ZodOptional<z.ZodObject<{
         listChanged: z.ZodOptional<z.ZodBoolean>;
-    }, "strip", z.ZodTypeAny, {
-        listChanged?: boolean | undefined;
-    }, {
-        listChanged?: boolean | undefined;
-    }>>;
-}, "strip", z.ZodTypeAny, {
-    tools?: {
-        listChanged?: boolean | undefined;
-    } | undefined;
-    prompts?: {
-        listChanged?: boolean | undefined;
-    } | undefined;
-    resources?: {
-        listChanged?: boolean | undefined;
-        subscribe?: boolean | undefined;
-    } | undefined;
-}, {
-    tools?: {
-        listChanged?: boolean | undefined;
-    } | undefined;
-    prompts?: {
-        listChanged?: boolean | undefined;
-    } | undefined;
-    resources?: {
-        listChanged?: boolean | undefined;
-        subscribe?: boolean | undefined;
-    } | undefined;
-}>;
+    }, z.core.$strip>>;
+}, z.core.$strip>;
 export declare const MCPInitializeRequestSchema: z.ZodObject<{
     protocolVersion: z.ZodString;
     capabilities: z.ZodObject<{
         tools: z.ZodOptional<z.ZodObject<{
             listChanged: z.ZodOptional<z.ZodBoolean>;
-        }, "strip", z.ZodTypeAny, {
-            listChanged?: boolean | undefined;
-        }, {
-            listChanged?: boolean | undefined;
-        }>>;
+        }, z.core.$strip>>;
         resources: z.ZodOptional<z.ZodObject<{
             subscribe: z.ZodOptional<z.ZodBoolean>;
             listChanged: z.ZodOptional<z.ZodBoolean>;
-        }, "strip", z.ZodTypeAny, {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        }, {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        }>>;
+        }, z.core.$strip>>;
         prompts: z.ZodOptional<z.ZodObject<{
             listChanged: z.ZodOptional<z.ZodBoolean>;
-        }, "strip", z.ZodTypeAny, {
-            listChanged?: boolean | undefined;
-        }, {
-            listChanged?: boolean | undefined;
-        }>>;
-    }, "strip", z.ZodTypeAny, {
-        tools?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        prompts?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        resources?: {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        } | undefined;
-    }, {
-        tools?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        prompts?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        resources?: {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        } | undefined;
-    }>;
+        }, z.core.$strip>>;
+    }, z.core.$strip>;
     clientInfo: z.ZodObject<{
         name: z.ZodString;
         version: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        version: string;
-        name: string;
-    }, {
-        version: string;
-        name: string;
-    }>;
-}, "strip", z.ZodTypeAny, {
-    clientInfo: {
-        version: string;
-        name: string;
-    };
-    capabilities: {
-        tools?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        prompts?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        resources?: {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        } | undefined;
-    };
-    protocolVersion: string;
-}, {
-    clientInfo: {
-        version: string;
-        name: string;
-    };
-    capabilities: {
-        tools?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        prompts?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        resources?: {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        } | undefined;
-    };
-    protocolVersion: string;
-}>;
+    }, z.core.$strip>;
+}, z.core.$strip>;
 export declare const MCPInitializeResponseSchema: z.ZodObject<{
     protocolVersion: z.ZodString;
     capabilities: z.ZodObject<{
         tools: z.ZodOptional<z.ZodObject<{
             listChanged: z.ZodOptional<z.ZodBoolean>;
-        }, "strip", z.ZodTypeAny, {
-            listChanged?: boolean | undefined;
-        }, {
-            listChanged?: boolean | undefined;
-        }>>;
+        }, z.core.$strip>>;
         resources: z.ZodOptional<z.ZodObject<{
             subscribe: z.ZodOptional<z.ZodBoolean>;
             listChanged: z.ZodOptional<z.ZodBoolean>;
-        }, "strip", z.ZodTypeAny, {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        }, {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        }>>;
+        }, z.core.$strip>>;
         prompts: z.ZodOptional<z.ZodObject<{
             listChanged: z.ZodOptional<z.ZodBoolean>;
-        }, "strip", z.ZodTypeAny, {
-            listChanged?: boolean | undefined;
-        }, {
-            listChanged?: boolean | undefined;
-        }>>;
-    }, "strip", z.ZodTypeAny, {
-        tools?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        prompts?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        resources?: {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        } | undefined;
-    }, {
-        tools?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        prompts?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        resources?: {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        } | undefined;
-    }>;
+        }, z.core.$strip>>;
+    }, z.core.$strip>;
     serverInfo: z.ZodObject<{
         name: z.ZodString;
         version: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        version: string;
-        name: string;
-    }, {
-        version: string;
-        name: string;
-    }>;
-}, "strip", z.ZodTypeAny, {
-    capabilities: {
-        tools?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        prompts?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        resources?: {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        } | undefined;
-    };
-    protocolVersion: string;
-    serverInfo: {
-        version: string;
-        name: string;
-    };
-}, {
-    capabilities: {
-        tools?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        prompts?: {
-            listChanged?: boolean | undefined;
-        } | undefined;
-        resources?: {
-            listChanged?: boolean | undefined;
-            subscribe?: boolean | undefined;
-        } | undefined;
-    };
-    protocolVersion: string;
-    serverInfo: {
-        version: string;
-        name: string;
-    };
-}>;
+    }, z.core.$strip>;
+}, z.core.$strip>;
 export type MCPTool = z.infer<typeof MCPToolSchema>;
 export type MCPToolCall = z.infer<typeof MCPToolCallSchema>;
 export type MCPToolResult = z.infer<typeof MCPToolResultSchema>;

package/dist/core/mcp/types.js

@@ -2,11 +2,11 @@ import { z } from 'zod';
 export const MCPToolSchema = z.object({
     name: z.string(),
     description: z.string(),
-    inputSchema: z.record(z.any()),
+    inputSchema: z.record(z.string(), z.any()),
 });
 export const MCPToolCallSchema = z.object({
     name: z.string(),
-    arguments: z.record(z.any()),
+    arguments: z.record(z.string(), z.any()),
 });
 export const MCPToolResultSchema = z.object({
     content: z.array(z.object({

package/dist/core/memory/categorizer.js

@@ -80,6 +80,8 @@ export async function categorizeMemory(content, existingTags = []) {
         decision: 0,
         context: 0,
         preference: 0,
+        note: 0,
+        task: 0,
     };
     for (const { type, patterns, weight } of MEMORY_TYPE_PATTERNS) {
         for (const pattern of patterns) {

package/dist/core/memory/conflict-detector.js

@@ -1,7 +1,7 @@
 import { eq, and, sql } from 'drizzle-orm';
 import { getDb } from '../../db/index.js';
 import { getSchema } from '../../db/schema.js';
-import { createDatabaseClient } from '../../core/database.js';
+import { createDatabaseClient } from '../storage/database.js';
 export async function detectConflicts(memoryId, proposedContent) {
     const db = createDatabaseClient(await getDb());
     const schema = await getSchema();

package/dist/core/memory/consolidation.d.ts

@@ -1,15 +1,6 @@
 /**
  * Memory Consolidation System
- *
- * Implements experience replay and memory consolidation for Squish.
- * Based on research from ReMe (Memory Replay) and LIGHT framework.
- *
- * Consolidation process:
- * 1. Find clusters of old, low-importance memories
- * 2. Group related memories by similarity
- * 3. Summarize each cluster using extractive summarization
- * 4. Create consolidated memory with summary
- * 5. Mark originals as consolidated (not deleted - reversible)
+ * Implements experience replay and memory consolidation
  */
 export interface ConsolidationOptions {
     projectId: string;

package/dist/core/memory/consolidation.js

@@ -1,21 +1,13 @@
 /**
  * Memory Consolidation System
- *
- * Implements experience replay and memory consolidation for Squish.
- * Based on research from ReMe (Memory Replay) and LIGHT framework.
- *
- * Consolidation process:
- * 1. Find clusters of old, low-importance memories
- * 2. Group related memories by similarity
- * 3. Summarize each cluster using extractive summarization
- * 4. Create consolidated memory with summary
- * 5. Mark originals as consolidated (not deleted - reversible)
+ * Implements experience replay and memory consolidation
  */
 import { eq, inArray } from 'drizzle-orm';
-import { createDatabaseClient } from '../database.js';
+import { createDatabaseClient } from '../storage/database.js';
 import { getDb } from '../../db/index.js';
 import { getSchema } from '../../db/schema.js';
-import { cosineSimilarity } from './importance.js';
+import { cosineSimilarity } from '../utils/vector-operations.js';
+import { parseEmbedding } from '../lib/parse-embedding.js';
 import { getLowImportanceMemories } from './importance.js';
 import { rememberMemory } from './memories.js';
 import { logger } from '../logger.js';
@@ -107,33 +99,6 @@ async function calculateMemorySimilarity(memory1, memory2) {
     }
     return cosineSimilarity(embedding1, embedding2);
 }
-/**
- * Parse embedding from storage
- */
-function parseEmbedding(embeddingData) {
-    if (!embeddingData)
-        return null;
-    if (Array.isArray(embeddingData))
-        return embeddingData;
-    if (typeof embeddingData === 'string') {
-        try {
-            return JSON.parse(embeddingData);
-        }
-        catch {
-            return null;
-        }
-    }
-    if (Buffer.isBuffer(embeddingData)) {
-        try {
-            const floatArray = new Float32Array(embeddingData.buffer || embeddingData);
-            return Array.from(floatArray);
-        }
-        catch {
-            return null;
-        }
-    }
-    return null;
-}
 /**
  * Simple text similarity as fallback (Jaccard similarity of word sets)
  */

package/dist/core/memory/context-collector.js

@@ -1,7 +1,7 @@
 /** Context Collector - Collect recent conversation context for query rewriting */
 import { logger } from '../logger.js';
 import { getDb } from '../../db/index.js';
-import { messages, conversations } from '../../drizzle/schema-sqlite.js';
+import { messages, conversations } from '../../db/drizzle/schema-sqlite.js';
 import { eq, desc } from 'drizzle-orm';
 export async function collectRecentContext(sessionId, count = 5) {
     const db = await getDb();

package/dist/core/memory/edit-workflow.js

@@ -2,7 +2,7 @@ import { randomUUID } from 'crypto';
 import { eq, and, sql } from 'drizzle-orm';
 import { getDb } from '../../db/index.js';
 import { getSchema } from '../../db/schema.js';
-import { createDatabaseClient } from '../../core/database.js';
+import { createDatabaseClient } from '../storage/database.js';
 export async function createEditProposal(memoryId, currentContent, proposedContent, reason, userId) {
     const db = createDatabaseClient(await getDb());
     const schema = await getSchema();

package/dist/core/memory/entity-extractor.d.ts

@@ -13,6 +13,10 @@ export interface ExtractedEntity {
     context: string;
     normalized?: string;
 }
+/**
+ * Extract unique entity names for auto-linking
+ */
+export declare function extractEntityNames(content: string): string[];
 /**
  * Extract entities from content
  */