From de259a25316ea91d2ae74b5f495bada2e0632ade Mon Sep 17 00:00:00 2001
From: geekan <geekan@foxmail.com>
Date: Fri, 29 Dec 2023 10:25:35 +0800
Subject: [PATCH] fix translate

---
 tests/metagpt/tools/test_translate.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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