Merge branch 'feat_qianfan' of github.com:better629/MetaGPT into feat_qianfan

tests/metagpt/actions/test_rebuild_class_view.py

@@ -14,6 +14,7 @@ from metagpt.actions.rebuild_class_view import RebuildClassView
 from metagpt.llm import LLM
 
 
+@pytest.mark.skip
 @pytest.mark.asyncio
 async def test_rebuild(context):
     action = RebuildClassView(

tests/metagpt/actions/test_summarize_code.py

@@ -176,6 +176,7 @@ class Snake:
 """
 
 
+@pytest.mark.skip
 @pytest.mark.asyncio
 async def test_summarize_code(context):
     git_dir = Path(__file__).parent / f"unittest/{uuid.uuid4().hex}"

tests/metagpt/provider/test_zhipuai_api.py

@@ -19,7 +19,7 @@ resp_cont = resp_cont_tmpl.format(name=name)
 default_resp = get_part_chat_completion(name)
 
 
-async def mock_zhipuai_acreate_stream(**kwargs):
+async def mock_zhipuai_acreate_stream(self, **kwargs):
     class MockResponse(object):
         async def _aread(self):
             class Iterator(object):
@@ -39,7 +39,7 @@ async def mock_zhipuai_acreate_stream(**kwargs):
     return MockResponse()
 
 
-async def mock_zhipuai_acreate(**kwargs) -> dict:
+async def mock_zhipuai_acreate(self, **kwargs) -> dict:
     return default_resp
 
 

tests/metagpt/roles/ci/test_code_interpreter.py

@@ -17,3 +17,7 @@ async def test_code_interpreter(mocker, auto_run):
     rsp = await ci.run(requirement)
     logger.info(rsp)
     assert len(rsp.content) > 0
+
+    finished_tasks = ci.planner.plan.get_finished_tasks()
+    assert len(finished_tasks) > 0
+    assert len(finished_tasks[0].code) > 0  # check one task to see if code is recorded

tests/metagpt/utils/test_text.py

@@ -42,6 +42,7 @@ def test_reduce_message_length(msgs, model_name, system_text, reserved, expected
         (" ".join("Hello World." for _ in range(1000)), "Prompt: {}", "gpt-3.5-turbo-16k", "System", 3000, 1),
         (" ".join("Hello World." for _ in range(4000)), "Prompt: {}", "gpt-4", "System", 2000, 2),
         (" ".join("Hello World." for _ in range(8000)), "Prompt: {}", "gpt-4-32k", "System", 4000, 1),
+        (" ".join("Hello World" for _ in range(8000)), "Prompt: {}", "gpt-3.5-turbo", "System", 1000, 8),
     ],
 )
 def test_generate_prompt_chunk(text, prompt_template, model_name, system_text, reserved, expected):