Merge pull request #465 from B1gM8c/master

支持自定义openai_api_base
2026-02-19 17:10:03 +08:00 · 2023-03-15 00:24:04 +08:00
parent 9e3a5395c7 2c8da59b47
commit 1dc39af423
3 changed files with 3 additions and 0 deletions
--- a/bot/chatgpt/chat_gpt_bot.py
+++ b/bot/chatgpt/chat_gpt_bot.py
@@ -16,6 +16,7 @@ else:
 class ChatGPTBot(Bot):
    def __init__(self):
        openai.api_key = conf().get('open_ai_api_key')
+        openai.api_base = conf().get('open_ai_api_base')
        proxy = conf().get('proxy')
        if proxy:
            openai.proxy = proxy
--- a/bot/openai/open_ai_bot.py
+++ b/bot/openai/open_ai_bot.py
@@ -12,6 +12,7 @@ user_session = dict()
 class OpenAIBot(Bot):
    def __init__(self):
        openai.api_key = conf().get('open_ai_api_key')
+        openai.api_base = conf().get('open_ai_api_base')


    def reply(self, query, context=None):
--- a/config-template.json
+++ b/config-template.json
@@ -1,5 +1,6 @@
 {
  "open_ai_api_key": "YOUR API KEY",
+  "open_ai_api_base":"",
  "proxy": "",
  "single_chat_prefix": ["bot", "@bot"],
  "single_chat_reply_prefix": "[bot] ",