From 086fdb5856cb0ace93e684a4fd66aa3c27e46486 Mon Sep 17 00:00:00 2001 From: cmgzn Date: Thu, 26 Sep 2024 02:49:52 +0100 Subject: [PATCH] fix gemini logger --- bot/gemini/google_gemini_bot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bot/gemini/google_gemini_bot.py b/bot/gemini/google_gemini_bot.py index 4c488a1..12435b0 100644 --- a/bot/gemini/google_gemini_bot.py +++ b/bot/gemini/google_gemini_bot.py @@ -24,7 +24,7 @@ class GoogleGeminiBot(Bot): def __init__(self): super().__init__() self.api_key = conf().get("gemini_api_key") - # 复用文心的token计算方式 + # 复用chatGPT的token计算方式 self.sessions = SessionManager(ChatGPTSession, model=conf().get("model") or "gpt-3.5-turbo") self.model = conf().get("model") or "gemini-pro" if self.model == "gemini": @@ -38,7 +38,7 @@ class GoogleGeminiBot(Bot): session_id = context["session_id"] session = self.sessions.session_query(query, session_id) gemini_messages = self._convert_to_gemini_messages(self.filter_messages(session.messages)) - logger.info(f"[Gemini] messages={gemini_messages}") + logger.debug(f"[Gemini] messages={gemini_messages}") genai.configure(api_key=self.api_key) model = genai.GenerativeModel(self.model)