llm-ie 1.2.1__py3-none-any.whl → 1.2.3__py3-none-any.whl

llm_ie/prompt_editor.py

@@ -85,7 +85,7 @@ class PromptEditor:
         messages = [{"role": "system", "content": self.system_prompt},
                     {"role": "user", "content": prompt}]
         res = self.inference_engine.chat(messages, verbose=True)
-        return res
+        return res["response"]
     
     def comment(self, draft:str) -> str:
         """
@@ -101,7 +101,7 @@ class PromptEditor:
         messages = [{"role": "system", "content": self.system_prompt},
                     {"role": "user", "content": prompt}]
         res = self.inference_engine.chat(messages, verbose=True)
-        return res
+        return res["response"]
     
     def clear_messages(self):
         """
@@ -175,7 +175,7 @@ class PromptEditor:
             self.messages.append({"role": "user", "content": user_input})
             print(f"{Fore.BLUE}Assistant: {Style.RESET_ALL}", end="")
             response = self.inference_engine.chat(self.messages, verbose=True)
-            self.messages.append({"role": "assistant", "content": response})
+            self.messages.append({"role": "assistant", "content": response["response"]})
             
 
     def _IPython_chat(self):
@@ -229,7 +229,7 @@ class PromptEditor:
             # Get assistant's response and append it to conversation
             print("Assistant: ", end="")
             response = self.inference_engine.chat(self.messages, verbose=True)
-            self.messages.append({"role": "assistant", "content": response})
+            self.messages.append({"role": "assistant", "content": response["response"]})
 
             # Display the assistant's response
             with output_area:

{llm_ie-1.2.1.dist-info → llm_ie-1.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm-ie
-Version: 1.2.1
+Version: 1.2.3
 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

{llm_ie-1.2.1.dist-info → llm_ie-1.2.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-llm_ie/__init__.py,sha256=rLP01qXkIisX0WLzZOv6y494Braw89g5JLmA6ZyrGGA,1590
+llm_ie/__init__.py,sha256=zG23MsRXnjShG-LhUl1tOSaEECXuhfZrS6Xt35XtrMw,1799
 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
@@ -18,11 +18,11 @@ llm_ie/asset/prompt_guide/MultiClassRelationExtractor_prompt_guide.txt,sha256=EQ
 llm_ie/asset/prompt_guide/ReviewFrameExtractor_prompt_guide.txt,sha256=rBRIXg8JQWUHTRdoluTS0zkbTkBAacEtHHvr3lZaQCw,10437
 llm_ie/asset/prompt_guide/SentenceFrameExtractor_prompt_guide.txt,sha256=97_-y_vHMLG4Kb8fLsGgibLxB-3mest8k3LHfLo5h-I,10465
 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=x8ZAzYkSfnFGyZP9fBjKp3eHSkuGneq94PDP4Zinvq0,36250
-llm_ie/extractors.py,sha256=aCRqKhjSoKTAWZ3WhX_O6V-S_rIvYhPsk78nZLDpQw8,95149
-llm_ie/prompt_editor.py,sha256=x900oDyAozdKlXrF8SDz7wAQ-ilE7orcKLdk_aM25fM,13150
-llm_ie-1.2.1.dist-info/METADATA,sha256=F-dAtCt3xjBX3yh1qm9E07fWWG9q0VAp0cZx9pBfbZU,728
-llm_ie-1.2.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-llm_ie-1.2.1.dist-info/RECORD,,
+llm_ie/chunkers.py,sha256=BTtqj2ZbAzl5s4nk9y5T12x7iTljSLfIio_u4s6Dq0Q,7611
+llm_ie/data_types.py,sha256=6vefyGTgZcJBYgiuyfcbJN1ZKK4tNvOZf6HFpxFZngY,17792
+llm_ie/engines.py,sha256=L28e4X6Qx0pcOzMMoRW1ecqudUbemF-KHrRLsYy9FRU,61987
+llm_ie/extractors.py,sha256=MRZEF9YSnFAkDCYWYeUysIaRxBSN71MmUpiW3M0Jejk,103890
+llm_ie/prompt_editor.py,sha256=nAgCJQY5kVWTAhmrngdWRG-JKxCCPBh0dyaUcIk_-c0,13198
+llm_ie-1.2.3.dist-info/METADATA,sha256=ww7UImUpSUWKHjzM3UJpylgO00h_glXttMykdoRNc0g,728
+llm_ie-1.2.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+llm_ie-1.2.3.dist-info/RECORD,,