Fix: hanging when query memory

Author: codeaholicguy

packages/cli/src/commands/skill.ts

@@ -37,7 +37,7 @@ export function registerSkillCommand(program: Command): void {
         if (skills.length === 0) {
           ui.warning('No skills installed in this project.');
           ui.info('Install a skill with: ai-devkit skill add <registry>/<repo> <skill-name>');
-          process.exit(0);
+          return;
         }
 
         ui.text('Installed Skills:', { breakline: true });
@@ -53,7 +53,6 @@ export function registerSkillCommand(program: Command): void {
         });
 
         ui.text(`Total: ${skills.length} skill(s)`, { breakline: true });
-        process.exit(0);
       } catch (error: any) {
         ui.error(`Failed to list skills: ${error.message}`);
         process.exit(1);

packages/memory/src/api.ts

@@ -1,5 +1,6 @@
 import { storeKnowledge } from './handlers/store';
 import { searchKnowledge } from './handlers/search';
+import { closeDatabase } from './database';
 import type { StoreKnowledgeInput, SearchKnowledgeInput, StoreKnowledgeResult, SearchKnowledgeResult } from './types';
 
 export { storeKnowledge, searchKnowledge };
@@ -21,23 +22,31 @@ export interface MemorySearchOptions {
 }
 
 export function memoryStoreCommand(options: MemoryStoreOptions): StoreKnowledgeResult {
-    const input: StoreKnowledgeInput = {
-        title: options.title,
-        content: options.content,
-        tags: options.tags ? options.tags.split(',').map(t => t.trim()) : undefined,
-        scope: options.scope,
-    };
+    try {
+        const input: StoreKnowledgeInput = {
+            title: options.title,
+            content: options.content,
+            tags: options.tags ? options.tags.split(',').map(t => t.trim()) : undefined,
+            scope: options.scope,
+        };
 
-    return storeKnowledge(input);
+        return storeKnowledge(input);
+    } finally {
+        closeDatabase();
+    }
 }
 
 export function memorySearchCommand(options: MemorySearchOptions): SearchKnowledgeResult {
-    const input: SearchKnowledgeInput = {
-        query: options.query,
-        contextTags: options.tags ? options.tags.split(',').map(t => t.trim()) : undefined,
-        scope: options.scope,
-        limit: options.limit,
-    };
+    try {
+        const input: SearchKnowledgeInput = {
+            query: options.query,
+            contextTags: options.tags ? options.tags.split(',').map(t => t.trim()) : undefined,
+            scope: options.scope,
+            limit: options.limit,
+        };
 
-    return searchKnowledge(input);
+        return searchKnowledge(input);
+    } finally {
+        closeDatabase();
+    }
 }

packages/memory/src/index.ts

@@ -4,7 +4,11 @@ import { runServer } from './server';
 
 export * from './api';
 
-runServer().catch((error: Error) => {
-    console.error('Failed to start server:', error);
-    process.exit(1);
-});
+// Only start MCP server when this file is run directly as a binary
+// Not when imported as a library (e.g., by CLI commands)
+if (require.main === module) {
+    runServer().catch((error: Error) => {
+        console.error('Failed to start server:', error);
+        process.exit(1);
+    });
+}