mirror of
https://github.com/TriliumNext/Notes.git
synced 2025-09-03 14:19:58 +08:00
refactor(llm): update NoteEmbeddingRow structure and add handling in LoadResults class
This commit is contained in:
parent
cc0795f812
commit
b0d60f8004
@ -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}'`);
|
||||||
}
|
}
|
||||||
|
@ -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
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user