apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-20 15:38:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Bugfix] Set openai proxy for class ZhiPuAPTAPI

Browse source 
When using ZHIPUAI as the large model provider, it is not possible to access ZHIPUAI in an HTTP proxy environment, and the following error will be reported:
openai.error.APIConnectionError: Error communicating with OpenAI

So we need set proxy for class ZhiPuAPTAPI.
This commit is contained in:
xiaofenggang 2023-12-25 16:04:58 +00:00
parent 4f52b47610
commit a87b5056d7
2 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/metagpt/provider/test_zhipuai_api.py
View file

@ -35,3 +35,10 @@ async def test_zhipuai_acompletion(mocker):
assert resp["code"] == 200
assert "chatglm-turbo" in resp["data"]["choices"][0]["content"]
def test_zhipuai_proxy(mocker):
import openai
from metagpt.config import CONFIG
CONFIG.openai_proxy = 'http://127.0.0.1:8080'
_ = ZhiPuAIGPTAPI()
assert openai.proxy == CONFIG.openai_proxy