summaryrefslogtreecommitdiff
path: root/rag/retriever
diff options
context:
space:
mode:
Diffstat (limited to 'rag/retriever')
-rw-r--r--rag/retriever/document.py2
-rw-r--r--rag/retriever/encoder.py2
-rw-r--r--rag/retriever/parser/pdf.py4
-rw-r--r--rag/retriever/retriever.py7
4 files changed, 8 insertions, 7 deletions
diff --git a/rag/retriever/document.py b/rag/retriever/document.py
index 54ac451..67ff936 100644
--- a/rag/retriever/document.py
+++ b/rag/retriever/document.py
@@ -1,8 +1,8 @@
import hashlib
import os
-from langchain_community.document_loaders.blob_loaders import Blob
import psycopg
+from langchain_community.document_loaders.blob_loaders import Blob
from loguru import logger as log
TABLES = """
diff --git a/rag/retriever/encoder.py b/rag/retriever/encoder.py
index 753157f..db69ee5 100644
--- a/rag/retriever/encoder.py
+++ b/rag/retriever/encoder.py
@@ -1,6 +1,6 @@
import os
from pathlib import Path
-from typing import List, Dict
+from typing import Dict, List
from uuid import uuid4
import ollama
diff --git a/rag/retriever/parser/pdf.py b/rag/retriever/parser/pdf.py
index 410f027..4c5addc 100644
--- a/rag/retriever/parser/pdf.py
+++ b/rag/retriever/parser/pdf.py
@@ -3,11 +3,11 @@ from pathlib import Path
from typing import List, Optional
from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain_core.documents import Document
+from langchain_community.document_loaders.blob_loaders import Blob
from langchain_community.document_loaders.parsers.pdf import (
PyPDFParser,
)
-from langchain_community.document_loaders.blob_loaders import Blob
+from langchain_core.documents import Document
class PDFParser:
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: