Fix extraction

griptape-ai · Aug 14, 2024 · 2a61f43 · 2a61f43
1 parent f6e3554
commit 2a61f43
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/griptape/engines/extraction/json_extraction_engine.py b/griptape/engines/extraction/json_extraction_engine.py
@@ -18,7 +18,7 @@
 
 @define
 class JsonExtractionEngine(BaseExtractionEngine):
-    JSON_PATTERN = r"(?s)[^\[]*(\[.*\])"
+    JSON_PATTERN = r"(?s)(\{.*?\}|\[.*?\])"
 
     template_schema: dict = field(default=Factory(dict), kw_only=True)
     system_template_generator: J2 = field(

diff --git a/griptape/templates/engines/extraction/json/user.j2 b/griptape/templates/engines/extraction/json/user.j2
@@ -1,4 +1,4 @@
-Extract information from the Text based on the Extraction Template JSON Schema into an array of JSON objects.
+Extract information from the Text based on the Extraction Template JSON Schema into valid JSON.
 Text: """{{ text }}"""
 
-JSON array:
+JSON: