mirror of
https://github.com/yihong0618/bilingual_book_maker.git
synced 2025-06-10 13:27:20 +00:00
9261d92e20
* chore: Bump google-generativeai and related dependencies * feat: add support for --temperature option to gemini * feat: add support for --interval option to gemini * feat: add support for --model_list option to gemini * feat: add support for --prompt option to gemini * modify: model settings * feat: add support for --use_context option to gemini * feat: add support for rotate_key to gemini * feat: add exponential backoff to gemini * Update README.md * fix: typos and apply black formatting * Update make_test_ebook.yaml * fix: cli * fix: interval option implementation * fix: interval for geminipro * fix: recreate convo after rotating key
30 lines
1.0 KiB
Python
30 lines
1.0 KiB
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.groq_translator import GroqClient
|
|
from book_maker.translator.tencent_transmart_translator import TencentTranSmart
|
|
from book_maker.translator.custom_api_translator import CustomAPI
|
|
|
|
MODEL_DICT = {
|
|
"openai": ChatGPTAPI,
|
|
"chatgptapi": ChatGPTAPI,
|
|
"gpt4": ChatGPTAPI,
|
|
"gpt4omini": ChatGPTAPI,
|
|
"gpt4o": ChatGPTAPI,
|
|
"google": Google,
|
|
"caiyun": Caiyun,
|
|
"deepl": DeepL,
|
|
"deeplfree": DeepLFree,
|
|
"claude": Claude,
|
|
"gemini": Gemini,
|
|
"geminipro": Gemini,
|
|
"groq": GroqClient,
|
|
"tencentransmart": TencentTranSmart,
|
|
"customapi": CustomAPI,
|
|
# add more here
|
|
}
|