Merge pull request #42 from arjbingly/ruff-reformat

ruff format

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"]
 
 

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"]
 

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

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