WIP job queue

src/query.rs

@@ -3,7 +3,10 @@
 use snafu::{ResultExt, Snafu};
 
 use crate::{
-    storage::{self, DocumentMatch, Postgres},
+    storage::{
+        self,
+        queries::{DocumentMatch, Queries},
+    },
     text_encoder::{self, TextEncoder},
     tokenize::{self, Tokenizer},
 };
@@ -16,7 +19,9 @@ pub enum AskError {
     #[snafu(display("Failed to embed query."))]
     Embed { source: text_encoder::EmbedError },
     #[snafu(display("Failed to retrieve similar documents."))]
-    Query { source: storage::QueryError },
+    Query {
+        source: storage::queries::QueryError,
+    },
     #[snafu(display("Failed to rerank documents."))]
     Rerank { source: text_encoder::RerankError },
 }
@@ -24,7 +29,7 @@ pub enum AskError {
 /// Process a user query and return ranked document matches.
 pub async fn ask(
     query: &str,
-    db: &Postgres,
+    db: &Queries,
     tokenizer: &Tokenizer,
     embedder: &TextEncoder,
     reranker: &TextEncoder,