Add system

system/ee.py

@@ -60,8 +60,8 @@ def run_gpt4_event_extraction(data_dir, max_tokens=100000):
                 text = " ".join(doc["text"])
                 input_text = (
                     f"{ICL}\nInput:\n\nTitle: {doc['metadata']['title']}\n"
-                    f"Date: {doc['metadata']['date']}\nArticle: {text}\n\n"
-                    f"Please only summarize events that are useful for verifying the claim '{claim}', and their dates in the JSON format.\n\nOutput:\n"
+                    f"Date: {doc['metadata']['date']}\nArticle: {text}\nPledge: {claim}\n\n"
+                    f"Please only summarize events that are useful for verifying the pledge '{claim}', and their dates in the JSON format.\n\nOutput:\n"
                 )
 
                 urls.append(doc["url"])