feat: 通过 one-api 支持google gemini

2026-06-02 03:55:55 +00:00 · 2023-12-17 22:36:30 +08:00
parent 707afdcdf9
commit d3f401c54d
1 changed files with 2 additions and 0 deletions
--- 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