apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-27 14:25:20 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: consumers -> members

Browse source
This commit is contained in:
莘权 马 2023-11-22 15:59:35 +08:00
parent b73cbe7379
commit 5142cb59f7
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
metagpt/environment.py
View file

@ -29,7 +29,7 @@ class Environment(BaseModel):
"""
roles: dict[str, Role] = Field(default_factory=dict)
consumers: dict[Role, Set] = Field(default_factory=dict)
members: dict[Role, Set] = Field(default_factory=dict)
history: str = Field(default="") # For debug
class Config:
@ -61,7 +61,7 @@ class Environment(BaseModel):
logger.info(f"publish_message: {message.dump()}")
found = False
# According to the routing feature plan in Chapter 2.2.3.2 of RFC 113
for role, subscription in self.consumers.items():
for role, subscription in self.members.items():
if is_subscribed(message, subscription):
role.put_message(message)
found = True
@ -106,8 +106,8 @@ class Environment(BaseModel):
def get_subscription(self, obj):
"""Get the labels for messages to be consumed by the object."""
return self.consumers.get(obj, {})
return self.members.get(obj, {})
def set_subscription(self, obj, tags):
"""Set the labels for message to be consumed by the object"""
self.consumers[obj] = tags
self.members[obj] = tags