@@ -4,7 +4,7 @@ import { MikroORM, EntityManager, EntitySchema } from '@mikro-orm/postgresql';
44import { VectorType , HalfvecType , BitType , SparsevecType , l2Distance , maxInnerProduct , cosineDistance , l1Distance , hammingDistance , jaccardDistance } from 'pgvector/mikro-orm' ;
55import { SparseVector } from 'pgvector' ;
66
7- test ( 'example' , async ( ) => {
7+ test ( 'mikro-orm example' , async ( ) => {
88 const Item = new EntitySchema ( {
99 name : 'Item' ,
1010 tableName : 'mikro_items' ,
@@ -25,9 +25,10 @@ test('example', async () => {
2525 const em = orm . em . fork ( ) ;
2626
2727 await em . execute ( 'CREATE EXTENSION IF NOT EXISTS vector' ) ;
28+ await em . execute ( 'DROP TABLE IF EXISTS mikro_items' ) ;
2829
2930 const generator = orm . getSchemaGenerator ( ) ;
30- await generator . refreshDatabase ( ) ;
31+ await generator . createSchema ( ) ;
3132
3233 em . create ( Item , { embedding : [ 1 , 1 , 1 ] , half_embedding : [ 1 , 1 , 1 ] , binary_embedding : '000' , sparse_embedding : new SparseVector ( [ 1 , 1 , 1 ] ) } ) ;
3334 em . create ( Item , { embedding : [ 2 , 2 , 2 ] , half_embedding : [ 2 , 2 , 2 ] , binary_embedding : '101' , sparse_embedding : new SparseVector ( [ 2 , 2 , 2 ] ) } ) ;
0 commit comments