refactor: Override cause_by

This commit is contained in:
莘权 马 2023-11-04 16:20:47 +08:00
parent c4eb028a83
commit 1febf168e7
2 changed files with 43 additions and 0 deletions

View file

@ -11,6 +11,7 @@ import json
import pytest
from metagpt.actions import Action
from metagpt.schema import AIMessage, Message, Routes, SystemMessage, UserMessage
@ -50,6 +51,15 @@ def test_message():
assert m.cause_by == "c"
assert m.get_meta("x") == "d"
m.cause_by = "Message"
assert m.cause_by == "Message"
m.cause_by = Action
assert m.cause_by == Action.get_class_name()
m.cause_by = Action()
assert m.cause_by == Action.get_class_name()
m.content = "b"
assert m.content == "b"
@pytest.mark.asyncio
def test_routes():