llm-ie 0.1.3__tar.gz → 0.1.4__tar.gz

{llm_ie-0.1.3 → llm_ie-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm-ie
-Version: 0.1.3
+Version: 0.1.4
 Summary: An LLM-powered tool that transforms everyday language into robust information extraction pipelines.
 License: MIT
 Author: Enshuo (David) Hsu

{llm_ie-0.1.3 → llm_ie-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "llm-ie"
-version = "0.1.3"
+version = "0.1.4"
 description = "An LLM-powered tool that transforms everyday language into robust information extraction pipelines."
 authors = ["Enshuo (David) Hsu"]
 license = "MIT"

{llm_ie-0.1.3 → llm_ie-0.1.4}/src/llm_ie/extractors.py

@@ -1,6 +1,6 @@
-import os
 import abc
 import re
+import json
 import importlib.resources
 from typing import List, Dict, Tuple, Union
 from llm_ie.data_types import LLMInformationExtractionFrame
@@ -70,12 +70,12 @@ class FrameExtractor:
         """
         pattern = r'\{.*?\}'
         out = []
-        for tup in re.findall(pattern, gen_text):
+        for match in re.findall(pattern, gen_text, re.DOTALL):
             try:
-                tup_dict = eval(tup)
+                tup_dict = json.loads(match)
                 out.append(tup_dict)
-            except:
-                print(f'Post-processing failed at:\n{tup}')
+            except json.JSONDecodeError:
+                print(f'Post-processing failed at:\n{match}')
         return out