Jelajahi Sumber

migration fixes

mertalev 1 tahun lalu
induk
melakukan
375d0320fc

+ 1 - 1
server/src/infra/database.config.ts

@@ -33,7 +33,7 @@ export async function initDataSource(): Promise<DataSource> {
   const dataSource = await new DataSource(databaseConfig).initialize();
   
   const hasVectorExtension = (await dataSource.query(
-    `SELECT * FROM pg_extension WHERE name = 'vectors'`,
+    `SELECT * FROM pg_available_extensions WHERE name = 'vectors'`,
   )).length > 0;
 
   if (hasVectorExtension) {

+ 2 - 2
server/src/infra/migrations/vector/1699746198141-UsePgVectors.ts

@@ -5,11 +5,11 @@ export class UsePgVectors1699746198141 implements MigrationInterface {
 
   public async up(queryRunner: QueryRunner): Promise<void> {
     const faceDimQuery = await queryRunner.query(`
-      SELECT CARDINALITY(embedding) as dimsize
+      SELECT CARDINALITY(embedding::real[]) as dimsize
       FROM asset_faces
       LIMIT 1`);
     const clipDimQuery = await queryRunner.query(`
-      SELECT CARDINALITY("clipEmbedding") as dimsize
+      SELECT CARDINALITY("clipEmbedding"::real[]) as dimsize
       FROM smart_info
       LIMIT 1`);