reorganize all imports
This commit is contained in:
parent
7181bd5ffb
commit
14fb489f49
3 changed files with 9 additions and 13 deletions
|
@ -1,12 +1,10 @@
|
||||||
use snafu::Report;
|
|
||||||
use snafu::{ResultExt, Snafu};
|
|
||||||
use std::{sync::Arc, time::Duration};
|
use std::{sync::Arc, time::Duration};
|
||||||
use tracing::error;
|
|
||||||
|
|
||||||
|
use snafu::{Report, ResultExt, Snafu};
|
||||||
|
use tracing::error;
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
use crate::storage;
|
use crate::{storage, storage::jobs::AddError};
|
||||||
use crate::storage::jobs::AddError;
|
|
||||||
|
|
||||||
/// Manage background job execution and storage.
|
/// Manage background job execution and storage.
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
|
|
|
@ -1,8 +1,7 @@
|
||||||
//! Shared application state for HTTP handlers.
|
//! Shared application state for HTTP handlers.
|
||||||
|
|
||||||
use crate::{storage::Postgres, text_encoder::TextEncoder, tokenize::Tokenizer};
|
|
||||||
|
|
||||||
use super::jobs::JobManager;
|
use super::jobs::JobManager;
|
||||||
|
use crate::{storage::Postgres, text_encoder::TextEncoder, tokenize::Tokenizer};
|
||||||
|
|
||||||
/// Application state shared across all HTTP request handlers.
|
/// Application state shared across all HTTP request handlers.
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
|
|
11
src/query.rs
11
src/query.rs
|
@ -58,12 +58,11 @@ pub async fn ask(
|
||||||
let reranker = reranker.clone();
|
let reranker = reranker.clone();
|
||||||
let tokenizer = tokenizer.clone();
|
let tokenizer = tokenizer.clone();
|
||||||
let query = query.to_string();
|
let query = query.to_string();
|
||||||
let reranked_docs = tokio::task::spawn_blocking(move || {
|
let reranked_docs =
|
||||||
reranker.rerank(&query, documents, &tokenizer, limit)
|
tokio::task::spawn_blocking(move || reranker.rerank(&query, documents, &tokenizer, limit))
|
||||||
})
|
.await
|
||||||
.await
|
.context(RerankJoinSnafu)?
|
||||||
.context(RerankJoinSnafu)?
|
.context(RerankSnafu)?;
|
||||||
.context(RerankSnafu)?;
|
|
||||||
|
|
||||||
Ok(reranked_docs)
|
Ok(reranked_docs)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue