mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-07-17 18:06:06 +00:00
huanghuoguoguo
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
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'
|