소스 검색

migration fixes

mertalev 1 년 전
부모
커밋
375d0320fc
2개의 변경된 파일3개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 1
      server/src/infra/database.config.ts
  2. 2 2
      server/src/infra/migrations/vector/1699746198141-UsePgVectors.ts

+ 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`);