mirror of
https://github.com/meilisearch/meilisearch.git
synced 2025-09-04 11:46:30 +00:00
Simplify conditions
This commit is contained in:
@ -1020,7 +1020,7 @@ pub fn prepare_search<'t>(
|
|||||||
.map_err(milli::Error::from)?
|
.map_err(milli::Error::from)?
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
search.semantic_auto_embedded(
|
search.semantic(
|
||||||
embedder_name.clone(),
|
embedder_name.clone(),
|
||||||
embedder.clone(),
|
embedder.clone(),
|
||||||
*quantized,
|
*quantized,
|
||||||
|
@ -201,6 +201,10 @@ async fn simple_search() {
|
|||||||
}
|
}
|
||||||
],
|
],
|
||||||
"query": "Captain",
|
"query": "Captain",
|
||||||
|
"queryVector": [
|
||||||
|
1.0,
|
||||||
|
1.0
|
||||||
|
],
|
||||||
"processingTimeMs": "[duration]",
|
"processingTimeMs": "[duration]",
|
||||||
"limit": 20,
|
"limit": 20,
|
||||||
"offset": 0,
|
"offset": 0,
|
||||||
@ -272,6 +276,10 @@ async fn simple_search() {
|
|||||||
}
|
}
|
||||||
],
|
],
|
||||||
"query": "Captain",
|
"query": "Captain",
|
||||||
|
"queryVector": [
|
||||||
|
1.0,
|
||||||
|
1.0
|
||||||
|
],
|
||||||
"processingTimeMs": "[duration]",
|
"processingTimeMs": "[duration]",
|
||||||
"limit": 20,
|
"limit": 20,
|
||||||
"offset": 0,
|
"offset": 0,
|
||||||
@ -343,6 +351,10 @@ async fn simple_search() {
|
|||||||
}
|
}
|
||||||
],
|
],
|
||||||
"query": "Captain",
|
"query": "Captain",
|
||||||
|
"queryVector": [
|
||||||
|
1.0,
|
||||||
|
1.0
|
||||||
|
],
|
||||||
"processingTimeMs": "[duration]",
|
"processingTimeMs": "[duration]",
|
||||||
"limit": 20,
|
"limit": 20,
|
||||||
"offset": 0,
|
"offset": 0,
|
||||||
|
@ -234,7 +234,6 @@ impl Search<'_> {
|
|||||||
// no embedder, no semantic search
|
// no embedder, no semantic search
|
||||||
let Some(SemanticSearch {
|
let Some(SemanticSearch {
|
||||||
vector,
|
vector,
|
||||||
mut auto_embedded,
|
|
||||||
embedder_name,
|
embedder_name,
|
||||||
embedder,
|
embedder,
|
||||||
quantized,
|
quantized,
|
||||||
@ -262,10 +261,7 @@ impl Search<'_> {
|
|||||||
let deadline = std::time::Instant::now() + std::time::Duration::from_secs(3);
|
let deadline = std::time::Instant::now() + std::time::Duration::from_secs(3);
|
||||||
|
|
||||||
match embedder.embed_search(query, Some(deadline)) {
|
match embedder.embed_search(query, Some(deadline)) {
|
||||||
Ok(embedding) => {
|
Ok(embedding) => embedding,
|
||||||
auto_embedded = true;
|
|
||||||
embedding
|
|
||||||
}
|
|
||||||
Err(error) => {
|
Err(error) => {
|
||||||
tracing::error!(error=%error, "Embedding failed");
|
tracing::error!(error=%error, "Embedding failed");
|
||||||
return Ok(return_keyword_results(
|
return Ok(return_keyword_results(
|
||||||
@ -280,7 +276,6 @@ impl Search<'_> {
|
|||||||
|
|
||||||
search.semantic = Some(SemanticSearch {
|
search.semantic = Some(SemanticSearch {
|
||||||
vector: Some(vector_query.clone()),
|
vector: Some(vector_query.clone()),
|
||||||
auto_embedded,
|
|
||||||
embedder_name,
|
embedder_name,
|
||||||
embedder,
|
embedder,
|
||||||
quantized,
|
quantized,
|
||||||
|
@ -33,7 +33,6 @@ pub mod similar;
|
|||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct SemanticSearch {
|
pub struct SemanticSearch {
|
||||||
vector: Option<Vec<f32>>,
|
vector: Option<Vec<f32>>,
|
||||||
auto_embedded: bool,
|
|
||||||
media: Option<serde_json::Value>,
|
media: Option<serde_json::Value>,
|
||||||
embedder_name: String,
|
embedder_name: String,
|
||||||
embedder: Arc<Embedder>,
|
embedder: Arc<Embedder>,
|
||||||
@ -105,26 +104,6 @@ impl<'a> Search<'a> {
|
|||||||
) -> &mut Search<'a> {
|
) -> &mut Search<'a> {
|
||||||
self.semantic = Some(SemanticSearch {
|
self.semantic = Some(SemanticSearch {
|
||||||
embedder_name,
|
embedder_name,
|
||||||
auto_embedded: false,
|
|
||||||
embedder,
|
|
||||||
quantized,
|
|
||||||
vector,
|
|
||||||
media,
|
|
||||||
});
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn semantic_auto_embedded(
|
|
||||||
&mut self,
|
|
||||||
embedder_name: String,
|
|
||||||
embedder: Arc<Embedder>,
|
|
||||||
quantized: bool,
|
|
||||||
vector: Option<Embedding>,
|
|
||||||
media: Option<serde_json::Value>,
|
|
||||||
) -> &mut Search<'a> {
|
|
||||||
self.semantic = Some(SemanticSearch {
|
|
||||||
embedder_name,
|
|
||||||
auto_embedded: true,
|
|
||||||
embedder,
|
embedder,
|
||||||
quantized,
|
quantized,
|
||||||
vector,
|
vector,
|
||||||
@ -278,13 +257,12 @@ impl<'a> Search<'a> {
|
|||||||
} = match self.semantic.as_ref() {
|
} = match self.semantic.as_ref() {
|
||||||
Some(SemanticSearch {
|
Some(SemanticSearch {
|
||||||
vector: Some(vector),
|
vector: Some(vector),
|
||||||
auto_embedded,
|
|
||||||
embedder_name,
|
embedder_name,
|
||||||
embedder,
|
embedder,
|
||||||
quantized,
|
quantized,
|
||||||
media: _,
|
media: _,
|
||||||
}) => {
|
}) => {
|
||||||
if *auto_embedded && self.retrieve_vectors {
|
if self.retrieve_vectors {
|
||||||
query_vector = Some(vector.clone());
|
query_vector = Some(vector.clone());
|
||||||
}
|
}
|
||||||
execute_vector_search(
|
execute_vector_search(
|
||||||
|
Reference in New Issue
Block a user