From cc0795f812fc28429ae90db1ae081c34f1e59f61 Mon Sep 17 00:00:00 2001
From: perf3ct <jonfuller2012@gmail.com>
Date: Mon, 2 Jun 2025 15:49:45 +0000
Subject: [PATCH] fix(llm): also add note_embeddings to froca and consistency
 tests

---
 apps/client/src/services/froca_updater.ts      | 2 +-
 apps/client/src/services/load_results.ts       | 2 ++
 apps/server/src/services/consistency_checks.ts | 1 +
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/apps/client/src/services/froca_updater.ts b/apps/client/src/services/froca_updater.ts
index 1f8eaa541..d01aa7a73 100644
--- a/apps/client/src/services/froca_updater.ts
+++ b/apps/client/src/services/froca_updater.ts
@@ -35,7 +35,7 @@ async function processEntityChanges(entityChanges: EntityChange[]) {
                 loadResults.addOption(attributeEntity.name);
             } else if (ec.entityName === "attachments") {
                 processAttachment(loadResults, ec);
-            } else if (ec.entityName === "blobs" || ec.entityName === "etapi_tokens") {
+            } else if (ec.entityName === "blobs" || ec.entityName === "etapi_tokens" || ec.entityName === "note_embeddings") {
                 // NOOP
             } else {
                 throw new Error(`Unknown entityName '${ec.entityName}'`);
diff --git a/apps/client/src/services/load_results.ts b/apps/client/src/services/load_results.ts
index 11f9a1a11..3f034a1e2 100644
--- a/apps/client/src/services/load_results.ts
+++ b/apps/client/src/services/load_results.ts
@@ -48,6 +48,7 @@ interface ContentNoteIdToComponentIdRow {
     noteId: string;
     componentId: string;
 }
+interface NoteEmbeddingRow {}
 
 type EntityRowMappings = {
     notes: NoteRow;
@@ -56,6 +57,7 @@ type EntityRowMappings = {
     options: OptionRow;
     revisions: RevisionRow;
     note_reordering: NoteReorderingRow;
+    note_embeddings: NoteEmbeddingRow;
 };
 
 export type EntityRowNames = keyof EntityRowMappings;
diff --git a/apps/server/src/services/consistency_checks.ts b/apps/server/src/services/consistency_checks.ts
index ec7850572..8022c74df 100644
--- a/apps/server/src/services/consistency_checks.ts
+++ b/apps/server/src/services/consistency_checks.ts
@@ -799,6 +799,7 @@ class ConsistencyChecks {
         this.runEntityChangeChecks("attributes", "attributeId");
         this.runEntityChangeChecks("etapi_tokens", "etapiTokenId");
         this.runEntityChangeChecks("options", "name");
+        this.runEntityChangeChecks("note_embeddings", "embedId");
     }
 
     findWronglyNamedAttributes() {