fixbug: intent detect

This commit is contained in:
莘权 马 2024-03-30 16:02:01 +08:00
parent de77fccc9a
commit 32223805a0
3 changed files with 62 additions and 36 deletions

View file

@ -0,0 +1,24 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from typing import List
import pytest
from metagpt.context import Context
from metagpt.roles.di.mgx import MGX
from metagpt.schema import Message
from tests.metagpt.actions.test_intent_detect import DEMO_CONTENT
@pytest.mark.asyncio
@pytest.mark.parametrize("user_messages", [[Message.model_validate(i) for i in DEMO_CONTENT if i["role"] == "user"]])
async def test_mgx(user_messages: List[Message]):
ctx = Context()
mgx = MGX(context=ctx)
for i in user_messages:
await mgx.run(i)
if __name__ == "__main__":
pytest.main([__file__, "-s"])