apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-24 14:15:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #630 from springasa/main

Browse source 
[Bugfix] Set openai proxy for class ZhiPuAPTAPI
This commit is contained in:
geekan 2023-12-26 22:12:51 +08:00 • committed by GitHub
commit b70cd88af2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
metagpt/provider/zhipuai_api.py
View file

@ -50,6 +50,8 @@ class ZhiPuAIGPTAPI(BaseGPTAPI):
assert config.zhipuai_api_key
zhipuai.api_key = config.zhipuai_api_key
openai.api_key = zhipuai.api_key # due to use openai sdk, set the api_key but it will't be used.
if config.openai_proxy:
openai.proxy = config.openai_proxy
def _const_kwargs(self, messages: list[dict]) -> dict:
kwargs = {"model": self.model, "prompt": messages, "temperature": 0.3}

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