From 0f6e5f5b103afced92ce5cd63741097f0a1ad357 Mon Sep 17 00:00:00 2001 From: JMARyA Date: Tue, 31 Dec 2024 17:42:56 +0100 Subject: [PATCH] search --- src/ai.rs | 1 + src/pages/mod.rs | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/ai.rs b/src/ai.rs index dc9a060..a0d72d6 100644 --- a/src/ai.rs +++ b/src/ai.rs @@ -234,6 +234,7 @@ impl EmbedStore { // TODO : fix search // + new ranked algorithm // + better repr + // limit should cover SearchResults not the query -> rework let results: Vec = sqlx::query_as( "SELECT *, 1 / (1 + (embed_mxbai_embed_large <-> $1)) AS similarity FROM doc_embedding ORDER BY embed_mxbai_embed_large <-> $1 LIMIT $2 OFFSET $3", diff --git a/src/pages/mod.rs b/src/pages/mod.rs index 6a2f77e..e4bc4d1 100644 --- a/src/pages/mod.rs +++ b/src/pages/mod.rs @@ -236,7 +236,7 @@ pub async fn vector_search( EmbedStore::search_vector(&input, limit as i64, offset as i64).await }) }, - 50, + 1500, ) .pager(page as u64, vector) .await;