cjones/immich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handle missing face embedding column

Browse Source
This commit is contained in:
mertalev
2024-07-03 19:43:14 -04:00
parent 43f1868598
commit 90709b4788
1 changed files with 25 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/src/migrations/1718486162779-AddFaceSearchRelation.ts
+8 -1
View File
@@ -18,12 +18,19 @@ export class AddFaceSearchRelation1718486162779 implements MigrationInterface {
await queryRunner.query(`ALTER TABLE smart_search ADD COLUMN IF NOT EXISTS embedding vector(512)`); await queryRunner.query(`ALTER TABLE smart_search ADD COLUMN IF NOT EXISTS embedding vector(512)`);
await queryRunner.query(`ALTER TABLE smart_search ALTER COLUMN embedding SET STORAGE EXTERNAL`); await queryRunner.query(`ALTER TABLE smart_search ALTER COLUMN embedding SET STORAGE EXTERNAL`);
const assetFacesColumns = await queryRunner.query(
`SELECT column_name as name
FROM information_schema.columns
WHERE table_name = 'asset_faces'`);
const hasFaceEmbeddings = assetFacesColumns.some((column: { name: string }) => column.name === 'embedding');
if (hasFaceEmbeddings) {
await queryRunner.query(` await queryRunner.query(`
INSERT INTO face_search("faceId", embedding) INSERT INTO face_search("faceId", embedding)
SELECT id, embedding SELECT id, embedding
FROM asset_faces faces`); FROM asset_faces faces`);
}
await queryRunner.query(`ALTER TABLE asset_faces DROP COLUMN "embedding"`); await queryRunner.query(`ALTER TABLE asset_faces DROP COLUMN IF EXISTS embedding`);
await queryRunner.query(`ALTER TABLE face_search ALTER COLUMN embedding SET DATA TYPE real[]`); await queryRunner.query(`ALTER TABLE face_search ALTER COLUMN embedding SET DATA TYPE real[]`);
await queryRunner.query(`ALTER TABLE face_search ALTER COLUMN embedding SET DATA TYPE vector(512)`); await queryRunner.query(`ALTER TABLE face_search ALTER COLUMN embedding SET DATA TYPE vector(512)`);