支持自定义openai_api_base

支持自定义openai_api_base 解决国内API被墙的问题，可以自定义使用自己的中转API
2026-02-22 08:28:42 +08:00 · 2023-03-15 00:14:39 +08:00
parent 9e3a5395c7
commit 2c8da59b47
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] ",