fix(agent-runner): close sandbox skill authoring flow

tests/unit_tests/api/service/test_monitoring_service.py

@@ -0,0 +1,48 @@
+from __future__ import annotations
+
+from types import SimpleNamespace
+from unittest.mock import AsyncMock, Mock
+
+import pytest
+
+from langbot.pkg.api.http.service.monitoring import MonitoringService
+
+
+class _MonitoringMessageRow:
+    """Match the Row shape returned by Core connections for ORM selects."""
+
+    id = 'message-1'
+    bot_id = 'bot-1'
+    bot_name = 'QA Bot'
+    pipeline_id = 'pipeline-1'
+    pipeline_name = 'QA Pipeline'
+    session_id = 'person_1'
+
+    def __getitem__(self, index: int) -> str:
+        assert index == 0
+        return self.id
+
+
+@pytest.mark.asyncio
+async def test_record_tool_call_uses_full_monitoring_message_row_for_context():
+    message_row = _MonitoringMessageRow()
+    select_result = Mock(first=Mock(return_value=message_row))
+    insert_result = Mock()
+    persistence_mgr = SimpleNamespace(
+        execute_async=AsyncMock(side_effect=[select_result, insert_result]),
+    )
+    service = MonitoringService(SimpleNamespace(persistence_mgr=persistence_mgr))
+
+    await service.record_tool_call(
+        tool_name='exec',
+        tool_source='native',
+        duration=12,
+        message_id='message-1',
+    )
+
+    insert_statement = persistence_mgr.execute_async.await_args_list[1].args[0]
+    values = insert_statement.compile().params
+    assert values['bot_id'] == 'bot-1'
+    assert values['pipeline_id'] == 'pipeline-1'
+    assert values['session_id'] == 'person_1'
+    assert values['message_id'] == 'message-1'