shen/bot-on-anything
1
0
Fork 0
mirror of https://github.com/zhayujie/bot-on-anything.git synced 2026-01-19 09:41:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #219 from wujiyu115/api_base

Browse Source 
update docker image
This commit is contained in:
zhayujie
2023-03-30 00:47:38 +08:00
committed by GitHub
parent 6ecdded007 7bfbbe8854
commit ea32ae42f9
6 changed files with 71 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
model/openai/chatgpt_model.py
View File

@@ -13,10 +13,14 @@ user_session = dict()
class ChatGPTModel(Model):
def __init__(self):
openai.api_key = model_conf(const.OPEN_AI).get('api_key')
api_base = model_conf(const.OPEN_AI).get('api_base')
if api_base:
openai.api_base = api_base
proxy = model_conf(const.OPEN_AI).get('proxy')
if proxy:
openai.proxy = proxy
log.info("[CHATGPT] api_base={} proxy={}".format(
api_base, proxy))
def reply(self, query, context=None):
# acquire reply content
if not context or not context.get('type') or context.get('type') == 'TEXT':

5
model/openai/open_ai_model.py
View File

@@ -13,6 +13,11 @@ user_session = dict()
class OpenAIModel(Model):
def __init__(self):
openai.api_key = model_conf(const.OPEN_AI).get('api_key')
api_base = model_conf(const.OPEN_AI).get('api_base')
if api_base:
openai.api_base = api_base
proxy = model_conf(const.OPEN_AI).get('proxy')
log.info("[OPEN_AI] api_base={}".format(openai.api_base))
def reply(self, query, context=None):