1 files changed, 4 insertions, 3 deletions
diff --git a/rag/retriever/retriever.py b/rag/retriever/retriever.py
index 885dafe..deffae5 100644
--- a/rag/retriever/retriever.py
+++ b/rag/retriever/retriever.py
@@ -1,12 +1,13 @@
+from io import BytesIO
 from pathlib import Path
-from typing import Optional, List
+from typing import List, Optional
+
 from loguru import logger as log
 
-from io import BytesIO
 from .document import DocumentDB
 from .encoder import Encoder
 from .parser.pdf import PDFParser
-from .vector import VectorDB, Document
+from .vector import Document, VectorDB
 
 
 class Retriever: