mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 05:54:26 +01:00
fix(agent/json_utils): Make extract_dict_from_response more robust
* Accommodate for both ```json and ```JSON blocks in responses
This commit is contained in:
@@ -9,13 +9,11 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
def extract_dict_from_response(response_content: str) -> dict[str, Any]:
|
||||
# Sometimes the response includes the JSON in a code block with ```
|
||||
pattern = r"```([\s\S]*?)```"
|
||||
pattern = r"```(?:json|JSON)*([\s\S]*?)```"
|
||||
match = re.search(pattern, response_content)
|
||||
|
||||
if match:
|
||||
response_content = match.group(1).strip()
|
||||
# Remove language names in code blocks
|
||||
response_content = response_content.lstrip("json")
|
||||
else:
|
||||
# The string may contain JSON.
|
||||
json_pattern = r"{[\s\S]*}"
|
||||
|
||||
Reference in New Issue
Block a user