fix translate

This commit is contained in:
geekan 2023-12-29 10:25:35 +08:00
parent 235b5b5189
commit de259a2531

View file

@ -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