From e09f5ff897543a93f561af68e3173050b1fe4b45 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 9 Apr 2024 01:06:26 +0200 Subject: Fix import order --- rag/retriever/document.py | 2 +- rag/retriever/encoder.py | 2 +- rag/retriever/parser/pdf.py | 4 ++-- rag/retriever/retriever.py | 7 ++++--- 4 files changed, 8 insertions(+), 7 deletions(-) (limited to 'rag/retriever') 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: -- cgit v1.2.3-70-g09d2