llm-ie 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

llm_ie/data_types.py

@@ -3,7 +3,7 @@ import yaml
 
 
 class LLMInformationExtractionFrame:
-    def __init__(self, frame_id:str, start:int, end:int, entity_text:str, attr:Dict[str,str]):
+    def __init__(self, frame_id:str, start:int, end:int, entity_text:str, attr:Dict[str,str]=None):
         """
         This class holds a frame (entity) extracted by LLM. 
         A frame contains the span (start and end character positions), a entity text, and 
@@ -19,7 +19,7 @@ class LLMInformationExtractionFrame:
             entity end character position
         entity_text : str
             entity string. Should be the exact string by [start:end]
-        attr : Dict[str,str]
+        attr : Dict[str,str], Optional
             dict of attributes
         """
         assert isinstance(frame_id, str), "frame_id must be a string."
@@ -27,7 +27,10 @@ class LLMInformationExtractionFrame:
         self.start = start
         self.end = end
         self.entity_text = entity_text
-        self.attr = attr.copy()
+        if attr:
+            self.attr = attr.copy()
+        else:
+            self.attr = None
 
     def is_equal(self, frame:"LLMInformationExtractionFrame") -> bool:
         """ 

llm_ie/extractors.py

@@ -197,9 +197,13 @@ class BasicFrameExtractor(FrameExtractor):
         Return : str
             the output from LLM. Need post-processing.
         """
+        messages = []
+        if self.system_prompt:
+            messages.append({'role': 'system', 'content': self.system_prompt})
+
+        messages.append({'role': 'user', 'content': self._get_user_prompt(text_content)})
         response = self.inference_engine.chat(
-                    messages=[{'role': 'system', 'content': self.system_prompt},
-                              {'role': 'user', 'content': self._get_user_prompt(text_content)}],
+                    messages=messages,
                     max_new_tokens=max_new_tokens, 
                     temperature=temperature,
                     stream=stream,
@@ -312,9 +316,11 @@ class ReviewFrameExtractor(BasicFrameExtractor):
             the output from LLM. Need post-processing.
         """
         # Pormpt extraction
-        messages=[{'role': 'system', 'content': self.system_prompt},
-                  {'role': 'user', 'content': self._get_user_prompt(text_content)}]
-        
+        messages = []
+        if self.system_prompt:
+            messages.append({'role': 'system', 'content': self.system_prompt})
+
+        messages.append({'role': 'user', 'content': self._get_user_prompt(text_content)})
         initial = self.inference_engine.chat(
                         messages=messages,
                         max_new_tokens=max_new_tokens, 

{llm_ie-0.1.4.dist-info → llm_ie-0.1.5.dist-info}/METADATA

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

{llm_ie-0.1.4.dist-info → llm_ie-0.1.5.dist-info}/RECORD

@@ -4,10 +4,10 @@ llm_ie/asset/PromptEditor_prompts/rewrite.txt,sha256=bYLOix7DUBlcWv-Q0JZ5kDnZ9OE
 llm_ie/asset/prompt_guide/BasicFrameExtractor_prompt_guide.txt,sha256=XbnU8byLGGUA3A3lT0bb2Hw-ggzhcqD3ZuKzduod2ww,1944
 llm_ie/asset/prompt_guide/ReviewFrameExtractor_prompt_guide.txt,sha256=XbnU8byLGGUA3A3lT0bb2Hw-ggzhcqD3ZuKzduod2ww,1944
 llm_ie/asset/prompt_guide/SentenceFrameExtractor_prompt_guide.txt,sha256=8nj9OLPJMtr9Soi5JU3Xk-HC7pKNoI54xA_A4u7I5j4,2620
-llm_ie/data_types.py,sha256=AxqgfmPkYySDz7VuTWh8yDWofvZgdjgFiW9hihqInHc,6605
+llm_ie/data_types.py,sha256=MnpyXFviFWhxeC5mqbaPdAxGx6vV_PhnUIFfUamq3D8,6687
 llm_ie/engines.py,sha256=TuxM56_u6-dsAAuNdfuKSH23nb9UfFbg6T60e-OXEA8,9294
-llm_ie/extractors.py,sha256=v_VZnpb7PPRlZC4X55k_w2t6X2W3NFWe8ZHo57q7s4Y,22401
+llm_ie/extractors.py,sha256=rpHJhlV3A9-9nldIutxd8rtgf7903Ke6QkwbCIVdUdY,22546
 llm_ie/prompt_editor.py,sha256=dbu7A3O7O7Iw2v-xCgrTFH1-wTLAGf4SHDqdeS-He2Q,1869
-llm_ie-0.1.4.dist-info/METADATA,sha256=-P_iDB6jMLm8Qb-paUAiwGsDu4d_SZIcaDXISnN36H4,28028
-llm_ie-0.1.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-llm_ie-0.1.4.dist-info/RECORD,,
+llm_ie-0.1.5.dist-info/METADATA,sha256=2z2VDBlBq7GLws7aEcyjRmw0NC3gvKJIXS4LB6AaeI4,28028
+llm_ie-0.1.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+llm_ie-0.1.5.dist-info/RECORD,,