47 lines
1.7 KiB
JavaScript
47 lines
1.7 KiB
JavaScript
/**
|
|
* Copyright (c) 2024-2025 Тарабанов Александр Викторович
|
|
* All rights reserved.
|
|
*
|
|
* This software is proprietary and confidential.
|
|
* Unauthorized copying, modification, or distribution is prohibited.
|
|
*
|
|
* For licensing inquiries: info@hb3-accelerator.com
|
|
* Website: https://hb3-accelerator.com
|
|
* GitHub: https://github.com/HB3-ACCELERATOR
|
|
*/
|
|
|
|
const vectorSearch = require('../services/vectorSearchClient');
|
|
|
|
const TEST_TABLE_ID = 'test_table_1';
|
|
|
|
const rows = [
|
|
{ row_id: '1', text: 'Как тебя зовут?', metadata: { answer: 'Алексей' } },
|
|
{ row_id: '2', text: 'Где ты живёшь?', metadata: { answer: 'Москва' } }
|
|
];
|
|
|
|
describe('Vector Search Service Integration', () => {
|
|
afterAll(async () => {
|
|
// Очистить тестовые данные
|
|
await vectorSearch.remove(TEST_TABLE_ID, rows.map(r => r.row_id));
|
|
});
|
|
|
|
test('Upsert and search', async () => {
|
|
await vectorSearch.upsert(TEST_TABLE_ID, rows);
|
|
const results = await vectorSearch.search(TEST_TABLE_ID, 'Как зовут?', 1);
|
|
expect(results.length).toBeGreaterThan(0);
|
|
expect(results[0].metadata.answer).toBe('Алексей');
|
|
});
|
|
|
|
test('Delete', async () => {
|
|
await vectorSearch.remove(TEST_TABLE_ID, ['1']);
|
|
const results = await vectorSearch.search(TEST_TABLE_ID, 'Как зовут?', 1);
|
|
expect(results.length === 0 || results[0].metadata.answer !== 'Алексей').toBe(true);
|
|
});
|
|
|
|
test('Rebuild', async () => {
|
|
await vectorSearch.rebuild(TEST_TABLE_ID, [rows[1]]);
|
|
const results = await vectorSearch.search(TEST_TABLE_ID, 'Где ты живёшь?', 1);
|
|
expect(results.length).toBeGreaterThan(0);
|
|
expect(results[0].metadata.answer).toBe('Москва');
|
|
});
|
|
});
|