refactor(llm): update NoteEmbeddingRow structure and add handling in LoadResults class

This commit is contained in:
perf3ct 2025-06-02 19:01:34 +00:00
parent cc0795f812
commit b0d60f8004
No known key found for this signature in database
GPG Key ID: 569C4EEC436F5232
2 changed files with 24 additions and 5 deletions

View File

@ -36,7 +36,7 @@ async function processEntityChanges(entityChanges: EntityChange[]) {
} else if (ec.entityName === "attachments") { } else if (ec.entityName === "attachments") {
processAttachment(loadResults, ec); processAttachment(loadResults, ec);
} else if (ec.entityName === "blobs" || ec.entityName === "etapi_tokens" || ec.entityName === "note_embeddings") { } else if (ec.entityName === "blobs" || ec.entityName === "etapi_tokens" || ec.entityName === "note_embeddings") {
// NOOP // NOOP - these entities don't require frontend processing
} else { } else {
throw new Error(`Unknown entityName '${ec.entityName}'`); throw new Error(`Unknown entityName '${ec.entityName}'`);
} }

View File

@ -44,11 +44,18 @@ interface OptionRow {}
interface NoteReorderingRow {} interface NoteReorderingRow {}
interface ContentNoteIdToComponentIdRow { interface NoteEmbeddingRow {
embedId: string;
noteId: string; noteId: string;
componentId: string; providerId: string;
modelId: string;
dimension: number;
version: number;
dateCreated: string;
utcDateCreated: string;
dateModified: string;
utcDateModified: string;
} }
interface NoteEmbeddingRow {}
type EntityRowMappings = { type EntityRowMappings = {
notes: NoteRow; notes: NoteRow;
@ -73,6 +80,7 @@ export default class LoadResults {
private contentNoteIdToComponentId: ContentNoteIdToComponentIdRow[]; private contentNoteIdToComponentId: ContentNoteIdToComponentIdRow[];
private optionNames: string[]; private optionNames: string[];
private attachmentRows: AttachmentRow[]; private attachmentRows: AttachmentRow[];
private noteEmbeddingRows: NoteEmbeddingRow[];
constructor(entityChanges: EntityChange[]) { constructor(entityChanges: EntityChange[]) {
const entities: Record<string, Record<string, any>> = {}; const entities: Record<string, Record<string, any>> = {};
@ -101,6 +109,8 @@ export default class LoadResults {
this.optionNames = []; this.optionNames = [];
this.attachmentRows = []; this.attachmentRows = [];
this.noteEmbeddingRows = [];
} }
getEntityRow<T extends EntityRowNames>(entityName: T, entityId: string): EntityRowMappings[T] { getEntityRow<T extends EntityRowNames>(entityName: T, entityId: string): EntityRowMappings[T] {
@ -203,6 +213,14 @@ export default class LoadResults {
return this.attachmentRows; return this.attachmentRows;
} }
addNoteEmbedding(embedding: NoteEmbeddingRow) {
this.noteEmbeddingRows.push(embedding);
}
getNoteEmbeddingRows() {
return this.noteEmbeddingRows;
}
/** /**
* @returns {boolean} true if there are changes which could affect the attributes (including inherited ones) * @returns {boolean} true if there are changes which could affect the attributes (including inherited ones)
* notably changes in note itself should not have any effect on attributes * notably changes in note itself should not have any effect on attributes
@ -220,7 +238,8 @@ export default class LoadResults {
this.revisionRows.length === 0 && this.revisionRows.length === 0 &&
this.contentNoteIdToComponentId.length === 0 && this.contentNoteIdToComponentId.length === 0 &&
this.optionNames.length === 0 && this.optionNames.length === 0 &&
this.attachmentRows.length === 0 this.attachmentRows.length === 0 &&
this.noteEmbeddingRows.length === 0
); );
} }