cozo-memory 1.0.6 → 1.0.8

package/dist/test-discovery.js

@@ -0,0 +1,53 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const index_1 = require("../src/index");
+const path_1 = __importDefault(require("path"));
+const fs_1 = __importDefault(require("fs"));
+async function runTest() {
+    const testDbPath = path_1.default.resolve(__dirname, "discovery_test.cozo");
+    // Cleanup previous test DB
+    if (fs_1.default.existsSync(testDbPath + ".db"))
+        fs_1.default.unlinkSync(testDbPath + ".db");
+    const server = new index_1.MemoryServer(testDbPath);
+    await server.initPromise;
+    console.log("1. Creating test entities...");
+    const aliceRes = await server.createEntity({
+        name: "Alice",
+        type: "Person",
+        metadata: { role: "Developer" }
+    });
+    const aliceId = aliceRes.id;
+    const phoenixRes = await server.createEntity({
+        name: "Project Phoenix",
+        type: "Project",
+        metadata: { status: "active" }
+    });
+    const phoenixId = phoenixRes.id;
+    console.log("2. Adding overlapping observations...");
+    await server.addObservation({
+        entity_id: aliceId,
+        text: "Alice is currently focusing all her time on Project Phoenix."
+    });
+    console.log("3. Running reflection in 'discovery' mode...");
+    // We specify the model to ensure it uses one that supports JSON mode if possible
+    const reflectRes = await server.reflectMemory({
+        entity_id: aliceId,
+        mode: "discovery"
+    });
+    console.log("Result:", JSON.stringify(reflectRes, null, 2));
+    console.log("4. Verifying relationship creation...");
+    const relationsRes = await server.db.run(`
+    ?[to_name, rel_type] := *relationship{from_id, to_id, relation_type: rel_type, @ "NOW"},
+      from_id = $alice,
+      *entity{id: to_id, name: to_name, @ "NOW"}
+  `, { alice: aliceId });
+    console.log("Relationships from Alice:", relationsRes.rows);
+    // Cleanup
+    if (fs_1.default.existsSync(testDbPath + ".db"))
+        fs_1.default.unlinkSync(testDbPath + ".db");
+    process.exit(0);
+}
+runTest().catch(console.error);

package/dist/test-graphrag.js

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const index_1 = require("./index");
+async function run() {
+    const server = new index_1.MemoryServer();
+    await server.initPromise;
+    console.log("Setting up Test Clusters for GraphRAG Community Summaries...");
+    // Cluster 1: Frontend
+    const fe1 = await server.createEntity({ name: "ReactJS", type: "technology", metadata: {} });
+    const fe2 = await server.createEntity({ name: "Redux", type: "technology", metadata: {} });
+    const fe3 = await server.createEntity({ name: "Frontend Project A", type: "project", metadata: {} });
+    await server.addObservation({ entity_id: fe1.id, text: "ReactJS is used for building UIs." });
+    await server.addObservation({ entity_id: fe2.id, text: "Redux is used for state management in React." });
+    await server.addObservation({ entity_id: fe3.id, text: "Project A is a heavy frontend SPA using React and Redux." });
+    await server.createRelation({ from_id: fe3.id, to_id: fe1.id, relation_type: "uses", strength: 1.0 });
+    await server.createRelation({ from_id: fe3.id, to_id: fe2.id, relation_type: "uses", strength: 1.0 });
+    await server.createRelation({ from_id: fe1.id, to_id: fe2.id, relation_type: "integrates_with", strength: 1.0 });
+    console.log("Created Frontend Cluster");
+    // Cluster 2: Backend
+    const be1 = await server.createEntity({ name: "PostgreSQL", type: "database", metadata: {} });
+    const be2 = await server.createEntity({ name: "CozoDB", type: "database", metadata: {} });
+    const be3 = await server.createEntity({ name: "Backend Project B", type: "project", metadata: {} });
+    await server.addObservation({ entity_id: be1.id, text: "PostgreSQL is a robust relational database." });
+    await server.addObservation({ entity_id: be2.id, text: "CozoDB is a graph database with Datalog." });
+    await server.addObservation({ entity_id: be3.id, text: "Project B relies heavily on complex queries across PG and CozoDB." });
+    await server.createRelation({ from_id: be3.id, to_id: be1.id, relation_type: "uses", strength: 1.0 });
+    await server.createRelation({ from_id: be3.id, to_id: be2.id, relation_type: "uses", strength: 1.0 });
+    await server.createRelation({ from_id: be1.id, to_id: be2.id, relation_type: "migrating_to", strength: 1.0 });
+    console.log("Created Backend Cluster");
+    console.log("Initiating Community Summarization...");
+    const result = await server.summarizeCommunities({ min_community_size: 3 });
+    console.log("Community Summarization Result:", JSON.stringify(result, null, 2));
+    process.exit(0);
+}
+run().catch(console.error);

package/dist/test-user-profile.js

@@ -0,0 +1,59 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const index_1 = require("./index");
+async function testUserProfileEditing() {
+    console.log("=== Testing User Profile Editing ===\n");
+    const server = new index_1.MemoryServer();
+    await server.start();
+    try {
+        // 1. View current profile
+        console.log("1. Current user profile:");
+        const currentProfile = await server.editUserProfile({});
+        console.log(JSON.stringify(currentProfile, null, 2));
+        // 2. Update metadata
+        console.log("\n2. Updating profile metadata:");
+        const metadataUpdate = await server.editUserProfile({
+            metadata: {
+                timezone: "Europe/Berlin",
+                language: "de",
+                work_hours: "09:00-17:00"
+            }
+        });
+        console.log(JSON.stringify(metadataUpdate, null, 2));
+        // 3. Add new preferences
+        console.log("\n3. Adding new preferences:");
+        const addPrefs = await server.editUserProfile({
+            observations: [
+                { text: "Prefers TypeScript over JavaScript for type safety" },
+                { text: "Likes clean, minimal code without unnecessary comments" },
+                { text: "Prefers functional programming patterns" }
+            ]
+        });
+        console.log(JSON.stringify(addPrefs, null, 2));
+        // 4. Clear and reset preferences
+        console.log("\n4. Clearing and resetting preferences:");
+        const resetPrefs = await server.editUserProfile({
+            clear_observations: true,
+            observations: [
+                { text: "Works primarily with Node.js and TypeScript", metadata: { category: "tech_stack" } },
+                { text: "Prefers concise documentation", metadata: { category: "style" } }
+            ]
+        });
+        console.log(JSON.stringify(resetPrefs, null, 2));
+        // 5. Update name and type
+        console.log("\n5. Updating profile name and type:");
+        const nameUpdate = await server.editUserProfile({
+            name: "Developer Profile",
+            type: "UserProfile"
+        });
+        console.log(JSON.stringify(nameUpdate, null, 2));
+        console.log("\n=== Test completed successfully ===");
+    }
+    catch (error) {
+        console.error("Test failed:", error);
+    }
+    finally {
+        process.exit(0);
+    }
+}
+testUserProfileEditing();

package/dist/tui.js

@@ -63,13 +63,13 @@ async function main() {
                     else if (key.downArrow) {
                         setState(prev => ({
                             ...prev,
-                            selectedIndex: Math.min(5, prev.selectedIndex + 1)
+                            selectedIndex: Math.min(6, prev.selectedIndex + 1)
                         }));
                     }
                     else if (key.return) {
-                        const screens = ['entity', 'search', 'graph', 'system', 'result', 'menu'];
+                        const screens = ['entity', 'profile', 'search', 'graph', 'system', 'result', 'menu'];
                         const newScreen = screens[state.selectedIndex];
-                        if (state.selectedIndex === 4) {
+                        if (state.selectedIndex === 5) {
                             setState(prev => ({ ...prev, loading: true }));
                             cli.health().then(result => {
                                 setState(prev => ({ ...prev, screen: 'result', result, loading: false }));
@@ -89,7 +89,7 @@ async function main() {
             if (state.error) {
                 return React.createElement(Box, { flexDirection: 'column', padding: 1 }, React.createElement(Text, { color: 'red' }, `❌ Error: ${state.error}`), React.createElement(Text, { dimColor: true }, 'Press ESC to return to menu'));
             }
-            return React.createElement(Box, { flexDirection: 'column', padding: 1 }, React.createElement(Header), state.screen === 'menu' && React.createElement(MainMenu, { selectedIndex: state.selectedIndex }), state.screen === 'entity' && React.createElement(EntityScreen), state.screen === 'search' && React.createElement(SearchScreen), state.screen === 'graph' && React.createElement(GraphScreen), state.screen === 'system' && React.createElement(SystemScreen), state.screen === 'result' && React.createElement(ResultScreen, { result: state.result }), React.createElement(Footer));
+            return React.createElement(Box, { flexDirection: 'column', padding: 1 }, React.createElement(Header), state.screen === 'menu' && React.createElement(MainMenu, { selectedIndex: state.selectedIndex }), state.screen === 'entity' && React.createElement(EntityScreen), state.screen === 'profile' && React.createElement(ProfileScreen), state.screen === 'search' && React.createElement(SearchScreen), state.screen === 'graph' && React.createElement(GraphScreen), state.screen === 'system' && React.createElement(SystemScreen), state.screen === 'result' && React.createElement(ResultScreen, { result: state.result }), React.createElement(Footer));
         };
         const Header = () => React.createElement(Box, {
             flexDirection: 'column',
@@ -107,6 +107,7 @@ async function main() {
         const MainMenu = ({ selectedIndex }) => {
             const menuItems = [
                 { icon: '📦', label: 'Entity Operations', desc: 'Create, read, update, delete entities' },
+                { icon: '👤', label: 'User Profile', desc: 'Manage user preferences and profile' },
                 { icon: '🔍', label: 'Search & Context', desc: 'Hybrid search, context retrieval' },
                 { icon: '🕸️', label: 'Graph Operations', desc: 'Explore, PageRank, communities' },
                 { icon: '⚙️', label: 'System Management', desc: 'Health, metrics, export/import' },
@@ -116,6 +117,7 @@ async function main() {
             return React.createElement(Box, { flexDirection: 'column' }, React.createElement(Box, { marginBottom: 1 }, React.createElement(Text, { bold: true, color: 'yellow' }, 'Main Menu')), ...menuItems.map((item, index) => React.createElement(Box, { key: index, marginLeft: 2 }, React.createElement(Text, { color: selectedIndex === index ? 'green' : 'white' }, `${selectedIndex === index ? '▶ ' : '  '}${item.icon} ${item.label}`), selectedIndex === index && React.createElement(Text, { dimColor: true }, ` - ${item.desc}`))));
         };
         const EntityScreen = () => React.createElement(Box, { flexDirection: 'column' }, React.createElement(Text, { bold: true, color: 'cyan' }, '📦 Entity Operations'), React.createElement(Box, { marginTop: 1, flexDirection: 'column', marginLeft: 2 }, React.createElement(Text, null, '• Create Entity: ', React.createElement(Text, { dimColor: true }, 'cozo-memory entity create -n "Name" -t "Type"')), React.createElement(Text, null, '• Get Entity: ', React.createElement(Text, { dimColor: true }, 'cozo-memory entity get -i <id>')), React.createElement(Text, null, '• Delete Entity: ', React.createElement(Text, { dimColor: true }, 'cozo-memory entity delete -i <id>')), React.createElement(Text, null, '• Add Observation: ', React.createElement(Text, { dimColor: true }, 'cozo-memory obs add -i <id> -t "Text"'))), React.createElement(Box, { marginTop: 1 }, React.createElement(Text, { color: 'yellow' }, '💡 Use the CLI commands shown above for entity operations')));
+        const ProfileScreen = () => React.createElement(Box, { flexDirection: 'column' }, React.createElement(Text, { bold: true, color: 'cyan' }, '👤 User Profile Management'), React.createElement(Box, { marginTop: 1, flexDirection: 'column', marginLeft: 2 }, React.createElement(Text, null, '• Show Profile: ', React.createElement(Text, { dimColor: true }, 'cozo-memory profile show')), React.createElement(Text, null, '• Update Metadata: ', React.createElement(Text, { dimColor: true }, 'cozo-memory profile update -m \'{"key":"value"}\'')), React.createElement(Text, null, '• Add Preference: ', React.createElement(Text, { dimColor: true }, 'cozo-memory profile add-preference -t "I prefer TypeScript"')), React.createElement(Text, null, '• Reset Preferences: ', React.createElement(Text, { dimColor: true }, 'cozo-memory profile reset'))), React.createElement(Box, { marginTop: 1 }, React.createElement(Text, { color: 'yellow' }, '💡 User profile preferences get 50% search boost automatically')));
         const SearchScreen = () => React.createElement(Box, { flexDirection: 'column' }, React.createElement(Text, { bold: true, color: 'cyan' }, '🔍 Search & Context'), React.createElement(Box, { marginTop: 1, flexDirection: 'column', marginLeft: 2 }, React.createElement(Text, null, '• Search: ', React.createElement(Text, { dimColor: true }, 'cozo-memory search query -q "your query"')), React.createElement(Text, null, '• Context: ', React.createElement(Text, { dimColor: true }, 'cozo-memory search context -q "query" -w 5')), React.createElement(Text, null, '• Advanced: ', React.createElement(Text, { dimColor: true }, 'With filters, entity types, time ranges'))), React.createElement(Box, { marginTop: 1 }, React.createElement(Text, { color: 'yellow' }, '💡 Hybrid search combines vector, keyword, and graph signals')));
         const GraphScreen = () => React.createElement(Box, { flexDirection: 'column' }, React.createElement(Text, { bold: true, color: 'cyan' }, '🕸️ Graph Operations'), React.createElement(Box, { marginTop: 1, flexDirection: 'column', marginLeft: 2 }, React.createElement(Text, null, '• Explore: ', React.createElement(Text, { dimColor: true }, 'cozo-memory graph explore -s <id> -h 3')), React.createElement(Text, null, '• PageRank: ', React.createElement(Text, { dimColor: true }, 'cozo-memory graph pagerank')), React.createElement(Text, null, '• Communities: ', React.createElement(Text, { dimColor: true }, 'cozo-memory graph communities')), React.createElement(Text, null, '• Path Finding: ', React.createElement(Text, { dimColor: true }, 'cozo-memory graph explore -s <id1> -e <id2>'))), React.createElement(Box, { marginTop: 1 }, React.createElement(Text, { color: 'yellow' }, '💡 Graph algorithms help discover implicit relationships')));
         const SystemScreen = () => React.createElement(Box, { flexDirection: 'column' }, React.createElement(Text, { bold: true, color: 'cyan' }, '⚙️ System Management'), React.createElement(Box, { marginTop: 1, flexDirection: 'column', marginLeft: 2 }, React.createElement(Text, null, '• Health: ', React.createElement(Text, { dimColor: true }, 'cozo-memory system health')), React.createElement(Text, null, '• Metrics: ', React.createElement(Text, { dimColor: true }, 'cozo-memory system metrics')), React.createElement(Text, null, '• Export JSON: ', React.createElement(Text, { dimColor: true }, 'cozo-memory export json -o backup.json')), React.createElement(Text, null, '• Export Markdown: ', React.createElement(Text, { dimColor: true }, 'cozo-memory export markdown -o notes.md')), React.createElement(Text, null, '• Export Obsidian: ', React.createElement(Text, { dimColor: true }, 'cozo-memory export obsidian -o vault.zip')), React.createElement(Text, null, '• Import: ', React.createElement(Text, { dimColor: true }, 'cozo-memory import file -i data.json -f cozo'))), React.createElement(Box, { marginTop: 1 }, React.createElement(Text, { color: 'yellow' }, '💡 Regular exports recommended for backup')));

package/dist/verify-agentic-logic.js

@@ -0,0 +1,64 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const hybrid_search_1 = require("./hybrid-search");
+const db_service_1 = require("./db-service");
+const path = __importStar(require("path"));
+async function testAgenticSearch() {
+    console.log("Starting agentic_search direct test...");
+    // Create a minimal environment to test HybridSearch
+    const dbPath = path.resolve(process.cwd(), 'memory.db');
+    console.log(`Using database at: ${dbPath}`);
+    const dbService = new db_service_1.CozoDbService(dbPath, 'sqlite');
+    await dbService.init();
+    const hybridSearch = new hybrid_search_1.HybridSearch(dbService);
+    await hybridSearch.init();
+    const query = "Was ist der Status der Frontend und Backend Projekte?";
+    console.log(`Querying: "${query}"`);
+    try {
+        const results = await hybridSearch.agenticRetrieve({
+            query,
+            limit: 5
+        });
+        console.log("Agentic Search Results:", JSON.stringify(results, null, 2));
+    }
+    catch (error) {
+        console.error("Error during agentic search:", error);
+    }
+    finally {
+        await dbService.close();
+    }
+}
+testAgenticSearch();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cozo-memory",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "mcpName": "io.github.tobs-code/cozo-memory",
   "description": "Local-first persistent memory system for AI agents with hybrid search, graph reasoning, and MCP integration",
   "main": "dist/index.js",
@@ -37,6 +37,7 @@
     "rebuild": "npm run clean && npm run build",
     "test": "echo \"Error: no test specified\" && exit 1",
     "benchmark": "ts-node src/benchmark.ts",
+    "eval": "ts-node src/eval-suite.ts",
     "download-model": "ts-node src/download-model.ts"
   },
   "keywords": [

package/dist/test-mcp-search.js

@@ -1,47 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const index_1 = require("./index");
-async function testMCPSearch() {
-    console.log('Initializing MemoryServer...');
-    const server = new index_1.MemoryServer();
-    await server.initPromise;
-    console.log('\n=== Test 1: Create Entity via MCP mutate_memory ===');
-    const createResult = await server.mutateMemory({
-        action: 'create_entity',
-        name: 'Test Search Entity',
-        type: 'test',
-        metadata: { purpose: 'search_test' }
-    });
-    console.log('Created:', createResult);
-    const entityId = createResult.id;
-    console.log('\n=== Test 2: Add Observation via MCP mutate_memory ===');
-    const obsResult = await server.mutateMemory({
-        action: 'add_observation',
-        entity_id: entityId,
-        text: 'This is a test observation for searching with keywords like authentication and OAuth',
-        metadata: { test: true }
-    });
-    console.log('Added observation:', obsResult);
-    console.log('\n=== Test 3: Search via MCP query_memory ===');
-    const searchResult = await server.queryMemory({
-        action: 'search',
-        query: 'authentication',
-        limit: 5
-    });
-    console.log('Search result:', JSON.stringify(searchResult, null, 2));
-    console.log('\n=== Test 4: Search for "Alice" ===');
-    const aliceResult = await server.queryMemory({
-        action: 'search',
-        query: 'Alice',
-        limit: 5
-    });
-    console.log('Alice search result:', JSON.stringify(aliceResult, null, 2));
-    console.log('\n=== Test 5: Search for "TypeScript" ===');
-    const tsResult = await server.queryMemory({
-        action: 'search',
-        query: 'TypeScript',
-        limit: 5
-    });
-    console.log('TypeScript search result:', JSON.stringify(tsResult, null, 2));
-}
-testMCPSearch().catch(console.error);