def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
def role_from_first_message( # pylint: disable=unused-argument
message: Message,
receiver_address: Address) -> BaseDialogue.Role:
"""Infer the role of the agent from an incoming/outgoing first message
:param message: an incoming/outgoing first message
:param receiver_address: the address of the receiving agent
:return: The role of the agent
"""
return BaseFipaDialogue.Role.SELLER
BaseFipaDialogues.__init__(
self,
self_address=self.context.agent_address,
role_from_first_message=role_from_first_message,
dialogue_class=FipaDialogue,
)
def __init__(self, **kwargs: Any) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
def role_from_first_message( # pylint: disable=unused-argument
message: Message, receiver_address: Address
) -> BaseDialogue.Role:
"""Infer the role of the agent from an incoming/outgoing first message
:param message: an incoming/outgoing first message
:param receiver_address: the address of the receiving agent
:return: The role of the agent
"""
return BaseLedgerApiDialogue.Role.AGENT
BaseLedgerApiDialogues.__init__(
self,
self_address=str(self.skill_id),
role_from_first_message=role_from_first_message,
dialogue_class=LedgerApiDialogue,
)
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
def role_from_first_message( # pylint: disable=unused-argument
message: Message, receiver_address: Address) -> Dialogue.Role:
"""Infer the role of the agent from an incoming/outgoing first message
:param message: an incoming/outgoing first message
:param receiver_address: the address of the receiving agent
:return: The role of the agent
"""
return 1 # type: ignore
Dialogues.__init__(
self,
self_address=self.context.agent_name,
end_states=[Mock()], # type: ignore
message_class=Message,
dialogue_class=Dialogue,
role_from_first_message=role_from_first_message,
)
def __init__(self, **kwargs: Any) -> None:
"""
Initialize dialogues.
:param agent_address: the address of the agent for whom dialogues are maintained
:return: None
"""
Model.__init__(self, **kwargs)
def role_from_first_message( # pylint: disable=unused-argument
message: Message,
receiver_address: Address) -> BaseDialogue.Role:
"""Infer the role of the agent from an incoming/outgoing first message
:param message: an incoming/outgoing first message
:param receiver_address: the address of the receiving agent
:return: The role of the agent
"""
return BaseHttpDialogue.Role.CLIENT
BaseHttpDialogues.__init__(
self,
self_address=self.context.agent_address,
role_from_first_message=role_from_first_message,
)
def __init__(self, **kwargs: Any) -> None:
"""
Initialize dialogues.
:return: None
"""
self.enabled = kwargs.pop("enabled", False)
self.metrics = kwargs.pop("metrics", [])
Model.__init__(self, **kwargs)
def role_from_first_message( # pylint: disable=unused-argument
message: Message, receiver_address: Address
) -> BaseDialogue.Role:
"""Infer the role of the agent from an incoming/outgoing first message
:param message: an incoming/outgoing first message
:param receiver_address: the address of the receiving agent
:return: The role of the agent
"""
return PrometheusDialogue.Role.AGENT
BasePrometheusDialogues.__init__(
self,
self_address=str(self.skill_id),
role_from_first_message=role_from_first_message,
)
def __init__(self, **kwargs: Any) -> None:
"""
Initialize dialogues.
:param agent_address: the address of the agent for whom dialogues are maintained
:return: None
"""
Model.__init__(self, **kwargs)
def role_from_first_message( # pylint: disable=unused-argument
message: Message,
receiver_address: Address) -> BaseDialogue.Role:
"""Infer the role of the agent from an incoming/outgoing first message
:param message: an incoming/outgoing first message
:param receiver_address: the address of the receiving agent
:return: The role of the agent in this dialogue
"""
if (message.performative == HttpMessage.Performative.REQUEST
and message.sender != receiver_address
) or (message.performative == HttpMessage.Performative.RESPONSE
and message.sender == receiver_address):
return BaseHttpDialogue.Role.SERVER
return BaseHttpDialogue.Role.CLIENT
BaseHttpDialogues.__init__(
self,
self_address=str(self.skill_id),
role_from_first_message=role_from_first_message,
)
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
FipaDialogues.__init__(self, agent_address=self.context.agent_address)
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
self._items: Dict[OnChainItemId, OnChainItem] = {}
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
FIPADialogues.__init__(self)
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
BaseDefaultDialogues.__init__(self, self.context.agent_name)
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
BaseSigningDialogues.__init__(self, self.context.agent_address)
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
BaseTacDialogues.__init__(self, self.context.agent_address)
self._dialogue_by_address = {} # type: Dict[str, Dialogue]
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:param agent_address: the address of the agent for whom dialogues are maintained
:return: None
"""
Model.__init__(self, **kwargs)
BaseSigningDialogues.__init__(self, self.context.agent_address)
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:param agent_address: the address of the agent for whom dialogues are maintained
:return: None
"""
Model.__init__(self, **kwargs)
BaseOefSearchDialogues.__init__(
self,
self.context.agent_address + "_" + str(self.context.skill_id))
def __init__(self, **kwargs) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
self._token_whitelist: List[PoolAddress] = []
self._pool_statuses: Dict[Pool, PoolStatus]
self._active_pool: Optional[Pool] = None
self._next_active_pool: Optional[Pool] = None
self._agent_status: AgentStatus = AgentStatus.PAUSED
def __init__(self, **kwargs: Any) -> None:
"""
Initialize dialogues.
:return: None
"""
self.url = kwargs.pop("url", DEFAULT_URL)
self.coin_id = kwargs.pop("coin_id", DEFAULT_COIN_ID)
self.currency = kwargs.pop("currency", DEFAULT_CURRENCY)
self.decimals = kwargs.pop("decimals", DEFAULT_DECIMALS)
self.use_http_server = kwargs.pop("use_http_server",
DEFAULT_USE_HTTP_SERVER)
Model.__init__(self, **kwargs)
def __init__(self, **kwargs: Any) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
def role_from_first_message(
message: Message, receiver_address: Address) -> Dialogue.Role:
"""Infer the role of the agent from an incoming/outgoing first message
:param message: an incoming/outgoing first message
:param receiver_address: the address of the receiving agent
:return: The role of the agent
"""
fipa_message = cast(FipaMessage, message)
if fipa_message.performative != FipaMessage.Performative.CFP:
raise ValueError(
"First message must be a CFP!") # pragma: nocover
query = fipa_message.query
if query.model is None:
raise ValueError(
"Query must have a data model!") # pragma: nocover
if query.model.name not in [
SUPPLY_DATAMODEL_NAME,
DEMAND_DATAMODEL_NAME,
]:
raise ValueError( # pragma: nocover
"Query data model name must be in [{},{}]".format(
SUPPLY_DATAMODEL_NAME, DEMAND_DATAMODEL_NAME))
if message.sender != receiver_address: # message is by other
is_seller = (
query.model.name == SUPPLY_DATAMODEL_NAME
) # the counterparty is querying for supply/sellers (this agent is receiving their CFP so is the seller)
else: # message is by self
is_seller = (
query.model.name == DEMAND_DATAMODEL_NAME
) # the agent is querying for demand/buyers (this agent is sending the CFP so it is the seller)
role = FipaDialogue.Role.SELLER if is_seller else FipaDialogue.Role.BUYER
return role
BaseFipaDialogues.__init__(
self,
self_address=self.context.agent_address,
role_from_first_message=role_from_first_message,
dialogue_class=FipaDialogue,
)
def __init__(self, **kwargs: Any) -> None:
"""
Initialize dialogues.
:return: None
"""
self.url = kwargs.pop("url", DEFAULT_URL)
self.method = kwargs.pop("method", DEFAULT_METHOD)
self.body = kwargs.pop("body", DEFAULT_BODY)
self.outputs = kwargs.pop("outputs", DEFAULT_OUTPUTS)
self.decimals = kwargs.pop("decimals", DEFAULT_DECIMALS)
self.use_http_server = kwargs.pop("use_http_server",
DEFAULT_USE_HTTP_SERVER)
Model.__init__(self, **kwargs)
self._validate_config()
def __init__(self, **kwargs: Any) -> None:
"""
Initialize dialogues.
:return: None
"""
Model.__init__(self, **kwargs)
def role_from_first_message( # pylint: disable=unused-argument
message: Message, receiver_address: Address) -> Dialogue.Role:
"""Infer the role of the agent from an incoming/outgoing first message
:param message: an incoming/outgoing first message
:param receiver_address: the address of the receiving agent
:return: The role of the agent
"""
return AgentEnvironmentDialogue.Role.ENVIRONMENT
BaseAgentEnvironmentDialogues.__init__(
self,
self_address=self.context.agent_address,
role_from_first_message=role_from_first_message,
)
def test_model_parse_module_missing_class():
"""Test Model.parse_module when a class is missing."""
skill_context = SkillContext(skill=MagicMock(
skill_id=PublicId.from_str("author/name:0.1.0")))
dummy_models_path = Path(ROOT_DIR, "tests", "data", "dummy_skill")
with unittest.mock.patch.object(aea.skills.base._default_logger,
"warning") as mock_logger_warning:
models_by_id = Model.parse_module(
dummy_models_path,
{
"dummy_model": SkillComponentConfiguration("DummyModel"),
"unknown_model": SkillComponentConfiguration("UnknownModel"),
},
skill_context,
)
mock_logger_warning.assert_called_with(
"Model 'UnknownModel' cannot be found.")
assert "dummy_model" in models_by_id
def setup_class(cls):
"""Set the tests up."""
skill_context = SkillContext()
skill_config = SkillConfig(name="simple_skill",
author="fetchai",
version="0.1.0")
class MyHandler(Handler):
def setup(self):
pass
def handle(self, message: Message):
pass
def teardown(self):
pass
class MyBehaviour(Behaviour):
def setup(self):
pass
def act(self):
pass
def teardown(self):
pass
cls.handler_name = "some_handler"
cls.handler = MyHandler(skill_context=skill_context,
name=cls.handler_name)
cls.model_name = "some_model"
cls.model = Model(skill_context=skill_context, name=cls.model_name)
cls.behaviour_name = "some_behaviour"
cls.behaviour = MyBehaviour(skill_context=skill_context,
name=cls.behaviour_name)
cls.skill = Skill(
skill_config,
skill_context,
handlers={cls.handler.name: cls.handler},
models={cls.model.name: cls.model},
behaviours={cls.behaviour.name: cls.behaviour},
)
def test_model_parse_module_without_configs():
"""Call Model.parse_module without configurations."""
assert Model.parse_module(MagicMock(), {}, MagicMock()) == {}
def __init__(self, name, skill_context):
Model.__init__(self, name, skill_context)
Dialogues.__init__(self, "addr", MagicMock(), Message, Dialogue,
role_from_first_message)
