apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-21 14:05:17 +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
metagpt/context.py
View file

@ -112,6 +112,7 @@ class Context(BaseModel):
return {
"workdir": str(self.repo.workdir) if self.repo else "",
"kwargs": {k: v for k, v in self.kwargs.__dict__.items()},
"cost_manager": self.cost_manager.model_dump_json(),
}
def deserialize(self, serialized_data: Dict[str, Any]):
@ -133,3 +134,6 @@ class Context(BaseModel):
if kwargs:
for k, v in kwargs.items():
self.kwargs.set(k, v)
cost_manager = serialized_data.get("cost_manager")
if cost_manager:
self.cost_manager.model_validate_json(cost_manager)