diff --git a/README.md b/README.md index b07772f..9b7c8a9 100644 --- a/README.md +++ b/README.md @@ -61,10 +61,11 @@ - HuggingChat, 由[插件](https://github.com/RockChinQ/revLibs)接入, 仅支持英文 - Claude, 由[插件](https://github.com/RockChinQ/revLibs)接入 - Google Bard, 由[插件](https://github.com/RockChinQ/revLibs)接入 +- Google Gemini Pro、Azure、Anthropic Claude、智谱 ChatGLM、百度文心一言、讯飞星火认知、阿里通义千问、360 智脑等官方接口, 通过[One API](https://github.com/songquanpeng/one-api)接入 ### 模型聚合平台 -- [One API](https://github.com/songquanpeng/one-api), Azure、Anthropic Claude、Google PaLM 2、智谱 ChatGLM、百度文心一言、讯飞星火认知、阿里通义千问以及 360 智脑等模型的官方接口转换成 OpenAI API 接入,QChatGPT 原生支持,您需要先配置 One API,之后在`config.py`中设置反向代理和`One API`的密钥后使用。 +- [One API](https://github.com/songquanpeng/one-api), Azure、Anthropic Claude、Google Gemini Pro、智谱 ChatGLM、百度文心一言、讯飞星火认知、阿里通义千问以及 360 智脑等模型的官方接口转换成 OpenAI API 接入,QChatGPT 原生支持,您需要先配置 One API,之后在`config.py`中设置反向代理和`One API`的密钥后使用。 - [gpt4free](https://github.com/xtekky/gpt4free), 破解以免费使用多个平台的各种文字模型, 由[插件](https://github.com/RockChinQ/revLibs)接入, 无需鉴权, 稳定性较差。 - [Poe](https://poe.com), 破解免费使用Poe上多个平台的模型, 由[oliverkirk-sudo/ChatPoeBot](https://github.com/oliverkirk-sudo/ChatPoeBot)接入(由于 Poe 上可用的大部分模型现已通过[revLibs插件](https://github.com/RockChinQ/revLubs)或其他方式接入,此插件现已停止维护)。 diff --git a/config-template.py b/config-template.py index 87234f2..a6e5638 100644 --- a/config-template.py +++ b/config-template.py @@ -254,6 +254,7 @@ auto_reset = True # "qwen-plus-v1", # "ERNIE-Bot", # "ERNIE-Bot-turbo", +# "gemini-pro", completion_api_params = { "model": "gpt-3.5-turbo", "temperature": 0.9, # 数值越低得到的回答越理性,取值范围[0, 1] diff --git a/pkg/openai/modelmgr.py b/pkg/openai/modelmgr.py index 6e7947b..699e79d 100644 --- a/pkg/openai/modelmgr.py +++ b/pkg/openai/modelmgr.py @@ -49,6 +49,7 @@ CHAT_COMPLETION_MODELS = { "qwen-plus-v1", "ERNIE-Bot", "ERNIE-Bot-turbo", + "gemini-pro", } EDIT_MODELS = { @@ -90,6 +91,7 @@ def count_chat_completion_tokens(messages: list, model: str) -> int: "qwen-plus-v1", "ERNIE-Bot", "ERNIE-Bot-turbo", + "gemini-pro", }: tokens_per_message = 3 tokens_per_name = 1