mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-25 21:36:52 +08:00
25d9cf2890
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Graham Neubig <neubig@gmail.com> Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
"""Tests for the conversation summary generator."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from openhands.core.config import LLMConfig
|
|
from openhands.utils.conversation_summary import generate_conversation_title
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_conversation_title_empty_message():
|
|
"""Test that an empty message returns None."""
|
|
mock_llm_registry = MagicMock()
|
|
mock_llm_config = LLMConfig(model='test-model')
|
|
|
|
result = await generate_conversation_title('', mock_llm_config, mock_llm_registry)
|
|
assert result is None
|
|
|
|
result = await generate_conversation_title(
|
|
' ', mock_llm_config, mock_llm_registry
|
|
)
|
|
assert result is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_conversation_title_success():
|
|
"""Test successful title generation."""
|
|
# Create a mock LLM registry that returns a title
|
|
mock_llm_registry = MagicMock()
|
|
mock_llm_registry.request_extraneous_completion.return_value = 'Generated Title'
|
|
|
|
mock_llm_config = LLMConfig(model='test-model')
|
|
|
|
result = await generate_conversation_title(
|
|
'Can you help me with Python?', mock_llm_config, mock_llm_registry
|
|
)
|
|
|
|
assert result == 'Generated Title'
|
|
# Verify the mock was called with the expected arguments
|
|
mock_llm_registry.request_extraneous_completion.assert_called_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_conversation_title_long_title():
|
|
"""Test that long titles are truncated."""
|
|
# Create a mock LLM registry that returns a long title
|
|
mock_llm_registry = MagicMock()
|
|
mock_llm_registry.request_extraneous_completion.return_value = 'This is a very long title that should be truncated because it exceeds the maximum length'
|
|
|
|
mock_llm_config = LLMConfig(model='test-model')
|
|
|
|
result = await generate_conversation_title(
|
|
'Can you help me with Python?',
|
|
mock_llm_config,
|
|
mock_llm_registry,
|
|
max_length=30,
|
|
)
|
|
|
|
# Verify the title is truncated correctly
|
|
assert len(result) <= 30
|
|
assert result.endswith('...')
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_conversation_title_exception():
|
|
"""Test that exceptions are handled gracefully."""
|
|
# Create a mock LLM registry that raises an exception
|
|
mock_llm_registry = MagicMock()
|
|
mock_llm_registry.request_extraneous_completion.side_effect = Exception(
|
|
'Test error'
|
|
)
|
|
|
|
mock_llm_config = LLMConfig(model='test-model')
|
|
|
|
result = await generate_conversation_title(
|
|
'Can you help me with Python?', mock_llm_config, mock_llm_registry
|
|
)
|
|
|
|
# Verify that None is returned when an exception occurs
|
|
assert result is None
|