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

revert pytest.MonkeyPatch

Browse source
This commit is contained in:
seehi 2023-12-07 10:23:08 +08:00
parent f4505d0e39
commit 97f156b10d
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -15,8 +15,8 @@ def mock_llm_ask(self, messages: list[dict]) -> dict:
return default_resp
def test_zhipuai_completion(monkeypatch: pytest.MonkeyPatch):
monkeypatch.setattr(ZhiPuAIGPTAPI, "completion", mock_llm_ask)
def test_zhipuai_completion(mocker):
mocker.patch("metagpt.provider.zhipuai_api.ZhiPuAIGPTAPI.completion", mock_llm_ask)
resp = ZhiPuAIGPTAPI().completion(messages)
assert resp["code"] == 200
@ -28,8 +28,8 @@ async def mock_llm_aask(self, messgaes: list[dict], stream: bool = False) -> dic
@pytest.mark.asyncio
async def test_zhipuai_acompletion(monkeypatch: pytest.MonkeyPatch):
monkeypatch.setattr(ZhiPuAIGPTAPI, "acompletion_text", mock_llm_aask)
async def test_zhipuai_acompletion(mocker):
mocker.patch("metagpt.provider.zhipuai_api.ZhiPuAIGPTAPI.acompletion_text", mock_llm_aask)
resp = await ZhiPuAIGPTAPI().acompletion_text(messages, stream=False)