apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-11 15:15:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add cost_manager

Browse source
This commit is contained in:
莘权 马 2024-03-25 23:23:43 +08:00
parent 630eecd18b
commit 6c2d413664
2 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/metagpt/serialize_deserialize/test_team.py
View file

@ -150,6 +150,7 @@ async def test_team_recover_multi_roles_save(mocker, context):
@pytest.mark.asyncio
async def test_context(context):
context.kwargs.set("a", "a")
context.cost_manager.max_budget = 9
company = Team(context=context)
save_to = context.repo.workdir / "serial"
@ -158,7 +159,8 @@ async def test_context(context):
company.deserialize(save_to, Context())
assert company.env.context.repo
assert company.env.context.repo.workdir == context.repo.workdir
assert context.kwargs.a == "a"
assert company.env.context.kwargs.a == "a"
assert company.env.context.cost_manager.max_budget == context.cost_manager.max_budget
if __name__ == "__main__":