summaryrefslogtreecommitdiff
path: root/rag/generator/__init__.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2024-06-19 02:07:06 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2024-06-19 02:07:06 +0200
commitaac821b148c6c0d35b940609dc9b0ddcb053b28e (patch)
tree5c125045b2b60ead39e093327d664adf43d1d35b /rag/generator/__init__.py
parentf2846429310452bebbf0d07203b1e53978c439c7 (diff)
Still wip on rewrite
Diffstat (limited to 'rag/generator/__init__.py')
-rw-r--r--rag/generator/__init__.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/rag/generator/__init__.py b/rag/generator/__init__.py
index a776231..770db16 100644
--- a/rag/generator/__init__.py
+++ b/rag/generator/__init__.py
@@ -1,8 +1,8 @@
from typing import Type
-from .abstract import AbstractGenerator
-from .cohere import Cohere
-from .ollama import Ollama
+from rag.generator.abstract import AbstractGenerator
+from rag.generator.cohere import Cohere
+from rag.generator.ollama import Ollama
MODELS = ["local", "cohere"]