Fix import order

5 files changed, 8 insertions, 11 deletions

diff --git a/rag/generator/__init__.py b/rag/generator/__init__.py
index 541eff8..ba23ffc 100644
--- a/rag/generator/__init__.py
+++ b/rag/generator/__init__.py
@@ -1,8 +1,8 @@
 from typing import Type
 
 from .abstract import AbstractGenerator
-from .ollama import Ollama
 from .cohere import Cohere
+from .ollama import Ollama
 
 MODELS = ["ollama", "cohere"]
 
diff --git a/rag/generator/abstract.py b/rag/generator/abstract.py
index 5b336ea..1beacfb 100644
--- a/rag/generator/abstract.py
+++ b/rag/generator/abstract.py
@@ -1,5 +1,4 @@
 from abc import abstractmethod
-
 from typing import Any, Generator
 
 from .prompt import Prompt
diff --git a/rag/generator/cohere.py b/rag/generator/cohere.py
index cf95c18..7028b21 100644
--- a/rag/generator/cohere.py
+++ b/rag/generator/cohere.py
@@ -1,13 +1,12 @@
 import os
+from dataclasses import asdict
 from typing import Any, Generator
-import cohere
 
-from dataclasses import asdict
+import cohere
+from loguru import logger as log
 
-from .prompt import Prompt
 from .abstract import AbstractGenerator
-
-from loguru import logger as log
+from .prompt import Prompt
 
 
 class Cohere(metaclass=AbstractGenerator):
diff --git a/rag/generator/ollama.py b/rag/generator/ollama.py
index 1e968f6..cd8533b 100644
--- a/rag/generator/ollama.py
+++ b/rag/generator/ollama.py
@@ -4,11 +4,11 @@ from typing import Any, Generator, List
 import ollama
 from loguru import logger as log
 
-from .prompt import Prompt
-from .abstract import AbstractGenerator
-
 from rag.retriever.vector import Document
 
+from .abstract import AbstractGenerator
+from .prompt import Prompt
+
 SYSTEM_PROMPT = (
     "# System Preamble"
     "## Basic Rules"
diff --git a/rag/generator/prompt.py b/rag/generator/prompt.py
index 059592a..41f264e 100644
--- a/rag/generator/prompt.py
+++ b/rag/generator/prompt.py
@@ -1,7 +1,6 @@
 from dataclasses import dataclass
 from typing import List
 
-
 from rag.retriever.vector import Document