snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2025-11-22 04:36:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Early return if time budget is already exceeded

Browse Source
This commit is contained in:
ManyTheFish
2025-11-06 11:40:01 +01:00
parent fe1e4814fa
commit 5f69a43846
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
crates/meilisearch/src/personalization/mod.rs
View File

@@ -77,6 +77,12 @@ impl CohereService {
query: Option<&str>, query: Option<&str>,
time_budget: TimeBudget, time_budget: TimeBudget,
) -> Result<SearchResult, ResponseError> { ) -> Result<SearchResult, ResponseError> {
if time_budget.exceeded() {
warn!("Could not rerank due to deadline");
// If the deadline is exceeded, return the original search result instead of an error
return Ok(search_result);
}
// Extract user context from personalization // Extract user context from personalization
let user_context = personalize.user_context.as_str(); let user_context = personalize.user_context.as_str();
@@ -146,6 +152,7 @@ impl CohereService {
warn!("Cohere rerank attempt #{} failed: {}", attempt, retry.error); warn!("Cohere rerank attempt #{} failed: {}", attempt, retry.error);
if time_budget.exceeded() { if time_budget.exceeded() {
warn!("Could not rerank due to deadline");
return Err(PersonalizationError::DeadlineExceeded); return Err(PersonalizationError::DeadlineExceeded);
} else { } else {
match retry.into_duration(attempt) { match retry.into_duration(attempt) {