diff options
Diffstat (limited to 'rag/generator/ollama.py')
| -rw-r--r-- | rag/generator/ollama.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/rag/generator/ollama.py b/rag/generator/ollama.py index ec6a55f..1e968f6 100644 --- a/rag/generator/ollama.py +++ b/rag/generator/ollama.py @@ -7,10 +7,7 @@ from loguru import logger as log from .prompt import Prompt from .abstract import AbstractGenerator -try: - from rag.retriever.vector import Document -except ModuleNotFoundError: - from retriever.vector import Document +from rag.retriever.vector import Document SYSTEM_PROMPT = ( "# System Preamble" |