From 3d0ad1c9ca0d2f2a1a3983ca18630db30f0b29b4 Mon Sep 17 00:00:00 2001
From: Risin <5331402+risin42@users.noreply.github.com>
Date: Tue, 15 Oct 2024 18:52:49 +0900
Subject: [PATCH] feat: add support for rotate_key to gemini

---
 book_maker/translator/gemini_translator.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/book_maker/translator/gemini_translator.py b/book_maker/translator/gemini_translator.py
index ff1f75e..07b0327 100644
--- a/book_maker/translator/gemini_translator.py
+++ b/book_maker/translator/gemini_translator.py
@@ -78,6 +78,7 @@ class Gemini(Base):
             or None  # Allow None, but not empty string
         )
 
+        genai.configure(api_key=next(self.keys))
         generation_config["temperature"] = temperature
 
     def create_convo(self):
@@ -96,7 +97,7 @@ class Gemini(Base):
         print(f"Using model {self.model}")
 
     def rotate_key(self):
-        pass
+        genai.configure(api_key=next(self.keys))
 
     def translate(self, text):
         t_text = ""