chore: making tests pass after merge

I had to make use of monkeypatch to simulate using the config file

ref: N25B-236
This commit is contained in:
Pim Hutting
2025-11-09 14:45:08 +01:00
5 changed files with 246 additions and 25 deletions

View File

@@ -1,4 +1,5 @@
import json
import logging
from spade.behaviour import CyclicBehaviour
from spade.message import Message
@@ -7,6 +8,8 @@ from control_backend.core.config import settings
class BeliefFromText(CyclicBehaviour):
logger = logging.getLogger(__name__)
# TODO: LLM prompt nog hardcoded
llm_instruction_prompt = """
You are an information extraction assistent for a BDI agent. Your task is to extract values \
@@ -36,6 +39,9 @@ class BeliefFromText(CyclicBehaviour):
async def run(self):
msg = await self.receive()
if msg is None:
return
sender = msg.sender.node
match sender:
case settings.agent_settings.transcription_agent_name:
@@ -62,10 +68,14 @@ class BeliefFromText(CyclicBehaviour):
# Verify by trying to parse
try:
json.loads(response)
belief_message = Message(
to=settings.agent_settings.bdi_core_agent_name + "@" + settings.agent_settings.host,
body=response,
belief_message = Message()
belief_message.to = (
settings.agent_settings.belief_collector_agent_name
+ "@"
+ settings.agent_settings.host
)
belief_message.body = response
belief_message.thread = "beliefs"
await self.send(belief_message)
@@ -82,12 +92,12 @@ class BeliefFromText(CyclicBehaviour):
"""
belief = {"beliefs": {"user_said": [txt]}, "type": "belief_extraction_text"}
payload = json.dumps(belief)
belief_msg = Message(
to=settings.agent_settings.belief_collector_agent_name
+ "@"
+ settings.agent_settings.host,
body=payload,
belief_msg = Message()
belief_msg.to = (
settings.agent_settings.belief_collector_agent_name + "@" + settings.agent_settings.host
)
belief_msg.body = payload
belief_msg.thread = "beliefs"
await self.send(belief_msg)

View File

@@ -3,7 +3,6 @@ import logging
import zmq
from fastapi import APIRouter
from fastapi.responses import StreamingResponse
from pyjabber.server_parameters import json
from zmq.asyncio import Context
from control_backend.core.config import settings
@@ -13,6 +12,7 @@ logger = logging.getLogger(__name__)
router = APIRouter()
# DO NOT LOG INSIDE THIS FUNCTION
@router.get("/logs/stream")
async def log_stream():
context = Context.instance()
@@ -27,7 +27,6 @@ async def log_stream():
while True:
_, message = await socket.recv_multipart()
message = message.decode().strip()
json_data = json.dumps(message)
yield f"data: {json_data}\n\n"
yield f"data: {message}\n\n"
return StreamingResponse(gen(), media_type="text/event-stream")