1 files changed, 33 insertions, 0 deletions
diff --git a/autogpts/autogpt/autogpt/json_utils/utilities.py b/autogpts/autogpt/autogpt/json_utils/utilities.py
new file mode 100644
index 000000000..0521372e0
--- /dev/null
+++ b/autogpts/autogpt/autogpt/json_utils/utilities.py
@@ -0,0 +1,33 @@
+"""Utilities for the json_fixes package."""
+import ast
+import logging
+import re
+from typing import Any
+
+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"```(?:json|JSON)*([\s\S]*?)```"
+    match = re.search(pattern, response_content)
+
+    if match:
+        response_content = match.group(1).strip()
+    else:
+        # The string may contain JSON.
+        json_pattern = r"{[\s\S]*}"
+        match = re.search(json_pattern, response_content)
+
+        if match:
+            response_content = match.group()
+
+    # Response content comes from OpenAI as a Python `str(content_dict)`.
+    # `literal_eval` does the reverse of `str(dict)`.
+    result = ast.literal_eval(response_content)
+    if not isinstance(result, dict):
+        raise ValueError(
+            f"Response '''{response_content}''' evaluated to "
+            f"non-dict value {repr(result)}"
+        )
+    return result