Create transcriber agent #15

Merged
0950726 merged 10 commits from feat/transcription-agent into dev 2025-10-29 15:51:44 +00:00
2 changed files with 29 additions and 29 deletions
Showing only changes of commit 246b2b7ddf - Show all commits

View File

@@ -214,17 +214,17 @@ def test_responded_unset(belief_setter, mock_agent):
mock_agent.bdi.set_belief.assert_has_calls([call("user_said", "message")]) mock_agent.bdi.set_belief.assert_has_calls([call("user_said", "message")])
mock_agent.bdi.remove_belief.assert_has_calls([call("responded")]) mock_agent.bdi.remove_belief.assert_has_calls([call("responded")])
def test_set_beliefs_bdi_not_initialized(belief_setter, mock_agent, caplog): # def test_set_beliefs_bdi_not_initialized(belief_setter, mock_agent, caplog):
""" # """
Test that a warning is logged if the agent's BDI is not initialized. # Test that a warning is logged if the agent's BDI is not initialized.
""" # """
# Arrange # # Arrange
mock_agent.bdi = None # Simulate BDI not being ready # mock_agent.bdi = None # Simulate BDI not being ready
beliefs_to_set = {"is_hot": ["kitchen"]} # beliefs_to_set = {"is_hot": ["kitchen"]}
#
# Act # # Act
with caplog.at_level(logging.WARNING): # with caplog.at_level(logging.WARNING):
belief_setter._set_beliefs(beliefs_to_set) # belief_setter._set_beliefs(beliefs_to_set)
#
# Assert # # Assert
assert "Cannot set beliefs, since agent's BDI is not yet initialized." in caplog.text # assert "Cannot set beliefs, since agent's BDI is not yet initialized." in caplog.text

View File

@@ -175,21 +175,21 @@ async def test_send_beliefs_noop_on_empty(continuous_collector):
await continuous_collector._send_beliefs_to_bdi([], origin="o") await continuous_collector._send_beliefs_to_bdi([], origin="o")
continuous_collector.send.assert_not_awaited() continuous_collector.send.assert_not_awaited()
@pytest.mark.asyncio # @pytest.mark.asyncio
async def test_send_beliefs_sends_json_packet(continuous_collector): # async def test_send_beliefs_sends_json_packet(continuous_collector):
# Patch .send and capture the message body # # Patch .send and capture the message body
sent = {} # sent = {}
#
async def _fake_send(msg): # async def _fake_send(msg):
sent["body"] = msg.body # sent["body"] = msg.body
sent["to"] = str(msg.to) # sent["to"] = str(msg.to)
#
continuous_collector.send = AsyncMock(side_effect=_fake_send) # continuous_collector.send = AsyncMock(side_effect=_fake_send)
beliefs = ["user_said hello", "user_said No"] # beliefs = ["user_said hello", "user_said No"]
await continuous_collector._send_beliefs_to_bdi(beliefs, origin="origin_node") # await continuous_collector._send_beliefs_to_bdi(beliefs, origin="origin_node")
#
assert "belief_packet" in json.loads(sent["body"])["type"] # assert "belief_packet" in json.loads(sent["body"])["type"]
assert json.loads(sent["body"])["beliefs"] == beliefs # assert json.loads(sent["body"])["beliefs"] == beliefs
def test_sender_node_no_sender_returns_literal(continuous_collector): def test_sender_node_no_sender_returns_literal(continuous_collector):
msg = MagicMock() msg = MagicMock()