diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-09 00:41:55 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-09 00:41:55 +0200 |
| commit | 3f447bff69c20109474c455f1ad52bd547ab49e9 (patch) | |
| tree | 66695ca0e3423e2973c5b24ec1ae7096019b5dd0 /rag/generator/__init__.py | |
| parent | c05eae81f9aaa0a764203446ec54d7dd7cbeb66f (diff) | |
Update
Diffstat (limited to 'rag/generator/__init__.py')
| -rw-r--r-- | rag/generator/__init__.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/rag/generator/__init__.py b/rag/generator/__init__.py index 7da603c..541eff8 100644 --- a/rag/generator/__init__.py +++ b/rag/generator/__init__.py @@ -4,6 +4,7 @@ from .abstract import AbstractGenerator from .ollama import Ollama from .cohere import Cohere +MODELS = ["ollama", "cohere"] def get_generator(model: str) -> Type[AbstractGenerator]: match model: |