diff options
Diffstat (limited to 'rag/generator/cohere.py')
| -rw-r--r-- | rag/generator/cohere.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/rag/generator/cohere.py b/rag/generator/cohere.py index cf95c18..7028b21 100644 --- a/rag/generator/cohere.py +++ b/rag/generator/cohere.py @@ -1,13 +1,12 @@ import os +from dataclasses import asdict from typing import Any, Generator -import cohere -from dataclasses import asdict +import cohere +from loguru import logger as log -from .prompt import Prompt from .abstract import AbstractGenerator - -from loguru import logger as log +from .prompt import Prompt class Cohere(metaclass=AbstractGenerator): |