| 323 | } |
| 324 | } |
| 325 | async function knnQuery(searchQuery: string, k = 3) { |
| 326 | if (documentStore?.index) { |
| 327 | const vector = await models.vectorize( |
| 328 | searchQuery, |
| 329 | vecStoreUploaderConfigurationState.value.modelName |
| 330 | ); |
| 331 | const res = documentStore.index.searchKnn(vector.tolist()[0], k, undefined); |
| 332 | // You can also search the index with a label filter |
| 333 | /*const labelFilter = (label: number) => { |
| 334 | return label >= 10 && label < 20; |
| 335 | }; |
| 336 | const result2 = index.searchKnn(testVectorData.vectors[10], 10, labelFilter);*/ |
| 337 | return res; |
| 338 | } |
| 339 | } |
| 340 | |
| 341 | export interface SearchResult { |
| 342 | distance: number; |