@zabaca/lattice 1.0.21 → 1.0.23

package/dist/main.js

@@ -873,8 +873,7 @@ class GraphService {
     const allResults = [];
     const conn = await this.ensureConnected();
     const vectorStr = `[${queryVector.join(", ")}]`;
-    try {
-      const reader = await conn.runAndReadAll(`
+    const sql = `
 				SELECT
 					name,
 					label,
@@ -883,10 +882,11 @@ class GraphService {
 					array_cosine_similarity(embedding, ${vectorStr}::FLOAT[${this.embeddingDimensions}]) as similarity
 				FROM nodes
 				WHERE embedding IS NOT NULL
-				ORDER BY similarity DESC
-				LIMIT ${k}
-			`);
-      for (const row of reader.getRows()) {
+			`;
+    try {
+      const reader = await conn.runAndReadAll(sql);
+      const rows = reader.getRows();
+      for (const row of rows) {
         const [name, label, title, description, similarity] = row;
         allResults.push({
           name,
@@ -1289,13 +1289,18 @@ class VoyageEmbeddingProvider {
 class EmbeddingService {
   configService;
   logger = new Logger3(EmbeddingService.name);
-  provider;
+  provider = null;
   config;
   constructor(configService) {
     this.configService = configService;
     this.config = this.loadConfig();
-    this.provider = this.createProvider();
-    this.logger.log(`Initialized embedding service with provider: ${this.provider.name}`);
+  }
+  getProvider() {
+    if (!this.provider) {
+      this.provider = this.createProvider();
+      this.logger.log(`Initialized embedding provider: ${this.provider.name}`);
+    }
+    return this.provider;
   }
   loadConfig() {
     const providerEnv = this.configService.get("EMBEDDING_PROVIDER");
@@ -1342,35 +1347,36 @@ class EmbeddingService {
     }
   }
   getProviderName() {
-    return this.provider.name;
+    return this.getProvider().name;
   }
   getDimensions() {
-    return this.provider.dimensions;
+    return this.getProvider().dimensions;
   }
   async generateEmbedding(text) {
     if (!text || text.trim().length === 0) {
       throw new Error("Cannot generate embedding for empty text");
     }
-    return this.provider.generateEmbedding(text);
+    return this.getProvider().generateEmbedding(text);
   }
   async generateQueryEmbedding(text) {
     if (!text || text.trim().length === 0) {
       throw new Error("Cannot generate embedding for empty text");
     }
-    if (this.provider.generateQueryEmbedding) {
-      return this.provider.generateQueryEmbedding(text);
+    const provider = this.getProvider();
+    if (provider.generateQueryEmbedding) {
+      return provider.generateQueryEmbedding(text);
     }
-    return this.provider.generateEmbedding(text);
+    return provider.generateEmbedding(text);
   }
   async generateEmbeddings(texts) {
     const validTexts = texts.filter((t) => t && t.trim().length > 0);
     if (validTexts.length === 0) {
       return [];
     }
-    return this.provider.generateEmbeddings(validTexts);
+    return this.getProvider().generateEmbeddings(validTexts);
   }
   isRealProvider() {
-    return this.provider.name !== "mock";
+    return this.getProvider().name !== "mock";
   }
 }
 EmbeddingService = __legacyDecorateClassTS([
@@ -3367,26 +3373,36 @@ GraphValidatorService = __legacyDecorateClassTS([
 // src/commands/sync.command.ts
 class SyncCommand extends CommandRunner7 {
   syncService;
+  graphService;
   _graphValidator;
   watcher = null;
   isShuttingDown = false;
-  constructor(syncService, _graphValidator) {
+  constructor(syncService, graphService, _graphValidator) {
     super();
     this.syncService = syncService;
+    this.graphService = graphService;
     this._graphValidator = _graphValidator;
   }
+  async safeExit(code) {
+    try {
+      await this.graphService.checkpoint();
+    } catch (error) {
+      console.error("Warning: checkpoint failed during exit");
+    }
+    process.exit(code);
+  }
   async run(paths, options) {
     if (options.watch && options.dryRun) {
       console.log(`
 \u26A0\uFE0F  Watch mode is not compatible with --dry-run mode
 `);
-      process.exit(1);
+      await this.safeExit(1);
     }
     if (options.watch && options.force) {
       console.log(`
 \u26A0\uFE0F  Watch mode is not compatible with --force mode (for safety)
 `);
-      process.exit(1);
+      await this.safeExit(1);
     }
     if (options.force && paths.length === 0) {
       console.log(`
@@ -3394,7 +3410,7 @@ class SyncCommand extends CommandRunner7 {
 `);
       console.log(`   Usage: lattice sync --force <path1> [path2] ...
 `);
-      process.exit(1);
+      await this.safeExit(1);
     }
     const syncOptions = {
       force: options.force,
@@ -3441,12 +3457,12 @@ class SyncCommand extends CommandRunner7 {
       if (options.watch) {
         await this.enterWatchMode(syncOptions);
       } else {
-        process.exit(initialResult.errors.length > 0 ? 1 : 0);
+        await this.safeExit(initialResult.errors.length > 0 ? 1 : 0);
       }
     } catch (error) {
       console.error(`
 \u274C Sync failed:`, error instanceof Error ? error.message : String(error));
-      process.exit(1);
+      await this.safeExit(1);
     }
   }
   async enterWatchMode(syncOptions) {
@@ -3503,10 +3519,12 @@ class SyncCommand extends CommandRunner7 {
         debouncedSync();
       }
     });
-    process.on("SIGINT", () => this.shutdown());
+    process.on("SIGINT", () => {
+      this.shutdown().catch(console.error);
+    });
     await new Promise(() => {});
   }
-  shutdown() {
+  async shutdown() {
     if (this.isShuttingDown)
       return;
     this.isShuttingDown = true;
@@ -3516,7 +3534,7 @@ class SyncCommand extends CommandRunner7 {
     if (this.watcher) {
       this.watcher.close();
     }
-    process.exit(0);
+    await this.safeExit(0);
   }
   printSyncResults(result, isWatchMode = false) {
     console.log(`
@@ -3706,6 +3724,7 @@ SyncCommand = __legacyDecorateClassTS([
   }),
   __legacyMetadataTS("design:paramtypes", [
     typeof SyncService === "undefined" ? Object : SyncService,
+    typeof GraphService === "undefined" ? Object : GraphService,
     typeof GraphValidatorService === "undefined" ? Object : GraphValidatorService
   ])
 ], SyncCommand);

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@zabaca/lattice",
-	"version": "1.0.21",
+	"version": "1.0.23",
 	"description": "Human-initiated, AI-powered knowledge graph for markdown documentation",
 	"type": "module",
 	"bin": {
@@ -44,7 +44,7 @@
 	},
 	"dependencies": {
 		"@anthropic-ai/claude-agent-sdk": "^0.1.67",
-		"@duckdb/node-api": "1.3.1-alpha.23",
+		"@duckdb/node-api": "1.4.3-r.1",
 		"@nestjs/common": "^10.0.0",
 		"@nestjs/config": "^3.0.0",
 		"@nestjs/core": "^10.0.0",