refactor: restructure to make sure the Bel agents are also part of BDI.
ref: N25B-257
This commit is contained in:
@@ -1 +1,5 @@
|
||||
from .bdi_core_agent import BDICoreAgent as BDICoreAgent
|
||||
from .bdi_belief_collector_agent.bel_collector_agent import (
|
||||
BDIBeliefCollectorAgent as BDIBeliefCollectorAgent,
|
||||
)
|
||||
from .bdi_core_agent.bdi_core_agent import BDICoreAgent as BDICoreAgent
|
||||
from .bdi_text_belief_agent.bdi_text_belief_agent import BDITextBeliefAgent as BDITextBeliefAgent
|
||||
|
||||
@@ -3,9 +3,9 @@ from control_backend.agents.base import BaseAgent
|
||||
from .behaviours.continuous_collect import ContinuousBeliefCollector
|
||||
|
||||
|
||||
class BelCollectorAgent(BaseAgent):
|
||||
class BDIBeliefCollectorAgent(BaseAgent):
|
||||
async def setup(self):
|
||||
self.logger.info("BelCollectorAgent starting (%s)", self.jid)
|
||||
self.logger.info("BDIBeliefCollectorAgent starting (%s)", self.jid)
|
||||
# Attach the continuous collector behaviour (listens and forwards to BDI)
|
||||
self.add_behaviour(ContinuousBeliefCollector())
|
||||
self.logger.info("BelCollectorAgent ready.")
|
||||
self.logger.info("BDIBeliefCollectorAgent ready.")
|
||||
@@ -32,7 +32,7 @@ class BeliefSetterBehaviour(CyclicBehaviour):
|
||||
self.agent.logger.debug("Processing message from sender: %s", sender)
|
||||
|
||||
match sender:
|
||||
case settings.agent_settings.bel_collector_agent_name:
|
||||
case settings.agent_settings.bdi_belief_collector_agent_name:
|
||||
self.agent.logger.debug(
|
||||
"Message is from the belief collector agent. Processing as belief message."
|
||||
)
|
||||
@@ -3,6 +3,6 @@ from control_backend.agents.base import BaseAgent
|
||||
from .behaviours.text_belief_extractor import BeliefFromText
|
||||
|
||||
|
||||
class BelTextExtractAgent(BaseAgent):
|
||||
class BDITextBeliefAgent(BaseAgent):
|
||||
async def setup(self):
|
||||
self.add_behaviour(BeliefFromText())
|
||||
@@ -71,7 +71,7 @@ class BeliefFromText(CyclicBehaviour):
|
||||
belief_message = Message()
|
||||
|
||||
belief_message.to = (
|
||||
settings.agent_settings.bel_collector_agent_name
|
||||
settings.agent_settings.bdi_belief_collector_agent_name
|
||||
+ "@"
|
||||
+ settings.agent_settings.host
|
||||
)
|
||||
@@ -95,7 +95,9 @@ class BeliefFromText(CyclicBehaviour):
|
||||
belief_msg = Message()
|
||||
|
||||
belief_msg.to = (
|
||||
settings.agent_settings.bel_collector_agent_name + "@" + settings.agent_settings.host
|
||||
settings.agent_settings.bdi_belief_collector_agent_name
|
||||
+ "@"
|
||||
+ settings.agent_settings.host
|
||||
)
|
||||
belief_msg.body = payload
|
||||
belief_msg.thread = "beliefs"
|
||||
@@ -1,4 +0,0 @@
|
||||
from .bel_collector_agent.bel_collector_agent import BelCollectorAgent as BelCollectorAgent
|
||||
from .bel_text_extract_agent.bel_text_extract_agent import (
|
||||
BelTextExtractAgent as BelTextExtractAgent,
|
||||
)
|
||||
@@ -11,7 +11,7 @@ class BelTextAgent(Agent):
|
||||
class SendOnceBehaviourBlfText(OneShotBehaviour):
|
||||
async def run(self):
|
||||
to_jid = (
|
||||
settings.agent_settings.bel_collector_agent_name
|
||||
settings.agent_settings.bdi_belief_collector_agent_name
|
||||
+ "@"
|
||||
+ settings.agent_settings.host
|
||||
)
|
||||
|
||||
@@ -47,7 +47,7 @@ class PerTranscriptionAgent(BaseAgent):
|
||||
async def _share_transcription(self, transcription: str):
|
||||
"""Share a transcription to the other agents that depend on it."""
|
||||
receiver_jids = [
|
||||
settings.agent_settings.texbel_text_extractor_agent_name
|
||||
settings.agent_settings.texbdi_text_belief_agent_name
|
||||
+ "@"
|
||||
+ settings.agent_settings.host,
|
||||
] # Set message receivers here
|
||||
|
||||
Reference in New Issue
Block a user