summaryrefslogtreecommitdiff
path: root/rag/generator/ollama.py
diff options
context:
space:
mode:
Diffstat (limited to 'rag/generator/ollama.py')
-rw-r--r--rag/generator/ollama.py6
1 files changed, 3 insertions, 3 deletions
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"