diff options
Diffstat (limited to 'rag/generator/ollama.py')
-rw-r--r-- | rag/generator/ollama.py | 6 |
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" |