mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-04-29 03:00:45 -04:00
01ae22ef57
* Replace OpenDevin with OpenHands * Update CONTRIBUTING.md * Update README.md * Update README.md * update poetry lock; move opendevin folder to openhands * fix env var * revert image references in docs * revert permissions * revert permissions --------- Co-authored-by: Xingyao Wang <xingyao6@illinois.edu>
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
import json
|
|
from datetime import datetime
|
|
|
|
from json_repair import repair_json
|
|
|
|
from openhands.core.exceptions import LLMResponseError
|
|
from openhands.events.event import Event
|
|
from openhands.events.serialization import event_to_dict
|
|
|
|
|
|
def my_default_encoder(obj):
|
|
"""Custom JSON encoder that handles datetime and event objects"""
|
|
if isinstance(obj, datetime):
|
|
return obj.isoformat()
|
|
if isinstance(obj, Event):
|
|
return event_to_dict(obj)
|
|
return json.JSONEncoder().default(obj)
|
|
|
|
|
|
def dumps(obj, **kwargs):
|
|
"""Serialize an object to str format"""
|
|
return json.dumps(obj, default=my_default_encoder, **kwargs)
|
|
|
|
|
|
def loads(json_str, **kwargs):
|
|
"""Create a JSON object from str"""
|
|
try:
|
|
return json.loads(json_str, **kwargs)
|
|
except json.JSONDecodeError:
|
|
pass
|
|
depth = 0
|
|
start = -1
|
|
for i, char in enumerate(json_str):
|
|
if char == '{':
|
|
if depth == 0:
|
|
start = i
|
|
depth += 1
|
|
elif char == '}':
|
|
depth -= 1
|
|
if depth == 0 and start != -1:
|
|
response = json_str[start : i + 1]
|
|
try:
|
|
json_str = repair_json(response)
|
|
return json.loads(json_str, **kwargs)
|
|
except (json.JSONDecodeError, ValueError, TypeError) as e:
|
|
raise LLMResponseError(
|
|
'Invalid JSON in response. Please make sure the response is a valid JSON object.'
|
|
) from e
|
|
raise LLMResponseError('No valid JSON object found in response.')
|