src/control_backend/agents/__init__.py

@@ -1,17 +1,7 @@
-from .base import BaseAgent
-from .belief_collector.belief_collector import BeliefCollectorAgent
-from .llm.llm import LLMAgent
-from .ri_command_agent import RICommandAgent
-from .ri_communication_agent import RICommunicationAgent
-from .transcription.transcription_agent import TranscriptionAgent
-from .vad_agent import VADAgent
-
-__all__ = [
-    "BaseAgent",
-    "BeliefCollectorAgent",
-    "LLMAgent",
-    "RICommandAgent",
-    "RICommunicationAgent",
-    "TranscriptionAgent",
-    "VADAgent",
-]
+from .base import BaseAgent as BaseAgent
+from .belief_collector.belief_collector import BeliefCollectorAgent as BeliefCollectorAgent
+from .llm.llm import LLMAgent as LLMAgent
+from .ri_command_agent import RICommandAgent as RICommandAgent
+from .ri_communication_agent import RICommunicationAgent as RICommunicationAgent
+from .transcription.transcription_agent import TranscriptionAgent as TranscriptionAgent
+from .vad_agent import VADAgent as VADAgent

src/control_backend/agents/bdi/__init__.py

@@ -1,2 +1,2 @@
-from .bdi_core import BDICoreAgent
-from .text_extractor import TBeliefExtractorAgent
+from .bdi_core import BDICoreAgent as BDICoreAgent
+from .text_extractor import TBeliefExtractorAgent as TBeliefExtractorAgent

src/control_backend/agents/belief_collector/behaviours/continuous_collect.py

@@ -7,7 +7,6 @@ from spade.behaviour import CyclicBehaviour
 from control_backend.core.config import settings
 
 
-
 class ContinuousBeliefCollector(CyclicBehaviour):
     """
     Continuously collects beliefs/emotions from extractor agents:
@@ -24,8 +23,8 @@ class ContinuousBeliefCollector(CyclicBehaviour):
         # Parse JSON payload
         try:
             payload = json.loads(msg.body)
-        except Exception as e:
-            logger.warning(
+        except JSONDecodeError as e:
+            self.agent.logger.warning(
                 "BeliefCollector: failed to parse JSON from %s. Body=%r Error=%s",
                 sender_node,
                 msg.body,

src/control_backend/logging/__init__.py

@@ -1 +1 @@
-from .setup_logging import setup_logging
+from .setup_logging import setup_logging as setup_logging

src/control_backend/main.py

@@ -49,7 +49,8 @@ async def lifespan(app: FastAPI):
             RICommunicationAgent,
             {
                 "name": settings.agent_settings.ri_communication_agent_name,
-                "jid": f"{settings.agent_settings.ri_communication_agent_name}@{settings.agent_settings.host}",
+                "jid": f"{settings.agent_settings.ri_communication_agent_name}\
+                @{settings.agent_settings.host}",
                 "password": settings.agent_settings.ri_communication_agent_name,
                 "address": "tcp://*:5555",
                 "bind": True,
@@ -67,7 +68,8 @@ async def lifespan(app: FastAPI):
             BDICoreAgent,
             {
                 "name": settings.agent_settings.bdi_core_agent_name,
-                "jid": f"{settings.agent_settings.bdi_core_agent_name}@{settings.agent_settings.host}",
+                "jid": f"{settings.agent_settings.bdi_core_agent_name}@\
+                {settings.agent_settings.host}",
                 "password": settings.agent_settings.bdi_core_agent_name,
                 "asl": "src/control_backend/agents/bdi/rules.asl",
             },
@@ -76,7 +78,8 @@ async def lifespan(app: FastAPI):
             BeliefCollectorAgent,
             {
                 "name": settings.agent_settings.belief_collector_agent_name,
-                "jid": f"{settings.agent_settings.belief_collector_agent_name}@{settings.agent_settings.host}",
+                "jid": f"{settings.agent_settings.belief_collector_agent_name}@\
+                {settings.agent_settings.host}",
                 "password": settings.agent_settings.belief_collector_agent_name,
             },
         ),
@@ -84,7 +87,8 @@ async def lifespan(app: FastAPI):
             TBeliefExtractorAgent,
             {
                 "name": settings.agent_settings.text_belief_extractor_agent_name,
-                "jid": f"{settings.agent_settings.text_belief_extractor_agent_name}@{settings.agent_settings.host}",
+                "jid": f"{settings.agent_settings.text_belief_extractor_agent_name}@\
+                {settings.agent_settings.host}",
                 "password": settings.agent_settings.text_belief_extractor_agent_name,
             },
         ),