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

update gemini model

Browse Source
This commit is contained in:
6vision
2024-07-04 01:22:55 +08:00
parent b9e84fefbd
commit bb2880191a
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bot/gemini/google_gemini_bot.py
View File

@@ -24,7 +24,7 @@ class GoogleGeminiBot(Bot):
self.api_key = conf().get("gemini_api_key") self.api_key = conf().get("gemini_api_key")
# 复用文心的token计算方式 # 复用文心的token计算方式
self.sessions = SessionManager(BaiduWenxinSession, model=conf().get("model") or "gpt-3.5-turbo") self.sessions = SessionManager(BaiduWenxinSession, model=conf().get("model") or "gpt-3.5-turbo")
self.model = conf().get("model") or "gemini-pro"
def reply(self, query, context: Context = None) -> Reply: def reply(self, query, context: Context = None) -> Reply:
try: try:
if context.type != ContextType.TEXT: if context.type != ContextType.TEXT:
@@ -35,7 +35,7 @@ class GoogleGeminiBot(Bot):
session = self.sessions.session_query(query, session_id) session = self.sessions.session_query(query, session_id)
gemini_messages = self._convert_to_gemini_messages(self.filter_messages(session.messages)) gemini_messages = self._convert_to_gemini_messages(self.filter_messages(session.messages))
genai.configure(api_key=self.api_key) genai.configure(api_key=self.api_key)
model = genai.GenerativeModel('gemini-pro') model = genai.GenerativeModel(self.model)
response = model.generate_content(gemini_messages) response = model.generate_content(gemini_messages)
reply_text = response.text reply_text = response.text
self.sessions.session_reply(reply_text, session_id) self.sessions.session_reply(reply_text, session_id)

2
bridge/bridge.py
View File

@@ -36,7 +36,7 @@ class Bridge(object):
self.btype["chat"] = const.QWEN self.btype["chat"] = const.QWEN
if model_type in [const.QWEN_TURBO, const.QWEN_PLUS, const.QWEN_MAX]: if model_type in [const.QWEN_TURBO, const.QWEN_PLUS, const.QWEN_MAX]:
self.btype["chat"] = const.QWEN_DASHSCOPE self.btype["chat"] = const.QWEN_DASHSCOPE
if model_type in [const.GEMINI]: if model_type and model_type.startswith("gemini"):
self.btype["chat"] = const.GEMINI self.btype["chat"] = const.GEMINI
if model_type in [const.ZHIPU_AI]: if model_type in [const.ZHIPU_AI]:
self.btype["chat"] = const.ZHIPU_AI self.btype["chat"] = const.ZHIPU_AI