Просмотр исходного кода

Merge pull request #1816 from xywhnh/master

修复gemini 插件的两个问题
master
Saboteur7 GitHub 9 месяцев назад
Родитель
Сommit
2c4b8a44dc
Не найден GPG ключ соответствующий данной подписи Идентификатор GPG ключа: B5690EEEBB952194
1 измененных файлов: 3 добавлений и 0 удалений
  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:


Загрузка…
Отмена
Сохранить