update gemini model

2026-03-05 08:06:21 +08:00 · 2024-07-04 01:44:53 +08:00
parent bb2880191a
commit 87525bb383
2 changed files with 8 additions and 3 deletions
--- a/bot/gemini/google_gemini_bot.py
+++ b/bot/gemini/google_gemini_bot.py
@@ -25,6 +25,8 @@ class GoogleGeminiBot(Bot):
        # 复用文心的token计算方式
        self.sessions = SessionManager(BaiduWenxinSession, model=conf().get("model") or "gpt-3.5-turbo")
        self.model = conf().get("model") or "gemini-pro"
+        if self.model == "gemini":
+            self.model = "gemini-pro"
    def reply(self, query, context: Context = None) -> Reply:
        try:
            if context.type != ContextType.TEXT: