diff options
Diffstat (limited to 'rag/retriever')
-rw-r--r-- | rag/retriever/document.py | 2 | ||||
-rw-r--r-- | rag/retriever/encoder.py | 2 | ||||
-rw-r--r-- | rag/retriever/parser/pdf.py | 4 | ||||
-rw-r--r-- | rag/retriever/retriever.py | 7 |
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: |