diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-09 01:06:26 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-09 01:06:26 +0200 |
| commit | e09f5ff897543a93f561af68e3173050b1fe4b45 (patch) | |
| tree | 47b2047eb0071f782e77ae030a97edbd17f47785 /rag/retriever/retriever.py | |
| parent | 73a4adb6d4161ea222ed1ba56413571892da73d6 (diff) | |
Fix import order
Diffstat (limited to 'rag/retriever/retriever.py')
| -rw-r--r-- | rag/retriever/retriever.py | 7 |
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: |