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

merge main

Browse source
This commit is contained in:
geekan 2023-12-26 22:17:15 +08:00
commit 5c7cdf5ee7
2 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
metagpt/provider/zhipuai_api.py
View file

@ -49,6 +49,9 @@ class ZhiPuAIGPTAPI(BaseLLM):
zhipuai.api_key = config.zhipuai_api_key
# due to use openai sdk, set the api_key but it will't be used.
# 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:
# FIXME: openai v1.x sdk has no proxy support
openai.proxy = config.openai_proxy
def _const_kwargs(self, messages: list[dict]) -> dict:
kwargs = {"model": self.model, "prompt": messages, "temperature": 0.3}

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

@ -51,3 +51,13 @@ async def test_zhipuai_acompletion(mocker):
resp = await zhipu_gpt.aask(prompt_msg)
assert resp == resp_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