From 8dbefd6fc42bc2086aff9beb0b0daef7d567772f Mon Sep 17 00:00:00 2001
From: Risin <5331402+risin42@users.noreply.github.com>
Date: Tue, 15 Oct 2024 18:51:30 +0900
Subject: [PATCH] feat: add support for --use_context option to gemini

---
 book_maker/translator/gemini_translator.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/book_maker/translator/gemini_translator.py b/book_maker/translator/gemini_translator.py
index fc4eb61..ff1f75e 100644
--- a/book_maker/translator/gemini_translator.py
+++ b/book_maker/translator/gemini_translator.py
@@ -59,11 +59,13 @@ class Gemini(Base):
         language,
         prompt_template=None,
         prompt_sys_msg=None,
+        context_flag=False,
         temperature=1.0,
         interval=0.01,
         **kwargs,
     ) -> None:
         super().__init__(key, language)
+        self.context_flag = context_flag
         self.interval = interval
         self.prompt = (
             prompt_template
@@ -125,8 +127,12 @@ class Gemini(Base):
             print(str(e))
             t_text = "Can not translate by other reason.(因安全问题不能翻译)"
 
-        if len(self.convo.history) > 10:
-            self.convo.history = self.convo.history[2:]
+
+        if self.context_flag:
+            if len(self.convo.history) > 10:
+                self.convo.history = self.convo.history[2:]
+        else:
+            self.convo.history = []
 
         print("[bold green]" + re.sub("\n{3,}", "\n\n", t_text) + "[/bold green]")
         # for rate limit(RPM)