mirror of
https://github.com/perstarkse/minne.git
synced 2026-03-29 22:01:59 +02:00
retrieval-pipeline: v1
This commit is contained in:
@@ -473,7 +473,8 @@ pub(crate) async fn load_or_init_system_settings(
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::ingest::{CorpusManifest, CorpusMetadata, CorpusParagraph, CorpusQuestion};
|
||||
use crate::ingest::store::CorpusParagraph;
|
||||
use crate::ingest::{CorpusManifest, CorpusMetadata, CorpusQuestion};
|
||||
use chrono::Utc;
|
||||
use common::storage::types::text_content::TextContent;
|
||||
|
||||
|
||||
@@ -294,16 +294,16 @@ pub fn build_stage_latency_breakdown(samples: &[PipelineStageTimings]) -> StageL
|
||||
}
|
||||
|
||||
StageLatencyBreakdown {
|
||||
embed: compute_latency_stats(&collect_stage(samples, |entry| entry.embed_ms)),
|
||||
embed: compute_latency_stats(&collect_stage(samples, |entry| entry.embed_ms())),
|
||||
collect_candidates: compute_latency_stats(&collect_stage(samples, |entry| {
|
||||
entry.collect_candidates_ms
|
||||
entry.collect_candidates_ms()
|
||||
})),
|
||||
graph_expansion: compute_latency_stats(&collect_stage(samples, |entry| {
|
||||
entry.graph_expansion_ms
|
||||
entry.graph_expansion_ms()
|
||||
})),
|
||||
chunk_attach: compute_latency_stats(&collect_stage(samples, |entry| entry.chunk_attach_ms)),
|
||||
rerank: compute_latency_stats(&collect_stage(samples, |entry| entry.rerank_ms)),
|
||||
assemble: compute_latency_stats(&collect_stage(samples, |entry| entry.assemble_ms)),
|
||||
chunk_attach: compute_latency_stats(&collect_stage(samples, |entry| entry.chunk_attach_ms())),
|
||||
rerank: compute_latency_stats(&collect_stage(samples, |entry| entry.rerank_ms())),
|
||||
assemble: compute_latency_stats(&collect_stage(samples, |entry| entry.assemble_ms())),
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
mod config;
|
||||
mod orchestrator;
|
||||
mod store;
|
||||
pub(crate) mod store;
|
||||
|
||||
pub use config::{CorpusCacheConfig, CorpusEmbeddingProvider};
|
||||
pub use orchestrator::ensure_corpus;
|
||||
|
||||
Reference in New Issue
Block a user