diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-09 01:04:50 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-09 01:04:50 +0200 |
| commit | 73a4adb6d4161ea222ed1ba56413571892da73d6 (patch) | |
| tree | 19c99d10b36a84547ffc69c7876dd138d1636adb /rag/generator | |
| parent | 8fd9af66e9eba60e3dbf5cb07a6ce599c92b7565 (diff) | |
Fix imports
Diffstat (limited to 'rag/generator')
| -rw-r--r-- | rag/generator/ollama.py | 5 | ||||
| -rw-r--r-- | rag/generator/prompt.py | 5 |
2 files changed, 2 insertions, 8 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" diff --git a/rag/generator/prompt.py b/rag/generator/prompt.py index ed372c9..059592a 100644 --- a/rag/generator/prompt.py +++ b/rag/generator/prompt.py @@ -2,10 +2,7 @@ from dataclasses import dataclass from typing import List -try: - from rag.retriever.vector import Document -except ModuleNotFoundError: - from retriever.vector import Document +from rag.retriever.vector import Document @dataclass |