github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-25 19:39:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: proxy by openai side

Browse Source
This commit is contained in:
yihong0618 2023-09-23 14:48:12 +08:00
parent 3e52803704
commit 77072df371
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/model/proxy/llms/chatgpt.py
View File

@ -20,6 +20,8 @@ def chatgpt_generate_stream(
print(f"Model: {model}, model_params: {model_params}") print(f"Model: {model}, model_params: {model_params}")
proxy_api_key = model_params.proxy_api_key proxy_api_key = model_params.proxy_api_key
if model_params.http_proxy:
openai.proxy = model_params.http_proxy
openai.api_key = os.getenv("OPENAI_API_KEY") or proxy_api_key openai.api_key = os.getenv("OPENAI_API_KEY") or proxy_api_key
proxyllm_backend = model_params.proxyllm_backend proxyllm_backend = model_params.proxyllm_backend
if not proxyllm_backend: if not proxyllm_backend: