xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-15 06:50:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: 通过 one-api 支持google gemini

Browse Source
This commit is contained in:
RockChinQ
2023-12-17 22:36:30 +08:00
parent 707afdcdf9
commit d3f401c54d
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/openai/modelmgr.py
+2
View File
@@ -49,6 +49,7 @@ CHAT_COMPLETION_MODELS = {
"qwen-plus-v1", "qwen-plus-v1",
"ERNIE-Bot", "ERNIE-Bot",
"ERNIE-Bot-turbo", "ERNIE-Bot-turbo",
"gemini-pro",
} }
EDIT_MODELS = { EDIT_MODELS = {
@@ -90,6 +91,7 @@ def count_chat_completion_tokens(messages: list, model: str) -> int:
"qwen-plus-v1", "qwen-plus-v1",
"ERNIE-Bot", "ERNIE-Bot",
"ERNIE-Bot-turbo", "ERNIE-Bot-turbo",
"gemini-pro",
}: }:
tokens_per_message = 3 tokens_per_message = 3
tokens_per_name = 1 tokens_per_name = 1