breaking up query fn

This commit is contained in:
Per Stark
2024-11-27 12:38:39 +01:00
parent 2bc46e5dfe
commit bb57844a68
2 changed files with 27 additions and 15 deletions

View File

@@ -58,7 +58,17 @@ pub async fn query_handler(
// Process response
let answer = process_llm_response(response).await?;
info!("{:?}", answer);
debug!("{:?}", answer);
Ok(answer.answer.into_response())
let references: Vec<String> = answer
.references
.into_iter()
.map(|reference| reference.reference)
.collect();
info!("{:?}", references);
Ok(
Json(serde_json::json!({"answer": answer.answer, "references": references}))
.into_response(),
)
}