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

support new openai package

Browse source
This commit is contained in:
seehi 2023-12-05 11:26:54 +08:00
parent 5f69878a08
commit eaf531e0ac
9 changed files with 866 additions and 137 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,27 +6,17 @@ import pytest
from metagpt.provider.zhipuai_api import ZhiPuAIGPTAPI
default_resp = {"code": 200, "data": {"choices": [{"role": "assistant", "content": "I'm chatglm-turbo"}]}}
default_resp = {
"code": 200,
"data": {
"choices": [
{"role": "assistant", "content": "I'm chatglm-turbo"}
]
}
}
messages = [
{"role": "user", "content": "who are you"}
]
messages = [{"role": "user", "content": "who are you"}]
def mock_llm_ask(self, messages: list[dict]) -> dict:
return default_resp
def test_zhipuai_completion(mocker):
mocker.patch("metagpt.provider.zhipuai_api.ZhiPuAIGPTAPI.completion", mock_llm_ask)
def test_zhipuai_completion(monkeypatch: pytest.MonkeyPatch):
monkeypatch.setattr(ZhiPuAIGPTAPI, "completion", mock_llm_ask)
resp = ZhiPuAIGPTAPI().completion(messages)
assert resp["code"] == 200
@ -38,8 +28,8 @@ async def mock_llm_aask(self, messgaes: list[dict], stream: bool = False) -> dic
@pytest.mark.asyncio
async def test_zhipuai_acompletion(mocker):
mocker.patch("metagpt.provider.zhipuai_api.ZhiPuAIGPTAPI.acompletion_text", mock_llm_aask)
async def test_zhipuai_acompletion(monkeypatch: pytest.MonkeyPatch):
monkeypatch.setattr(ZhiPuAIGPTAPI, "acompletion_text", mock_llm_aask)
resp = await ZhiPuAIGPTAPI().acompletion_text(messages, stream=False)