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