yihong 9a20b17970
feat Gemini (#366)
* feat: gemini init

Signed-off-by: yihong0618 <zouzou0208@gmail.com>

* fix: useless code

Signed-off-by: yihong0618 <zouzou0208@gmail.com>

* feat: gemini model

---------

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
2024-01-06 22:21:53 +08:00

22 lines
748 B
Python

from book_maker.translator.caiyun_translator import Caiyun
from book_maker.translator.chatgptapi_translator import ChatGPTAPI
from book_maker.translator.deepl_translator import DeepL
from book_maker.translator.deepl_free_translator import DeepLFree
from book_maker.translator.google_translator import Google
from book_maker.translator.claude_translator import Claude
from book_maker.translator.gemini_translator import Gemini
from book_maker.translator.custom_api_translator import CustomAPI
MODEL_DICT = {
"chatgptapi": ChatGPTAPI,
"google": Google,
"caiyun": Caiyun,
"deepl": DeepL,
"deeplfree": DeepLFree,
"gpt4": ChatGPTAPI,
"claude": Claude,
"gemini": Gemini,
"customapi": CustomAPI,
# add more here
}