Re-integrate embedder stats

This commit is contained in:
ManyTheFish 2025-06-30 09:46:19 +02:00
parent d35b2d8d33
commit 6db5939f84
17 changed files with 45 additions and 23 deletions

View file

@ -245,7 +245,11 @@ impl IndexScheduler {
let must_stop_processing = self.scheduler.must_stop_processing.clone();
builder
.execute(&|| must_stop_processing.get(), &progress)
.execute(
&|| must_stop_processing.get(),
&progress,
current_batch.embedder_stats.clone(),
)
.map_err(|e| Error::from_milli(e, Some(index_uid.to_string())))?;
index_wtxn.commit()?;
}

View file

@ -475,7 +475,7 @@ impl IndexScheduler {
progress.update_progress(SettingsProgress::ApplyTheSettings);
let congestion = builder
.execute(&|| must_stop_processing.get(), progress)
.execute(&|| must_stop_processing.get(), progress, embedder_stats)
.map_err(|err| Error::from_milli(err, Some(index_uid.clone())))?;
Ok((tasks, congestion))