llm-ie 0.2.2__tar.gz → 0.3.1__tar.gz

{llm_ie-0.2.2 → llm_ie-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm-ie
-Version: 0.2.2
+Version: 0.3.1
 Summary: An LLM-powered tool that transforms everyday language into robust information extraction pipelines.
 License: MIT
 Author: Enshuo (David) Hsu
@@ -9,6 +9,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: colorama (>=0.4.6,<0.5.0)
 Requires-Dist: nltk (>=3.8,<4.0)
 Description-Content-Type: text/markdown
 
@@ -20,6 +21,13 @@ Description-Content-Type: text/markdown
 
 An LLM-powered tool that transforms everyday language into robust information extraction pipelines. 
 
+| Features | Support |
+|----------|----------|
+| **LLM Agent for prompt writing** | :white_check_mark:  Interactive chat, Python functions |
+| **Named Entity Recognition (NER)** | :white_check_mark: Document-level, Sentence-level |
+| **Entity Attributes Extraction** | :white_check_mark: Flexible formats |
+| **Relation Extraction (RE)** | :white_check_mark: Binary & Multiclass relations |
+
 ## Table of Contents
 - [Overview](#overview)
 - [Prerequisite](#prerequisite)
@@ -35,12 +43,12 @@ An LLM-powered tool that transforms everyday language into robust information ex
         - [RelationExtractor](#relationextractor)
 
 ## Overview
-LLM-IE is a toolkit that provides robust information extraction utilities for frame-based information extraction. Since prompt design has a significant impact on generative information extraction with LLMs, it also provides a built-in LLM editor to help with prompt writing. The flowchart below demonstrates the workflow starting from a casual language request.
+LLM-IE is a toolkit that provides robust information extraction utilities for named entity, entity attributes, and entity relation extraction. Since prompt design has a significant impact on generative information extraction with LLMs, it has a built-in LLM agent ("editor") to help with prompt writing. The flowchart below demonstrates the workflow starting from a casual language request to output visualization.
 
 <div align="center"><img src="doc_asset/readme_img/LLM-IE flowchart.png" width=800 ></div>
 
 ## Prerequisite
-At least one LLM inference engine is required. There are built-in supports for 🦙 [Llama-cpp-python](https://github.com/abetlen/llama-cpp-python), <img src="https://avatars.githubusercontent.com/u/151674099?s=48&v=4" alt="Icon" width="20"/> [Ollama](https://github.com/ollama/ollama), 🤗 [Huggingface_hub](https://github.com/huggingface/huggingface_hub), <img src=doc_asset/readme_img/openai-logomark.png width=16 /> [OpenAI API](https://platform.openai.com/docs/api-reference/introduction), and <img src=doc_asset/readme_img/vllm-logo.png width=20 /> vLLM. For installation guides, please refer to those projects. Other inference engines can be configured through the [InferenceEngine](src/llm_ie/engines.py) abstract class. See [LLM Inference Engine](#llm-inference-engine) section below.
+At least one LLM inference engine is required. There are built-in supports for 🦙 [Llama-cpp-python](https://github.com/abetlen/llama-cpp-python), <img src="https://avatars.githubusercontent.com/u/151674099?s=48&v=4" alt="Icon" width="20"/> [Ollama](https://github.com/ollama/ollama), 🤗 [Huggingface_hub](https://github.com/huggingface/huggingface_hub), <img src=doc_asset/readme_img/openai-logomark.png width=16 /> [OpenAI API](https://platform.openai.com/docs/api-reference/introduction), and <img src=doc_asset/readme_img/vllm-logo.png width=20 /> [vLLM](https://github.com/vllm-project/vllm). For installation guides, please refer to those projects. Other inference engines can be configured through the [InferenceEngine](src/llm_ie/engines.py) abstract class. See [LLM Inference Engine](#llm-inference-engine) section below.
 
 ## Installation
 The Python package is available on PyPI. 
@@ -125,21 +133,26 @@ We start with a casual description:
 
 *"Extract diagnosis from the clinical note. Make sure to include diagnosis date and status."* 
 
-The ```PromptEditor``` rewrites it following the schema required by the ```BasicFrameExtractor```. 
-
-```python 
+Define the AI prompt editor.
+```python
+from llm_ie.engines import OllamaInferenceEngine
 from llm_ie.extractors import BasicFrameExtractor
 from llm_ie.prompt_editor import PromptEditor
 
-# Describe the task in casual language
-prompt_draft = "Extract diagnosis from the clinical note. Make sure to include diagnosis date and status."
-
-# Use LLM editor to generate a formal prompt template with standard extraction schema
+# Define a LLM inference engine
+llm = OllamaInferenceEngine(model_name="llama3.1:8b-instruct-q8_0")
+# Define LLM prompt editor
 editor = PromptEditor(llm, BasicFrameExtractor)
-prompt_template = editor.rewrite(prompt_draft)
+# Start chat
+editor.chat()
 ```
 
-The editor generates a prompt template as below:
+This opens an interactive session:
+<div align="left"><img src=doc_asset/readme_img/terminal_chat.PNG width=1000 ></div>
+
+
+The ```PromptEditor``` drafts a prompt template following the schema required by the ```BasicFrameExtractor```:
+
 ```
 # Task description
 The paragraph below contains a clinical note with diagnoses listed. Please carefully review it and extract the diagnoses, including the diagnosis date and status.
@@ -165,6 +178,8 @@ If there is no specific date or status, just omit those keys.
 Below is the clinical note:
 {{input}}
 ```
+
+
 #### Information extraction pipeline
 Now we apply the prompt template to build an information extraction pipeline.
 
@@ -202,15 +217,33 @@ from llm_ie.data_types import LLMInformationExtractionDocument
 doc = LLMInformationExtractionDocument(doc_id="Synthesized medical note",
                                        text=note_text)
 # Add frames to a document
-for frame in frames:
-    doc.add_frame(frame, valid_mode="span", create_id=True)
+doc.add_frames(frames, create_id=True)
 
 # Save document to file (.llmie)
 doc.save("<your filename>.llmie")
 ```
 
+To visualize the extracted frames, we use the ```viz_serve()``` method. 
+```python
+doc.viz_serve()
+```
+A Flask APP starts at port 5000 (default).
+```
+* Serving Flask app 'ie_viz.utilities'
+* Debug mode: off
+WARNING: This is a development server. Do not use it in a production deployment. Use a production WSGI server instead.
+ * Running on all addresses (0.0.0.0)
+ * Running on http://127.0.0.1:5000
+Press CTRL+C to quit
+127.0.0.1 - - [03/Oct/2024 23:36:22] "GET / HTTP/1.1" 200 -
+```
+
+<div align="left"><img src="doc_asset/readme_img/llm-ie_demo.PNG" width=1000 ></div>
+
+
 ## Examples
-  - [Write prompt templates with AI editors](demo/prompt_template_writing.ipynb)
+  - [Interactive chat with LLM prompt editors](demo/prompt_template_writing_via_chat.ipynb)
+  - [Write prompt templates with LLM prompt editors](demo/prompt_template_writing.ipynb)
   - [NER + RE for Drug, Strength, Frequency](demo/medication_relation_extraction.ipynb)
 
 ## User Guide
@@ -435,7 +468,30 @@ print(BasicFrameExtractor.get_prompt_guide())
 ```
 
 ### Prompt Editor
-The prompt editor is an LLM agent that reviews, comments and rewrites a prompt following the defined schema of each extractor. It is recommended to use prompt editor iteratively: 
+The prompt editor is an LLM agent that help users write prompt templates following the defined schema and guideline of each extractor. Chat with the promtp editor:
+
+```python
+from llm_ie.prompt_editor import PromptEditor
+from llm_ie.extractors import BasicFrameExtractor
+from llm_ie.engines import OllamaInferenceEngine
+
+# Define an LLM inference engine
+ollama = OllamaInferenceEngine(model_name="llama3.1:8b-instruct-q8_0")
+
+# Define editor
+editor = PromptEditor(ollama, BasicFrameExtractor)
+
+editor.chat()
+```
+
+In a terminal environment, an interactive chat session will start:
+<div align="left"><img src=doc_asset/readme_img/terminal_chat.PNG width=1000 ></div>
+
+In the Jupyter/IPython environment, an ipywidgets session will start:
+<div align="left"><img src=doc_asset/readme_img/IPython_chat.PNG width=1000 ></div>
+
+
+We can also use the `rewrite()` and `comment()` methods to programmingly interact with the prompt editor: 
 1. start with a casual description of the task
 2. have the prompt editor generate a prompt template as the starting point
 3. manually revise the prompt template
@@ -581,40 +637,29 @@ print(BasicFrameExtractor.get_prompt_guide())
 ```
 
 ```
-Prompt template design:
-    1. Task description
-    2. Schema definition
-    3. Output format definition
-    4. Additional hints
-    5. Input placeholder
+Prompt Template Design:
 
-Example:
+1. Task Description:  
+   Provide a detailed description of the task, including the background and the type of task (e.g., named entity recognition).
 
-    # Task description
-    The paragraph below is from the Food and Drug Administration (FDA) Clinical Pharmacology Section of Labeling for Human Prescription Drug and Biological Products, Adverse reactions section. Please carefully review it and extract the adverse reactions and percentages. Note that each adverse reaction is nested under a clinical trial and potentially an arm. Your output should take that into consideration.
+2. Schema Definition:  
+   List the key concepts that should be extracted, and provide clear definitions for each one.
 
-    # Schema definition
-    Your output should contain: 
-        "ClinicalTrial" which is the name of the trial, 
-        If applicable, "Arm" which is the arm within the clinical trial, 
-        "AdverseReaction" which is the name of the adverse reaction,
-        If applicable, "Percentage" which is the occurance of the adverse reaction within the trial and arm,
-        "Evidence" which is the EXACT sentence in the text where you found the AdverseReaction from
+3. Output Format Definition:  
+   The output should be a JSON list, where each element is a dictionary representing a frame (an entity along with its attributes). Each dictionary must include a key that holds the entity text. This key can be named "entity_text" or anything else depend on the context. The attributes can either be flat (e.g., {"entity_text": "<entity_text>", "attr1": "<attr1>", "attr2": "<attr2>"}) or nested (e.g., {"entity_text": "<entity_text>", "attributes": {"attr1": "<attr1>", "attr2": "<attr2>"}}).
 
-    # Output format definition
-    Your output should follow JSON format, for example:
-    [
-        {"ClinicalTrial": "<Clinical trial name or number>", "Arm": "<name of arm>", "AdverseReaction": "<Adverse reaction text>", "Percentage": "<a percent>", "Evidence": "<exact sentence from the text>"},
-        {"ClinicalTrial": "<Clinical trial name or number>", "Arm": "<name of arm>", "AdverseReaction": "<Adverse reaction text>", "Percentage": "<a percent>", "Evidence": "<exact sentence from the text>"} 
-    ]
+4. Optional: Hints:  
+   Provide itemized hints for the information extractors to guide the extraction process.
+
+5. Optional: Examples:  
+   Include examples in the format:  
+    Input: ...  
+    Output: ...
 
-    # Additional hints
-    Your output should be 100% based on the provided content. DO NOT output fake numbers. 
-    If there is no specific arm, just omit the "Arm" key. If the percentage is not reported, just omit the "Percentage" key. The "Evidence" should always be provided.
+6. Input Placeholder:  
+   The template must include a placeholder in the format {{<placeholder_name>}} for the input text. The placeholder name can be customized as needed.
 
-    # Input placeholder
-    Below is the Adverse reactions section:
-    {{input}}
+......
 ```
 </details>
 

{llm_ie-0.2.2 → llm_ie-0.3.1}/README.md

@@ -6,6 +6,13 @@
 
 An LLM-powered tool that transforms everyday language into robust information extraction pipelines. 
 
+| Features | Support |
+|----------|----------|
+| **LLM Agent for prompt writing** | :white_check_mark:  Interactive chat, Python functions |
+| **Named Entity Recognition (NER)** | :white_check_mark: Document-level, Sentence-level |
+| **Entity Attributes Extraction** | :white_check_mark: Flexible formats |
+| **Relation Extraction (RE)** | :white_check_mark: Binary & Multiclass relations |
+
 ## Table of Contents
 - [Overview](#overview)
 - [Prerequisite](#prerequisite)
@@ -21,12 +28,12 @@ An LLM-powered tool that transforms everyday language into robust information ex
         - [RelationExtractor](#relationextractor)
 
 ## Overview
-LLM-IE is a toolkit that provides robust information extraction utilities for frame-based information extraction. Since prompt design has a significant impact on generative information extraction with LLMs, it also provides a built-in LLM editor to help with prompt writing. The flowchart below demonstrates the workflow starting from a casual language request.
+LLM-IE is a toolkit that provides robust information extraction utilities for named entity, entity attributes, and entity relation extraction. Since prompt design has a significant impact on generative information extraction with LLMs, it has a built-in LLM agent ("editor") to help with prompt writing. The flowchart below demonstrates the workflow starting from a casual language request to output visualization.
 
 <div align="center"><img src="doc_asset/readme_img/LLM-IE flowchart.png" width=800 ></div>
 
 ## Prerequisite
-At least one LLM inference engine is required. There are built-in supports for 🦙 [Llama-cpp-python](https://github.com/abetlen/llama-cpp-python), <img src="https://avatars.githubusercontent.com/u/151674099?s=48&v=4" alt="Icon" width="20"/> [Ollama](https://github.com/ollama/ollama), 🤗 [Huggingface_hub](https://github.com/huggingface/huggingface_hub), <img src=doc_asset/readme_img/openai-logomark.png width=16 /> [OpenAI API](https://platform.openai.com/docs/api-reference/introduction), and <img src=doc_asset/readme_img/vllm-logo.png width=20 /> vLLM. For installation guides, please refer to those projects. Other inference engines can be configured through the [InferenceEngine](src/llm_ie/engines.py) abstract class. See [LLM Inference Engine](#llm-inference-engine) section below.
+At least one LLM inference engine is required. There are built-in supports for 🦙 [Llama-cpp-python](https://github.com/abetlen/llama-cpp-python), <img src="https://avatars.githubusercontent.com/u/151674099?s=48&v=4" alt="Icon" width="20"/> [Ollama](https://github.com/ollama/ollama), 🤗 [Huggingface_hub](https://github.com/huggingface/huggingface_hub), <img src=doc_asset/readme_img/openai-logomark.png width=16 /> [OpenAI API](https://platform.openai.com/docs/api-reference/introduction), and <img src=doc_asset/readme_img/vllm-logo.png width=20 /> [vLLM](https://github.com/vllm-project/vllm). For installation guides, please refer to those projects. Other inference engines can be configured through the [InferenceEngine](src/llm_ie/engines.py) abstract class. See [LLM Inference Engine](#llm-inference-engine) section below.
 
 ## Installation
 The Python package is available on PyPI. 
@@ -111,21 +118,26 @@ We start with a casual description:
 
 *"Extract diagnosis from the clinical note. Make sure to include diagnosis date and status."* 
 
-The ```PromptEditor``` rewrites it following the schema required by the ```BasicFrameExtractor```. 
-
-```python 
+Define the AI prompt editor.
+```python
+from llm_ie.engines import OllamaInferenceEngine
 from llm_ie.extractors import BasicFrameExtractor
 from llm_ie.prompt_editor import PromptEditor
 
-# Describe the task in casual language
-prompt_draft = "Extract diagnosis from the clinical note. Make sure to include diagnosis date and status."
-
-# Use LLM editor to generate a formal prompt template with standard extraction schema
+# Define a LLM inference engine
+llm = OllamaInferenceEngine(model_name="llama3.1:8b-instruct-q8_0")
+# Define LLM prompt editor
 editor = PromptEditor(llm, BasicFrameExtractor)
-prompt_template = editor.rewrite(prompt_draft)
+# Start chat
+editor.chat()
 ```
 
-The editor generates a prompt template as below:
+This opens an interactive session:
+<div align="left"><img src=doc_asset/readme_img/terminal_chat.PNG width=1000 ></div>
+
+
+The ```PromptEditor``` drafts a prompt template following the schema required by the ```BasicFrameExtractor```:
+
 ```
 # Task description
 The paragraph below contains a clinical note with diagnoses listed. Please carefully review it and extract the diagnoses, including the diagnosis date and status.
@@ -151,6 +163,8 @@ If there is no specific date or status, just omit those keys.
 Below is the clinical note:
 {{input}}
 ```
+
+
 #### Information extraction pipeline
 Now we apply the prompt template to build an information extraction pipeline.
 
@@ -188,15 +202,33 @@ from llm_ie.data_types import LLMInformationExtractionDocument
 doc = LLMInformationExtractionDocument(doc_id="Synthesized medical note",
                                        text=note_text)
 # Add frames to a document
-for frame in frames:
-    doc.add_frame(frame, valid_mode="span", create_id=True)
+doc.add_frames(frames, create_id=True)
 
 # Save document to file (.llmie)
 doc.save("<your filename>.llmie")
 ```
 
+To visualize the extracted frames, we use the ```viz_serve()``` method. 
+```python
+doc.viz_serve()
+```
+A Flask APP starts at port 5000 (default).
+```
+* Serving Flask app 'ie_viz.utilities'
+* Debug mode: off
+WARNING: This is a development server. Do not use it in a production deployment. Use a production WSGI server instead.
+ * Running on all addresses (0.0.0.0)
+ * Running on http://127.0.0.1:5000
+Press CTRL+C to quit
+127.0.0.1 - - [03/Oct/2024 23:36:22] "GET / HTTP/1.1" 200 -
+```
+
+<div align="left"><img src="doc_asset/readme_img/llm-ie_demo.PNG" width=1000 ></div>
+
+
 ## Examples
-  - [Write prompt templates with AI editors](demo/prompt_template_writing.ipynb)
+  - [Interactive chat with LLM prompt editors](demo/prompt_template_writing_via_chat.ipynb)
+  - [Write prompt templates with LLM prompt editors](demo/prompt_template_writing.ipynb)
   - [NER + RE for Drug, Strength, Frequency](demo/medication_relation_extraction.ipynb)
 
 ## User Guide
@@ -421,7 +453,30 @@ print(BasicFrameExtractor.get_prompt_guide())
 ```
 
 ### Prompt Editor
-The prompt editor is an LLM agent that reviews, comments and rewrites a prompt following the defined schema of each extractor. It is recommended to use prompt editor iteratively: 
+The prompt editor is an LLM agent that help users write prompt templates following the defined schema and guideline of each extractor. Chat with the promtp editor:
+
+```python
+from llm_ie.prompt_editor import PromptEditor
+from llm_ie.extractors import BasicFrameExtractor
+from llm_ie.engines import OllamaInferenceEngine
+
+# Define an LLM inference engine
+ollama = OllamaInferenceEngine(model_name="llama3.1:8b-instruct-q8_0")
+
+# Define editor
+editor = PromptEditor(ollama, BasicFrameExtractor)
+
+editor.chat()
+```
+
+In a terminal environment, an interactive chat session will start:
+<div align="left"><img src=doc_asset/readme_img/terminal_chat.PNG width=1000 ></div>
+
+In the Jupyter/IPython environment, an ipywidgets session will start:
+<div align="left"><img src=doc_asset/readme_img/IPython_chat.PNG width=1000 ></div>
+
+
+We can also use the `rewrite()` and `comment()` methods to programmingly interact with the prompt editor: 
 1. start with a casual description of the task
 2. have the prompt editor generate a prompt template as the starting point
 3. manually revise the prompt template
@@ -567,40 +622,29 @@ print(BasicFrameExtractor.get_prompt_guide())
 ```
 
 ```
-Prompt template design:
-    1. Task description
-    2. Schema definition
-    3. Output format definition
-    4. Additional hints
-    5. Input placeholder
+Prompt Template Design:
 
-Example:
+1. Task Description:  
+   Provide a detailed description of the task, including the background and the type of task (e.g., named entity recognition).
 
-    # Task description
-    The paragraph below is from the Food and Drug Administration (FDA) Clinical Pharmacology Section of Labeling for Human Prescription Drug and Biological Products, Adverse reactions section. Please carefully review it and extract the adverse reactions and percentages. Note that each adverse reaction is nested under a clinical trial and potentially an arm. Your output should take that into consideration.
+2. Schema Definition:  
+   List the key concepts that should be extracted, and provide clear definitions for each one.
 
-    # Schema definition
-    Your output should contain: 
-        "ClinicalTrial" which is the name of the trial, 
-        If applicable, "Arm" which is the arm within the clinical trial, 
-        "AdverseReaction" which is the name of the adverse reaction,
-        If applicable, "Percentage" which is the occurance of the adverse reaction within the trial and arm,
-        "Evidence" which is the EXACT sentence in the text where you found the AdverseReaction from
+3. Output Format Definition:  
+   The output should be a JSON list, where each element is a dictionary representing a frame (an entity along with its attributes). Each dictionary must include a key that holds the entity text. This key can be named "entity_text" or anything else depend on the context. The attributes can either be flat (e.g., {"entity_text": "<entity_text>", "attr1": "<attr1>", "attr2": "<attr2>"}) or nested (e.g., {"entity_text": "<entity_text>", "attributes": {"attr1": "<attr1>", "attr2": "<attr2>"}}).
 
-    # Output format definition
-    Your output should follow JSON format, for example:
-    [
-        {"ClinicalTrial": "<Clinical trial name or number>", "Arm": "<name of arm>", "AdverseReaction": "<Adverse reaction text>", "Percentage": "<a percent>", "Evidence": "<exact sentence from the text>"},
-        {"ClinicalTrial": "<Clinical trial name or number>", "Arm": "<name of arm>", "AdverseReaction": "<Adverse reaction text>", "Percentage": "<a percent>", "Evidence": "<exact sentence from the text>"} 
-    ]
+4. Optional: Hints:  
+   Provide itemized hints for the information extractors to guide the extraction process.
+
+5. Optional: Examples:  
+   Include examples in the format:  
+    Input: ...  
+    Output: ...
 
-    # Additional hints
-    Your output should be 100% based on the provided content. DO NOT output fake numbers. 
-    If there is no specific arm, just omit the "Arm" key. If the percentage is not reported, just omit the "Percentage" key. The "Evidence" should always be provided.
+6. Input Placeholder:  
+   The template must include a placeholder in the format {{<placeholder_name>}} for the input text. The placeholder name can be customized as needed.
 
-    # Input placeholder
-    Below is the Adverse reactions section:
-    {{input}}
+......
 ```
 </details>
 

{llm_ie-0.2.2 → llm_ie-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "llm-ie"
-version = "0.2.2"
+version = "0.3.1"
 description = "An LLM-powered tool that transforms everyday language into robust information extraction pipelines."
 authors = ["Enshuo (David) Hsu"]
 license = "MIT"
@@ -14,6 +14,7 @@ exclude = [
 [tool.poetry.dependencies]
 python = "^3.11"
 nltk = "^3.8"
+colorama = "^0.4.6"
 
 
 [build-system]

llm_ie-0.3.1/src/llm_ie/asset/PromptEditor_prompts/chat.txt

@@ -0,0 +1,5 @@
+# Task description
+Chat with the user following the prompt guideline below. 
+
+# Prompt guideline
+{{prompt_guideline}}

llm_ie-0.3.1/src/llm_ie/asset/PromptEditor_prompts/rewrite.txt

@@ -0,0 +1,9 @@
+# Task description
+Rewrite the draft prompt following the prompt guideline below. 
+DO NOT explain your answer. 
+
+# Prompt guideline
+{{prompt_guideline}}
+
+# Draft prompt
+{{draft}}

llm_ie-0.3.1/src/llm_ie/asset/PromptEditor_prompts/system.txt

@@ -0,0 +1 @@
+You are an AI assistant specializing in prompt writing and improvement. Your role is to help users refine, rewrite, and generate effective prompts based on guidelines provided. You are highly knowledgeable in extracting key information and adhering to structured formats. During interactions, you will engage in clear, insightful, and context-aware conversations, providing thoughtful responses to assist the user. Maintain a polite, professional tone and ensure each response adds value to the conversation, promoting clarity and creativity in the user's prompts. If users ask about irrelevant topics (not related to prompt development), you will politely decline to answer and guide the conversation back to prompt development.

llm_ie-0.3.1/src/llm_ie/asset/default_prompts/ReviewFrameExtractor_addition_review_prompt.txt

@@ -0,0 +1,3 @@
+Review the input text and your output carefully. If anything was missed, add it to your output following the defined output formats. 
+You should ONLY adding new items. Do NOT re-generate the entire answer.
+Your output should strictly adheres to the defined output formats. 

llm_ie-0.3.1/src/llm_ie/asset/default_prompts/ReviewFrameExtractor_revision_review_prompt.txt

@@ -0,0 +1,2 @@
+Review the input text and your output carefully. If you find any omissions or errors, correct them by generating a revised output following the defined output formats.
+Your output should strictly adheres to the defined output formats. 

llm_ie-0.3.1/src/llm_ie/asset/default_prompts/SentenceReviewFrameExtractor_addition_review_prompt.txt

@@ -0,0 +1,4 @@
+Review the input sentence and your output carefully. If anything was missed, add it to your output following the defined output formats. 
+You should ONLY adding new items. Do NOT re-generate the entire answer.
+Your output should be based on the input sentence. 
+Your output should strictly adheres to the defined output formats. 

llm_ie-0.3.1/src/llm_ie/asset/default_prompts/SentenceReviewFrameExtractor_revision_review_prompt.txt

@@ -0,0 +1,3 @@
+Review the input sentence and your output carefully. If you find any omissions or errors, correct them by generating a revised output following the defined output formats.
+Your output should be based on the input sentence. 
+Your output should strictly adheres to the defined output formats. 

llm_ie-0.3.1/src/llm_ie/asset/prompt_guide/BasicFrameExtractor_prompt_guide.txt

@@ -0,0 +1,145 @@
+Prompt Template Design:
+
+1. Task Description:  
+   Provide a detailed description of the task, including the background and the type of task (e.g., named entity recognition).
+
+2. Schema Definition:  
+   List the key concepts that should be extracted, and provide clear definitions for each one.
+
+3. Output Format Definition:  
+   The output should be a JSON list, where each element is a dictionary representing a frame (an entity along with its attributes). Each dictionary must include a key that holds the entity text. This key can be named "entity_text" or anything else depend on the context. The attributes can either be flat (e.g., {"entity_text": "<entity_text>", "attr1": "<attr1>", "attr2": "<attr2>"}) or nested (e.g., {"entity_text": "<entity_text>", "attributes": {"attr1": "<attr1>", "attr2": "<attr2>"}}).
+
+4. Optional: Hints:  
+   Provide itemized hints for the information extractors to guide the extraction process.
+
+5. Optional: Examples:  
+   Include examples in the format:  
+    Input: ...  
+    Output: ...
+
+6. Input Placeholder:  
+   The template must include a placeholder in the format {{<placeholder_name>}} for the input text. The placeholder name can be customized as needed.
+
+
+Example 1 (single entity type with attributes):
+
+    # Task description
+    The paragraph below is from the Food and Drug Administration (FDA) Clinical Pharmacology Section of Labeling for Human Prescription Drug and Biological Products, Adverse reactions section. Please carefully review it and extract the adverse reactions and percentages. Note that each adverse reaction is nested under a clinical trial and potentially an arm. Your output should take that into consideration.
+
+    # Schema definition
+    Your output should contain: 
+        "ClinicalTrial" which is the name of the trial, 
+        If applicable, "Arm" which is the arm within the clinical trial, 
+        "AdverseReaction" which is the name of the adverse reaction,
+        If applicable, "Percentage" which is the occurance of the adverse reaction within the trial and arm,
+        "Evidence" which is the EXACT sentence in the text where you found the AdverseReaction from
+
+    # Output format definition
+    Your output should follow JSON format, for example:
+    [
+        {"ClinicalTrial": "<Clinical trial name or number>", "Arm": "<name of arm>", "AdverseReaction": "<Adverse reaction text>", "Percentage": "<a percent>", "Evidence": "<exact sentence from the text>"},
+        {"ClinicalTrial": "<Clinical trial name or number>", "Arm": "<name of arm>", "AdverseReaction": "<Adverse reaction text>", "Percentage": "<a percent>", "Evidence": "<exact sentence from the text>"} 
+    ]
+
+    # Additional hints
+    Your output should be 100% based on the provided content. DO NOT output fake numbers. 
+    If there is no specific arm, just omit the "Arm" key. If the percentage is not reported, just omit the "Percentage" key. The "Evidence" should always be provided.
+
+    # Input placeholder
+    Below is the Adverse reactions section:
+    {{input}}
+
+
+Example 2 (multiple entity types):
+
+    # Task description
+    This is a named entity recognition task. Given medical note, annotate the Drug, Form, Strength, Frequency, Route, Dosage, Reason, ADE, and Duration.
+
+    # Schema definition
+    Your output should contain: 
+        "entity_text": the exact wording as mentioned in the note.
+        "entity_type": type of the entity. It should be one of the "Drug", "Form", "Strength", "Frequency", "Route", "Dosage", "Reason", "ADE", or "Duration".
+
+    # Output format definition
+    Your output should follow JSON format, 
+    if there are one of the entity mentions: Drug, Form, Strength, Frequency, Route, Dosage, Reason, ADE, or Duration:
+        [{"entity_text": "<Exact entity mentions as in the note>", "entity_type": "<entity type as listed above>"},
+        {"entity_text": "<Exact entity mentions as in the note>", "entity_type": "<entity type as listed above>"}]
+    if there is no entity mentioned in the given note, just output an empty list:
+        []
+
+    I am only interested in the extracted contents in []. Do NOT explain your answer.
+
+    # Examples
+    Below are some examples:
+
+    Input: Acetaminophen 650 mg PO BID 5.
+    Output: [{"entity_text": "Acetaminophen", "entity_type": "Drug"}, {"entity_text": "650 mg", "entity_type": "Strength"}, {"entity_text": "PO", "entity_type": "Route"}, {"entity_text": "BID", "entity_type": "Frequency"}]
+
+    Input: Mesalamine DR 1200 mg PO BID 2.
+    Output: [{"entity_text": "Mesalamine DR", "entity_type": "Drug"}, {"entity_text": "1200 mg", "entity_type": "Strength"}, {"entity_text": "BID", "entity_type": "Frequency"}, {"entity_text": "PO", "entity_type": "Route"}]
+
+
+    # Input placeholder
+    Below is the medical note:
+    "{{input}}"
+
+
+Example 3 (multiple entity types with corresponding attributes):
+
+    # Task description
+    This is a named entity recognition task. Given a medical note, annotate the events (EVENT) and time expressions (TIMEX3):
+
+    # Schema definition
+    Your output should contain: 
+        "entity_text": the exact wording as mentioned in the note.
+        "entity_type": type of the entity. It should be one of the "EVENT" or "TIMEX3".
+        if entity_type is "EVENT",
+            "type": the event type as one of the "TEST", "PROBLEM", "TREATMENT", "CLINICAL_DEPT", "EVIDENTIAL", or "OCCURRENCE".
+            "polarity": whether an EVENT is positive ("POS") or negative ("NAG"). For example, in “the patient reports headache, and denies chills”, the EVENT [headache] is positive in its polarity, and the EVENT [chills] is negative in its polarity.
+            "modality": whether an EVENT actually occurred or not. Must be one of the "FACTUAL", "CONDITIONAL", "POSSIBLE", or "PROPOSED".
+
+        if entity_type is "TIMEX3",
+            "type": the type as one of the "DATE", "TIME", "DURATION", or "FREQUENCY".
+            "val": the numeric value 1) DATE: [YYYY]-[MM]-[DD], 2) TIME: [hh]:[mm]:[ss], 3) DURATION: P[n][Y/M/W/D]. So, “for eleven days” will be 
+    represented as “P11D”, meaning a period of 11 days. 4)  R[n][duration], where n denotes the number of repeats. When the n is omitted, the expression denotes an unspecified amount of repeats. For example, “once a day for 3 days” is “R3P1D” (repeat the time interval of 1 day (P1D) for 3 times (R3)), twice every day is “RP12H” (repeat every 12 hours)
+            "mod": additional information regarding the temporal value of a time expression. Must be one of the:
+                “NA”: the default value, no relevant modifier is present;  
+                “MORE”, means “more than”, e.g. over 2 days (val = P2D, mod = MORE);  
+                “LESS”, means “less than”, e.g. almost 2 months (val = P2M, mod=LESS); 
+                “APPROX”, means “approximate”, e.g. nearly a week (val = P1W, mod=APPROX);  
+                “START”, describes the beginning of a period of time, e.g.  Christmas morning, 2005 (val= 2005-12-25, mod= START).  
+                “END”, describes the end of a period of time, e.g. late last year, (val = 2010, mod = END)
+                “MIDDLE”, describes the middle of a period of time, e.g. mid-September 2001 (val = 2001-09, mod = MIDDLE) 
+
+    # Output format definition
+    Your output should follow JSON format, 
+    if there are one of the EVENT or TIMEX3 entity mentions:
+        [
+            {"entity_text": "<Exact entity mentions as in the note>", "entity_type": "EVENT", "type": "<event type>", "polarity": "<event polarity>", "modality": "<event modality>"},
+            {"entity_text": "<Exact entity mentions as in the note>", "entity_type": "TIMEX3", "type": "<TIMEX3 type>", "val": "<time value>", "mod": "<additional information>"}
+            ...
+        ]
+    if there is no entity mentioned in the given note, just output an empty list:
+        []
+
+    I am only interested in the extracted contents in []. Do NOT explain your answer.
+
+    # Examples
+    Below are some examples:
+
+    Input: At 9/7/93 , 1:00 a.m. , intravenous fluids rate was decreased to 50 cc&apos;s per hour , total fluids given during the first 24 hours were 140 to 150 cc&apos;s per kilo per day .
+    Output: [{"entity_text": "intravenous fluids", "entity_type": "EVENT", "type": "TREATMENT", "polarity": "POS", "modality": "FACTUAL"},
+            {"entity_text": "decreased", "entity_type": "EVENT", "type": "OCCURRENCE", "polarity": "POS", "modality": "FACTUAL"},
+            {"entity_text": "total fluids", "entity_type": "EVENT", "type": "TREATMENT", "polarity": "POS", "modality": "FACTUAL"}, 
+            {"entity_text": "9/7/93 , 1:00 a.m.", "entity_type": "TIMEX3", "type": "TIME", "val": "1993-09-07T01:00", "mod": "NA"},
+            {"entity_text": "24 hours", "entity_type": "TIMEX3", "type": "DURATION", "val": "PT24H", "mod": "NA"}]
+
+    Input: At that time it appeared well adhered to the underlying skin .
+    Output: [{"entity_text": "it", "entity_type": "EVENT", "type": "TREATMENT", "polarity": "POS", "modality": "FACTUAL"},
+            {"entity_text": "well adhered", "entity_type": "EVENT", "type": "OCCURRENCE", "polarity": "POS", "modality": "FACTUAL"}]
+
+
+    # Input placeholder
+    Below is the entire medical note:
+    "{{input}}"