kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-21 22:27:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1816 from xywhnh/master

Browse Source 
修复gemini 插件的两个问题
This commit is contained in:
Saboteur7
2024-03-21 15:34:42 +08:00
committed by GitHub
parent f45ff5fd0a 800419e7cc
commit 2c4b8a44dc
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
bot/gemini/google_gemini_bot.py
View File

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