소스 검색
Merge pull request #465 from B1gM8c/master
支持自定义openai_api_base
master
zhayujie
GitHub
1 년 전
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3개의 변경된 파일과
3개의 추가작업 그리고
0개의 파일을 삭제
-
bot/chatgpt/chat_gpt_bot.py
-
bot/openai/open_ai_bot.py
-
config-template.json
|
|
|
@@ -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 |
|
|
|
|
|
|
|
@@ -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): |
|
|
|
|
|
|
|
@@ -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] ", |
|
|
|
|
