github/bilingual_book_maker
1
1
Fork 0
mirror of https://github.com/yihong0618/bilingual_book_maker.git synced 2025-06-06 11:35:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add support for rotate_key to gemini

Browse Source
This commit is contained in:
Risin 2024-10-15 18:52:49 +09:00
parent 8dbefd6fc4
commit 3d0ad1c9ca
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
book_maker/translator/gemini_translator.py
View File

@ -78,6 +78,7 @@ class Gemini(Base):
or None # Allow None, but not empty string or None # Allow None, but not empty string
) )
genai.configure(api_key=next(self.keys))
generation_config["temperature"] = temperature generation_config["temperature"] = temperature
def create_convo(self): def create_convo(self):
@ -96,7 +97,7 @@ class Gemini(Base):
print(f"Using model {self.model}") print(f"Using model {self.model}")
def rotate_key(self): def rotate_key(self):
pass genai.configure(api_key=next(self.keys))
def translate(self, text): def translate(self, text):
t_text = "" t_text = ""