compound-agent 1.6.2 → 1.6.4

package/dist/index.d.ts

@@ -3477,14 +3477,14 @@ declare const AuditFindingSchema: z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     file: string;
     source: "lesson" | "pattern" | "rule";
-    severity: "warning" | "error" | "info";
+    severity: "error" | "warning" | "info";
     issue: string;
     relatedLessonId?: string | undefined;
     suggestedFix?: string | undefined;
 }, {
     file: string;
     source: "lesson" | "pattern" | "rule";
-    severity: "warning" | "error" | "info";
+    severity: "error" | "warning" | "info";
     issue: string;
     relatedLessonId?: string | undefined;
     suggestedFix?: string | undefined;
@@ -3501,14 +3501,14 @@ declare const AuditReportSchema: z.ZodObject<{
     }, "strip", z.ZodTypeAny, {
         file: string;
         source: "lesson" | "pattern" | "rule";
-        severity: "warning" | "error" | "info";
+        severity: "error" | "warning" | "info";
         issue: string;
         relatedLessonId?: string | undefined;
         suggestedFix?: string | undefined;
     }, {
         file: string;
         source: "lesson" | "pattern" | "rule";
-        severity: "warning" | "error" | "info";
+        severity: "error" | "warning" | "info";
         issue: string;
         relatedLessonId?: string | undefined;
         suggestedFix?: string | undefined;
@@ -3535,7 +3535,7 @@ declare const AuditReportSchema: z.ZodObject<{
     findings: {
         file: string;
         source: "lesson" | "pattern" | "rule";
-        severity: "warning" | "error" | "info";
+        severity: "error" | "warning" | "info";
         issue: string;
         relatedLessonId?: string | undefined;
         suggestedFix?: string | undefined;
@@ -3551,7 +3551,7 @@ declare const AuditReportSchema: z.ZodObject<{
     findings: {
         file: string;
         source: "lesson" | "pattern" | "rule";
-        severity: "warning" | "error" | "info";
+        severity: "error" | "warning" | "info";
         issue: string;
         relatedLessonId?: string | undefined;
         suggestedFix?: string | undefined;

package/dist/index.js

@@ -888,23 +888,44 @@ async function getEmbedding() {
   })();
   return pendingInit;
 }
-function unloadEmbedding() {
-  if (embeddingContext) {
-    embeddingContext.dispose().catch(() => {
-    });
-    embeddingContext = null;
+async function unloadEmbeddingResources() {
+  const pending = pendingInit;
+  if (pending) {
+    try {
+      await pending;
+    } catch {
+    }
   }
-  if (modelInstance) {
-    modelInstance.dispose().catch(() => {
-    });
-    modelInstance = null;
+  const context = embeddingContext;
+  const model = modelInstance;
+  const llama = llamaInstance;
+  embeddingContext = null;
+  modelInstance = null;
+  llamaInstance = null;
+  pendingInit = null;
+  const disposals = [];
+  if (context) {
+    disposals.push(context.dispose());
   }
-  if (llamaInstance) {
-    llamaInstance.dispose().catch(() => {
-    });
-    llamaInstance = null;
+  if (model) {
+    disposals.push(model.dispose());
+  }
+  if (llama) {
+    disposals.push(llama.dispose());
+  }
+  if (disposals.length > 0) {
+    await Promise.allSettled(disposals);
+  }
+}
+function unloadEmbedding() {
+  void unloadEmbeddingResources();
+}
+async function withEmbedding(fn) {
+  try {
+    return await fn();
+  } finally {
+    await unloadEmbeddingResources();
   }
-  pendingInit = null;
 }
 async function embedText(text) {
   const ctx = await getEmbedding();
@@ -2105,7 +2126,7 @@ async function runBackgroundEmbed(repoRoot) {
   const start = Date.now();
   writeEmbedStatus(repoRoot, { state: "running", startedAt: (/* @__PURE__ */ new Date()).toISOString() });
   try {
-    const result = await embedChunks(repoRoot, { onlyMissing: true });
+    const result = await withEmbedding(async () => embedChunks(repoRoot, { onlyMissing: true }));
     writeEmbedStatus(repoRoot, {
       state: "completed",
       chunksEmbedded: result.chunksEmbedded,
@@ -2120,7 +2141,6 @@ async function runBackgroundEmbed(repoRoot) {
       durationMs: Date.now() - start
     });
   } finally {
-    unloadEmbedding();
     closeKnowledgeDb();
     lock.release();
   }
@@ -2988,6 +3008,7 @@ var AuditReportSchema = z.object({
 });
 
 // src/commands/knowledge.ts
+init_embeddings();
 init_sqlite_knowledge();
 
 // src/commands/knowledge-index.ts
@@ -2998,6 +3019,7 @@ init_sqlite_knowledge();
 init_embeddings();
 init_search2();
 init_storage();
+init_embeddings();
 init_storage();
 init_search2();