fix: cleanup DocumentEmbeddings in database (#7958)

Signed-off-by: Alexander Onnikov <Alexander.Onnikov@xored.com>
This commit is contained in:
Alexander Onnikov 2025-02-07 18:01:33 +07:00 committed by GitHub
parent 346ee28eeb
commit 6c969cb194
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -355,6 +355,10 @@ export const documentOperation: MigrateOperation = {
{ {
state: 'migrateEmbeddings', state: 'migrateEmbeddings',
func: migrateEmbeddings func: migrateEmbeddings
},
{
state: 'migrateEmbeddingsRefs',
func: migrateEmbeddingsRefs
} }
]) ])
}, },
@ -370,3 +374,13 @@ async function migrateEmbeddings (client: MigrationClient): Promise<void> {
) )
await client.move(DOMAIN_DOCUMENT, { _class: attachment.class.Embedding }, DOMAIN_ATTACHMENT) await client.move(DOMAIN_DOCUMENT, { _class: attachment.class.Embedding }, DOMAIN_ATTACHMENT)
} }
async function migrateEmbeddingsRefs (client: MigrationClient): Promise<void> {
const _class = 'document:class:DocumentEmbedding'
await client.update(DOMAIN_ACTIVITY, { attachedToClass: _class }, { attachedToClass: attachment.class.Embedding })
await client.update(DOMAIN_ACTIVITY, { objectClass: _class }, { objectClass: attachment.class.Embedding })
await client.update(DOMAIN_NOTIFICATION, { attachedToClass: _class }, { attachedToClass: attachment.class.Embedding })
await client.update(DOMAIN_TX, { objectClass: _class }, { objectClass: attachment.class.Embedding })
await client.update(DOMAIN_TX, { 'tx.objectClass': _class }, { 'tx.objectClass': attachment.class.Embedding })
}