File size: 2,995 Bytes
246d201 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
import pytest
from openhands.agenthub.micro.agent import parse_response as parse_response_micro
from openhands.core.exceptions import LLMResponseError
from openhands.core.utils.json import loads as custom_loads
from openhands.events.action import (
FileWriteAction,
MessageAction,
)
@pytest.mark.parametrize(
'parse_response_module',
[parse_response_micro],
)
def test_parse_single_complete_json(parse_response_module):
input_response = """
{
"action": "message",
"args": {
"content": "The following typos were fixed:\\n* 'futur' -> 'future'\\n* 'imagin' -> 'imagine'\\n* 'techological' -> 'technological'\\n* 'responsability' -> 'responsibility'\\nThe corrected file is ./short_essay.txt."
}
}
"""
expected = MessageAction(
"The following typos were fixed:\n* 'futur' -> 'future'\n* 'imagin' -> 'imagine'\n* 'techological' -> 'technological'\n* 'responsability' -> 'responsibility'\nThe corrected file is ./short_essay.txt."
)
result = parse_response_module(input_response)
assert result == expected
@pytest.mark.parametrize(
'parse_response_module',
[parse_response_micro],
)
def test_parse_json_with_surrounding_text(parse_response_module):
input_response = """
Some initial text that is not JSON formatted.
{
"action": "write",
"args": {
"path": "./updated_file.txt",
"content": "Updated text content here..."
}
}
Some trailing text that is also not JSON formatted.
"""
expected = FileWriteAction(
path='./updated_file.txt', content='Updated text content here...'
)
result = parse_response_module(input_response)
assert result == expected
@pytest.mark.parametrize(
'parse_response_module',
[parse_response_micro],
)
def test_parse_first_of_multiple_jsons(parse_response_module):
input_response = """
I will firstly do
{
"action": "write",
"args": {
"path": "./short_essay.txt",
"content": "Text content here..."
}
}
Then I will continue with
{
"action": "think",
"args": {
"thought": "This should not be parsed."
}
}
"""
expected = FileWriteAction(path='./short_essay.txt', content='Text content here...')
result = parse_response_module(input_response)
assert result == expected
def test_invalid_json_raises_error():
# This should fail if repair_json is able to fix this faulty JSON
input_response = '{"action": "write", "args": { "path": "./short_essay.txt", "content": "Missing closing brace" }'
with pytest.raises(LLMResponseError):
custom_loads(input_response)
def test_no_json_found():
input_response = 'This is just a string with no JSON object.'
with pytest.raises(LLMResponseError):
custom_loads(input_response)
|