llm-ie 1.0.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

llm_ie/prompt_editor.py

@@ -67,7 +67,7 @@ class PromptEditor:
         return prompt
     
 
-    def rewrite(self, draft:str, **kwrs) -> str:
+    def rewrite(self, draft:str) -> str:
         """
         This method inputs a prompt draft and rewrites it following the extractor's guideline.
         """
@@ -79,10 +79,10 @@ class PromptEditor:
                                              prompt_template=rewrite_prompt_template)
         messages = [{"role": "system", "content": self.system_prompt},
                     {"role": "user", "content": prompt}]
-        res = self.inference_engine.chat(messages, verbose=True, **kwrs)
+        res = self.inference_engine.chat(messages, verbose=True)
         return res
     
-    def comment(self, draft:str, **kwrs) -> str:
+    def comment(self, draft:str) -> str:
         """
         This method inputs a prompt draft and comment following the extractor's guideline.
         """
@@ -94,11 +94,11 @@ class PromptEditor:
                                              prompt_template=comment_prompt_template)
         messages = [{"role": "system", "content": self.system_prompt},
                     {"role": "user", "content": prompt}]
-        res = self.inference_engine.chat(messages, verbose=True, **kwrs)
+        res = self.inference_engine.chat(messages, verbose=True)
         return res
     
 
-    def _terminal_chat(self, **kwrs):
+    def _terminal_chat(self):
         """
         This method runs an interactive chat session in the terminal to help users write prompt templates.
         """
@@ -126,11 +126,11 @@ class PromptEditor:
             # Chat
             messages.append({"role": "user", "content": user_input})
             print(f"{Fore.BLUE}Assistant: {Style.RESET_ALL}", end="")
-            response = self.inference_engine.chat(messages, verbose=True, **kwrs)
+            response = self.inference_engine.chat(messages, verbose=True)
             messages.append({"role": "assistant", "content": response})
             
 
-    def _IPython_chat(self, **kwrs):
+    def _IPython_chat(self):
         """
         This method runs an interactive chat session in Jupyter/IPython using ipywidgets to help users write prompt templates.
         """
@@ -186,7 +186,7 @@ class PromptEditor:
 
             # Get assistant's response and append it to conversation
             print("Assistant: ", end="")
-            response = self.inference_engine.chat(messages, verbose=True, **kwrs)
+            response = self.inference_engine.chat(messages, verbose=True)
             messages.append({"role": "assistant", "content": response})
 
             # Display the assistant's response
@@ -200,16 +200,16 @@ class PromptEditor:
         display(input_box)
         display(output_area)
 
-    def chat(self, **kwrs):
+    def chat(self):
         """
         External method that detects the environment and calls the appropriate chat method.
         """
         if 'ipykernel' in sys.modules:
-            self._IPython_chat(**kwrs)
+            self._IPython_chat()
         else:
-            self._terminal_chat(**kwrs)
+            self._terminal_chat()
 
-    def chat_stream(self, messages: List[Dict[str, str]], **kwrs) -> Generator[str, None, None]:
+    def chat_stream(self, messages: List[Dict[str, str]]) -> Generator[str, None, None]:
         """
         This method processes messages and yields response chunks from the inference engine.
         This is for frontend App.
@@ -239,5 +239,5 @@ class PromptEditor:
                     {"role": "user", "content": prompt}] + messages
         
 
-        stream_generator = self.inference_engine.chat(messages, stream=True, **kwrs)
+        stream_generator = self.inference_engine.chat(messages, stream=True)
         yield from stream_generator

{llm_ie-1.0.0.dist-info → llm_ie-1.2.0.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: llm-ie
-Version: 1.0.0
-Summary: An LLM-powered tool that transforms everyday language into robust information extraction pipelines.
+Version: 1.2.0
+Summary: A comprehensive toolkit that provides building blocks for LLM-based named entity recognition, attribute extraction, and relation extraction pipelines.
 License: MIT
 Author: Enshuo (David) Hsu
 Requires-Python: >=3.11,<4.0

{llm_ie-1.0.0.dist-info → llm_ie-1.2.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-llm_ie/__init__.py,sha256=47E7fLY4vrNwZlvUmWdE_CkA024uFuQuovjNeuf59rQ,1426
+llm_ie/__init__.py,sha256=rLP01qXkIisX0WLzZOv6y494Braw89g5JLmA6ZyrGGA,1590
 llm_ie/asset/PromptEditor_prompts/chat.txt,sha256=Fq62voV0JQ8xBRcxS1Nmdd7DkHs1fGYb-tmNwctZZK0,118
 llm_ie/asset/PromptEditor_prompts/comment.txt,sha256=C_lxx-dlOlFJ__jkHKosZ8HsNAeV1aowh2B36nIipBY,159
 llm_ie/asset/PromptEditor_prompts/rewrite.txt,sha256=JAwY9vm1jSmKf2qcLBYUvrSmME2EJH36bALmkwZDWYQ,178
@@ -9,6 +9,7 @@ llm_ie/asset/default_prompts/ReviewFrameExtractor_addition_review_prompt.txt,sha
 llm_ie/asset/default_prompts/ReviewFrameExtractor_revision_review_prompt.txt,sha256=lGGjdeFpzZEc56w-EtQDMyYFs7A3DQAM32sT42Nf_08,293
 llm_ie/asset/default_prompts/SentenceReviewFrameExtractor_addition_review_prompt.txt,sha256=Of11LFuXLB249oekFelzlIeoAB0cATReqWgFTvhNz_8,329
 llm_ie/asset/default_prompts/SentenceReviewFrameExtractor_revision_review_prompt.txt,sha256=kNJQK7NdoCx13TXGY8HYGrW_v4SEaErK8j9qIzd70CM,291
+llm_ie/asset/prompt_guide/AttributeExtractor_prompt_guide.txt,sha256=w2amKipinuJtCiyPsgWsjaJRwTpS1qOBDuPPtPCMeQA,2120
 llm_ie/asset/prompt_guide/BasicFrameExtractor_prompt_guide.txt,sha256=-Cli7rwu4wM4vSmkG0nInNkpStUhRqKESQ3oqD38pbE,10395
 llm_ie/asset/prompt_guide/BasicReviewFrameExtractor_prompt_guide.txt,sha256=-Cli7rwu4wM4vSmkG0nInNkpStUhRqKESQ3oqD38pbE,10395
 llm_ie/asset/prompt_guide/BinaryRelationExtractor_prompt_guide.txt,sha256=Z6Yc2_QRqroWcJ13owNJbo78I0wpS4XXDsOjXFR-aPk,2166
@@ -19,9 +20,9 @@ llm_ie/asset/prompt_guide/SentenceFrameExtractor_prompt_guide.txt,sha256=97_-y_v
 llm_ie/asset/prompt_guide/SentenceReviewFrameExtractor_prompt_guide.txt,sha256=97_-y_vHMLG4Kb8fLsGgibLxB-3mest8k3LHfLo5h-I,10465
 llm_ie/chunkers.py,sha256=24h9l-Ldyx3EgfYicFqGhV_b-XofUS3yovC1nBWdDoo,5143
 llm_ie/data_types.py,sha256=72-3bzzYpo7KZpD9bjoroWT2eiM0zmWyDkBr2nHoBV0,18559
-llm_ie/engines.py,sha256=04wZgWc9a0wwZCLeWVr1dqrDU2-cDAxGB4b4rYik6gU,27438
-llm_ie/extractors.py,sha256=iTF2U0_KmOlJiMX27Xr3-AKYckzK7Et-NE-__XFZ6tU,104520
-llm_ie/prompt_editor.py,sha256=2sunCNLVlpmoF71tdUkCFDYp_YRcVYMvU7-n07Og93U,11062
-llm_ie-1.0.0.dist-info/METADATA,sha256=EhpvJKYUEFPLIsDndhPgsnveCmhcb1mQyRh99tXEgMg,677
-llm_ie-1.0.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-llm_ie-1.0.0.dist-info/RECORD,,
+llm_ie/engines.py,sha256=uE5sag1YeKBYBFF4gY7rYZK9e1ttatf9T7bV_xSg9Pk,36075
+llm_ie/extractors.py,sha256=aCRqKhjSoKTAWZ3WhX_O6V-S_rIvYhPsk78nZLDpQw8,95149
+llm_ie/prompt_editor.py,sha256=zh7Es5Ta2qSTgHtfF9Y9ZKXs4DMue6XlyRt9O6_Uk6c,10962
+llm_ie-1.2.0.dist-info/METADATA,sha256=X9zsMDwBAq1QzIkX8SSbmwLsEFiiAVeNeA0GTiNkAkQ,728
+llm_ie-1.2.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+llm_ie-1.2.0.dist-info/RECORD,,