63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
import asyncio
|
|
import json
|
|
import logging
|
|
|
|
from spade.agent import Message
|
|
from spade.behaviour import CyclicBehaviour
|
|
from spade_bdi.bdi import BDIAgent
|
|
|
|
from control_backend.core.config import settings
|
|
|
|
|
|
class BeliefSetter(CyclicBehaviour):
|
|
"""
|
|
This is the behaviour that the BDI agent runs. This behaviour waits for incoming
|
|
message and processes it based on sender. Currently, it only waits for messages
|
|
containing beliefs from BeliefCollector and adds these to its KB.
|
|
"""
|
|
|
|
agent: BDIAgent
|
|
logger = logging.getLogger("BDI/Belief Setter")
|
|
|
|
async def run(self):
|
|
msg = await self.receive(timeout=0.1)
|
|
if msg:
|
|
self.logger.info(f"Received message {msg.body}")
|
|
self._process_message(msg)
|
|
await asyncio.sleep(1)
|
|
|
|
def _process_message(self, message: Message):
|
|
sender = message.sender.node # removes host from jid and converts to str
|
|
self.logger.debug("Sender: %s", sender)
|
|
|
|
match sender:
|
|
case settings.agent_settings.belief_collector_agent_name:
|
|
self.logger.debug("Processing message from belief collector.")
|
|
self._process_belief_message(message)
|
|
case _:
|
|
pass
|
|
|
|
def _process_belief_message(self, message: Message):
|
|
if not message.body:
|
|
return
|
|
|
|
match message.thread:
|
|
case "beliefs":
|
|
try:
|
|
beliefs: dict[str, list[list[str]]] = json.loads(message.body)
|
|
self._set_beliefs(beliefs)
|
|
except json.JSONDecodeError as e:
|
|
self.logger.error("Could not decode beliefs into JSON format: %s", e)
|
|
case _:
|
|
pass
|
|
|
|
def _set_beliefs(self, beliefs: dict[str, list[list[str]]]):
|
|
if self.agent.bdi is None:
|
|
self.logger.warning("Cannot set beliefs, since agent's BDI is not yet initialized.")
|
|
return
|
|
|
|
for belief, arguments_list in beliefs.items():
|
|
for arguments in arguments_list:
|
|
self.agent.bdi.set_belief(belief, *arguments)
|
|
self.logger.info("Set belief %s with arguments %s", belief, arguments)
|