24 lines
896 B
TypeScript
24 lines
896 B
TypeScript
import { vectorExt } from 'src/database.config';
|
|
import { DatabaseExtension } from 'src/interfaces/database.interface';
|
|
import { MigrationInterface, QueryRunner } from 'typeorm';
|
|
|
|
export class AddCLIPEmbeddingIndex1700713994428 implements MigrationInterface {
|
|
name = 'AddCLIPEmbeddingIndex1700713994428';
|
|
|
|
public async up(queryRunner: QueryRunner): Promise<void> {
|
|
if (vectorExt === DatabaseExtension.VECTORS) {
|
|
await queryRunner.query(`SET vectors.pgvector_compatibility=on`);
|
|
}
|
|
await queryRunner.query(`SET search_path TO "$user", public, vectors`);
|
|
|
|
await queryRunner.query(`
|
|
CREATE INDEX IF NOT EXISTS clip_index ON smart_search
|
|
USING hnsw (embedding vector_cosine_ops)
|
|
WITH (ef_construction = 300, m = 16)`);
|
|
}
|
|
|
|
public async down(queryRunner: QueryRunner): Promise<void> {
|
|
await queryRunner.query(`DROP INDEX IF EXISTS clip_index`);
|
|
}
|
|
}
|