Automate fetching the current year (#828)

Author: harryvu-futurehouse

paperqa/prompts.py

@@ -1,3 +1,5 @@
+from datetime import datetime
+
 summary_prompt = (
     "Summarize the excerpt below to help answer a question.\n\nExcerpt from"
     " {citation}\n\n----\n\n{text}\n\n----\n\nQuestion: {question}\n\nDo not directly"
@@ -47,10 +49,11 @@
     "Papers: {papers}\n\n"
     "Selected keys:"
 )
+
 citation_prompt = (
     "Provide the citation for the following text in MLA Format. "
     "Do not write an introductory sentence. "
-    "If reporting date accessed, the current year is 2024\n\n"
+    f"If reporting date accessed, the current year is {datetime.now().year}\n\n"
     "{text}\n\n"
     "Citation:"
 )

tests/test_configs.py

@@ -5,6 +5,7 @@
 from pydantic import ValidationError
 from pytest_subtests import SubTests
 
+from paperqa.prompts import citation_prompt
 from paperqa.settings import (
     AgentSettings,
     IndexSettings,
@@ -15,6 +16,7 @@
     get_settings,
 )
 from paperqa.types import Doc, DocDetails
+from paperqa.utils import get_year
 
 
 def test_prompt_settings_validation() -> None:
@@ -167,3 +169,11 @@ def test_o1_requires_temp_equals_1() -> None:
 def test_matches_filter_criteria(doc_class, doc_data, filter_criteria, expected_result):
     doc = doc_class(**doc_data)
     assert doc.matches_filter_criteria(filter_criteria) == expected_result
+
+
+def test_citation_prompt_current_year():
+    expected_year_text = f"the current year is {get_year()}"
+
+    assert (
+        expected_year_text in citation_prompt
+    ), f"Citation prompt should contain '{expected_year_text}' but got: {citation_prompt}"