search
Some checks failed
ci/woodpecker/push/build Pipeline failed

This commit is contained in:
JMARyA 2024-12-31 17:42:56 +01:00
parent 3ed87eaba2
commit 0f6e5f5b10
Signed by: jmarya
GPG key ID: 901B2ADDF27C2263
2 changed files with 2 additions and 1 deletions

View file

@ -234,6 +234,7 @@ impl EmbedStore {
// TODO : fix search // TODO : fix search
// + new ranked algorithm // + new ranked algorithm
// + better repr // + better repr
// limit should cover SearchResults not the query -> rework
let results: Vec<DocEmbedding> = sqlx::query_as( let results: Vec<DocEmbedding> = 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", "SELECT *, 1 / (1 + (embed_mxbai_embed_large <-> $1)) AS similarity FROM doc_embedding ORDER BY embed_mxbai_embed_large <-> $1 LIMIT $2 OFFSET $3",

View file

@ -236,7 +236,7 @@ pub async fn vector_search(
EmbedStore::search_vector(&input, limit as i64, offset as i64).await EmbedStore::search_vector(&input, limit as i64, offset as i64).await
}) })
}, },
50, 1500,
) )
.pager(page as u64, vector) .pager(page as u64, vector)
.await; .await;