diff --git a/tests/metagpt/tools/test_translate.py b/tests/metagpt/tools/test_translate.py index 024bda3ca..53f00a88a 100644 --- a/tests/metagpt/tools/test_translate.py +++ b/tests/metagpt/tools/test_translate.py @@ -12,14 +12,15 @@ from metagpt.logs import logger from metagpt.tools.translator import Translator +@pytest.mark.asyncio @pytest.mark.usefixtures("llm_api") -def test_translate(llm_api): +async def test_translate(llm_api): poetries = [ ("Let life be beautiful like summer flowers", "花"), ("The ancient Chinese poetries are all songs.", "中国"), ] for i, j in poetries: prompt = Translator.translate_prompt(i) - rsp = llm_api.ask_batch([prompt]) + rsp = await llm_api.aask(prompt) logger.info(rsp) assert j in rsp