From 5e51add842665ffe3f6b9088233a5b3db4a01de9 Mon Sep 17 00:00:00 2001 From: brianreicher Date: Tue, 10 Oct 2023 15:01:49 -0400 Subject: [PATCH] Fmt --- ingestion/src/mongo_utils.rs | 5 ++++- tokenizer/src/oai.rs | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/ingestion/src/mongo_utils.rs b/ingestion/src/mongo_utils.rs index bccc783..029972b 100644 --- a/ingestion/src/mongo_utils.rs +++ b/ingestion/src/mongo_utils.rs @@ -161,7 +161,10 @@ impl MongoDriver { } let collection: mongodb::Collection = - self.client.as_ref().unwrap().database(&self.db_name).collection(collection_name); + self.client.as_ref() + .unwrap() + .database(&self.db_name) + .collection(collection_name); let cursor: mongodb::Cursor = collection.find(Document::new(), None).await?; let mut result: Vec = Vec::new(); diff --git a/tokenizer/src/oai.rs b/tokenizer/src/oai.rs index 6c30de9..ea2a79b 100644 --- a/tokenizer/src/oai.rs +++ b/tokenizer/src/oai.rs @@ -23,7 +23,10 @@ impl OpenAIClient { pub async fn tokenize_collection(&self, collection: &str) -> std::io::Result<>{ let result Vec = self.mongo_client.get_all_documents(self.collection); - let col: mongodb::Collection= self.mongo_model.client.as_ref().unwrap().database(&self.mongo_model.db_name).collection(collection); + let col: mongodb::Collection= self.mongo_model.client.as_ref() + .unwrap() + .database(&self.mongo_model.db_name) + .collection(collection); while let Some(result) = cursor.next().await { match result {