retrieval-pipeline: v0

html-router/src/routes/knowledge/handlers.rs

@@ -24,7 +24,7 @@ use common::{
     },
     utils::embedding::generate_embedding,
 };
-use composite_retrieval::{retrieve_entities, RetrievedEntity};
+use retrieval_pipeline::{retrieve_entities, RetrievalConfig, RetrievedEntity, StrategyOutput};
 use tracing::debug;
 use uuid::Uuid;
 
@@ -284,11 +284,15 @@ pub async fn suggest_knowledge_relationships(
             None => None,
         };
 
-        if let Ok(results) = retrieve_entities(
+        let mut retrieval_config = RetrievalConfig::default();
+        retrieval_config.strategy = state.retrieval_strategy();
+
+        if let Ok(StrategyOutput::Entities(results)) = retrieve_entities(
             &state.db,
             &state.openai_client,
             &query,
             &user.id,
+            retrieval_config,
             rerank_lease,
         )
         .await