feat: use pgvecto.rs (#3605)

server/src/infra/entities/asset-face.entity.ts

@@ -2,7 +2,7 @@ import { Column, Entity, Index, ManyToOne, PrimaryGeneratedColumn } from 'typeor
 import { AssetEntity } from './asset.entity';
 import { PersonEntity } from './person.entity';
 
-@Entity('asset_faces')
+@Entity('asset_faces', { synchronize: false })
 @Index(['personId', 'assetId'])
 export class AssetFaceEntity {
   @PrimaryGeneratedColumn('uuid')
@@ -14,12 +14,9 @@ export class AssetFaceEntity {
   @Column({ nullable: true, type: 'uuid' })
   personId!: string | null;
 
-  @Column({
-    type: 'float4',
-    array: true,
-    nullable: true,
-  })
-  embedding!: number[] | null;
+  @Index('face_index', { synchronize: false })
+  @Column({ type: 'float4', array: true, select: false })
+  embedding!: number[];
 
   @Column({ default: 0, type: 'int' })
   imageWidth!: number;

server/src/infra/entities/asset.entity.ts

@@ -20,6 +20,7 @@ import { ExifEntity } from './exif.entity';
 import { LibraryEntity } from './library.entity';
 import { SharedLinkEntity } from './shared-link.entity';
 import { SmartInfoEntity } from './smart-info.entity';
+import { SmartSearchEntity } from './smart-search.entity';
 import { TagEntity } from './tag.entity';
 import { UserEntity } from './user.entity';
 
@@ -137,6 +138,9 @@ export class AssetEntity {
   @OneToOne(() => SmartInfoEntity, (smartInfoEntity) => smartInfoEntity.asset)
   smartInfo?: SmartInfoEntity;
 
+  @OneToOne(() => SmartSearchEntity, (smartSearchEntity) => smartSearchEntity.asset)
+  smartSearch?: SmartSearchEntity;
+
   @ManyToMany(() => TagEntity, (tag) => tag.assets, { cascade: true })
   @JoinTable({ name: 'tag_asset' })
   tags!: TagEntity[];

server/src/infra/entities/exif.entity.ts

@@ -46,6 +46,7 @@ export class ExifEntity {
   @Column({ type: 'varchar', nullable: true })
   projectionType!: string | null;
 
+  @Index('exif_city')
   @Column({ type: 'varchar', nullable: true })
   city!: string | null;
 
@@ -98,6 +99,7 @@ export class ExifEntity {
   @Column({
     type: 'tsvector',
     generatedType: 'STORED',
+    select: false,
     asExpression: `TO_TSVECTOR('english',
                          COALESCE(make, '') || ' ' ||
                          COALESCE(model, '') || ' ' ||

server/src/infra/entities/index.ts

@@ -15,6 +15,7 @@ import { PartnerEntity } from './partner.entity';
 import { PersonEntity } from './person.entity';
 import { SharedLinkEntity } from './shared-link.entity';
 import { SmartInfoEntity } from './smart-info.entity';
+import { SmartSearchEntity } from './smart-search.entity';
 import { SystemConfigEntity } from './system-config.entity';
 import { SystemMetadataEntity } from './system-metadata.entity';
 import { TagEntity } from './tag.entity';
@@ -38,6 +39,7 @@ export * from './partner.entity';
 export * from './person.entity';
 export * from './shared-link.entity';
 export * from './smart-info.entity';
+export * from './smart-search.entity';
 export * from './system-config.entity';
 export * from './system-metadata.entity';
 export * from './tag.entity';
@@ -61,6 +63,7 @@ export const databaseEntities = [
   PersonEntity,
   SharedLinkEntity,
   SmartInfoEntity,
+  SmartSearchEntity,
   SystemConfigEntity,
   SystemMetadataEntity,
   TagEntity,

server/src/infra/entities/smart-info.entity.ts

@@ -1,7 +1,7 @@
 import { Column, Entity, JoinColumn, OneToOne, PrimaryColumn } from 'typeorm';
 import { AssetEntity } from './asset.entity';
 
-@Entity('smart_info')
+@Entity('smart_info', { synchronize: false })
 export class SmartInfoEntity {
   @OneToOne(() => AssetEntity, { onDelete: 'CASCADE', nullable: true })
   @JoinColumn({ name: 'assetId', referencedColumnName: 'id' })
@@ -15,11 +15,4 @@ export class SmartInfoEntity {
 
   @Column({ type: 'text', array: true, nullable: true })
   objects!: string[] | null;
-
-  @Column({
-    type: 'float4',
-    array: true,
-    nullable: true,
-  })
-  clipEmbedding!: number[] | null;
 }

server/src/infra/entities/smart-search.entity.ts

@@ -0,0 +1,20 @@
+import { Column, Entity, Index, JoinColumn, OneToOne, PrimaryColumn } from 'typeorm';
+import { AssetEntity } from './asset.entity';
+
+@Entity('smart_search', { synchronize: false })
+export class SmartSearchEntity {
+  @OneToOne(() => AssetEntity, { onDelete: 'CASCADE', nullable: true })
+  @JoinColumn({ name: 'assetId', referencedColumnName: 'id' })
+  asset?: AssetEntity;
+
+  @PrimaryColumn()
+  assetId!: string;
+
+  @Index('clip_index', { synchronize: false })
+  @Column({
+    type: 'float4',
+    array: true,
+    select: false,
+  })
+  embedding!: number[];
+}