diff --git a/src/grag/components/chroma_client.py b/src/grag/components/chroma_client.py index 1d816bd..7efd7c3 100644 --- a/src/grag/components/chroma_client.py +++ b/src/grag/components/chroma_client.py @@ -1,15 +1,14 @@ from typing import List import chromadb +from grag.components.embedding import Embedding +from grag.components.utils import get_config from langchain_community.vectorstores import Chroma from langchain_community.vectorstores.utils import filter_complex_metadata from langchain_core.documents import Document from tqdm import tqdm from tqdm.asyncio import tqdm as atqdm -from grag.components.embedding import Embedding -from grag.components.utils import get_config - chroma_conf = get_config()["chroma"] diff --git a/src/grag/components/multivec_retriever.py b/src/grag/components/multivec_retriever.py index 9946a3a..18ed752 100644 --- a/src/grag/components/multivec_retriever.py +++ b/src/grag/components/multivec_retriever.py @@ -2,13 +2,12 @@ import uuid from typing import List -from langchain.retrievers.multi_vector import MultiVectorRetriever -from langchain.storage import LocalFileStore -from langchain_core.documents import Document - from grag.components.chroma_client import ChromaClient from grag.components.text_splitter import TextSplitter from grag.components.utils import get_config +from langchain.retrievers.multi_vector import MultiVectorRetriever +from langchain.storage import LocalFileStore +from langchain_core.documents import Document multivec_retriever_conf = get_config()["multivec_retriever"] diff --git a/src/grag/rag/basic_rag.py b/src/grag/rag/basic_rag.py index 9589920..a99ecdd 100644 --- a/src/grag/rag/basic_rag.py +++ b/src/grag/rag/basic_rag.py @@ -4,7 +4,7 @@ from grag import prompts from grag.components.llm import LLM from grag.components.multivec_retriever import Retriever -from grag.components.prompt import Prompt, FewShotPrompt +from grag.components.prompt import FewShotPrompt, Prompt from grag.components.utils import get_config from importlib_resources import files from langchain_core.documents import Document diff --git a/src/tests/rag/basic_rag_test.py b/src/tests/rag/basic_rag_test.py index 2249028..06db25e 100644 --- a/src/tests/rag/basic_rag_test.py +++ b/src/tests/rag/basic_rag_test.py @@ -1,4 +1,4 @@ -from typing import Text, List +from typing import List, Text from grag.rag.basic_rag import BasicRAG