Spaces:
Build error
Build error
File size: 3,298 Bytes
01523b5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
from __future__ import annotations
from abc import abstractmethod
from typing import TYPE_CHECKING, Any, List, Optional
from agentverse.environments.simulation_env.rules.describer import (
BaseDescriber,
describer_registry,
)
from agentverse.environments.simulation_env.rules.order import BaseOrder, order_registry
from agentverse.environments.simulation_env.rules.selector import (
BaseSelector,
selector_registry,
)
from agentverse.environments.simulation_env.rules.updater import (
BaseUpdater,
updater_registry,
)
from agentverse.environments.simulation_env.rules.visibility import (
BaseVisibility,
visibility_registry,
)
from agentverse.environments import BaseRule
if TYPE_CHECKING:
from agentverse.environments.base import BaseEnvironment
from agentverse.message import Message
# class Rule(BaseModel):
class SimulationRule(BaseRule):
"""
Rule for the environment. It controls the speaking order of the agents
and maintain the set of visible agents for each agent.
"""
order: BaseOrder
visibility: BaseVisibility
selector: BaseSelector
updater: BaseUpdater
describer: BaseDescriber
def __init__(
self,
order_config,
visibility_config,
selector_config,
updater_config,
describer_config,
):
order = order_registry.build(**order_config)
visibility = visibility_registry.build(**visibility_config)
selector = selector_registry.build(**selector_config)
updater = updater_registry.build(**updater_config)
describer = describer_registry.build(**describer_config)
super().__init__(
order=order,
visibility=visibility,
selector=selector,
updater=updater,
describer=describer,
)
def get_next_agent_idx(
self, environment: BaseEnvironment, *args, **kwargs
) -> List[int]:
"""Return the index of the next agent to speak"""
return self.order.get_next_agent_idx(environment, *args, **kwargs)
def update_visible_agents(
self, environment: BaseEnvironment, *args, **kwargs
) -> None:
"""Update the set of visible agents for the agent"""
self.visibility.update_visible_agents(environment, *args, **kwargs)
def select_message(
self, environment: BaseEnvironment, messages: List[Message], *args, **kwargs
) -> List[Message]:
"""Select a set of valid messages from all the generated messages"""
return self.selector.select_message(environment, messages, *args, **kwargs)
def update_memory(self, environment: BaseEnvironment, *args, **kwargs) -> None:
"""For each message, add it to the memory of the agent who is able to see that message"""
self.updater.update_memory(environment, *args, **kwargs)
def get_env_description(
self, environment: BaseEnvironment, *args, **kwargs
) -> List[str]:
"""Return the description of the environment for each agent"""
return self.describer.get_env_description(environment, *args, **kwargs)
def reset(self) -> None:
self.order.reset()
self.visibility.reset()
self.selector.reset()
self.updater.reset()
self.describer.reset()
|