stratagate-dsh 0.3.9 → 0.3.10

package/dist/plugin.js

@@ -13126,6 +13126,30 @@ async function topicEvents(runtime, url) {
     items: page.map(({ id, title, status, createdAt }) => ({ id, title, status, createdAt }))
   };
 }
+function liveGraphProjection(snapshot) {
+  const eventIds = new Set([...snapshot.events, ...snapshot.agentEvents].filter(({ status: status2 }) => status2 !== "forgotten" && status2 !== "archived").map(({ id }) => id));
+  const jobs = snapshot.graphProjectionJobs.filter(({ sourceEventIds }) => sourceEventIds.some((id) => eventIds.has(id)));
+  const status = summarizeDerivationJobs(jobs, (job) => job.status === "completed");
+  const projected = new Set(jobs.filter(({ status: status2, projectorVersion }) => status2 === "completed" && projectorVersion === KNOWLEDGE_GRAPH_PROJECTOR_VERSION).flatMap(({ sourceEventIds }) => sourceEventIds.filter((id) => eventIds.has(id)))).size;
+  const total = eventIds.size;
+  const complete = projected === total && status.processing === 0 && status.terminalFailed === 0 && status.blocked === 0;
+  const state = status.processing > 0 ? "processing" : status.terminalFailed > 0 ? "failed" : complete ? "complete" : "incomplete";
+  return {
+    jobs,
+    status,
+    migration: {
+      projected,
+      total,
+      pending: jobs.filter(({ status: status2 }) => status2 === "pending").length,
+      running: jobs.filter(({ status: status2 }) => status2 === "running").length,
+      failed: status.terminalFailed,
+      retryable: status.retryable,
+      processing: status.processing,
+      state,
+      complete
+    }
+  };
+}
 async function overview(runtime, cachedEntries) {
   const entries = cachedEntries ?? await Promise.all((await runtime.adminNamespaces()).map(async (namespace) => ({
     namespace,
@@ -13231,20 +13255,21 @@ async function overview(runtime, cachedEntries) {
     const topicStatus = summarizeDerivationJobs(topicJobs, (job) => job.status === "completed");
     const summaryStatus = summarizeDerivationJobs(snapshot.summaryJobs, (job) => job.status === "succeeded");
     const extractionStatus = summarizeDerivationJobs(snapshot.extractionJobs, (job) => job.status === "succeeded" || job.status === "skipped");
-    const graphStatus = summarizeDerivationJobs(snapshot.graphProjectionJobs, (job) => job.status === "completed");
+    const graphWork = liveGraphProjection(snapshot);
+    const graphStatus = graphWork.status;
     const failedJobs = summaryStatus.terminalFailed + extractionStatus.terminalFailed + graphStatus.terminalFailed + topicStatus.terminalFailed;
     const processingJobs = summaryStatus.processing + extractionStatus.processing + graphStatus.processing + topicStatus.processing;
     const failedJobDetails = [
       ...topicJobs.filter((job) => job.status === "failed").map(describeTopicJob),
       ...snapshot.summaryJobs.filter(({ status }) => status === "failed").map((job) => describeBlockJob("block-summary", job)),
       ...snapshot.extractionJobs.filter(({ status }) => status === "failed").map((job) => describeBlockJob("event-extraction", job)),
-      ...snapshot.graphProjectionJobs.filter(({ status }) => status === "failed").map(describeGraphJob)
+      ...graphWork.jobs.filter(({ status }) => status === "failed").map(describeGraphJob)
     ];
     const processingJobDetails = [
       ...topicJobs.filter((job) => ["processing", "retryable"].includes(derivationJobState(job, job.status === "completed"))).map(describeTopicJob),
       ...snapshot.summaryJobs.filter((job) => derivationJobState(job, job.status === "succeeded") === "processing" || derivationJobState(job, job.status === "succeeded") === "retryable").map((job) => describeBlockJob("block-summary", job)),
       ...snapshot.extractionJobs.filter((job) => derivationJobState(job, job.status === "succeeded" || job.status === "skipped") === "processing" || derivationJobState(job, job.status === "succeeded" || job.status === "skipped") === "retryable").map((job) => describeBlockJob("event-extraction", job)),
-      ...snapshot.graphProjectionJobs.filter(graphProjectionIsProcessing).map(describeGraphJob)
+      ...graphWork.jobs.filter(graphProjectionIsProcessing).map(describeGraphJob)
     ];
     const timestamps = [
       ...snapshot.blocks.map(({ createdAt }) => createdAt),
@@ -13274,23 +13299,7 @@ async function overview(runtime, cachedEntries) {
         graphProjection: graphStatus,
         topicProjection: topicStatus
       },
-      graphMigration: (() => {
-        const projected = new Set(snapshot.graphProjectionJobs.filter(({ status, projectorVersion }) => status === "completed" && projectorVersion === KNOWLEDGE_GRAPH_PROJECTOR_VERSION).flatMap(({ sourceEventIds }) => sourceEventIds)).size;
-        const total = snapshot.events.filter(({ status }) => status !== "forgotten" && status !== "archived").length;
-        const complete = projected >= total;
-        const state = complete ? "complete" : graphStatus.processing > 0 ? "processing" : graphStatus.terminalFailed > 0 ? "failed" : "incomplete";
-        return {
-          projected,
-          total,
-          pending: snapshot.graphProjectionJobs.filter(({ status }) => status === "pending").length,
-          running: snapshot.graphProjectionJobs.filter(({ status }) => status === "running").length,
-          failed: graphStatus.terminalFailed,
-          retryable: graphStatus.retryable,
-          processing: graphStatus.processing,
-          state,
-          complete
-        };
-      })(),
+      graphMigration: graphWork.migration,
       usageReceipts: snapshot.usageReceipts.length,
       memoryUseCount: snapshot.usageReceipts.filter((receipt) => receipt.eventIds.length > 0 || receipt.elementIds.length > 0).length,
       failedJobs,
@@ -13677,23 +13686,7 @@ async function memories(runtime, url) {
       })),
       edges: snapshot.graphEdges,
       clusters: clusterKnowledgeGraph(snapshot.graphNodes, snapshot.graphEdges),
-      migration: (() => {
-        const projected = new Set(snapshot.graphProjectionJobs.filter(({ status, projectorVersion }) => status === "completed" && projectorVersion === KNOWLEDGE_GRAPH_PROJECTOR_VERSION).flatMap(({ sourceEventIds }) => sourceEventIds)).size;
-        const graphStatus = summarizeDerivationJobs(snapshot.graphProjectionJobs, (job) => job.status === "completed");
-        const total = snapshot.events.filter(({ status }) => status !== "forgotten" && status !== "archived").length;
-        const complete = projected >= total;
-        return {
-          projected,
-          total,
-          pending: snapshot.graphProjectionJobs.filter(({ status }) => status === "pending").length,
-          running: snapshot.graphProjectionJobs.filter(({ status }) => status === "running").length,
-          failed: graphStatus.terminalFailed,
-          retryable: graphStatus.retryable,
-          processing: graphStatus.processing,
-          state: complete ? "complete" : graphStatus.processing > 0 ? "processing" : graphStatus.terminalFailed > 0 ? "failed" : "incomplete",
-          complete
-        };
-      })()
+      migration: liveGraphProjection(snapshot).migration
     };
   } else if (kind === "elements") values = snapshot.elements.map(elementSummary);
   else if (kind === "blocks") {