Selaa lähdekoodia

Merge pull request #1816 from xywhnh/master

修复gemini 插件的两个问题
Saboteur7 2 vuotta sitten
vanhempi
säilyke
2c4b8a44dc
1 muutettua tiedostoa jossa 3 lisäystä ja 0 poistoa
  1. 3 0
      bot/gemini/google_gemini_bot.py

+ 3 - 0
bot/gemini/google_gemini_bot.py

@@ -44,6 +44,7 @@ class GoogleGeminiBot(Bot):
         except Exception as e:
             logger.error("[Gemini] fetch reply error, may contain unsafe content")
             logger.error(e)
+            return Reply(ReplyType.ERROR, "invoke [Gemini] api failed!")
 
     def _convert_to_gemini_messages(self, messages: list):
         res = []
@@ -63,6 +64,8 @@ class GoogleGeminiBot(Bot):
     def _filter_messages(self, messages: list):
         res = []
         turn = "user"
+        if not messages:
+            return res
         for i in range(len(messages) - 1, -1, -1):
             message = messages[i]
             if message.get("role") != turn: